vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

add cache_key to workflows table (#3112)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-08-06 08:32:14 -07:00
committed by GitHub
parent 800c030e5c
commit 31aa7d6973
6 changed files with 46 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
skyvern/forge/sdk/workflow/models/workflow.py
View File

@@ -80,6 +80,8 @@ class Workflow(BaseModel):
status: WorkflowStatus = WorkflowStatus.published
max_screenshot_scrolls: int | None = None
extra_http_headers: dict[str, str] | None = None
use_cache: bool = False
cache_key: str | None = None
created_at: datetime
modified_at: datetime

2
skyvern/forge/sdk/workflow/models/yaml.py
View File

@@ -444,4 +444,4 @@ class WorkflowCreateYAMLRequest(BaseModel):
extra_http_headers: dict[str, str] | None = None
status: WorkflowStatus = WorkflowStatus.published
use_cache: bool = False
cache_project_id: str | None = None
cache_key: str | None = None

8
skyvern/forge/sdk/workflow/service.py
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