From f0758c8ee6c9c9813abcc8eb54f6d912063da868 Mon Sep 17 00:00:00 2001
From: SebastianKrupinski <krupinskis05@gmail.com>
Date: Wed, 29 Jan 2025 12:50:04 -0500
Subject: [PATCH] fix: Adjust TaskProcessingListener Again

Signed-off-by: SebastianKrupinski <krupinskis05@gmail.com>
---
 lib/Listener/TaskProcessingListener.php              | 4 ++--
 lib/Service/AiIntegrations/AiIntegrationsService.php | 6 +++---
 tests/Unit/Service/AiIntegrationsServiceTest.php     | 6 +++---
 3 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/lib/Listener/TaskProcessingListener.php b/lib/Listener/TaskProcessingListener.php
index 5a08f78e2c..cfac190196 100644
--- a/lib/Listener/TaskProcessingListener.php
+++ b/lib/Listener/TaskProcessingListener.php
@@ -14,7 +14,7 @@
 use OCP\EventDispatcher\Event;
 use OCP\EventDispatcher\IEventListener;
 use OCP\TaskProcessing\Events\TaskSuccessfulEvent;
-use OCP\TaskProcessing\TaskTypes\TextToTextSummary;
+use OCP\TaskProcessing\TaskTypes\TextToText;
 use Psr\Log\LoggerInterface;
 
 /**
@@ -40,7 +40,7 @@ public function handle(Event $event): void {
 			return;
 		}
 
-		if ($task->getTaskTypeId() !== TextToTextSummary::ID) {
+		if ($task->getTaskTypeId() !== TextToText::ID) {
 			return;
 		}
 
diff --git a/lib/Service/AiIntegrations/AiIntegrationsService.php b/lib/Service/AiIntegrations/AiIntegrationsService.php
index a5a84e84dc..a10169dab7 100644
--- a/lib/Service/AiIntegrations/AiIntegrationsService.php
+++ b/lib/Service/AiIntegrations/AiIntegrationsService.php
@@ -22,7 +22,7 @@
 use OCP\IConfig;
 use OCP\TaskProcessing\IManager as TaskProcessingManager;
 use OCP\TaskProcessing\Task as TaskProcessingTask;
-use OCP\TaskProcessing\TaskTypes\TextToTextSummary;
+use OCP\TaskProcessing\TaskTypes\TextToText;
 use OCP\TextProcessing\FreePromptTaskType;
 use OCP\TextProcessing\IManager;
 use OCP\TextProcessing\SummaryTaskType;
@@ -66,7 +66,7 @@ public function __construct(
 	 */
 	public function summarizeMessages(Account $account, array $messages): void {
 		$availableTaskTypes = $this->taskProcessingManager->getAvailableTaskTypes();
-		if (!isset($availableTaskTypes[TextToTextSummary::ID])) {
+		if (!isset($availableTaskTypes[TextToText::ID])) {
 			$this->logger->info('No text summary provider available');
 			return;
 		}
@@ -101,7 +101,7 @@ public function summarizeMessages(Account $account, array $messages): void {
 						  "Here is the ***E-MAIL*** for which you must generate a helpful summary: \r\n" .
 						  "***START_OF_E-MAIL***\r\n$messageBody\r\n***END_OF_E-MAIL***\r\n";
 				$task = new TaskProcessingTask(
-					TextToTextSummary::ID,
+					TextToText::ID,
 					[
 						'max_tokens' => 1024,
 						'input' => $prompt,
diff --git a/tests/Unit/Service/AiIntegrationsServiceTest.php b/tests/Unit/Service/AiIntegrationsServiceTest.php
index 0b1914f2ba..a96d27f02a 100644
--- a/tests/Unit/Service/AiIntegrationsServiceTest.php
+++ b/tests/Unit/Service/AiIntegrationsServiceTest.php
@@ -402,7 +402,7 @@ public function testSummarizeMessagesContainsSummary() {
 
 		$this->taskProcessingManager->expects(self::once())
 			->method('getAvailableTaskTypes')
-			->willReturn(['core:text2text:summary' => $this->taskProcessingProvider]);
+			->willReturn(['core:text2text' => $this->taskProcessingProvider]);
 		$this->clientFactory->expects(self::once())
 			->method('getClient');
 		
@@ -440,7 +440,7 @@ public function testSummarizeMessagesIsEncrypted() {
 
 		$this->taskProcessingManager->expects(self::once())
 			->method('getAvailableTaskTypes')
-			->willReturn(['core:text2text:summary' => $this->taskProcessingProvider]);
+			->willReturn(['core:text2text' => $this->taskProcessingProvider]);
 		$this->taskProcessingManager->expects(self::never())
 			->method('scheduleTask');
 
@@ -499,7 +499,7 @@ public function testSummarizeMessages() {
 
 		$this->taskProcessingManager->expects(self::once())
 			->method('getAvailableTaskTypes')
-			->willReturn(['core:text2text:summary' => $this->taskProcessingProvider]);
+			->willReturn(['core:text2text' => $this->taskProcessingProvider]);
 		$this->taskProcessingManager->expects(self::once())
 			->method('scheduleTask');