[SKY-6974] Browser Profiles [2/3] Marc/backend browser session profiles (#3923)

skyvern/forge/sdk/db/client.py

@@ -2239,6 +2239,7 @@ class AgentDB:
         workflow_id: str,
         organization_id: str,
         browser_session_id: str | None = None,
+        browser_profile_id: str | None = None,
         proxy_location: ProxyLocation | None = None,
         webhook_callback_url: str | None = None,
         totp_verification_url: str | None = None,
@@ -2260,6 +2261,7 @@ class AgentDB:
                     workflow_id=workflow_id,
                     organization_id=organization_id,
                     browser_session_id=browser_session_id,
+                    browser_profile_id=browser_profile_id,
                     proxy_location=proxy_location,
                     status="created",
                     webhook_callback_url=webhook_callback_url,

skyvern/forge/sdk/db/models.py

@@ -304,6 +304,7 @@ class WorkflowRunModel(Base):
     parent_workflow_run_id = Column(String, nullable=True, index=True)
     organization_id = Column(String, nullable=False, index=True)
     browser_session_id = Column(String, nullable=True, index=True)
+    browser_profile_id = Column(String, nullable=True, index=True)
     status = Column(String, nullable=False)
     failure_reason = Column(String)
     proxy_location = Column(String)

skyvern/forge/sdk/db/utils.py

@@ -309,6 +309,7 @@ def convert_to_workflow_run(
         workflow_id=workflow_run_model.workflow_id,
         organization_id=workflow_run_model.organization_id,
         browser_session_id=workflow_run_model.browser_session_id,
+        browser_profile_id=workflow_run_model.browser_profile_id,
         status=WorkflowRunStatus[workflow_run_model.status],
         failure_reason=workflow_run_model.failure_reason,
         proxy_location=(