blaxel 0.2.5__tar.gz → 0.2.7__tar.gz

{blaxel-0.2.5 → blaxel-0.2.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: blaxel
-Version: 0.2.5
+Version: 0.2.7
 Summary: Blaxel - AI development platform SDK
 Project-URL: Homepage, https://blaxel.ai
 Project-URL: Documentation, https://docs.blaxel.ai

{blaxel-0.2.5 → blaxel-0.2.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "blaxel"
-version = "0.2.5"
+version = "0.2.7"
 description = "Blaxel - AI development platform SDK"
 readme = "README.md"
 authors = [{ name = "cploujoux", email = "cploujoux@blaxel.ai" }]

{blaxel-0.2.5 → blaxel-0.2.7}/src/blaxel/telemetry/instrumentation/blaxel_core.py

@@ -73,7 +73,7 @@ class BlaxelCoreInstrumentor(BaseInstrumentor):
 
         async def traced_list_tools(self, *args, **kwargs):
             span_attributes = {
-                "tool.server": self.url,
+                "tool.server": self._url,
                 "tool.server_name": self.name,
                 "span.type": "tool.list",
                 **SpanManager.get_default_attributes(),
@@ -88,8 +88,8 @@ class BlaxelCoreInstrumentor(BaseInstrumentor):
         # Patch convert_mcp_tool_to_blaxel_tool to wrap tool calls
         orig_convert = blaxel.core.tools.convert_mcp_tool_to_blaxel_tool
 
-        def traced_convert_mcp_tool_to_blaxel_tool(websocket_client, name, url, tool):
-            Tool = orig_convert(websocket_client, name, url, tool)
+        def traced_convert_mcp_tool_to_blaxel_tool(websocket_client, tool):
+            Tool = orig_convert(websocket_client, tool)
             orig_coroutine = Tool.coroutine
             orig_sync_coroutine = Tool.sync_coroutine
 
@@ -97,8 +97,8 @@ class BlaxelCoreInstrumentor(BaseInstrumentor):
                 span_attributes = {
                     "tool.name": tool.name,
                     "tool.args": json.dumps(kwargs),
-                    "tool.server": url,
-                    "tool.server_name": name,
+                    "tool.server": websocket_client._url,
+                    "tool.server_name": websocket_client.name,
                     "span.type": "tool.call",
                     **SpanManager.get_default_attributes(),
                 }
@@ -110,8 +110,8 @@ class BlaxelCoreInstrumentor(BaseInstrumentor):
                 span_attributes = {
                     "tool.name": tool.name,
                     "tool.args": json.dumps(kwargs),
-                    "tool.server": url,
-                    "tool.server_name": name,
+                    "tool.server": websocket_client._url,
+                    "tool.server_name": websocket_client.name,
                     "span.type": "tool.call",
                     **SpanManager.get_default_attributes(),
                 }