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

alita_sdk/runtime/toolkits/mcp.py

@@ -346,7 +346,20 @@ class McpToolkit(BaseToolkit):
             if response.status_code != 200:
                 raise Exception(f"HTTP {response.status_code}: {response.text}")
 
-            data = response.json()
+            # Check content type and parse accordingly
+            content_type = response.headers.get('Content-Type', '')
+            
+            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']}")
@@ -361,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

@@ -65,21 +65,17 @@ class McpInspectTool(BaseTool):
     def _run(self, resource_type: str = "all") -> str:
         """Inspect the MCP server for available resources."""
         try:
-            # Run the async inspection
-            loop = asyncio.get_event_loop()
-            if loop.is_running():
-                # If we're in an async context, we need to create a new event loop
-                import concurrent.futures
-                with concurrent.futures.ThreadPoolExecutor() as executor:
-                    future = executor.submit(self._run_async_inspection, resource_type)
-                    return future.result(timeout=self.timeout)
-            else:
-                return loop.run_until_complete(self._run_async_inspection(resource_type))
+            # Always create a new event loop for sync context
+            # This avoids issues with existing event loops in threads
+            import concurrent.futures
+            with concurrent.futures.ThreadPoolExecutor() as executor:
+                future = executor.submit(self._run_in_new_loop, resource_type)
+                return future.result(timeout=self.timeout)
         except Exception as e:
             logger.error(f"Error inspecting MCP server '{self.server_name}': {e}")
             return f"Error inspecting MCP server: {e}"
 
-    def _run_async_inspection(self, resource_type: str) -> str:
+    def _run_in_new_loop(self, resource_type: str) -> str:
         """Run the async inspection in a new event loop."""
         return asyncio.run(self._inspect_server(resource_type))
 
@@ -123,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 = {
@@ -132,13 +137,24 @@ class McpInspectTool(BaseTool):
             "params": {}
         }
 
-        headers = {"Content-Type": "application/json", **self.server_headers}
+        headers = {
+            "Content-Type": "application/json",
+            "Accept": "application/json, text/event-stream",
+            **self.server_headers
+        }
 
         async with session.post(self.server_url, json=request, headers=headers) as response:
             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']}")
@@ -154,13 +170,23 @@ class McpInspectTool(BaseTool):
             "params": {}
         }
 
-        headers = {"Content-Type": "application/json", **self.server_headers}
+        headers = {
+            "Content-Type": "application/json",
+            "Accept": "application/json, text/event-stream",
+            **self.server_headers
+        }
 
         async with session.post(self.server_url, json=request, headers=headers) as response:
             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']}")
@@ -176,13 +202,23 @@ class McpInspectTool(BaseTool):
             "params": {}
         }
 
-        headers = {"Content-Type": "application/json", **self.server_headers}
+        headers = {
+            "Content-Type": "application/json",
+            "Accept": "application/json, text/event-stream",
+            **self.server_headers
+        }
 
         async with session.post(self.server_url, json=request, headers=headers) as response:
             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.431.dist-info → alita_sdk-0.3.433.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.431
+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.431.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=BcXNhsAs92lgu-ApMu__C6COc60CR9OqkuAviPTPVCw,25256
+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=EW81_XlXX5C4lxT-gHMehHcZhtbuG-8DQzo0NuDkCA4,10160
+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.431.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.431.dist-info/METADATA,sha256=x-p6VpEH5cOUrBEOW-cES2TON6nVbBSuQny5aBsWTAQ,19071
-alita_sdk-0.3.431.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.431.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.431.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,,