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

Signed-off-by: Marcel Klehr <mklehr@gmx.net>
This commit is contained in:
Marcel Klehr 2023-06-30 13:20:15 +02:00
parent c568c4a464
commit 1747068e51

View file

@ -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);