add cache_key to workflows table (#3112)

This commit is contained in:
Shuchang Zheng 2025-08-06 08:32:14 -07:00 committed by GitHub
parent 800c030e5c
commit 31aa7d6973
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 46 additions and 11 deletions

View file

@ -636,7 +636,7 @@ class WorkflowService:
status: WorkflowStatus = WorkflowStatus.published,
extra_http_headers: dict[str, str] | None = None,
use_cache: bool = False,
cache_project_id: str | None = None,
cache_key: str | None = None,
) -> Workflow:
return await app.DATABASE.create_workflow(
title=title,
@ -656,7 +656,7 @@ class WorkflowService:
status=status,
extra_http_headers=extra_http_headers,
use_cache=use_cache,
cache_project_id=cache_project_id,
cache_key=cache_key,
)
async def get_workflow(self, workflow_id: str, organization_id: str | None = None) -> Workflow:
@ -1539,7 +1539,7 @@ class WorkflowService:
is_saved_task=request.is_saved_task,
status=request.status,
use_cache=request.use_cache,
cache_project_id=request.cache_project_id,
cache_key=request.cache_key,
)
else:
workflow = await self.create_workflow(
@ -1558,7 +1558,7 @@ class WorkflowService:
is_saved_task=request.is_saved_task,
status=request.status,
use_cache=request.use_cache,
cache_project_id=request.cache_project_id,
cache_key=request.cache_key,
)
# Keeping track of the new workflow id to delete it if an error occurs during the creation process
new_workflow_id = workflow.workflow_id