From 1747068e518240edec22ff3c4f6946ac925918fe Mon Sep 17 00:00:00 2001
From: Marcel Klehr <mklehr@gmx.net>
Date: Fri, 30 Jun 2023 13:20:15 +0200
Subject: [PATCH] Fix LanguageModelManager#runTask: Insert task into db if it
 doesn't exist

Signed-off-by: Marcel Klehr <mklehr@gmx.net>
---
 lib/private/LanguageModel/LanguageModelManager.php | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/lib/private/LanguageModel/LanguageModelManager.php b/lib/private/LanguageModel/LanguageModelManager.php
index 9117b131578..e07762cf8cc 100644
--- a/lib/private/LanguageModel/LanguageModelManager.php
+++ b/lib/private/LanguageModel/LanguageModelManager.php
@@ -120,8 +120,12 @@ class LanguageModelManager implements ILanguageModelManager {
 			}
 			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);