alita-sdk 0.3.446__py3-none-any.whl → 0.3.448__py3-none-any.whl

alita_sdk/runtime/toolkits/mcp.py

@@ -402,7 +402,10 @@ class McpToolkit(BaseToolkit):
                 ]
 
             # Create BaseTool instances from discovered metadata
-            # Use session_id from connection_config (passed from UI) instead of discovery
+            # Use session_id from frontend (passed via connection_config)
+            if session_id:
+                logger.info(f"[MCP Session] Using session_id from frontend: {session_id}")
+            
             for tool_metadata in tool_metadata_list:
                 server_tool = cls._create_tool_from_dict(
                     tool_dict=tool_metadata,
@@ -410,7 +413,7 @@ class McpToolkit(BaseToolkit):
                     connection_config=connection_config,
                     timeout=timeout,
                     client=client,
-                    session_id=connection_config.session_id  # Use session from connection config
+                    session_id=session_id  # Use session from discovery
                 )
 
                 if server_tool:
@@ -462,11 +465,13 @@ class McpToolkit(BaseToolkit):
         """
         all_tools = []
         
-        # Note: We don't initialize MCP session here because:
-        # 1. Discovery happens before we have OAuth tokens
-        # 2. Sessions are initialized on-demand during first tool call (when we have auth)
-        # 3. Sessions are stored in UI sessionStorage and passed back via mcp_tokens
-        session_id = None
+        # Use session_id from UI (passed via connection_config)
+        # For SSE-based MCP servers, frontend generates a UUID and sends it with the OAuth token
+        session_id = connection_config.session_id
+        
+        if not session_id:
+            logger.warning(f"[MCP Session] No session_id provided for '{toolkit_name}' - server may require it")
+            logger.warning(f"[MCP Session] Frontend should generate a UUID and include it with mcp_tokens")
         
         # Discover regular tools
         tools_data = cls._discover_mcp_endpoint(
@@ -521,106 +526,6 @@ class McpToolkit(BaseToolkit):
         logger.info(f"Total discovered {len(all_tools)} tools+prompts from MCP toolkit '{toolkit_name}'")
         return all_tools, session_id
 
-    @classmethod
-    def _initialize_mcp_session(
-        cls,
-        toolkit_name: str,
-        connection_config: McpConnectionConfig,
-        timeout: int,
-        extra_headers: Optional[Dict[str, str]] = None
-    ) -> Optional[str]:
-        """
-        Initialize an MCP session for stateful SSE servers.
-        Returns sessionId if successful, None if server doesn't require sessions.
-        
-        Note: This session should be stored by the caller (e.g., UI sessionStorage)
-        and passed back for subsequent requests. Sessions should NOT be cached
-        on the backend for security reasons (multi-tenant environment).
-        
-        Args:
-            toolkit_name: Name of the toolkit
-            connection_config: MCP connection configuration
-            timeout: Request timeout in seconds
-            extra_headers: Additional headers (e.g., Authorization) to include
-        """
-        import time
-
-        mcp_request = {
-            "jsonrpc": "2.0",
-            "id": f"initialize_{int(time.time())}",
-            "method": "initialize",
-            "params": {
-                "protocolVersion": "2024-11-05",
-                "capabilities": {
-                    "roots": {"listChanged": True},
-                    "sampling": {}
-                },
-                "clientInfo": {
-                    "name": "Alita MCP Client",
-                    "version": "1.0.0"
-                }
-            }
-        }
-
-        headers = {
-            "Content-Type": "application/json",
-            "Accept": "application/json, text/event-stream"
-        }
-        if connection_config.headers:
-            headers.update(connection_config.headers)
-        if extra_headers:
-            headers.update(extra_headers)
-            logger.debug(f"[MCP Session] Added extra headers: {list(extra_headers.keys())}")
-
-        try:
-            logger.info(f"[MCP Session] Attempting to initialize session for {connection_config.url}")
-            logger.debug(f"[MCP Session] Initialize request: {mcp_request}")
-            response = requests.post(
-                connection_config.url,
-                json=mcp_request,
-                headers=headers,
-                timeout=timeout
-            )
-
-            logger.info(f"[MCP Session] Initialize response status: {response.status_code}")
-            if response.status_code == 200:
-                # Parse the response to extract sessionId
-                content_type = response.headers.get('Content-Type', '')
-                logger.debug(f"[MCP Session] Response Content-Type: {content_type}")
-                
-                if 'text/event-stream' in content_type:
-                    data = cls._parse_sse_response(response.text)
-                elif 'application/json' in content_type:
-                    data = response.json()
-                else:
-                    logger.warning(f"[MCP Session] Unexpected Content-Type during initialize: {content_type}")
-                    logger.debug(f"[MCP Session] Response text: {response.text[:500]}")
-                    return None
-
-                logger.debug(f"[MCP Session] Parsed response: {data}")
-                
-                # Extract sessionId from response
-                result = data.get("result", {})
-                session_id = result.get("sessionId")
-                if session_id:
-                    logger.info(f"[MCP Session] ✓ Session initialized for '{toolkit_name}': {session_id}")
-                    logger.info(f"[MCP Session] This sessionId should be stored by the caller and sent back for subsequent requests")
-                    return session_id
-                else:
-                    logger.info(f"[MCP Session] No sessionId in initialize response for '{toolkit_name}' - server may not require sessions")
-                    logger.debug(f"[MCP Session] Full result: {result}")
-                    return None
-            else:
-                logger.warning(f"[MCP Session] Initialize returned {response.status_code} for '{toolkit_name}' - server may not support sessions")
-                logger.debug(f"[MCP Session] Response: {response.text[:500]}")
-                return None
-
-        except Exception as e:
-            logger.warning(f"[MCP Session] Failed to initialize MCP session for '{toolkit_name}': {e} - proceeding without session")
-            import traceback
-            logger.debug(f"[MCP Session] Traceback: {traceback.format_exc()}")
-            return None
-
     @classmethod
     def _discover_mcp_endpoint(
         cls,

alita_sdk/runtime/tools/mcp_remote_tool.py

@@ -188,49 +188,11 @@ class McpRemoteTool(McpServerTool):
                         error_text = await response.text()
                         
                         # Check if this is a "Missing sessionId" error
-                        if response.status == 404 and "sessionId" in error_text and not self.session_id:
-                            logger.info(f"[MCP Session] Server requires session but none provided - attempting to initialize")
-                            # Try to initialize session with current auth headers
-                            from ..toolkits.mcp import McpToolkit
-                            from ..models.mcp_models import McpConnectionConfig
-                            
-                            connection_config = McpConnectionConfig(
-                                url=self.server_url,
-                                headers=self.server_headers
-                            )
-                            
-                            # Initialize session with auth headers
-                            session_id = McpToolkit._initialize_mcp_session(
-                                toolkit_name=self.server,
-                                connection_config=connection_config,
-                                timeout=self.tool_timeout_sec,
-                                extra_headers=headers  # Include Authorization header
-                            )
-                            
-                            if session_id:
-                                # Retry the request with session
-                                self.session_id = session_id
-                                # Update metadata so callbacks can access session info
-                                self._update_metadata_with_session()
-                                separator = '&' if '?' in self.server_url else '?'
-                                retry_url = f"{self.server_url}{separator}sessionId={session_id}"
-                                logger.info(f"[MCP Session] Retrying tool call with new session: {session_id}")
-                                logger.info(f"[MCP Session] Created session - server: {canonical_resource(self.server_url)}, session_id: {session_id}")
-                                
-                                async with session.post(retry_url, json=mcp_request, headers=headers) as retry_response:
-                                    if retry_response.status != 200:
-                                        retry_error = await retry_response.text()
-                                        raise Exception(f"HTTP {retry_response.status} (after session init): {retry_error}")
-                                    
-                                    # Process successful retry response
-                                    content_type = retry_response.headers.get('Content-Type', '')
-                                    if 'text/event-stream' in content_type:
-                                        text = await retry_response.text()
-                                        data = self._parse_sse(text)
-                                    else:
-                                        data = await retry_response.json()
-                            else:
-                                raise Exception(f"HTTP {response.status}: {error_text} (session initialization failed)")
+                        if response.status == 404 and "sessionId" in error_text:
+                            logger.error(f"[MCP Session] Server requires session but none provided")
+                            logger.error(f"[MCP Session] Frontend must generate a UUID and send it with mcp_tokens")
+                            logger.error(f"[MCP Session] Example: mcp_tokens = {{'server_url': {{'access_token': '...', 'session_id': crypto.randomUUID()}}}}")
+                            raise Exception(f"HTTP {response.status}: {error_text} - sessionId required but not provided by frontend")
                         else:
                             raise Exception(f"HTTP {response.status}: {error_text}")
                     

{alita_sdk-0.3.446.dist-info → alita_sdk-0.3.448.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.446
+Version: 0.3.448
 Summary: SDK for building langchain agents using resources from Alita
 Author-email: Artem Rozumenko <artyom.rozumenko@gmail.com>, Mikalai Biazruchka <mikalai_biazruchka@epam.com>, Roman Mitusov <roman_mitusov@epam.com>, Ivan Krakhmaliuk <lifedj27@gmail.com>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.446.dist-info → alita_sdk-0.3.448.dist-info}/RECORD

@@ -103,7 +103,7 @@ alita_sdk/runtime/toolkits/application.py,sha256=HHAKgwKOckxc7EQG-AV7rz4POOzQJKF
 alita_sdk/runtime/toolkits/artifact.py,sha256=YChNCX4QhVpaQG7Jk4TS-Wl0Aruc4slQ2K21zh9nNO0,3176
 alita_sdk/runtime/toolkits/configurations.py,sha256=kIDAlnryPQfbZyFxV-9SzN2-Vefzx06TX1BBdIIpN90,141
 alita_sdk/runtime/toolkits/datasource.py,sha256=qk78OdPoReYPCWwahfkKLbKc4pfsu-061oXRryFLP6I,2498
-alita_sdk/runtime/toolkits/mcp.py,sha256=bBfV-iOJ481SzMt14vMgs8BqZzzXwpa3111cGSB3810,46391
+alita_sdk/runtime/toolkits/mcp.py,sha256=2RpT00q6qKkXHPMkFN6wnqUHYGjYpDAmZ5dWokEXA_s,42172
 alita_sdk/runtime/toolkits/prompt.py,sha256=WIpTkkVYWqIqOWR_LlSWz3ug8uO9tm5jJ7aZYdiGRn0,1192
 alita_sdk/runtime/toolkits/subgraph.py,sha256=wwUK8JjPXkGzyVZ3tAukmvST6eGbqx_U11rpnmbrvtg,2105
 alita_sdk/runtime/toolkits/tools.py,sha256=74R0SadMiUJFUe3U5_PJcxzw5m1KMMBiVfVeiowZK6I,13434
@@ -122,7 +122,7 @@ alita_sdk/runtime/tools/llm.py,sha256=iRG_wU4T01LRsjEMPZe5Uah7LiMqDc-vspwkMuQtlt
 alita_sdk/runtime/tools/loop.py,sha256=uds0WhZvwMxDVFI6MZHrcmMle637cQfBNg682iLxoJA,8335
 alita_sdk/runtime/tools/loop_output.py,sha256=U4hO9PCQgWlXwOq6jdmCGbegtAxGAPXObSxZQ3z38uk,8069
 alita_sdk/runtime/tools/mcp_inspect_tool.py,sha256=38X8euaxDbEGjcfp6ElvExZalpZun6QEr6ZEW4nU5pQ,11496
-alita_sdk/runtime/tools/mcp_remote_tool.py,sha256=ypOJE7MhdPAQqIIjtirH4OzVNFbZeZuwy0Sw1H2CQJ0,15866
+alita_sdk/runtime/tools/mcp_remote_tool.py,sha256=0-wqqq2ZY9cgGeTd6ao17e393YgD99n3X-k2pXlaR2g,13352
 alita_sdk/runtime/tools/mcp_server_tool.py,sha256=-xO3H6BM63KaIV1CdcQKPVE0WPigiqOgFZDX7m2_yGs,4419
 alita_sdk/runtime/tools/pgvector_search.py,sha256=NN2BGAnq4SsDHIhUcFZ8d_dbEOM8QwB0UwpsWCYruXU,11692
 alita_sdk/runtime/tools/prompt.py,sha256=nJafb_e5aOM1Rr3qGFCR-SKziU9uCsiP2okIMs9PppM,741
@@ -360,8 +360,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=kT0TbmMvuKhDUZc0i7KO18O38JM9S
 alita_sdk/tools/zephyr_squad/__init__.py,sha256=0ne8XLJEQSLOWfzd2HdnqOYmQlUliKHbBED5kW_Vias,2895
 alita_sdk/tools/zephyr_squad/api_wrapper.py,sha256=kmw_xol8YIYFplBLWTqP_VKPRhL_1ItDD0_vXTe_UuI,14906
 alita_sdk/tools/zephyr_squad/zephyr_squad_cloud_client.py,sha256=R371waHsms4sllHCbijKYs90C-9Yu0sSR3N4SUfQOgU,5066
-alita_sdk-0.3.446.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.446.dist-info/METADATA,sha256=akIEnIoOfAXzb6sKj1ef-dWGEr3ySjKbdQhNYkdHNdE,19071
-alita_sdk-0.3.446.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.446.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.446.dist-info/RECORD,,
+alita_sdk-0.3.448.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.448.dist-info/METADATA,sha256=FsoPHn8uhMeNXFvo4DkRBzyweY-GNMGH_-xJtnAVX2c,19071
+alita_sdk-0.3.448.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.448.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.448.dist-info/RECORD,,