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 CHANGED
@@ -2,4 +2,4 @@
2
2
  MBX AI package.
3
3
  """
4
4
 
5
- __version__ = "0.5.11"
5
+ __version__ = "0.5.12"
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.11",
34
+ version="0.5.12",
35
35
  )
36
36
 
37
37
  # Initialize MCP server
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 each tool call
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
- # Add the tool response to the messages
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
- messages.append(tool_response)
159
- logger.debug(f"Added tool response to messages: {tool_response}")
160
-
161
- # Get a new response from the model with this tool result
162
- response = self._client.chat_completion(
163
- messages=messages,
164
- model=model,
165
- stream=stream,
166
- **kwargs,
167
- )
168
-
169
- if stream:
170
- return response
171
-
172
- message = response.choices[0].message
173
- # Add the assistant's message with tool calls
174
- assistant_message = {
175
- "role": "assistant",
176
- "content": message.content or None, # Ensure content is None if empty
177
- }
178
- if message.tool_calls:
179
- assistant_message["tool_calls"] = [
180
- {
181
- "id": tool_call.id,
182
- "type": "function",
183
- "function": {
184
- "name": tool_call.function.name,
185
- "arguments": tool_call.function.arguments,
186
- },
187
- }
188
- for tool_call in message.tool_calls
189
- ]
190
- messages.append(assistant_message)
191
- logger.debug(f"Added assistant message: {assistant_message}")
192
-
193
- # If there are no more tool calls, we're done
194
- if not message.tool_calls:
195
- return response
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 each tool call
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
- # Add the tool response to the messages
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
- messages.append(tool_response)
306
- logger.debug(f"Added tool response to messages: {tool_response}")
307
-
308
- # Get a new response from the model with this tool result
309
- response = self._client.chat_completion_parse(
310
- messages=messages,
311
- response_format=response_format,
312
- model=model,
313
- stream=stream,
314
- **kwargs,
315
- )
316
-
317
- if stream:
318
- return response
319
-
320
- message = response.choices[0].message
321
- # Add the assistant's message with tool calls
322
- assistant_message = {
323
- "role": "assistant",
324
- "content": message.content or None, # Ensure content is None if empty
325
- }
326
- if message.tool_calls:
327
- assistant_message["tool_calls"] = [
328
- {
329
- "id": tool_call.id,
330
- "type": "function",
331
- "function": {
332
- "name": tool_call.function.name,
333
- "arguments": tool_call.function.arguments,
334
- },
335
- }
336
- for tool_call in message.tool_calls
337
- ]
338
- messages.append(assistant_message)
339
- logger.debug(f"Added assistant message: {assistant_message}")
340
-
341
- # If there are no more tool calls, we're done
342
- if not message.tool_calls:
343
- return response
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,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: mbxai
3
- Version: 0.5.11
3
+ Version: 0.5.12
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=Nrk-9hVjGptSGLRYdkZYhni7CQG40G6nbzn7DQCmeTE,48
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=kYErSSCh__6Fe5hanBCrgYbtT-5rcjTrP356d2EjuVQ,3463
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=0JyWijPzXuyeYBzUuTWndxh9GGmJ0hrEsmLFmXbtD-Q,12978
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.11.dist-info/METADATA,sha256=e5ZqNMVQfzDegWT_mQyCnd0WqrLsJ8L43Qi2eBuVSos,4108
16
- mbxai-0.5.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
17
- mbxai-0.5.11.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
18
- mbxai-0.5.11.dist-info/RECORD,,
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