mbxai 1.0.2__tar.gz → 1.0.4__tar.gz

{mbxai-1.0.2 → mbxai-1.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mbxai
-Version: 1.0.2
+Version: 1.0.4
 Summary: MBX AI SDK
 Project-URL: Homepage, https://www.mibexx.de
 Project-URL: Documentation, https://www.mibexx.de

{mbxai-1.0.2 → mbxai-1.0.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "mbxai"
-version = "1.0.2"
+version = "1.0.4"
 authors = [
     { name = "MBX AI" }
 ]

{mbxai-1.0.2 → mbxai-1.0.4}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="mbxai",
-    version="1.0.2",
+    version="1.0.4",
     author="MBX AI",
     description="MBX AI SDK",
     long_description=open("README.md").read(),

{mbxai-1.0.2 → mbxai-1.0.4}/src/mbxai/__init__.py

@@ -2,4 +2,4 @@
 MBX AI package.
 """
 
-__version__ = "1.0.2" 
+__version__ = "1.0.4" 

{mbxai-1.0.2 → mbxai-1.0.4}/src/mbxai/mcp/server.py

@@ -31,7 +31,7 @@ class MCPServer:
         self.app = FastAPI(
             title=self.name,
             description=self.description,
-            version="1.0.2",
+            version="1.0.4",
         )
         
         # Initialize MCP server

{mbxai-1.0.2 → mbxai-1.0.4}/src/mbxai/openrouter/client.py

@@ -268,6 +268,8 @@ class OpenRouterClient:
             
             response = self._client.beta.chat.completions.parse(**request)
             
+            logger.debug(f"Full Response: {response}")
+            
             if response is None:
                 logger.error("Received None response from OpenRouter API")
                 raise OpenRouterAPIError("Received None response from OpenRouter API")
@@ -275,10 +277,38 @@ class OpenRouterClient:
             logger.info(f"Response type: {type(response)}")
             logger.info(f"Response attributes: {dir(response)}")
             
+            # Validate response structure
             if not hasattr(response, 'choices'):
                 logger.error(f"Response missing 'choices' attribute. Available attributes: {dir(response)}")
                 raise OpenRouterAPIError("Invalid response format: missing 'choices' attribute")
-                
+            
+            if not response.choices:
+                logger.error("Response has empty choices list")
+                raise OpenRouterAPIError("Invalid response format: empty choices list")
+            
+            if not hasattr(response.choices[0], 'message'):
+                logger.error(f"First choice missing 'message' attribute. Available attributes: {dir(response.choices[0])}")
+                raise OpenRouterAPIError("Invalid response format: missing 'message' attribute in first choice")
+            
+            # Check if the message has a parsed attribute or content
+            if not hasattr(response.choices[0].message, 'parsed') and not hasattr(response.choices[0].message, 'content'):
+                logger.error(f"Message missing both 'parsed' and 'content' attributes. Available attributes: {dir(response.choices[0].message)}")
+                raise OpenRouterAPIError("Invalid response format: message must have either 'parsed' or 'content' attribute")
+            
+            # If there's no parsed attribute but there is content, try to parse it
+            if not hasattr(response.choices[0].message, 'parsed') and hasattr(response.choices[0].message, 'content'):
+                try:
+                    import json
+                    content = response.choices[0].message.content
+                    if isinstance(content, str):
+                        parsed = json.loads(content)
+                        response.choices[0].message.parsed = parsed
+                    else:
+                        response.choices[0].message.parsed = content
+                except Exception as e:
+                    logger.error(f"Failed to parse message content: {str(e)}")
+                    raise OpenRouterAPIError(f"Failed to parse message content: {str(e)}")
+            
             logger.info(f"Received response from OpenRouter: {len(response.choices)} choices")
             
             return response