Add persistent browser session model

skyvern/forge/sdk/db/id.py

@@ -46,7 +46,7 @@
 TASK_GENERATION_PREFIX = "tg"
 OBSERVER_CRUISE_ID = "oc"
 OBSERVER_THOUGHT_ID = "ot"
-
+PERSISTENT_BROWSER_SESSION_ID = "pbs"
 
 def generate_workflow_id() -> str:
     int_id = generate_id()
@@ -153,6 +153,11 @@ def generate_observer_thought_id() -> str:
     return f"{OBSERVER_THOUGHT_ID}_{int_id}"
 
 
+def generate_persistent_browser_session_id() -> str:
+    int_id = generate_id()
+    return f"{PERSISTENT_BROWSER_SESSION_ID}_{int_id}"
+
+
 def generate_id() -> int:
     """
     generate a 64-bit int ID

skyvern/forge/sdk/db/models.py

@@ -39,6 +39,7 @@
     generate_workflow_permanent_id,
     generate_workflow_run_block_id,
     generate_workflow_run_id,
+    generate_persistent_browser_session_id,
 )
 from skyvern.forge.sdk.schemas.tasks import ProxyLocation
 
@@ -540,3 +541,13 @@ class ObserverThoughtModel(Base):
 
     created_at = Column(DateTime, default=datetime.datetime.utcnow, nullable=False)
     modified_at = Column(DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow, nullable=False)
+
+class PersistentBrowserSessionModel(Base):
+    __tablename__ = "persistent_browser_sessions"
+
+    persistent_browser_session_id = Column(String, primary_key=True, default=generate_persistent_browser_session_id)
+    organization_id = Column(String, ForeignKey("organizations.organization_id"), nullable=False)
+    runnable_type = Column(String, nullable=False)
+    runnable_id = Column(String, nullable=False)
+    created_at = Column(DateTime, default=datetime.datetime.utcnow, nullable=False)
+    modified_at = Column(DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow, nullable=False)

skyvern/webeye/persistent_sessions_manager.py

@@ -1,11 +1,11 @@
 from __future__ import annotations
 
-import uuid
 from typing import Dict, List, Optional, Tuple
 
 import structlog
 from playwright.async_api import async_playwright
 
+from skyvern.forge.sdk.db.id import generate_persistent_browser_session_id
 from skyvern.forge.sdk.schemas.tasks import ProxyLocation
 from skyvern.webeye.browser_factory import BrowserContextFactory, BrowserState
 from skyvern.webeye.models import BrowserSessionResponse
@@ -37,7 +37,7 @@ async def create_session(
         url: str | None = None,
     ) -> Tuple[str, BrowserState]:
         """Create a new browser session for an organization and return its ID with the browser state."""
-        session_id = str(uuid.uuid4())
+        session_id = generate_persistent_browser_session_id()
 
         LOG.info(
             "Creating new browser session",