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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
mbxai/__init__.py CHANGED
@@ -2,4 +2,4 @@
2
2
  MBX AI package.
3
3
  """
4
4
 
5
- __version__ = "0.5.14"
5
+ __version__ = "0.5.15"
mbxai/mcp/server.py CHANGED
@@ -31,7 +31,7 @@ class MCPServer:
31
31
  self.app = FastAPI(
32
32
  title=self.name,
33
33
  description=self.description,
34
- version="0.5.14",
34
+ version="0.5.15",
35
35
  )
36
36
 
37
37
  # Initialize MCP server
mbxai/tools/client.py CHANGED
@@ -58,6 +58,20 @@ class ToolClient:
58
58
  return content
59
59
  return content[:max_length] + "..."
60
60
 
61
+ def _truncate_dict(self, data: dict[str, Any], max_length: int = 50) -> str:
62
+ """Truncate dictionary values for logging."""
63
+ if not data:
64
+ return "{}"
65
+ truncated = {}
66
+ for k, v in data.items():
67
+ if isinstance(v, str):
68
+ truncated[k] = self._truncate_content(v, max_length)
69
+ elif isinstance(v, dict):
70
+ truncated[k] = self._truncate_dict(v, max_length)
71
+ else:
72
+ truncated[k] = str(v)[:max_length] + "..." if len(str(v)) > max_length else v
73
+ return str(truncated)
74
+
61
75
  def _log_messages(self, messages: list[dict[str, Any]]) -> None:
62
76
  """Log the messages being sent to OpenRouter."""
63
77
  logger.info("Sending messages to OpenRouter:")
@@ -167,7 +181,6 @@ class ToolClient:
167
181
  return response
168
182
 
169
183
  # Handle all tool calls before getting the next model response
170
- tool_responses = []
171
184
  for tool_call in message.tool_calls:
172
185
  tool = self._tools.get(tool_call.function.name)
173
186
  if not tool:
@@ -183,7 +196,7 @@ class ToolClient:
183
196
  raise ValueError(f"Invalid tool arguments format: {arguments}")
184
197
 
185
198
  # Call the tool
186
- logger.info(f"Calling tool: {tool.name} with args: {self._truncate_content(json.dumps(arguments))}")
199
+ logger.info(f"Calling tool: {tool.name} with args: {self._truncate_dict(arguments)}")
187
200
  if inspect.iscoroutinefunction(tool.function):
188
201
  result = await tool.function(**arguments)
189
202
  else:
@@ -193,27 +206,15 @@ class ToolClient:
193
206
  if not isinstance(result, str):
194
207
  result = json.dumps(result)
195
208
 
196
- # Create the tool response
209
+ # Create and add the tool response immediately
197
210
  tool_response = {
198
211
  "role": "tool",
199
212
  "tool_call_id": tool_call.id,
200
213
  "content": result,
201
214
  }
202
- tool_responses.append(tool_response)
215
+ messages.append(tool_response)
203
216
  logger.info(f"Tool response for call ID {tool_call.id}: {self._truncate_content(result)}")
204
217
 
205
- # Add all tool responses to the messages
206
- messages.extend(tool_responses)
207
- logger.info(f"Added {len(tool_responses)} tool responses to messages")
208
-
209
- # Validate that all tool calls have responses
210
- tool_call_ids = {tc.id for tc in message.tool_calls}
211
- tool_response_ids = {tr["tool_call_id"] for tr in tool_responses}
212
- missing_responses = tool_call_ids - tool_response_ids
213
- if missing_responses:
214
- logger.error(f"Missing tool responses for call IDs: {missing_responses}")
215
- raise ValueError(f"Missing tool responses for call IDs: {missing_responses}")
216
-
217
218
  # Get a new response from the model with all tool results
218
219
  response = self._client.chat_completion(
219
220
  messages=messages,
@@ -336,7 +337,7 @@ class ToolClient:
336
337
  raise ValueError(f"Invalid tool arguments format: {arguments}")
337
338
 
338
339
  # Call the tool
339
- logger.info(f"Calling tool: {tool.name} with args: {self._truncate_content(json.dumps(arguments))}")
340
+ logger.info(f"Calling tool: {tool.name} with args: {self._truncate_dict(arguments)}")
340
341
  if inspect.iscoroutinefunction(tool.function):
341
342
  result = await tool.function(**arguments)
342
343
  else:
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: mbxai
3
- Version: 0.5.14
3
+ Version: 0.5.15
4
4
  Summary: MBX AI SDK
5
5
  Project-URL: Homepage, https://www.mibexx.de
6
6
  Project-URL: Documentation, https://www.mibexx.de
@@ -1,18 +1,18 @@
1
- mbxai/__init__.py,sha256=dffyFNDcTA_0NDroPZIJc7T7242wliyUpX1i47_bBmQ,48
1
+ mbxai/__init__.py,sha256=bRHyO3paNdo9cp-TPVr7zZ4cW-RvHzQm5rwAp-WDYb0,48
2
2
  mbxai/core.py,sha256=WMvmU9TTa7M_m-qWsUew4xH8Ul6xseCZ2iBCXJTW-Bs,196
3
3
  mbxai/mcp/__init__.py,sha256=_ek9iYdYqW5saKetj4qDci11jxesQDiHPJRpHMKkxgU,175
4
4
  mbxai/mcp/client.py,sha256=hEAVWIrIq758C1zm9aWGf-FiITB3LxtuxZEZ0CcjJ4s,5343
5
5
  mbxai/mcp/example.py,sha256=oaol7AvvZnX86JWNz64KvPjab5gg1VjVN3G8eFSzuaE,2350
6
- mbxai/mcp/server.py,sha256=1kmpDlay-9PQQ4-AXtBpCdyVJKWooYhTBJI-QGbWUXc,3463
6
+ mbxai/mcp/server.py,sha256=CMcgw3FKAZKmiKYfy09-Gs9vTGA9kWGR-O5nGw4F_Xo,3463
7
7
  mbxai/openrouter/__init__.py,sha256=Ito9Qp_B6q-RLGAQcYyTJVWwR2YAZvNqE-HIYXxhtD8,298
8
8
  mbxai/openrouter/client.py,sha256=XLRMRNRJH96Jl6_af0KkzRDdLJnixh8I3RvEEcFuXyg,10840
9
9
  mbxai/openrouter/config.py,sha256=MTX_YHsFrM7JYqovJSkEF6JzVyIdajeI5Dja2CALH58,2874
10
10
  mbxai/openrouter/models.py,sha256=b3IjjtZAjeGOf2rLsdnCD1HacjTnS8jmv_ZXorc-KJQ,2604
11
11
  mbxai/tools/__init__.py,sha256=QUFaXhDm-UKcuAtT1rbKzhBkvyRBVokcQIOf9cxIuwc,160
12
- mbxai/tools/client.py,sha256=ag6gTDPfY_M1Mrf1iUaBX4HLCcsFKYHDiGaukj940Jk,15613
12
+ mbxai/tools/client.py,sha256=jkP7HLlr-0SVklNnQRU9A6XX8-_ObixplNjheWlfZwY,15486
13
13
  mbxai/tools/example.py,sha256=1HgKK39zzUuwFbnp3f0ThyWVfA_8P28PZcTwaUw5K78,2232
14
14
  mbxai/tools/types.py,sha256=fo5t9UbsHGynhA88vD_ecgDqL8iLvt2E1h1ym43Rrgk,745
15
- mbxai-0.5.14.dist-info/METADATA,sha256=VkZHG5kjh8_dy_P2s1JmZnuwzlQgZCoMVVcx5y40zXM,4108
16
- mbxai-0.5.14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
17
- mbxai-0.5.14.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
18
- mbxai-0.5.14.dist-info/RECORD,,
15
+ mbxai-0.5.15.dist-info/METADATA,sha256=SE49yR-ukNm-dhRxwXrp1xyQYktQ7mZMWZ1O_hvDESU,4108
16
+ mbxai-0.5.15.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
17
+ mbxai-0.5.15.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
18
+ mbxai-0.5.15.dist-info/RECORD,,
File without changes