alita-sdk 0.3.431__tar.gz → 0.3.433__tar.gz

{alita_sdk-0.3.431 → alita_sdk-0.3.433}/PKG-INFO

@@ -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 → alita_sdk-0.3.433}/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-0.3.431 → alita_sdk-0.3.433}/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 → alita_sdk-0.3.433}/alita_sdk.egg-info/PKG-INFO

@@ -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 → alita_sdk-0.3.433}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "alita_sdk"
-version = "0.3.431"
+version = "0.3.433"
 description = "SDK for building langchain agents using resources from Alita"
 readme = "README.md"
 requires-python = ">=3.10"