Browser streaming: reorganize & rename (#4033)

skyvern/forge/sdk/routes/__init__.py

@@ -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 scripts  # 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.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

skyvern/forge/sdk/routes/streaming/agent.py

@@ -8,7 +8,7 @@ from contextlib import asynccontextmanager
 import structlog
 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
 
 LOG = structlog.get_logger()

skyvern/forge/sdk/routes/streaming/messages.py

@@ -8,11 +8,11 @@ import structlog
 from fastapi import WebSocket, WebSocketDisconnect
 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.streaming_agent import connected_agent
-from skyvern.forge.sdk.routes.streaming_auth import auth
-from skyvern.forge.sdk.routes.streaming_verify import (
+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.verify import (
     loop_verify_browser_session,
     loop_verify_workflow_run,
     verify_browser_session,

skyvern/forge/sdk/routes/streaming/screenshot.py

@@ -146,7 +146,11 @@ async def workflow_run_streaming(
         organization = await get_current_org(x_api_key=apikey, authorization=token)
         organization_id = organization.organization_id
     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:
             await websocket.send_text("Invalid credential provided")
         except ConnectionClosedOK:

skyvern/forge/sdk/routes/streaming/verify.py

@@ -3,7 +3,7 @@ from datetime import datetime
 
 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.forge import app
 from skyvern.forge.sdk.schemas.persistent_browser_sessions import AddressablePersistentBrowserSession

skyvern/forge/sdk/routes/streaming/vnc.py

@@ -19,14 +19,14 @@ from fastapi import WebSocket, WebSocketDisconnect
 from websockets import Data
 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.forge import app
 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.streaming_agent import connected_agent
-from skyvern.forge.sdk.routes.streaming_auth import auth
-from skyvern.forge.sdk.routes.streaming_verify import (
+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.verify import (
     loop_verify_browser_session,
     loop_verify_task,
     loop_verify_workflow_run,