feat(audio-chat): remove fallback task type, register provider only if the server task type exists

Signed-off-by: Julien Veyssier <julien-nc@posteo.net>

Author: julien-nc

lib/AppInfo/Application.php

@@ -23,7 +23,6 @@
 use OCA\Assistant\Reference\TaskOutputFileReferenceProvider;
 use OCA\Assistant\Reference\Text2ImageReferenceProvider;
 use OCA\Assistant\TaskProcessing\AudioToAudioChatProvider;
-use OCA\Assistant\TaskProcessing\AudioToAudioChatTaskType;
 use OCP\AppFramework\App;
 use OCP\AppFramework\Bootstrap\IBootContext;
 
@@ -73,8 +72,9 @@ public function register(IRegistrationContext $context): void {
 
 		$context->registerEventListener(AddContentSecurityPolicyEvent::class, CSPListener::class);
 
-		$context->registerTaskProcessingTaskType(AudioToAudioChatTaskType::class);
-		$context->registerTaskProcessingProvider(AudioToAudioChatProvider::class);
+		if (class_exists('OCP\\TaskProcessing\\TaskTypes\\AudioToAudioChat')) {
+			$context->registerTaskProcessingProvider(AudioToAudioChatProvider::class);
+		}
 	}
 
 	public function boot(IBootContext $context): void {

lib/TaskProcessing/AudioToAudioChatProvider.php

@@ -14,10 +14,9 @@
 use OCA\Assistant\Service\TaskProcessingService;
 use OCP\Files\File;
 use OCP\IL10N;
-use OCP\TaskProcessing\EShapeType;
 use OCP\TaskProcessing\ISynchronousProvider;
-use OCP\TaskProcessing\ShapeDescriptor;
 use OCP\TaskProcessing\Task;
+use OCP\TaskProcessing\TaskTypes\AudioToAudioChat;
 use OCP\TaskProcessing\TaskTypes\AudioToText;
 use OCP\TaskProcessing\TaskTypes\TextToSpeech;
 use OCP\TaskProcessing\TaskTypes\TextToTextChat;
@@ -42,10 +41,7 @@ public function getName(): string {
 	}
 
 	public function getTaskTypeId(): string {
-		if (class_exists('OCP\\TaskProcessing\\TaskTypes\\AudioToAudioChat')) {
-			return \OCP\TaskProcessing\TaskTypes\AudioToAudioChat::ID;
-		}
-		return AudioToAudioChatTaskType::ID;
+		return AudioToAudioChat::ID;
 	}
 
 	public function getExpectedRuntime(): int {
@@ -78,13 +74,7 @@ public function getOutputShapeEnumValues(): array {
 	}
 
 	public function getOptionalOutputShape(): array {
-		return [
-			'input_transcript' => new ShapeDescriptor(
-				$this->l->t('Input transcript'),
-				$this->l->t('Input transcription'),
-				EShapeType::Text,
-			),
-		];
+		return [];
 	}
 
 	public function getOptionalOutputShapeEnumValues(): array {

lib/TaskProcessing/AudioToAudioChatTaskType.php

@@ -72,14 +72,19 @@ public function getInputShape(): array {
 	 */
 	public function getOutputShape(): array {
 		return [
+			'input_transcript' => new ShapeDescriptor(
+				$this->l->t('Input transcript'),
+				$this->l->t('Transcription of the audio input'),
+				EShapeType::Text,
+			),
 			'output' => new ShapeDescriptor(
 				$this->l->t('Response voice message'),
-				$this->l->t('The generated response as part of the conversation'),
+				$this->l->t('The generated voice response as part of the conversation'),
 				EShapeType::Audio
 			),
 			'output_transcript' => new ShapeDescriptor(
 				$this->l->t('Output transcript'),
-				$this->l->t('Response transcription'),
+				$this->l->t('Transcription of the audio output'),
 				EShapeType::Text,
 			),
 		];