Skip to content

Commit

Permalink
remove unrelated sync changes
Browse files Browse the repository at this point in the history
  • Loading branch information
msalihaltun committed Dec 17, 2024
1 parent 8210d45 commit 389446c
Show file tree
Hide file tree
Showing 2 changed files with 19 additions and 0 deletions.
6 changes: 6 additions & 0 deletions skyvern/forge/sdk/db/id.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,6 +46,7 @@
TASK_GENERATION_PREFIX = "tg"
OBSERVER_CRUISE_ID = "oc"
OBSERVER_THOUGHT_ID = "ot"
PERSISTENT_BROWSER_SESSION_ID = "pbs"


def generate_workflow_id() -> str:
Expand Down Expand Up @@ -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
Expand Down
13 changes: 13 additions & 0 deletions skyvern/forge/sdk/db/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@
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,
Expand Down Expand Up @@ -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)

0 comments on commit 389446c

Please sign in to comment.