Fix LanguageModelManager#runTask: Insert task into db if it doesn't e…

…xist Signed-off-by: Marcel Klehr <mklehr@gmx.net>
nextcloud · Jul 7, 2023 · 1747068 · 1747068
1 parent c568c4a
commit 1747068
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/private/LanguageModel/LanguageModelManager.php b/lib/private/LanguageModel/LanguageModelManager.php
@@ -120,8 +120,12 @@ public function runTask(ILanguageModelTask $task): string {
 			}
 			try {
 				$task->setStatus(ILanguageModelTask::STATUS_RUNNING);
-				$taskEntity = $this->taskMapper->update(Task::fromLanguageModelTask($task));
-				$task->setId($taskEntity->getId());
+				if ($task->getId() === null) {
+					$taskEntity = $this->taskMapper->insert(Task::fromLanguageModelTask($task));
+					$task->setId($taskEntity->getId());
+				}else {
+					$this->taskMapper->update(Task::fromLanguageModelTask($task));
+				}
 				$output = $task->visitProvider($provider);
 				$task->setOutput($output);
 				$task->setStatus(ILanguageModelTask::STATUS_SUCCESSFUL);