alita-sdk 0.3.432__py3-none-any.whl → 0.3.433__py3-none-any.whl

alita_sdk/runtime/toolkits/mcp.py

@@ -346,15 +346,20 @@ class McpToolkit(BaseToolkit):
             if response.status_code != 200:
                 raise Exception(f"HTTP {response.status_code}: {response.text}")
 
-            # Check if response is actually JSON
+            # Check content type and parse accordingly
             content_type = response.headers.get('Content-Type', '')
-            if 'application/json' not in content_type:
-                raise Exception(f"Expected JSON response but got Content-Type: {content_type}. Response: {response.text[:200]}")
-
-            try:
-                data = response.json()
-            except ValueError as json_err:
-                raise Exception(f"Invalid JSON response: {json_err}. Response text: {response.text[:200]}")
+            
+            if 'text/event-stream' in content_type:
+                # Parse SSE (Server-Sent Events) format
+                data = cls._parse_sse_response(response.text)
+            elif 'application/json' in content_type:
+                # Parse regular JSON
+                try:
+                    data = response.json()
+                except ValueError as json_err:
+                    raise Exception(f"Invalid JSON response: {json_err}. Response text: {response.text[:200]}")
+            else:
+                raise Exception(f"Unexpected Content-Type: {content_type}. Response: {response.text[:200]}")
 
             if "error" in data:
                 raise Exception(f"MCP Error: {data['error']}")
@@ -369,6 +374,30 @@ class McpToolkit(BaseToolkit):
             logger.error(f"Failed to discover tools from MCP toolkit '{toolkit_name}': {e}")
             raise
 
+    @staticmethod
+    def _parse_sse_response(sse_text: str) -> Dict[str, Any]:
+        """
+        Parse Server-Sent Events (SSE) format response.
+        SSE format: event: message\ndata: {json}\n\n
+        """
+        import json
+        
+        lines = sse_text.strip().split('\n')
+        data_line = None
+        
+        for line in lines:
+            if line.startswith('data:'):
+                data_line = line[5:].strip()  # Remove 'data:' prefix
+                break
+        
+        if not data_line:
+            raise Exception(f"No data found in SSE response: {sse_text[:200]}")
+        
+        try:
+            return json.loads(data_line)
+        except json.JSONDecodeError as e:
+            raise Exception(f"Failed to parse SSE data as JSON: {e}. Data: {data_line[:200]}")
+
     @classmethod
     def _create_tool_from_dict(
         cls,

alita_sdk/runtime/tools/mcp_inspect_tool.py

@@ -119,6 +119,15 @@ class McpInspectTool(BaseTool):
 
         return self._format_results(results, resource_type)
 
+    def _parse_sse(self, text: str) -> Dict[str, Any]:
+        """Parse Server-Sent Events (SSE) format response."""
+        for line in text.split('\n'):
+            line = line.strip()
+            if line.startswith('data:'):
+                json_str = line[5:].strip()
+                return json.loads(json_str)
+        raise ValueError("No data found in SSE response")
+
     async def _list_tools(self, session: aiohttp.ClientSession) -> Dict[str, Any]:
         """List available tools from the MCP server."""
         request = {
@@ -138,7 +147,14 @@ class McpInspectTool(BaseTool):
             if response.status != 200:
                 raise Exception(f"HTTP {response.status}: {await response.text()}")
 
-            data = await response.json()
+            # Handle both JSON and SSE responses
+            content_type = response.headers.get('Content-Type', '')
+            if 'text/event-stream' in content_type:
+                # Parse SSE format
+                text = await response.text()
+                data = self._parse_sse(text)
+            else:
+                data = await response.json()
 
             if "error" in data:
                 raise Exception(f"MCP Error: {data['error']}")
@@ -164,7 +180,13 @@ class McpInspectTool(BaseTool):
             if response.status != 200:
                 raise Exception(f"HTTP {response.status}: {await response.text()}")
 
-            data = await response.json()
+            # Handle both JSON and SSE responses
+            content_type = response.headers.get('Content-Type', '')
+            if 'text/event-stream' in content_type:
+                text = await response.text()
+                data = self._parse_sse(text)
+            else:
+                data = await response.json()
 
             if "error" in data:
                 raise Exception(f"MCP Error: {data['error']}")
@@ -190,7 +212,13 @@ class McpInspectTool(BaseTool):
             if response.status != 200:
                 raise Exception(f"HTTP {response.status}: {await response.text()}")
 
-            data = await response.json()
+            # Handle both JSON and SSE responses
+            content_type = response.headers.get('Content-Type', '')
+            if 'text/event-stream' in content_type:
+                text = await response.text()
+                data = self._parse_sse(text)
+            else:
+                data = await response.json()
 
             if "error" in data:
                 raise Exception(f"MCP Error: {data['error']}")

{alita_sdk-0.3.432.dist-info → alita_sdk-0.3.433.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.432
+Version: 0.3.433
 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.432.dist-info → alita_sdk-0.3.433.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=lh1REOY680Q3yP4DfmzFLllmJIZy1IM90NZd1CwLrDY,25730
+alita_sdk/runtime/toolkits/mcp.py,sha256=5fdmY5s5lfw4NieovFYr0C9STiCHGFuOtAZ1uFKaj0U,26764
 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=YCTjrTJuwj2V2C8ZQqXhFvUbVr7NQcUHZlCQLLvoeGA,10946
@@ -121,7 +121,7 @@ alita_sdk/runtime/tools/indexer_tool.py,sha256=whSLPevB4WD6dhh2JDXEivDmTvbjiMV1M
 alita_sdk/runtime/tools/llm.py,sha256=iRG_wU4T01LRsjEMPZe5Uah7LiMqDc-vspwkMuQtltk,16136
 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=wgnv0z9MvB-PWriGKHHpY2IMxjBkd7Cbz4kvwGp9aq8,10243
+alita_sdk/runtime/tools/mcp_inspect_tool.py,sha256=38X8euaxDbEGjcfp6ElvExZalpZun6QEr6ZEW4nU5pQ,11496
 alita_sdk/runtime/tools/mcp_server_tool.py,sha256=y6y0j3JeUajNatpsLVJfP0JEknhlRcDRJ0PlJ3Ch8fA,6626
 alita_sdk/runtime/tools/pgvector_search.py,sha256=NN2BGAnq4SsDHIhUcFZ8d_dbEOM8QwB0UwpsWCYruXU,11692
 alita_sdk/runtime/tools/prompt.py,sha256=nJafb_e5aOM1Rr3qGFCR-SKziU9uCsiP2okIMs9PppM,741
@@ -358,8 +358,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.432.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.432.dist-info/METADATA,sha256=5ta9EQvl3qdFIGaqQCAQLW9VWo7PYJRHdNmffxCbdK8,19071
-alita_sdk-0.3.432.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.432.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.432.dist-info/RECORD,,
+alita_sdk-0.3.433.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.433.dist-info/METADATA,sha256=EIfa8_iHpAyNV8f5hADnzfRQEKDglXU7--aHZ8qxcxU,19071
+alita_sdk-0.3.433.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.433.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.433.dist-info/RECORD,,