browser address schema update (#2931)

skyvern/forge/sdk/routes/streaming_vnc.py

@@ -3,6 +3,7 @@ Streaming VNC WebSocket connections.
 """
 
 import asyncio
+from urllib.parse import urlparse
 
 import structlog
 import websockets
@@ -169,7 +170,9 @@ async def loop_stream_vnc(streaming: sc.Streaming) -> None:
         return
 
     browser_address = streaming.browser_session.browser_address
-    host, _ = browser_address.rsplit(":")
+
+    parsed_browser_address = urlparse(browser_address)
+    host = parsed_browser_address.hostname
     vnc_url = f"ws://{host}:{streaming.vnc_port}"
 
     LOG.info(

skyvern/webeye/persistent_sessions_manager.py

@@ -16,8 +16,6 @@ LOG = structlog.get_logger()
 @dataclass
 class BrowserSession:
     browser_state: BrowserState
-    cdp_port: int
-    cdp_host: str = "localhost"
 
 
 class PersistentSessionsManager:
@@ -139,16 +137,6 @@ class PersistentSessionsManager:
             organization_id=organization_id,
         )
 
-    async def get_network_info(self, session_id: str) -> tuple[int | None, str | None]:
-        """Returns cdp port and ip address of the browser session"""
-        browser_session = self._browser_sessions.get(session_id)
-        if browser_session:
-            return (
-                browser_session.cdp_port,
-                browser_session.cdp_host,
-            )
-        return None, None
-
     async def release_browser_session(self, session_id: str, organization_id: str) -> None:
         """Release a specific browser session."""
         await self.database.release_persistent_browser_session(session_id, organization_id)