LLM/docling-serve
1
0
Fork 0
mirror of https://github.com/docling-project/docling-serve.git synced 2025-12-01 09:33:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: clear results registry (#192)

Browse Source 
Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
This commit is contained in:
Michele Dolfi
2025-05-23 08:30:57 -04:00
committed by GitHub
parent abe5aa03f5
commit de002dfcdc
10 changed files with 229 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
docling_serve/engines/async_kfp/orchestrator.py
View File

@@ -130,13 +130,13 @@ class AsyncKfpOrchestrator(BaseAsyncOrchestrator):
# CANCELED = "CANCELED"
# PAUSED = "PAUSED"
if run_info.state == V2beta1RuntimeState.SUCCEEDED:
task.task_status = TaskStatus.SUCCESS
task.set_status(TaskStatus.SUCCESS)
elif run_info.state == V2beta1RuntimeState.PENDING:
task.task_status = TaskStatus.PENDING
task.set_status(TaskStatus.PENDING)
elif run_info.state == V2beta1RuntimeState.RUNNING:
task.task_status = TaskStatus.STARTED
task.set_status(TaskStatus.STARTED)
else:
task.task_status = TaskStatus.FAILURE
task.set_status(TaskStatus.FAILURE)
async def task_status(self, task_id: str, wait: float = 0.0) -> Task:
await self._update_task_from_run(task_id=task_id, wait=wait)