feat: migrate workflow use_cache field to generate_script (#3154)

Co-authored-by: devsy-bot <no-reply@devsy.ai>
Co-authored-by: Shuchang Zheng <wintonzheng0325@gmail.com>
This commit is contained in:
devsy-bot[bot]
2025-08-10 09:02:23 -07:00
committed by GitHub
parent 16596e5c61
commit 1a3938a756
13 changed files with 56 additions and 25 deletions

View File

@@ -1363,7 +1363,7 @@ class AgentDB:
version: int | None = None,
is_saved_task: bool = False,
status: WorkflowStatus = WorkflowStatus.published,
use_cache: bool = False,
generate_script: bool = False,
cache_key: str | None = None,
) -> Workflow:
async with self.Session() as session:
@@ -1382,7 +1382,7 @@ class AgentDB:
model=model,
is_saved_task=is_saved_task,
status=status,
use_cache=use_cache,
generate_script=generate_script,
cache_key=cache_key,
)
if workflow_permanent_id:
@@ -1562,7 +1562,7 @@ class AgentDB:
description: str | None = None,
workflow_definition: dict[str, Any] | None = None,
version: int | None = None,
use_cache: bool | None = None,
generate_script: bool | None = None,
cache_key: str | None = None,
) -> Workflow:
try:
@@ -1581,8 +1581,8 @@ class AgentDB:
workflow.workflow_definition = workflow_definition
if version is not None:
workflow.version = version
if use_cache is not None:
workflow.use_cache = use_cache
if generate_script is not None:
workflow.generate_script = generate_script
if cache_key is not None:
workflow.cache_key = cache_key
await session.commit()