hide ChatWithTools and ContextAgentInteraction task types

Signed-off-by: Julien Veyssier <[email protected]>
nextcloud · Jan 8, 2025 · f59ac8f · f59ac8f
1 parent 1ad93f7
commit f59ac8f
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/lib/Service/AssistantService.php b/lib/Service/AssistantService.php
@@ -173,7 +173,17 @@ public function getAvailableTaskTypes(): array {
 		}
 		/** @var string $typeId */
 		foreach ($availableTaskTypes as $typeId => $taskTypeArray) {
-			// skip chat task type (not directly useful to the end user)
+			// skip chat, chat with tools and ContextAgent task types (not directly useful to the end user)
+			if (!self::DEBUG) {
+				if (class_exists('OCP\\TaskProcessing\\TaskTypes\\TextToTextChatWithTools')
+					&& $typeId === \OCP\TaskProcessing\TaskTypes\TextToTextChatWithTools::ID) {
+					continue;
+				}
+				if (class_exists('OCP\\TaskProcessing\\TaskTypes\\ContextAgentInteraction')
+					&& $typeId === \OCP\TaskProcessing\TaskTypes\ContextAgentInteraction::ID) {
+					continue;
+				}
+			}
 			if ($typeId === TextToTextChat::ID) {
 				// add the chattyUI virtual task type
 				$types[] = [
@@ -191,7 +201,9 @@ public function getAvailableTaskTypes(): array {
 					'priority' => self::TASK_TYPE_PRIORITIES['chatty-llm'] ?? 1000,
 				];
 				// do not add the raw TextToTextChat type
-				continue;
+				if (!self::DEBUG) {
+					continue;
+				}
 			}
 			$taskTypeArray['id'] = $typeId;
 			$taskTypeArray['priority'] = self::TASK_TYPE_PRIORITIES[$typeId] ?? 1000;

diff --git a/tests/psalm-baseline.xml b/tests/psalm-baseline.xml
@@ -42,6 +42,8 @@
     </MoreSpecificReturnType>
     <RedundantCondition>
       <code><![CDATA[!self::DEBUG]]></code>
+      <code><![CDATA[!self::DEBUG]]></code>
+      <code><![CDATA[!self::DEBUG]]></code>
     </RedundantCondition>
     <TypeDoesNotContainType>
       <code><![CDATA[self::DEBUG]]></code>