mbxai 0.5.13__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 +1 -1
- mbxai/mcp/server.py +1 -1
- mbxai/tools/client.py +18 -8
- {mbxai-0.5.13.dist-info → mbxai-0.5.15.dist-info}/METADATA +1 -1
- {mbxai-0.5.13.dist-info → mbxai-0.5.15.dist-info}/RECORD +7 -7
- {mbxai-0.5.13.dist-info → mbxai-0.5.15.dist-info}/WHEEL +0 -0
- {mbxai-0.5.13.dist-info → mbxai-0.5.15.dist-info}/licenses/LICENSE +0 -0
mbxai/__init__.py
CHANGED
mbxai/mcp/server.py
CHANGED
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.
|
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,18 +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
|
-
|
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
|
-
|
208
218
|
# Get a new response from the model with all tool results
|
209
219
|
response = self._client.chat_completion(
|
210
220
|
messages=messages,
|
@@ -327,7 +337,7 @@ class ToolClient:
|
|
327
337
|
raise ValueError(f"Invalid tool arguments format: {arguments}")
|
328
338
|
|
329
339
|
# Call the tool
|
330
|
-
logger.info(f"Calling tool: {tool.name} with args: {self.
|
340
|
+
logger.info(f"Calling tool: {tool.name} with args: {self._truncate_dict(arguments)}")
|
331
341
|
if inspect.iscoroutinefunction(tool.function):
|
332
342
|
result = await tool.function(**arguments)
|
333
343
|
else:
|
@@ -1,18 +1,18 @@
|
|
1
|
-
mbxai/__init__.py,sha256=
|
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=
|
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=
|
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.
|
16
|
-
mbxai-0.5.
|
17
|
-
mbxai-0.5.
|
18
|
-
mbxai-0.5.
|
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
|
File without changes
|