blaxel 0.2.0rc1__tar.gz → 0.2.0rc2__tar.gz

{blaxel-0.2.0rc1 → blaxel-0.2.0rc2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: blaxel
-Version: 0.2.0rc1
+Version: 0.2.0rc2
 Summary: Blaxel - AI development platform SDK
 Project-URL: Homepage, https://blaxel.ai
 Project-URL: Documentation, https://docs.blaxel.ai
@@ -43,7 +43,7 @@ Requires-Dist: llama-index-llms-groq>=0.3.1; extra == 'all'
 Requires-Dist: llama-index-llms-mistralai>=0.4.0; extra == 'all'
 Requires-Dist: llama-index-llms-openai>=0.3.42; extra == 'all'
 Requires-Dist: llama-index>=0.12.37; extra == 'all'
-Requires-Dist: openai-agents>=0.0.7; extra == 'all'
+Requires-Dist: openai-agents>=0.0.14; extra == 'all'
 Requires-Dist: opentelemetry-exporter-otlp>=1.28.0; extra == 'all'
 Requires-Dist: opentelemetry-instrumentation-anthropic==0.40.6; extra == 'all'
 Requires-Dist: opentelemetry-instrumentation-cohere==0.40.6; extra == 'all'
@@ -95,7 +95,7 @@ Requires-Dist: llama-index-llms-openai>=0.3.42; extra == 'llamaindex'
 Requires-Dist: llama-index>=0.12.37; extra == 'llamaindex'
 Requires-Dist: opentelemetry-instrumentation-llamaindex>=0.40.7; extra == 'llamaindex'
 Provides-Extra: openai
-Requires-Dist: openai-agents>=0.0.7; extra == 'openai'
+Requires-Dist: openai-agents>=0.0.14; extra == 'openai'
 Provides-Extra: pydantic
 Requires-Dist: pydantic-ai>=0.0.48; extra == 'pydantic'
 Provides-Extra: telemetry

{blaxel-0.2.0rc1 → blaxel-0.2.0rc2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "blaxel"
-version = "0.2.0.pre-1"
+version = "0.2.0rc2"
 description = "Blaxel - AI development platform SDK"
 readme = "README.md"
 authors = [{ name = "cploujoux", email = "cploujoux@blaxel.ai" }]
@@ -38,7 +38,7 @@ telemetry = [
 crewai = ["crewai>=0.120.1", "opentelemetry-instrumentation-crewai>=0.40.6"]
 
 # OpenAI module
-openai = ["openai-agents>=0.0.7"]
+openai = ["openai-agents>=0.0.14"]
 
 # LangGraph module
 langgraph = [

{blaxel-0.2.0rc1 → blaxel-0.2.0rc2}/src/blaxel/core/mcp/server.py

@@ -2,7 +2,7 @@ import logging
 import traceback
 import uuid
 from contextlib import asynccontextmanager
-from typing import Dict, Literal
+from typing import Literal
 
 import anyio
 import mcp.types as types
@@ -10,63 +10,14 @@ from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStre
 from mcp.server.fastmcp import FastMCP as FastMCPBase
 from websockets.asyncio.server import ServerConnection, serve
 
-try:
-    from opentelemetry.trace import Span, StatusCode
-
-    HAS_OPENTELEMETRY = True
-except ImportError:
-    HAS_OPENTELEMETRY = False
-
-    # Create dummy classes for when opentelemetry is not available
-    class Span:
-        def set_attributes(self, *args, **kwargs):
-            pass
-
-        def set_status(self, *args, **kwargs):
-            pass
-
-        def record_exception(self, *args, **kwargs):
-            pass
-
-        def end(self, *args, **kwargs):
-            pass
-
-        def add_event(self, *args, **kwargs):
-            pass
-
-        def get_span_context(self, *args, **kwargs):
-            return None
-
-        def is_recording(self, *args, **kwargs):
-            return False
-
-        def set_attribute(self, *args, **kwargs):
-            pass
-
-        def update_name(self, *args, **kwargs):
-            pass
-
-    class StatusCode:
-        ERROR = "ERROR"
-
-
 from ..common.env import env
 
 logger = logging.getLogger(__name__)
 
 
-class DummySpanManager:
-    """Dummy span manager for when opentelemetry is not available."""
-
-    def create_span(self, name: str, attributes: dict = None):
-        return Span()
-
-
 class BlaxelMcpServerTransport:
     """WebSocket server transport for MCP."""
 
-    spans: Dict[str, Span] = {}
-
     def __init__(self, port: int = 8080):
         """Initialize the WebSocket server transport.
 
@@ -80,13 +31,6 @@ class BlaxelMcpServerTransport:
         self.clients = {}
         self.server = None
 
-        # Initialize span manager
-        if HAS_OPENTELEMETRY:
-            # TODO: Implement proper OpenTelemetry span manager when telemetry is available
-            self.span_manager = DummySpanManager()
-        else:
-            self.span_manager = DummySpanManager()
-
     @asynccontextmanager
     async def websocket_server(self):
         """Create and run a WebSocket server for MCP communication."""
@@ -106,30 +50,14 @@ class BlaxelMcpServerTransport:
 
             try:
                 async for message in websocket:
-                    span = self.span_manager.create_span("message", {"mcp.client.id": client_id})
                     try:
                         msg = types.JSONRPCMessage.model_validate_json(message)
                         # Modify message ID to include client ID
                         if hasattr(msg, "id") and msg.id is not None:
                             original_id = msg.id
                             msg.id = f"{client_id}:{original_id}"
-                            span.set_attributes(
-                                {
-                                    "mcp.message.parsed": True,
-                                    "mcp.method": getattr(msg, "method", None),
-                                    "mcp.messageId": getattr(msg, "id", None),
-                                    "mcp.toolName": getattr(
-                                        getattr(msg, "params", None), "name", None
-                                    ),
-                                    "span.type": "mcp.message",
-                                }
-                            )
-                            self.spans[client_id + ":" + msg.id] = span
                         await read_stream_writer.send(msg)
                     except Exception as exc:
-                        span.set_status(StatusCode.ERROR)
-                        span.record_exception(exc)
-                        span.end()
                         logger.error(f"Failed to parse message: {exc}\n{traceback.format_exc()}")
                         await read_stream_writer.send(exc)
             except Exception as e:
@@ -158,25 +86,12 @@ class BlaxelMcpServerTransport:
                     if client_id and client_id in self.clients:
                         # Send to specific client
                         websocket = self.clients[client_id]
-                        span = self.spans.get(client_id + ":" + msg_id)
                         try:
                             await websocket.send(data)
-                            if span:
-                                span.set_attributes(
-                                    {
-                                        "mcp.message.response_sent": True,
-                                    }
-                                )
                         except Exception as e:
-                            if span:
-                                span.set_status(StatusCode.ERROR)
-                                span.record_exception(e)
                             logger.error(f"Failed to send message to client {client_id}: {e}")
                             if client_id in self.clients:
                                 del self.clients[client_id]
-                        finally:
-                            if span:
-                                span.end()
                     else:
                         # Broadcast to all clients
                         dead_clients = []