add debug_session_id column to workflow_runs table, and accept from a… (#3571)

skyvern/forge/sdk/db/client.py

@@ -1698,6 +1698,7 @@ class AgentDB:
         browser_address: str | None = None,
         sequential_key: str | None = None,
         run_with: str | None = None,
+        debug_session_id: str | None = None,
     ) -> WorkflowRun:
         try:
             async with self.Session() as session:
@@ -1717,6 +1718,7 @@ class AgentDB:
                     browser_address=browser_address,
                     sequential_key=sequential_key,
                     run_with=run_with,
+                    debug_session_id=debug_session_id,
                 )
                 session.add(workflow_run)
                 await session.commit()

skyvern/forge/sdk/db/models.py

@@ -290,6 +290,7 @@ class WorkflowRunModel(Base):
     job_id = Column(String, nullable=True)
     sequential_key = Column(String, nullable=True)
     run_with = Column(String, nullable=True)  # 'agent' or 'code'
+    debug_session_id: Column = Column(String, nullable=True)
 
     queued_at = Column(DateTime, nullable=True)
     started_at = Column(DateTime, nullable=True)

skyvern/forge/sdk/routes/agent_protocol.py

@@ -991,7 +991,7 @@ async def run_block(
         organization=organization,
         template=template,
         workflow_permanent_id=block_run_request.workflow_id,
-        workflow_run_request=block_run_request,
+        block_run_request=block_run_request,
     )
 
     browser_session_id = block_run_request.browser_session_id

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

@@ -126,6 +126,7 @@ class WorkflowRun(BaseModel):
     workflow_permanent_id: str
     organization_id: str
     browser_session_id: str | None = None
+    debug_session_id: str | None = None
     status: WorkflowRunStatus
     extra_http_headers: dict[str, str] | None = None
     proxy_location: ProxyLocation | None = None

skyvern/forge/sdk/workflow/service.py

@@ -148,6 +148,7 @@ class WorkflowService:
         version: int | None = None,
         max_steps_override: int | None = None,
         parent_workflow_run_id: str | None = None,
+        debug_session_id: str | None = None,
     ) -> WorkflowRun:
         """
         Create a workflow run and its parameters. Validate the workflow and the organization. If there are missing
@@ -181,6 +182,7 @@ class WorkflowService:
             organization_id=organization.organization_id,
             parent_workflow_run_id=parent_workflow_run_id,
             sequential_key=workflow.sequential_key,
+            debug_session_id=debug_session_id,
         )
         LOG.info(
             f"Created workflow run {workflow_run.workflow_run_id} for workflow {workflow.workflow_id}",
@@ -950,6 +952,7 @@ class WorkflowService:
         organization_id: str,
         parent_workflow_run_id: str | None = None,
         sequential_key: str | None = None,
+        debug_session_id: str | None = None,
     ) -> WorkflowRun:
         # validate the browser session id
         if workflow_request.browser_session_id:
@@ -975,6 +978,7 @@ class WorkflowService:
             browser_address=workflow_request.browser_address,
             sequential_key=sequential_key,
             run_with=workflow_request.run_with,
+            debug_session_id=debug_session_id,
         )
 
     async def _update_workflow_run_status(