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()

View File

@@ -241,7 +241,7 @@ class WorkflowModel(Base):
persist_browser_session = Column(Boolean, default=False, nullable=False)
model = Column(JSON, nullable=True)
status = Column(String, nullable=False, default="published")
use_cache = Column(Boolean, default=False, nullable=False)
generate_script = Column(Boolean, default=False, nullable=False)
cache_key = Column(String, nullable=True)
created_at = Column(DateTime, default=datetime.datetime.utcnow, nullable=False)

View File

@@ -262,7 +262,7 @@ def convert_to_workflow(workflow_model: WorkflowModel, debug_enabled: bool = Fal
deleted_at=workflow_model.deleted_at,
status=WorkflowStatus(workflow_model.status),
extra_http_headers=workflow_model.extra_http_headers,
use_cache=workflow_model.use_cache,
generate_script=workflow_model.generate_script,
cache_key=workflow_model.cache_key,
)