Merge pull request #10629 from nextcloud/fix/10616-adjust-taskprocess…

…inglistener-v2 fix: Adjust TaskProcessingListener Again
nextcloud · Jan 30, 2025 · 2a2f82e · 2a2f82e
2 parents d3d7af0 + f0758c8
commit 2a2f82e
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git 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
@@ -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
@@ -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');