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

Use persistent browser session in runnables (#1510)

Browse Source 
Co-authored-by: Shuchang Zheng <wintonzheng0325@gmail.com>
Co-authored-by: Shuchang Zheng <shu@skyvern.com>
This commit is contained in:
Maksim Ivanov
2025-01-09 22:04:53 +01:00
committed by GitHub
parent 5ed7e5ad8e
commit a4744ed9f5
12 changed files with 506 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

194
scripts/test_persistent_browsers.py
View File

@@ -1,11 +1,12 @@
import json
import os
from typing import Any, Optional
from typing import Any, Optional, cast
import requests
from dotenv import load_dotenv
from skyvern.forge import app
from skyvern.forge.sdk.schemas.tasks import TaskRequest
load_dotenv("./skyvern-frontend/.env")
API_KEY = os.getenv("VITE_SKYVERN_API_KEY")
@@ -56,7 +57,7 @@ def list_sessions() -> None:
print(f"Error listing sessions: {str(e)}")
def create_session() -> Optional[str]:
def create_browser_session() -> Optional[str]:
"""Create a new browser session"""
try:
response = make_request("POST", "/browser_sessions")
@@ -162,14 +163,128 @@ def close_session(session_id: str) -> None:
print(f"Error closing session: {str(e)}")
def create_task(
url: str | None = None,
goal: str | None = None,
browser_session_id: str | None = None,
) -> Optional[str]:
"""Create a new task
Args:
url: URL to navigate to (default: https://news.ycombinator.com)
goal: Task goal/instructions (default: Extract top HN post)
browser_session_id: Optional browser session ID to use
"""
try:
default_url = "https://news.ycombinator.com"
default_goal = "Navigate to the Hacker News homepage and identify the top post. COMPLETE when the title and URL of the top post are extracted. Ensure that the top post is the first post listed on the page."
data = TaskRequest(
url=url or default_url,
goal=goal or default_goal,
browser_session_id=browser_session_id,
)
response = make_request("POST", "/tasks", data=data.model_dump())
task = cast(dict[str, Any], response.json())
print("\nCreated new task:")
try:
print(f" ID: {task.get('task_id', 'N/A')}")
print(f"Full response: {json.dumps(task, indent=2)}")
return task.get("task_id")
except Exception as e:
print(f"Error parsing response: {task}")
print(f"Error: {str(e)}")
return None
except Exception as e:
print(f"Error creating task: {str(e)}")
return None
def create_workflow_run(
workflow_permanent_id: str = "wpid_346464432851787586",
browser_session_id: str | None = None,
) -> Optional[str]:
"""Create a new workflow run
Args:
workflow_permanent_id: Workflow permanent ID (default: wpid_346464432851787586)
browser_session_id: Optional browser session ID to use
"""
try:
data: dict[str, Any] = {
"parameters": {}, # Add parameters if needed
"browser_session_id": browser_session_id,
}
response = make_request("POST", f"/workflows/{workflow_permanent_id}/run", data=data)
workflow_run = response.json()
print("\nCreated new workflow run:")
try:
print(f" Workflow Run ID: {workflow_run.get('workflow_run_id', 'N/A')}")
print(f" Workflow ID: {workflow_run.get('workflow_id', 'N/A')}")
print(f"Full response: {json.dumps(workflow_run, indent=2)}")
return workflow_run.get("workflow_run_id")
except Exception as e:
print(f"Error parsing response: {workflow_run}")
print(f"Error: {str(e)}")
return None
except Exception as e:
print(f"Error creating workflow run: {str(e)}")
return None
def create_cruise(
prompt: str | None = None,
url: str | None = None,
browser_session_id: str | None = None,
) -> Optional[str]:
"""Create a new observer cruise
Args:
prompt: Task prompt/instructions (default: Extract top HN post)
url: URL to navigate to (default: None)
browser_session_id: Optional browser session ID to use
"""
try:
default_prompt = "Navigate to the Hacker News homepage and identify the top post. COMPLETE when the title and URL of the top post are extracted. Ensure that the top post is the first post listed on the page."
data = {"user_prompt": prompt or default_prompt, "url": url, "browser_session_id": browser_session_id}
response = make_request("POST", "/cruise", data=data)
cruise = response.json()
print("\nCreated new observer cruise:")
try:
print(f" Cruise ID: {cruise.get('observer_cruise_id', 'N/A')}")
print(f" URL: {cruise.get('url', 'N/A')}")
print(f"Full response: {json.dumps(cruise, indent=2)}")
return cruise.get("observer_cruise_id")
except Exception as e:
print(f"Error parsing response: {cruise}")
print(f"Error: {str(e)}")
return None
except Exception as e:
print(f"Error creating cruise: {str(e)}")
return None
def print_help() -> None:
"""Print available commands"""
print("\nHTTP API Commands:")
print(" list - List all active browser sessions")
print(" create - Create a new browser session")
print(" get <session_id> - Get details of a specific session")
print(" close <session_id> - Close a specific session")
print(" close_all - Close all active browser sessions")
print(" list_sessions - List all active browser sessions")
print(" create_browser_session - Create a new browser session")
print(" get_session <session_id> - Get details of a specific session")
print(" close_session <session_id> - Close a specific session")
print(" close_all_sessions - Close all active browser sessions")
print(" create_task [args] - Create a new task")
print(" Optional args:")
print(" --url <url> - URL to navigate to")
print(" --goal <goal> - Task goal/instructions")
print(" --browser_session_id <id> - Browser session ID to use")
print(" create_workflow_run [args] - Create a new workflow run")
print(" Optional args:")
print(" --workflow_id <id> - Workflow permanent ID")
print(" --browser_session_id <id> - Browser session ID to use")
print(" create_cruise [args] - Create a new observer cruise")
print(" Optional args:")
print(" --prompt <prompt> - Task prompt/instructions")
print(" --url <url> - URL to navigate to")
print(" --browser_session_id <id> - Browser session ID to use")
print(" help - Show this help message")
print("\nDirect Method Commands:")
print(" direct_list <org_id> - List sessions directly")
@@ -200,22 +315,75 @@ async def main() -> None:
await handle_direct_command(cmd, args)
elif cmd == "help":
print_help()
elif cmd == "list":
elif cmd == "list_sessions":
list_sessions()
elif cmd == "create":
create_session()
elif cmd == "get":
elif cmd == "create_browser_session":
create_browser_session()
elif cmd == "create_task":
# Parse optional args
url = None
goal = None
browser_session_id = None
i = 0
while i < len(args):
if args[i] == "--url" and i + 1 < len(args):
url = args[i + 1]
i += 2
elif args[i] == "--goal" and i + 1 < len(args):
goal = args[i + 1]
i += 2
elif args[i] == "--browser_session_id" and i + 1 < len(args):
browser_session_id = args[i + 1]
i += 2
else:
i += 1
create_task(url=url, goal=goal, browser_session_id=browser_session_id)
elif cmd == "get_session":
if not args:
print("Error: session_id required")
continue
get_session(args[0])
elif cmd == "close":
elif cmd == "close_session":
if not args:
print("Error: session_id required")
continue
close_session(args[0])
elif cmd == "close_all":
elif cmd == "close_all_sessions":
close_all_sessions()
elif cmd == "create_workflow_run":
# Parse optional args
workflow_id = "wpid_346464432851787586" # Default workflow ID
browser_session_id = None
i = 0
while i < len(args):
if args[i] == "--workflow_id" and i + 1 < len(args):
workflow_id = args[i + 1]
i += 2
elif args[i] == "--browser_session_id" and i + 1 < len(args):
browser_session_id = args[i + 1]
i += 2
else:
i += 1
create_workflow_run(workflow_permanent_id=workflow_id, browser_session_id=browser_session_id)
elif cmd == "create_cruise":
# Parse optional args
prompt = None
url = None
browser_session_id = None
i = 0
while i < len(args):
if args[i] == "--prompt" and i + 1 < len(args):
prompt = args[i + 1]
i += 2
elif args[i] == "--url" and i + 1 < len(args):
url = args[i + 1]
i += 2
elif args[i] == "--browser_session_id" and i + 1 < len(args):
browser_session_id = args[i + 1]
i += 2
else:
i += 1
create_cruise(prompt=prompt, url=url, browser_session_id=browser_session_id)
else:
print(f"Unknown command: {cmd}")
print("Type 'help' for available commands")