mbxai 0.5.11__py3-none-any.whl → 0.5.12__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 +91 -81
- {mbxai-0.5.11.dist-info → mbxai-0.5.12.dist-info}/METADATA +1 -1
- {mbxai-0.5.11.dist-info → mbxai-0.5.12.dist-info}/RECORD +7 -7
- {mbxai-0.5.11.dist-info → mbxai-0.5.12.dist-info}/WHEEL +0 -0
- {mbxai-0.5.11.dist-info → mbxai-0.5.12.dist-info}/licenses/LICENSE +0 -0
mbxai/__init__.py
CHANGED
mbxai/mcp/server.py
CHANGED
mbxai/tools/client.py
CHANGED
@@ -113,7 +113,8 @@ class ToolClient:
|
|
113
113
|
if not message.tool_calls:
|
114
114
|
return response
|
115
115
|
|
116
|
-
# Handle
|
116
|
+
# Handle all tool calls before getting the next model response
|
117
|
+
tool_responses = []
|
117
118
|
for tool_call in message.tool_calls:
|
118
119
|
logger.debug(f"Processing tool call: {tool_call}")
|
119
120
|
logger.debug(f"Tool call ID: {tool_call.id}")
|
@@ -149,50 +150,54 @@ class ToolClient:
|
|
149
150
|
if not isinstance(result, str):
|
150
151
|
result = json.dumps(result)
|
151
152
|
|
152
|
-
#
|
153
|
+
# Create the tool response
|
153
154
|
tool_response = {
|
154
155
|
"role": "tool",
|
155
156
|
"tool_call_id": tool_call.id,
|
156
157
|
"content": result,
|
157
158
|
}
|
158
|
-
|
159
|
-
logger.debug(f"
|
160
|
-
|
161
|
-
|
162
|
-
|
163
|
-
|
164
|
-
|
165
|
-
|
166
|
-
|
167
|
-
|
168
|
-
|
169
|
-
|
170
|
-
|
171
|
-
|
172
|
-
|
173
|
-
|
174
|
-
|
175
|
-
|
176
|
-
|
177
|
-
|
178
|
-
|
179
|
-
|
180
|
-
|
181
|
-
|
182
|
-
|
183
|
-
|
184
|
-
|
185
|
-
|
186
|
-
|
187
|
-
|
188
|
-
|
189
|
-
|
190
|
-
|
191
|
-
|
192
|
-
|
193
|
-
|
194
|
-
|
195
|
-
|
159
|
+
tool_responses.append(tool_response)
|
160
|
+
logger.debug(f"Created tool response: {tool_response}")
|
161
|
+
|
162
|
+
# Add all tool responses to the messages
|
163
|
+
messages.extend(tool_responses)
|
164
|
+
logger.debug(f"Added all tool responses to messages: {tool_responses}")
|
165
|
+
|
166
|
+
# Get a new response from the model with all tool results
|
167
|
+
response = self._client.chat_completion(
|
168
|
+
messages=messages,
|
169
|
+
model=model,
|
170
|
+
stream=stream,
|
171
|
+
**kwargs,
|
172
|
+
)
|
173
|
+
|
174
|
+
if stream:
|
175
|
+
return response
|
176
|
+
|
177
|
+
message = response.choices[0].message
|
178
|
+
# Add the assistant's message with tool calls
|
179
|
+
assistant_message = {
|
180
|
+
"role": "assistant",
|
181
|
+
"content": message.content or None, # Ensure content is None if empty
|
182
|
+
}
|
183
|
+
if message.tool_calls:
|
184
|
+
assistant_message["tool_calls"] = [
|
185
|
+
{
|
186
|
+
"id": tool_call.id,
|
187
|
+
"type": "function",
|
188
|
+
"function": {
|
189
|
+
"name": tool_call.function.name,
|
190
|
+
"arguments": tool_call.function.arguments,
|
191
|
+
},
|
192
|
+
}
|
193
|
+
for tool_call in message.tool_calls
|
194
|
+
]
|
195
|
+
messages.append(assistant_message)
|
196
|
+
logger.debug(f"Added assistant message: {assistant_message}")
|
197
|
+
|
198
|
+
# If there are no more tool calls, we're done
|
199
|
+
if not message.tool_calls:
|
200
|
+
return response
|
196
201
|
|
197
202
|
async def parse(
|
198
203
|
self,
|
@@ -260,7 +265,8 @@ class ToolClient:
|
|
260
265
|
if not message.tool_calls:
|
261
266
|
return response
|
262
267
|
|
263
|
-
# Handle
|
268
|
+
# Handle all tool calls before getting the next model response
|
269
|
+
tool_responses = []
|
264
270
|
for tool_call in message.tool_calls:
|
265
271
|
logger.debug(f"Processing tool call: {tool_call}")
|
266
272
|
logger.debug(f"Tool call ID: {tool_call.id}")
|
@@ -296,48 +302,52 @@ class ToolClient:
|
|
296
302
|
if not isinstance(result, str):
|
297
303
|
result = json.dumps(result)
|
298
304
|
|
299
|
-
#
|
305
|
+
# Create the tool response
|
300
306
|
tool_response = {
|
301
307
|
"role": "tool",
|
302
308
|
"tool_call_id": tool_call.id,
|
303
309
|
"content": result,
|
304
310
|
}
|
305
|
-
|
306
|
-
logger.debug(f"
|
307
|
-
|
308
|
-
|
309
|
-
|
310
|
-
|
311
|
-
|
312
|
-
|
313
|
-
|
314
|
-
|
315
|
-
|
316
|
-
|
317
|
-
|
318
|
-
|
319
|
-
|
320
|
-
|
321
|
-
|
322
|
-
|
323
|
-
|
324
|
-
|
325
|
-
|
326
|
-
|
327
|
-
|
328
|
-
|
329
|
-
|
330
|
-
|
331
|
-
|
332
|
-
|
333
|
-
|
334
|
-
|
335
|
-
|
336
|
-
|
337
|
-
|
338
|
-
|
339
|
-
|
340
|
-
|
341
|
-
|
342
|
-
|
343
|
-
|
311
|
+
tool_responses.append(tool_response)
|
312
|
+
logger.debug(f"Created tool response: {tool_response}")
|
313
|
+
|
314
|
+
# Add all tool responses to the messages
|
315
|
+
messages.extend(tool_responses)
|
316
|
+
logger.debug(f"Added all tool responses to messages: {tool_responses}")
|
317
|
+
|
318
|
+
# Get a new response from the model with all tool results
|
319
|
+
response = self._client.chat_completion_parse(
|
320
|
+
messages=messages,
|
321
|
+
response_format=response_format,
|
322
|
+
model=model,
|
323
|
+
stream=stream,
|
324
|
+
**kwargs,
|
325
|
+
)
|
326
|
+
|
327
|
+
if stream:
|
328
|
+
return response
|
329
|
+
|
330
|
+
message = response.choices[0].message
|
331
|
+
# Add the assistant's message with tool calls
|
332
|
+
assistant_message = {
|
333
|
+
"role": "assistant",
|
334
|
+
"content": message.content or None, # Ensure content is None if empty
|
335
|
+
}
|
336
|
+
if message.tool_calls:
|
337
|
+
assistant_message["tool_calls"] = [
|
338
|
+
{
|
339
|
+
"id": tool_call.id,
|
340
|
+
"type": "function",
|
341
|
+
"function": {
|
342
|
+
"name": tool_call.function.name,
|
343
|
+
"arguments": tool_call.function.arguments,
|
344
|
+
},
|
345
|
+
}
|
346
|
+
for tool_call in message.tool_calls
|
347
|
+
]
|
348
|
+
messages.append(assistant_message)
|
349
|
+
logger.debug(f"Added assistant message: {assistant_message}")
|
350
|
+
|
351
|
+
# If there are no more tool calls, we're done
|
352
|
+
if not message.tool_calls:
|
353
|
+
return response
|
@@ -1,18 +1,18 @@
|
|
1
|
-
mbxai/__init__.py,sha256=
|
1
|
+
mbxai/__init__.py,sha256=IZikFZwxYq2ZzCj7jfjGct1is4gCcMAc590whjAThy4,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=MXGh1jn4tKLY49fAs_5rTaVMh1GT-azvwyXGHIDoSRg,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=8aTrRSljnjrKIaBzLspSc8OpqRar6LZn__SVNOBpgHM,13190
|
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.12.dist-info/METADATA,sha256=mga6ghbaA3DUBmlVAAGAyHiACU--F6dgiR038ausnec,4108
|
16
|
+
mbxai-0.5.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
|
17
|
+
mbxai-0.5.12.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
|
18
|
+
mbxai-0.5.12.dist-info/RECORD,,
|
File without changes
|
File without changes
|