blaxel 0.2.0rc1__py3-none-any.whl → 0.2.0rc2__py3-none-any.whl

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 = []

{blaxel-0.2.0rc1.dist-info → blaxel-0.2.0rc2.dist-info}/METADATA

@@ -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.dist-info → blaxel-0.2.0rc2.dist-info}/RECORD

@@ -302,7 +302,7 @@ blaxel/core/common/settings.py,sha256=7KTryuBdud0IfHqykX7xEEtpgq5M5h1Z8YEzYKsHB-
 blaxel/core/jobs/__init__.py,sha256=tApZ6t6Savd5c9yH1RhuiPDGq19B464Fakd-YqCOEtI,5441
 blaxel/core/mcp/__init__.py,sha256=5VjkiQFb1QWW5QKRgwPHARlxZJ9Xqaz0diJTpM8LLF0,142
 blaxel/core/mcp/client.py,sha256=dFjPHirybqA3IMecSvdDss-iU5ferPUaunNrPyhFm8o,5195
-blaxel/core/mcp/server.py,sha256=OU1sYcwepnvGtWrxH5IxZdcZfMqMF8hwjvyAzssM6bo,8741
+blaxel/core/mcp/server.py,sha256=lx-4C71PdPxbG_tGuk81j44jrl7a1ztSixC-k4ObyBo,5812
 blaxel/core/models/__init__.py,sha256=HbRDsMnUFHkPC-MMkzPXh4mUqkVjqO6p3j7m00N_XSo,1722
 blaxel/core/sandbox/__init__.py,sha256=eWTH7PWNQZe9efqParwgXzltc7L4OyPtT_REqCdQybU,551
 blaxel/core/sandbox/action.py,sha256=9Zjkco7YkLzBThD3N2Hr5SpeEiqU_-Ktk8HlKpkpiAg,2802
@@ -402,7 +402,7 @@ blaxel/telemetry/instrumentation/map.py,sha256=EVCw6ug1hemE-9kh77Vxs9cqr_p9TPzFw
 blaxel/telemetry/instrumentation/utils.py,sha256=KInMYZH-mu9_wvetmf0EmgrfN3Sw8IWk2Y95v2u90_U,1901
 blaxel/telemetry/log/log.py,sha256=RvQByRjZMoP_dRaAZu8oK6DTegsHs-xV4W-UIqis6CA,2461
 blaxel/telemetry/log/logger.py,sha256=NPAS3g82ryROjvc_DEZaTIfrcehoLEZoP-JkLxADxc0,4113
-blaxel-0.2.0rc1.dist-info/METADATA,sha256=n61p-dUYa8NGsZdSxAllJkrdiq2PzI1BvlI_UNXPRAQ,9876
-blaxel-0.2.0rc1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-blaxel-0.2.0rc1.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-blaxel-0.2.0rc1.dist-info/RECORD,,
+blaxel-0.2.0rc2.dist-info/METADATA,sha256=g08ZNxaAovSAaeMwRLMFeoY5YJ2HWAOmTL1e-QqIHko,9878
+blaxel-0.2.0rc2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+blaxel-0.2.0rc2.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+blaxel-0.2.0rc2.dist-info/RECORD,,