mbxai 0.5.12__py3-none-any.whl → 0.5.14__py3-none-any.whl

mbxai/__init__.py

@@ -2,4 +2,4 @@
 MBX AI package.
 """
 
-__version__ = "0.5.12" 
+__version__ = "0.5.14" 

mbxai/mcp/server.py

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

mbxai/tools/client.py

@@ -48,7 +48,57 @@ class ToolClient:
             schema=schema,
         )
         self._tools[name] = tool
-        logger.debug(f"Registered tool: {name}")
+        logger.info(f"Registered tool: {name}")
+
+    def _truncate_content(self, content: str | None, max_length: int = 100) -> str:
+        """Truncate content for logging."""
+        if not content:
+            return "None"
+        if len(content) <= max_length:
+            return content
+        return content[:max_length] + "..."
+
+    def _log_messages(self, messages: list[dict[str, Any]]) -> None:
+        """Log the messages being sent to OpenRouter."""
+        logger.info("Sending messages to OpenRouter:")
+        for msg in messages:
+            role = msg.get("role", "unknown")
+            content = self._truncate_content(msg.get("content"))
+            tool_calls = msg.get("tool_calls", [])
+            tool_call_id = msg.get("tool_call_id")
+            
+            if tool_calls:
+                tool_call_info = [
+                    f"{tc['function']['name']}(id={tc['id']})"
+                    for tc in tool_calls
+                ]
+                logger.info(f"  {role}: content='{content}', tool_calls={tool_call_info}")
+            elif tool_call_id:
+                logger.info(f"  {role}: content='{content}', tool_call_id={tool_call_id}")
+            else:
+                logger.info(f"  {role}: content='{content}'")
+
+        # Validate tool call responses
+        tool_call_ids = set()
+        tool_response_ids = set()
+        
+        for msg in messages:
+            if msg.get("role") == "assistant" and "tool_calls" in msg:
+                for tc in msg["tool_calls"]:
+                    tool_call_ids.add(tc["id"])
+            elif msg.get("role") == "tool":
+                tool_response_ids.add(msg["tool_call_id"])
+        
+        missing_responses = tool_call_ids - tool_response_ids
+        if missing_responses:
+            logger.error(f"Missing tool responses for call IDs: {missing_responses}")
+            logger.error("Message sequence:")
+            for msg in messages:
+                role = msg.get("role", "unknown")
+                if role == "assistant" and "tool_calls" in msg:
+                    logger.error(f"  Assistant message with tool calls: {[tc['id'] for tc in msg['tool_calls']]}")
+                elif role == "tool":
+                    logger.error(f"  Tool response for call ID: {msg['tool_call_id']}")
 
     async def chat(
         self,
@@ -72,11 +122,14 @@ class ToolClient:
         tools = [tool.to_openai_function() for tool in self._tools.values()]
         
         if tools:
-            logger.debug(f"Using tools: {tools}")
+            logger.info(f"Available tools: {[tool['function']['name'] for tool in tools]}")
             kwargs["tools"] = tools
             kwargs["tool_choice"] = "auto"
         
         while True:
+            # Log messages before sending to OpenRouter
+            self._log_messages(messages)
+            
             # Get the model's response
             response = self._client.chat_completion(
                 messages=messages,
@@ -107,7 +160,7 @@ class ToolClient:
                     for tool_call in message.tool_calls
                 ]
             messages.append(assistant_message)
-            logger.debug(f"Added assistant message: {assistant_message}")
+            logger.info(f"Assistant message: content='{self._truncate_content(message.content)}', tool_calls={[tc.function.name for tc in message.tool_calls] if message.tool_calls else None}")
 
             # If there are no tool calls, we're done
             if not message.tool_calls:
@@ -116,35 +169,25 @@ class ToolClient:
             # Handle all tool calls before getting the next model response
             tool_responses = []
             for tool_call in message.tool_calls:
-                logger.debug(f"Processing tool call: {tool_call}")
-                logger.debug(f"Tool call ID: {tool_call.id}")
-                logger.debug(f"Tool call function: {tool_call.function}")
-                logger.debug(f"Tool call arguments: {tool_call.function.arguments}")
-
                 tool = self._tools.get(tool_call.function.name)
                 if not tool:
                     raise ValueError(f"Unknown tool: {tool_call.function.name}")
 
                 # Parse arguments if they're a string
                 arguments = tool_call.function.arguments
-                logger.debug(f"Raw arguments type: {type(arguments)}")
-                logger.debug(f"Raw arguments: {arguments}")
-
                 if isinstance(arguments, str):
                     try:
                         arguments = json.loads(arguments)
-                        logger.debug(f"Parsed arguments: {arguments}")
                     except json.JSONDecodeError as e:
                         logger.error(f"Failed to parse tool arguments: {e}")
                         raise ValueError(f"Invalid tool arguments format: {arguments}")
 
                 # Call the tool
-                logger.debug(f"Calling tool {tool.name} with arguments: {arguments}")
+                logger.info(f"Calling tool: {tool.name} with args: {self._truncate_content(json.dumps(arguments))}")
                 if inspect.iscoroutinefunction(tool.function):
                     result = await tool.function(**arguments)
                 else:
                     result = tool.function(**arguments)
-                logger.debug(f"Tool result: {result}")
 
                 # Convert result to JSON string if it's not already
                 if not isinstance(result, str):
@@ -157,11 +200,19 @@ class ToolClient:
                     "content": result,
                 }
                 tool_responses.append(tool_response)
-                logger.debug(f"Created tool response: {tool_response}")
+                logger.info(f"Tool response for call ID {tool_call.id}: {self._truncate_content(result)}")
 
             # Add all tool responses to the messages
             messages.extend(tool_responses)
-            logger.debug(f"Added all tool responses to messages: {tool_responses}")
+            logger.info(f"Added {len(tool_responses)} tool responses to messages")
+
+            # Validate that all tool calls have responses
+            tool_call_ids = {tc.id for tc in message.tool_calls}
+            tool_response_ids = {tr["tool_call_id"] for tr in tool_responses}
+            missing_responses = tool_call_ids - tool_response_ids
+            if missing_responses:
+                logger.error(f"Missing tool responses for call IDs: {missing_responses}")
+                raise ValueError(f"Missing tool responses for call IDs: {missing_responses}")
 
             # Get a new response from the model with all tool results
             response = self._client.chat_completion(
@@ -193,7 +244,7 @@ class ToolClient:
                     for tool_call in message.tool_calls
                 ]
             messages.append(assistant_message)
-            logger.debug(f"Added assistant message: {assistant_message}")
+            logger.info(f"Assistant message: content='{self._truncate_content(message.content)}', tool_calls={[tc.function.name for tc in message.tool_calls] if message.tool_calls else None}")
 
             # If there are no more tool calls, we're done
             if not message.tool_calls:
@@ -223,11 +274,14 @@ class ToolClient:
         tools = [tool.to_openai_function() for tool in self._tools.values()]
         
         if tools:
-            logger.debug(f"Using tools: {tools}")
+            logger.info(f"Available tools: {[tool['function']['name'] for tool in tools]}")
             kwargs["tools"] = tools
             kwargs["tool_choice"] = "auto"
         
         while True:
+            # Log messages before sending to OpenRouter
+            self._log_messages(messages)
+            
             # Get the model's response
             response = self._client.chat_completion_parse(
                 messages=messages,
@@ -259,7 +313,7 @@ class ToolClient:
                     for tool_call in message.tool_calls
                 ]
             messages.append(assistant_message)
-            logger.debug(f"Added assistant message: {assistant_message}")
+            logger.info(f"Assistant message: content='{self._truncate_content(message.content)}', tool_calls={[tc.function.name for tc in message.tool_calls] if message.tool_calls else None}")
 
             # If there are no tool calls, we're done
             if not message.tool_calls:
@@ -268,35 +322,25 @@ class ToolClient:
             # Handle all tool calls before getting the next model response
             tool_responses = []
             for tool_call in message.tool_calls:
-                logger.debug(f"Processing tool call: {tool_call}")
-                logger.debug(f"Tool call ID: {tool_call.id}")
-                logger.debug(f"Tool call function: {tool_call.function}")
-                logger.debug(f"Tool call arguments: {tool_call.function.arguments}")
-
                 tool = self._tools.get(tool_call.function.name)
                 if not tool:
                     raise ValueError(f"Unknown tool: {tool_call.function.name}")
 
                 # Parse arguments if they're a string
                 arguments = tool_call.function.arguments
-                logger.debug(f"Raw arguments type: {type(arguments)}")
-                logger.debug(f"Raw arguments: {arguments}")
-
                 if isinstance(arguments, str):
                     try:
                         arguments = json.loads(arguments)
-                        logger.debug(f"Parsed arguments: {arguments}")
                     except json.JSONDecodeError as e:
                         logger.error(f"Failed to parse tool arguments: {e}")
                         raise ValueError(f"Invalid tool arguments format: {arguments}")
 
                 # Call the tool
-                logger.debug(f"Calling tool {tool.name} with arguments: {arguments}")
+                logger.info(f"Calling tool: {tool.name} with args: {self._truncate_content(json.dumps(arguments))}")
                 if inspect.iscoroutinefunction(tool.function):
                     result = await tool.function(**arguments)
                 else:
                     result = tool.function(**arguments)
-                logger.debug(f"Tool result: {result}")
 
                 # Convert result to JSON string if it's not already
                 if not isinstance(result, str):
@@ -309,11 +353,10 @@ class ToolClient:
                     "content": result,
                 }
                 tool_responses.append(tool_response)
-                logger.debug(f"Created tool response: {tool_response}")
+                logger.info(f"Tool response: {self._truncate_content(result)}")
 
             # Add all tool responses to the messages
             messages.extend(tool_responses)
-            logger.debug(f"Added all tool responses to messages: {tool_responses}")
 
             # Get a new response from the model with all tool results
             response = self._client.chat_completion_parse(
@@ -346,7 +389,7 @@ class ToolClient:
                     for tool_call in message.tool_calls
                 ]
             messages.append(assistant_message)
-            logger.debug(f"Added assistant message: {assistant_message}")
+            logger.info(f"Assistant message: content='{self._truncate_content(message.content)}', tool_calls={[tc.function.name for tc in message.tool_calls] if message.tool_calls else None}")
 
             # If there are no more tool calls, we're done
             if not message.tool_calls:

{mbxai-0.5.12.dist-info → mbxai-0.5.14.dist-info}/METADATA

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

{mbxai-0.5.12.dist-info → mbxai-0.5.14.dist-info}/RECORD

@@ -1,18 +1,18 @@
-mbxai/__init__.py,sha256=IZikFZwxYq2ZzCj7jfjGct1is4gCcMAc590whjAThy4,48
+mbxai/__init__.py,sha256=dffyFNDcTA_0NDroPZIJc7T7242wliyUpX1i47_bBmQ,48
 mbxai/core.py,sha256=WMvmU9TTa7M_m-qWsUew4xH8Ul6xseCZ2iBCXJTW-Bs,196
 mbxai/mcp/__init__.py,sha256=_ek9iYdYqW5saKetj4qDci11jxesQDiHPJRpHMKkxgU,175
 mbxai/mcp/client.py,sha256=hEAVWIrIq758C1zm9aWGf-FiITB3LxtuxZEZ0CcjJ4s,5343
 mbxai/mcp/example.py,sha256=oaol7AvvZnX86JWNz64KvPjab5gg1VjVN3G8eFSzuaE,2350
-mbxai/mcp/server.py,sha256=MXGh1jn4tKLY49fAs_5rTaVMh1GT-azvwyXGHIDoSRg,3463
+mbxai/mcp/server.py,sha256=1kmpDlay-9PQQ4-AXtBpCdyVJKWooYhTBJI-QGbWUXc,3463
 mbxai/openrouter/__init__.py,sha256=Ito9Qp_B6q-RLGAQcYyTJVWwR2YAZvNqE-HIYXxhtD8,298
 mbxai/openrouter/client.py,sha256=XLRMRNRJH96Jl6_af0KkzRDdLJnixh8I3RvEEcFuXyg,10840
 mbxai/openrouter/config.py,sha256=MTX_YHsFrM7JYqovJSkEF6JzVyIdajeI5Dja2CALH58,2874
 mbxai/openrouter/models.py,sha256=b3IjjtZAjeGOf2rLsdnCD1HacjTnS8jmv_ZXorc-KJQ,2604
 mbxai/tools/__init__.py,sha256=QUFaXhDm-UKcuAtT1rbKzhBkvyRBVokcQIOf9cxIuwc,160
-mbxai/tools/client.py,sha256=8aTrRSljnjrKIaBzLspSc8OpqRar6LZn__SVNOBpgHM,13190
+mbxai/tools/client.py,sha256=ag6gTDPfY_M1Mrf1iUaBX4HLCcsFKYHDiGaukj940Jk,15613
 mbxai/tools/example.py,sha256=1HgKK39zzUuwFbnp3f0ThyWVfA_8P28PZcTwaUw5K78,2232
 mbxai/tools/types.py,sha256=fo5t9UbsHGynhA88vD_ecgDqL8iLvt2E1h1ym43Rrgk,745
-mbxai-0.5.12.dist-info/METADATA,sha256=mga6ghbaA3DUBmlVAAGAyHiACU--F6dgiR038ausnec,4108
-mbxai-0.5.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mbxai-0.5.12.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
-mbxai-0.5.12.dist-info/RECORD,,
+mbxai-0.5.14.dist-info/METADATA,sha256=VkZHG5kjh8_dy_P2s1JmZnuwzlQgZCoMVVcx5y40zXM,4108
+mbxai-0.5.14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mbxai-0.5.14.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
+mbxai-0.5.14.dist-info/RECORD,,