diff --git a/lib/Service/AssistantService.php b/lib/Service/AssistantService.php
index 7ec1fe03..a3bff54c 100644
--- 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
index 7d91178c..4ee21a00 100644
--- a/tests/psalm-baseline.xml
+++ b/tests/psalm-baseline.xml
@@ -42,6 +42,8 @@
+
+