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

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

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

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