mbxai 0.5.1__py3-none-any.whl → 0.5.3__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.1"
5
+ __version__ = "0.5.3"
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.1",
34
+ version="0.5.3",
35
35
  )
36
36
 
37
37
  # Initialize MCP server
@@ -7,6 +7,9 @@ from openai import OpenAI, OpenAIError
7
7
  from pydantic import BaseModel, TypeAdapter
8
8
  from .models import OpenRouterModel, OpenRouterModelRegistry
9
9
  from .config import OpenRouterConfig
10
+ import logging
11
+
12
+ logger = logging.getLogger(__name__)
10
13
 
11
14
 
12
15
  class OpenRouterError(Exception):
@@ -76,12 +79,15 @@ class OpenRouterClient:
76
79
  OpenRouterAPIError: For API errors
77
80
  OpenRouterError: For other errors
78
81
  """
82
+ error_msg = str(error)
83
+ logger.error(f"API error during {operation}: {error_msg}")
84
+
79
85
  if isinstance(error, OpenAIError):
80
- raise OpenRouterAPIError(f"API error during {operation}: {str(error)}")
81
- elif "Connection" in str(error):
82
- raise OpenRouterConnectionError(f"Connection error during {operation}: {str(error)}")
86
+ raise OpenRouterAPIError(f"API error during {operation}: {error_msg}")
87
+ elif "Connection" in error_msg:
88
+ raise OpenRouterConnectionError(f"Connection error during {operation}: {error_msg}")
83
89
  else:
84
- raise OpenRouterError(f"Error during {operation}: {str(error)}")
90
+ raise OpenRouterError(f"Error during {operation}: {error_msg}")
85
91
 
86
92
  @property
87
93
  def model(self) -> str:
@@ -133,13 +139,50 @@ class OpenRouterClient:
133
139
  # Remove any incompatible parameters
134
140
  kwargs.pop("parse", None) # Remove parse parameter if present
135
141
 
136
- return self._client.chat.completions.create(
142
+ logger.debug(f"Making chat completion request with model {model or self.model}")
143
+ logger.debug(f"Request messages: {messages}")
144
+ logger.debug(f"Request kwargs: {kwargs}")
145
+
146
+ response = self._client.chat.completions.create(
137
147
  model=str(model or self.model),
138
148
  messages=messages,
139
149
  stream=stream,
140
150
  **kwargs,
141
151
  )
152
+
153
+ # Log response structure for debugging
154
+ logger.debug(f"Response type: {type(response)}")
155
+ logger.debug(f"Response attributes: {dir(response)}")
156
+ logger.debug(f"Raw response: {response}")
157
+
158
+ # Handle streaming response differently
159
+ if stream:
160
+ return response
161
+
162
+ # Validate response structure
163
+ if response is None:
164
+ logger.error("Received None response from API")
165
+ raise OpenRouterError("Received None response from API")
166
+
167
+ if not hasattr(response, 'choices'):
168
+ logger.error(f"Invalid response: missing choices attribute. Response: {response}")
169
+ raise OpenRouterError("Invalid response: missing choices attribute")
170
+
171
+ if not response.choices:
172
+ logger.error(f"Invalid response: empty choices list. Response: {response}")
173
+ raise OpenRouterError("Invalid response: empty choices list")
174
+
175
+ if not hasattr(response.choices[0], 'message'):
176
+ logger.error(f"Invalid response: missing message in first choice. Response: {response}")
177
+ raise OpenRouterError("Invalid response: missing message in first choice")
178
+
179
+ logger.debug(f"Response message: {response.choices[0].message}")
180
+ return response
181
+
142
182
  except Exception as e:
183
+ logger.error(f"Error in chat completion: {str(e)}")
184
+ if isinstance(e, OpenRouterError):
185
+ raise
143
186
  self._handle_api_error("chat completion", e)
144
187
 
145
188
  def chat_completion_parse(
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: mbxai
3
- Version: 0.5.1
3
+ Version: 0.5.3
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=kyNYdkqdxssgIHfUQJw8pEuP8HOGhC7hR1d9oaAcGco,47
1
+ mbxai/__init__.py,sha256=ujlMlgvQNlXgDy1c5g83G6iKTxzV6_pMRXKUjX-pNRs,47
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=CYm_5HiB3hZUYfahzM_senilPgH2oZwSevf2up1JTF0,5506
5
5
  mbxai/mcp/example.py,sha256=oaol7AvvZnX86JWNz64KvPjab5gg1VjVN3G8eFSzuaE,2350
6
- mbxai/mcp/server.py,sha256=11TgdqtWs_IhaqbQtLbeRbBpYwCYVo692JvQT0kbQJo,3462
6
+ mbxai/mcp/server.py,sha256=Kn4zY_aBiQMfcUAOnWehhITB_5usSYZi8zSNU4QhLMM,3462
7
7
  mbxai/openrouter/__init__.py,sha256=Ito9Qp_B6q-RLGAQcYyTJVWwR2YAZvNqE-HIYXxhtD8,298
8
- mbxai/openrouter/client.py,sha256=NPccB7Slou5tSc8DvXvItLdoDMgU0pcxW1lKcRKQXp0,8865
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
12
  mbxai/tools/client.py,sha256=PDkDTvpwAs8-F8CcF6wjjq2Z-VhYpjA9RXTK64Rwa3Y,5418
13
13
  mbxai/tools/example.py,sha256=1HgKK39zzUuwFbnp3f0ThyWVfA_8P28PZcTwaUw5K78,2232
14
14
  mbxai/tools/types.py,sha256=ZHnmiDXpH6wZhiZ-Tj9PiPMJaW1aiDAq5It2gpiwNp0,831
15
- mbxai-0.5.1.dist-info/METADATA,sha256=g61BMob_rXCUGO5GOo6yRT-0U0g0MfwRJhWst9KJ02g,4107
16
- mbxai-0.5.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
17
- mbxai-0.5.1.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
18
- mbxai-0.5.1.dist-info/RECORD,,
15
+ mbxai-0.5.3.dist-info/METADATA,sha256=HNDLbRGqN3rT-m0kA7DGI5fWQUH8uGiNh70xzriM6Bs,4107
16
+ mbxai-0.5.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
17
+ mbxai-0.5.3.dist-info/licenses/LICENSE,sha256=hEyhc4FxwYo3NQ40yNgZ7STqwVk-1_XcTXOnAPbGJAw,1069
18
+ mbxai-0.5.3.dist-info/RECORD,,
File without changes