add cache_key to workflows table (#3112)

skyvern/forge/sdk/db/client.py

@@ -1358,7 +1358,7 @@ class AgentDB:
         is_saved_task: bool = False,
         status: WorkflowStatus = WorkflowStatus.published,
         use_cache: bool = False,
-        cache_project_id: str | None = None,
+        cache_key: str | None = None,
     ) -> Workflow:
         async with self.Session() as session:
             workflow = WorkflowModel(
@@ -1377,7 +1377,7 @@ class AgentDB:
                 is_saved_task=is_saved_task,
                 status=status,
                 use_cache=use_cache,
-                cache_project_id=cache_project_id,
+                cache_key=cache_key,
             )
             if workflow_permanent_id:
                 workflow.workflow_permanent_id = workflow_permanent_id
@@ -1557,7 +1557,7 @@ class AgentDB:
         workflow_definition: dict[str, Any] | None = None,
         version: int | None = None,
         use_cache: bool | None = None,
-        cache_project_id: str | None = None,
+        cache_key: str | None = None,
     ) -> Workflow:
         try:
             async with self.Session() as session:
@@ -1577,8 +1577,8 @@ class AgentDB:
                         workflow.version = version
                     if use_cache is not None:
                         workflow.use_cache = use_cache
-                    if cache_project_id is not None:
-                        workflow.cache_project_id = cache_project_id
+                    if cache_key is not None:
+                        workflow.cache_key = cache_key
                     await session.commit()
                     await session.refresh(workflow)
                     return convert_to_workflow(workflow, self.debug_enabled)

skyvern/forge/sdk/db/models.py

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

skyvern/forge/sdk/workflow/models/workflow.py

@@ -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

skyvern/forge/sdk/workflow/models/yaml.py

@@ -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

skyvern/forge/sdk/workflow/service.py

@@ -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