Browser streaming: reorganize & rename (#4033)

This commit is contained in:
Jonathan Dobson
2025-11-19 09:35:05 -05:00
committed by GitHub
parent 1559160aef
commit 2253ca2004
8 changed files with 18 additions and 14 deletions

View File

@@ -8,7 +8,7 @@ from skyvern.forge.sdk.routes import pylon # noqa: F401
from skyvern.forge.sdk.routes import run_blocks # noqa: F401 from skyvern.forge.sdk.routes import run_blocks # noqa: F401
from skyvern.forge.sdk.routes import scripts # noqa: F401 from skyvern.forge.sdk.routes import scripts # noqa: F401
from skyvern.forge.sdk.routes import sdk # noqa: F401 from skyvern.forge.sdk.routes import sdk # noqa: F401
from skyvern.forge.sdk.routes import streaming # noqa: F401
from skyvern.forge.sdk.routes import streaming_messages # noqa: F401
from skyvern.forge.sdk.routes import streaming_vnc # noqa: F401
from skyvern.forge.sdk.routes import webhooks # noqa: F401 from skyvern.forge.sdk.routes import webhooks # noqa: F401
from skyvern.forge.sdk.routes.streaming import messages # noqa: F401
from skyvern.forge.sdk.routes.streaming import screenshot # noqa: F401
from skyvern.forge.sdk.routes.streaming import vnc # noqa: F401

View File

@@ -8,7 +8,7 @@ from contextlib import asynccontextmanager
import structlog import structlog
from playwright.async_api import Browser, BrowserContext, Page, Playwright, async_playwright from playwright.async_api import Browser, BrowserContext, Page, Playwright, async_playwright
import skyvern.forge.sdk.routes.streaming_clients as sc import skyvern.forge.sdk.routes.streaming.clients as sc
from skyvern.config import settings from skyvern.config import settings
LOG = structlog.get_logger() LOG = structlog.get_logger()

View File

@@ -8,11 +8,11 @@ import structlog
from fastapi import WebSocket, WebSocketDisconnect from fastapi import WebSocket, WebSocketDisconnect
from websockets.exceptions import ConnectionClosedError from websockets.exceptions import ConnectionClosedError
import skyvern.forge.sdk.routes.streaming_clients as sc import skyvern.forge.sdk.routes.streaming.clients as sc
from skyvern.forge.sdk.routes.routers import base_router, legacy_base_router from skyvern.forge.sdk.routes.routers import base_router, legacy_base_router
from skyvern.forge.sdk.routes.streaming_agent import connected_agent from skyvern.forge.sdk.routes.streaming.agent import connected_agent
from skyvern.forge.sdk.routes.streaming_auth import auth from skyvern.forge.sdk.routes.streaming.auth import auth
from skyvern.forge.sdk.routes.streaming_verify import ( from skyvern.forge.sdk.routes.streaming.verify import (
loop_verify_browser_session, loop_verify_browser_session,
loop_verify_workflow_run, loop_verify_workflow_run,
verify_browser_session, verify_browser_session,

View File

@@ -146,7 +146,11 @@ async def workflow_run_streaming(
organization = await get_current_org(x_api_key=apikey, authorization=token) organization = await get_current_org(x_api_key=apikey, authorization=token)
organization_id = organization.organization_id organization_id = organization.organization_id
except Exception: except Exception:
LOG.exception("WofklowRun Streaming: Error while getting organization", workflow_run_id=workflow_run_id) LOG.exception(
"WofklowRun Streaming: Error while getting organization",
workflow_run_id=workflow_run_id,
token=token,
)
try: try:
await websocket.send_text("Invalid credential provided") await websocket.send_text("Invalid credential provided")
except ConnectionClosedOK: except ConnectionClosedOK:

View File

@@ -3,7 +3,7 @@ from datetime import datetime
import structlog import structlog
import skyvern.forge.sdk.routes.streaming_clients as sc import skyvern.forge.sdk.routes.streaming.clients as sc
from skyvern.config import settings from skyvern.config import settings
from skyvern.forge import app from skyvern.forge import app
from skyvern.forge.sdk.schemas.persistent_browser_sessions import AddressablePersistentBrowserSession from skyvern.forge.sdk.schemas.persistent_browser_sessions import AddressablePersistentBrowserSession

View File

@@ -19,14 +19,14 @@ from fastapi import WebSocket, WebSocketDisconnect
from websockets import Data from websockets import Data
from websockets.exceptions import ConnectionClosedError from websockets.exceptions import ConnectionClosedError
import skyvern.forge.sdk.routes.streaming_clients as sc import skyvern.forge.sdk.routes.streaming.clients as sc
from skyvern.config import settings from skyvern.config import settings
from skyvern.forge import app from skyvern.forge import app
from skyvern.forge.sdk.db.enums import OrganizationAuthTokenType from skyvern.forge.sdk.db.enums import OrganizationAuthTokenType
from skyvern.forge.sdk.routes.routers import base_router, legacy_base_router from skyvern.forge.sdk.routes.routers import base_router, legacy_base_router
from skyvern.forge.sdk.routes.streaming_agent import connected_agent from skyvern.forge.sdk.routes.streaming.agent import connected_agent
from skyvern.forge.sdk.routes.streaming_auth import auth from skyvern.forge.sdk.routes.streaming.auth import auth
from skyvern.forge.sdk.routes.streaming_verify import ( from skyvern.forge.sdk.routes.streaming.verify import (
loop_verify_browser_session, loop_verify_browser_session,
loop_verify_task, loop_verify_task,
loop_verify_workflow_run, loop_verify_workflow_run,