vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Introduce persistent browser sessions model (#1408)

Browse Source
This commit is contained in:
Maksim Ivanov
2024-12-17 19:45:38 +01:00
committed by GitHub
parent 69687f121a
commit c97b261f96
3 changed files with 64 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
skyvern/forge/sdk/db/id.py
View File

@@ -46,6 +46,7 @@ BITWARDEN_CREDIT_CARD_DATA_PARAMETER_PREFIX = "bccd"
TASK_GENERATION_PREFIX = "tg"
OBSERVER_CRUISE_ID = "oc"
OBSERVER_THOUGHT_ID = "ot"
PERSISTENT_BROWSER_SESSION_ID = "pbs"
def generate_workflow_id() -> str:
@@ -153,6 +154,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

13
skyvern/forge/sdk/db/models.py
View File

@@ -30,6 +30,7 @@ from skyvern.forge.sdk.db.id import (
generate_org_id,
generate_organization_auth_token_id,
generate_output_parameter_id,
generate_persistent_browser_session_id,
generate_step_id,
generate_task_generation_id,
generate_task_id,
@@ -540,3 +541,15 @@ 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)
deleted_at = Column(DateTime, nullable=True)