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

support extension choice in pbs (#4364)

Browse Source
This commit is contained in:
LawyZheng
2025-12-24 13:01:52 +08:00
committed by GitHub
parent ef8cbddfeb
commit c2bf0f8913
8 changed files with 58 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
skyvern/forge/sdk/db/agent_db.py
View File

@@ -94,7 +94,7 @@ from skyvern.forge.sdk.schemas.organizations import (
Organization,
OrganizationAuthToken,
)
from skyvern.forge.sdk.schemas.persistent_browser_sessions import PersistentBrowserSession
from skyvern.forge.sdk.schemas.persistent_browser_sessions import Extensions, PersistentBrowserSession
from skyvern.forge.sdk.schemas.runs import Run
from skyvern.forge.sdk.schemas.task_generations import TaskGeneration
from skyvern.forge.sdk.schemas.task_v2 import TaskV2, TaskV2Status, Thought, ThoughtType
@@ -4540,8 +4540,10 @@ class AgentDB(BaseAlchemyDB):
runnable_id: str | None = None,
timeout_minutes: int | None = None,
proxy_location: ProxyLocationInput = ProxyLocation.RESIDENTIAL,
extensions: list[Extensions] | None = None,
) -> PersistentBrowserSession:
"""Create a new persistent browser session."""
extensions_str: list[str] | None = [extension.value for extension in extensions] if extensions else None
try:
async with self.Session() as session:
browser_session = PersistentBrowserSessionModel(
@@ -4550,6 +4552,7 @@ class AgentDB(BaseAlchemyDB):
runnable_id=runnable_id,
timeout_minutes=timeout_minutes,
proxy_location=_serialize_proxy_location(proxy_location),
extensions=extensions_str,
)
session.add(browser_session)
await session.commit()