blaxel 0.1.22rc70__py3-none-any.whl → 0.2.0rc2__py3-none-any.whl

blaxel/telemetry/instrumentation/map.py

@@ -0,0 +1,61 @@
+from typing import Dict, List
+
+from pydantic import BaseModel
+
+
+class InstrumentationMapping(BaseModel):
+    module_path: str
+    class_name: str
+    required_packages: List[str]
+    ignore_if_packages: List[str]
+
+MAPPINGS: Dict[str, InstrumentationMapping] = {
+    "anthropic": InstrumentationMapping(
+        module_path="opentelemetry.instrumentation.anthropic",
+        class_name="AnthropicInstrumentor",
+        required_packages=["anthropic"],
+        ignore_if_packages=[]
+    ),
+    "cohere": InstrumentationMapping(
+        module_path="opentelemetry.instrumentation.cohere",
+        class_name="CohereInstrumentor",
+        required_packages=["cohere"],
+        ignore_if_packages=[]
+    ),
+    "openai": InstrumentationMapping(
+        module_path="opentelemetry.instrumentation.openai",
+        class_name="OpenAIInstrumentor",
+        required_packages=["openai"],
+        ignore_if_packages=[]
+    ),
+    "gemini": InstrumentationMapping(
+        module_path="opentelemetry.instrumentation.google_generativeai",
+        class_name="GoogleGenerativeAIInstrumentor",
+        required_packages=["google-generativeai"],
+        ignore_if_packages=[]
+    ),
+    "blaxel.core": InstrumentationMapping(
+        module_path="blaxel_telemetry.instrumentation.blaxel.core",
+        class_name="BlaxelCoreInstrumentor",
+        required_packages=["blaxel.core"],
+        ignore_if_packages=[]
+    ),
+    "blaxel_langgraph": InstrumentationMapping(
+        module_path="blaxel_telemetry.instrumentation.blaxel_langgraph",
+        class_name="BlaxelLanggraphInstrumentor",
+        required_packages=["blaxel_langgraph"],
+        ignore_if_packages=[]
+    ),
+    "blaxel_langgraph_gemini": InstrumentationMapping(
+        module_path="blaxel_telemetry.instrumentation.blaxel_langgraph_gemini",
+        class_name="BlaxelLanggraphGeminiInstrumentor",
+        required_packages=["blaxel_langgraph"],
+        ignore_if_packages=[]
+    ),
+    "blaxel_llamaindex": InstrumentationMapping(
+        module_path="blaxel_telemetry.instrumentation.blaxel_llamaindex",
+        class_name="BlaxelLlamaIndexInstrumentor",
+        required_packages=["blaxel_llamaindex"],
+        ignore_if_packages=[]
+    ),
+}

blaxel/telemetry/instrumentation/utils.py

@@ -0,0 +1,74 @@
+import dataclasses
+import json
+import logging
+import os
+import traceback
+from contextlib import asynccontextmanager
+
+from opentelemetry.semconv_ai import SpanAttributes
+
+
+def dont_throw(func):
+    # Obtain a logger specific to the function's module
+    logger = logging.getLogger(func.__module__)
+
+    def wrapper(*args, **kwargs):
+        try:
+            return func(*args, **kwargs)
+        except Exception:
+            logger.debug(
+                "Instrumentation failed to trace in %s, error: %s",
+                func.__name__,
+                traceback.format_exc(),
+            )
+
+    return wrapper
+
+
+def with_tracer_wrapper(func):
+    def _with_tracer(tracer):
+        def wrapper(wrapped, instance, args, kwargs):
+            return func(tracer, wrapped, instance, args, kwargs)
+
+        return wrapper
+
+    return _with_tracer
+
+
+@asynccontextmanager
+async def start_as_current_span_async(tracer, *args, **kwargs):
+    with tracer.start_as_current_span(*args, **kwargs) as span:
+        yield span
+
+class JSONEncoder(json.JSONEncoder):
+    def default(self, o):
+        if dataclasses.is_dataclass(o):
+            return dataclasses.asdict(o)
+        elif hasattr(o, "json"):
+            return o.json()
+        elif hasattr(o, "to_json"):
+            return o.to_json()
+        return super().default(o)
+
+
+def should_send_prompts():
+    return (
+        os.getenv("TRACELOOP_TRACE_CONTENT") or "true"
+    ).lower() == "true"
+
+@dont_throw
+def process_request(span, args, kwargs):
+    if should_send_prompts():
+        span.set_attribute(
+            SpanAttributes.TRACELOOP_ENTITY_INPUT,
+            json.dumps({"args": args, "kwargs": kwargs}, cls=JSONEncoder),
+        )
+
+
+@dont_throw
+def process_response(span, res):
+    if should_send_prompts():
+        span.set_attribute(
+            SpanAttributes.TRACELOOP_ENTITY_OUTPUT,
+            json.dumps(res, cls=JSONEncoder),
+        )

blaxel/{common → telemetry/log}/logger.py

@@ -6,7 +6,7 @@ import json
 import logging
 import os
 
-from opentelemetry.trace import get_current_span
+from opentelemetry import trace
 
 
 class JsonFormatter(logging.Formatter):
@@ -35,18 +35,12 @@ class JsonFormatter(logging.Formatter):
             self.labels_name: {}
         }
 
-        # Get current active span - equivalent to trace.getActiveSpan() in JS
-        current_span = get_current_span()
-        
-        # Check if span exists and has valid context (equivalent to 'if (currentSpan)' in JS)
-        if current_span and current_span.get_span_context().is_valid:
+        # Add trace context if available
+        current_span = trace.get_current_span()
+        if current_span.is_recording():
             span_context = current_span.get_span_context()
-            # Format trace_id and span_id as hex strings (like JS does)
-            trace_id_hex = format(span_context.trace_id, '032x')
-            span_id_hex = format(span_context.span_id, '016x')
-            
-            log_entry[self.trace_id_name] = f"{self.trace_id_prefix}{trace_id_hex}"
-            log_entry[self.span_id_name] = f"{self.span_id_prefix}{span_id_hex}"
+            log_entry[self.trace_id_name] = f"{self.trace_id_prefix}{span_context.trace_id}"
+            log_entry[self.span_id_name] = f"{self.span_id_prefix}{span_context.span_id}"
 
         # Add task ID if available
         task_id = os.environ.get(self.task_index)

blaxel/{instrumentation → telemetry}/manager.py

@@ -21,16 +21,16 @@ from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import BatchSpanProcessor
 from opentelemetry.trace import NoOpTracerProvider
 
-from blaxel.instrumentation.exporters import (
+from blaxel.core.common import Settings
+
+from .exporters import (
     DynamicHeadersLogExporter,
     DynamicHeadersMetricExporter,
     DynamicHeadersSpanExporter,
 )
-from blaxel.instrumentation.span import DefaultAttributesSpanProcessor
-
-from ..common.settings import Settings
-from .log import AsyncLogRecordProcessor
-from .map import MAPPINGS
+from .instrumentation.map import MAPPINGS
+from .log.log import AsyncLogRecordProcessor
+from .span import DefaultAttributesSpanProcessor
 
 logger = logging.getLogger(__name__)
 
@@ -116,6 +116,7 @@ class TelemetryManager:
 
     def setup_signal_handler(self):
         """Set up signal handlers for graceful shutdown."""
+
         def handle_signal(signum, frame):
             logger.debug(f"Received signal {signum}")
             self.shutdown()
@@ -150,11 +151,15 @@ class TelemetryManager:
         # Set up the TracerProvider
         trace_provider = TracerProvider(resource=resource)
         span_processor = BatchSpanProcessor(self.get_span_exporter())
-        trace_provider.add_span_processor(DefaultAttributesSpanProcessor({
-            "workload.id": self.resource_name,
-            "workload.type": self.resource_type,
-            "workspace": self.resource_workspace,
-        }))
+        trace_provider.add_span_processor(
+            DefaultAttributesSpanProcessor(
+                {
+                    "workload.id": self.resource_name,
+                    "workload.type": self.resource_type,
+                    "workspace": self.resource_workspace,
+                }
+            )
+        )
         trace_provider.add_span_processor(span_processor)
         trace.set_tracer_provider(trace_provider)
         self.tracer = trace_provider.get_tracer(__name__)
@@ -217,10 +222,13 @@ class TelemetryManager:
             if time.time() - start_time < timeout:
                 logger.debug("Instrumentation shutdown complete")
             else:
-                logger.warning("Shutdown timed out after 5 seconds, skipping remaining shutdown tasks")
+                logger.warning(
+                    "Shutdown timed out after 5 seconds, skipping remaining shutdown tasks"
+                )
 
         except Exception as error:
             logger.error(f"Error during shutdown: {error}")
 
+
 # Create a singleton instance
 telemetry_manager = TelemetryManager()

blaxel/{instrumentation → telemetry}/span.py

@@ -10,7 +10,7 @@ from opentelemetry.sdk.trace import Span as SdkSpan
 from opentelemetry.sdk.trace.export import SpanProcessor
 from opentelemetry.trace import Span, Tracer
 
-from ..common.settings import settings
+from blaxel.core import settings
 
 T = TypeVar('T')
 
@@ -45,6 +45,16 @@ class SpanManager:
     def __init__(self, name: str):
         self.tracer: Tracer = trace.get_tracer(name)
 
+    @staticmethod
+    def get_default_attributes() -> Dict[str, Any]:
+        """Get default attributes for the span."""
+        return {
+            "blaxel.environment": settings.env,
+            "workload.id": settings.name,
+            "workload.type": f"{settings.type}s",
+            "workspace": settings.workspace
+        }
+
     def create_active_span(self, name: str, attributes: Dict[str, Any], parent: Optional[Span] = None) -> ContextManager[Span]:
         """
         Creates an active span and executes the provided function within its context.
@@ -92,3 +102,4 @@ class SpanManager:
         if parent:
             context = trace.set_span_in_context(parent)
         return self.tracer.start_span(name, attributes=full_attributes, context=context)
+        return self.tracer.start_span(name, attributes=full_attributes, context=context)

blaxel-0.2.0rc2.dist-info/METADATA

@@ -0,0 +1,224 @@
+Metadata-Version: 2.4
+Name: blaxel
+Version: 0.2.0rc2
+Summary: Blaxel - AI development platform SDK
+Project-URL: Homepage, https://blaxel.ai
+Project-URL: Documentation, https://docs.blaxel.ai
+Project-URL: Repository, https://github.com/blaxel-ai/sdk-python
+Project-URL: Changelog, https://docs.blaxel.ai/changelog
+Author-email: cploujoux <cploujoux@blaxel.ai>
+License-File: LICENSE
+Requires-Python: >=3.10
+Requires-Dist: attrs>=21.3.0
+Requires-Dist: httpx>=0.27.0
+Requires-Dist: mcp<=1.7.1
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: pyjwt>=2.0.0
+Requires-Dist: python-dateutil>=2.8.0
+Requires-Dist: pyyaml>=6.0.0
+Requires-Dist: requests>=2.32.3
+Requires-Dist: tomli>=2.2.1
+Requires-Dist: websockets<16.0.0
+Provides-Extra: all
+Requires-Dist: crewai>=0.120.1; extra == 'all'
+Requires-Dist: google-adk>=1.1.1; extra == 'all'
+Requires-Dist: langchain-anthropic>=0.3.10; extra == 'all'
+Requires-Dist: langchain-cerebras>=0.5.0; extra == 'all'
+Requires-Dist: langchain-cohere>=0.4.3; extra == 'all'
+Requires-Dist: langchain-community<0.4.0,>=0.3.3; extra == 'all'
+Requires-Dist: langchain-core<0.4.0,>=0.3.13; extra == 'all'
+Requires-Dist: langchain-deepseek-official>=0.1.0.post1; extra == 'all'
+Requires-Dist: langchain-openai>=0.3.10; extra == 'all'
+Requires-Dist: langchain-xai>=0.2.2; extra == 'all'
+Requires-Dist: langgraph<0.3.0,>=0.2.40; extra == 'all'
+Requires-Dist: litellm>=1.63.11; extra == 'all'
+Requires-Dist: livekit-agents[anthropic,cartesia,deepgram,elevenlabs,groq,openai,silero,turn-detector]~=1.0; extra == 'all'
+Requires-Dist: livekit-plugins-noise-cancellation~=0.2; extra == 'all'
+Requires-Dist: llama-index-llms-anthropic>=0.6.14; extra == 'all'
+Requires-Dist: llama-index-llms-cerebras>=0.2.2; extra == 'all'
+Requires-Dist: llama-index-llms-cohere>=0.4.1; extra == 'all'
+Requires-Dist: llama-index-llms-deepseek>=0.1.1; extra == 'all'
+Requires-Dist: llama-index-llms-google-genai>=0.1.13; extra == 'all'
+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.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'
+Requires-Dist: opentelemetry-instrumentation-crewai>=0.40.6; extra == 'all'
+Requires-Dist: opentelemetry-instrumentation-fastapi==0.54b1; extra == 'all'
+Requires-Dist: opentelemetry-instrumentation-google-generativeai>=0.40.7; extra == 'all'
+Requires-Dist: opentelemetry-instrumentation-langchain>=0.35.0; extra == 'all'
+Requires-Dist: opentelemetry-instrumentation-llamaindex>=0.40.7; extra == 'all'
+Requires-Dist: opentelemetry-instrumentation-ollama==0.40.6; extra == 'all'
+Requires-Dist: opentelemetry-instrumentation-openai==0.40.6; extra == 'all'
+Requires-Dist: pillow>=10.0.0; extra == 'all'
+Requires-Dist: pydantic-ai>=0.0.48; extra == 'all'
+Provides-Extra: core
+Provides-Extra: crewai
+Requires-Dist: crewai>=0.120.1; extra == 'crewai'
+Requires-Dist: opentelemetry-instrumentation-crewai>=0.40.6; extra == 'crewai'
+Provides-Extra: dev
+Requires-Dist: pyright; extra == 'dev'
+Requires-Dist: pytest; extra == 'dev'
+Requires-Dist: ruff; extra == 'dev'
+Requires-Dist: uv; extra == 'dev'
+Provides-Extra: googleadk
+Requires-Dist: google-adk>=1.1.1; extra == 'googleadk'
+Requires-Dist: litellm>=1.63.11; extra == 'googleadk'
+Provides-Extra: langgraph
+Requires-Dist: langchain-anthropic>=0.3.10; extra == 'langgraph'
+Requires-Dist: langchain-cerebras>=0.5.0; extra == 'langgraph'
+Requires-Dist: langchain-cohere>=0.4.3; extra == 'langgraph'
+Requires-Dist: langchain-community<0.4.0,>=0.3.3; extra == 'langgraph'
+Requires-Dist: langchain-core<0.4.0,>=0.3.13; extra == 'langgraph'
+Requires-Dist: langchain-deepseek-official>=0.1.0.post1; extra == 'langgraph'
+Requires-Dist: langchain-openai>=0.3.10; extra == 'langgraph'
+Requires-Dist: langchain-xai>=0.2.2; extra == 'langgraph'
+Requires-Dist: langgraph<0.3.0,>=0.2.40; extra == 'langgraph'
+Requires-Dist: opentelemetry-instrumentation-langchain>=0.35.0; extra == 'langgraph'
+Requires-Dist: pillow>=10.0.0; extra == 'langgraph'
+Provides-Extra: livekit
+Requires-Dist: livekit-agents[anthropic,cartesia,deepgram,elevenlabs,groq,openai,silero,turn-detector]~=1.0; extra == 'livekit'
+Requires-Dist: livekit-plugins-noise-cancellation~=0.2; extra == 'livekit'
+Provides-Extra: llamaindex
+Requires-Dist: llama-index-llms-anthropic>=0.6.14; extra == 'llamaindex'
+Requires-Dist: llama-index-llms-cerebras>=0.2.2; extra == 'llamaindex'
+Requires-Dist: llama-index-llms-cohere>=0.4.1; extra == 'llamaindex'
+Requires-Dist: llama-index-llms-deepseek>=0.1.1; extra == 'llamaindex'
+Requires-Dist: llama-index-llms-google-genai>=0.1.13; extra == 'llamaindex'
+Requires-Dist: llama-index-llms-groq>=0.3.1; extra == 'llamaindex'
+Requires-Dist: llama-index-llms-mistralai>=0.4.0; extra == 'llamaindex'
+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.14; extra == 'openai'
+Provides-Extra: pydantic
+Requires-Dist: pydantic-ai>=0.0.48; extra == 'pydantic'
+Provides-Extra: telemetry
+Requires-Dist: opentelemetry-exporter-otlp>=1.28.0; extra == 'telemetry'
+Requires-Dist: opentelemetry-instrumentation-anthropic==0.40.6; extra == 'telemetry'
+Requires-Dist: opentelemetry-instrumentation-cohere==0.40.6; extra == 'telemetry'
+Requires-Dist: opentelemetry-instrumentation-fastapi==0.54b1; extra == 'telemetry'
+Requires-Dist: opentelemetry-instrumentation-google-generativeai>=0.40.7; extra == 'telemetry'
+Requires-Dist: opentelemetry-instrumentation-ollama==0.40.6; extra == 'telemetry'
+Requires-Dist: opentelemetry-instrumentation-openai==0.40.6; extra == 'telemetry'
+Description-Content-Type: text/markdown
+
+<p align="center">
+  <img src="https://blaxel.ai/logo.png" alt="Blaxel" width=500/>
+</p>
+
+# Python SDK
+
+**Blaxel is a computing platform for AI agent builders, with all the services and infrastructure to build and deploy agents efficiently.** This repository contains the Python SDK to create and manage resources on Blaxel.
+
+## Table of Contents
+
+- [Installation](#installation)
+  - [Authentication](#authentication)
+- [Features](#features)
+- [Quickstart](#quickstart)
+- [Contributing](#contributing)
+- [License](#license)
+
+
+
+## Installation
+
+Install Blaxel SDK which lets you manage Blaxel resources.
+
+```bash
+# Base package (core functionality)
+pip install blaxel
+
+# With specific modules
+pip install "blaxel[telemetry]"
+pip install "blaxel[core,telemetry,crewai]"
+
+# Everything
+pip install "blaxel[all]"
+```
+
+### Available modules
+
+- `blaxel.core` - Core functionality (always available)
+- `blaxel.telemetry` - Telemetry and monitoring
+- `blaxel.crewai` - CrewAI integration
+- `blaxel.openai` - OpenAI integration
+- `blaxel.langgraph` - LangGraph integration
+- `blaxel.livekit` - LiveKit integration
+- `blaxel.llamaindex` - LlamaIndex integration
+- `blaxel.pydantic` - Pydantic AI integration
+- `blaxel.googleadk` - Google ADK integration
+
+
+
+### Authentication
+
+The Blaxel SDK authenticates with your workspace using credentials from these sources, in priority order:
+1. When running on Blaxel, authentication is handled automatically
+2. Variables in your .env file (`BL_WORKSPACE` and `BL_API_KEY`, or see [this page](https://docs.blaxel.ai/Agents/Variables-and-secrets) for other authentication options).
+3. Environment variables from your machine
+4. Configuration file created locally when you log in through Blaxel CLI (or deploy on Blaxel)
+
+When developing locally, the recommended method is to just log in to your workspace with Blaxel CLI. This allows you to run Blaxel SDK functions that will automatically connect to your workspace without additional setup. When you deploy on Blaxel, this connection persists automatically.
+
+When running Blaxel SDK from a remote server that is not Blaxel-hosted, we recommend using environment variables as described in the third option above.
+
+
+
+## Features
+- Agents & MCP servers
+  - [Create MCP servers](https://docs.blaxel.ai/Functions/Create-MCP-server)
+  - [Connect to MCP servers and model APIs hosted on Blaxel](https://docs.blaxel.ai/Agents/Develop-an-agent-ts)
+  - [Call agents from another agent](https://docs.blaxel.ai/Agents/Develop-an-agent-ts#connect-to-another-agent-multi-agent-chaining)
+  - [Deploy on Blaxel](https://docs.blaxel.ai/Agents/Deploy-an-agent)
+- Sandboxes
+  - [Create and update sandboxes and sandbox previews](https://docs.blaxel.ai/Sandboxes/Overview)
+  - [Run filesystem operations and processes on a sandbox](https://docs.blaxel.ai/Sandboxes/Processes)
+- [Use environment variables or secrets](https://docs.blaxel.ai/Agents/Variables-and-secrets)
+
+
+
+## Quickstart
+
+Blaxel CLI gives you a quick way to create new applications: agents, MCP servers, jobs, etc - and deploy them to Blaxel.
+
+**Prerequisites**:
+- **Node.js:** v18 or later.
+- **Blaxel CLI:** Make sure you have Blaxel CLI installed. If not, [install it](https://docs.blaxel.ai/cli-reference/introduction):
+  ```bash
+  curl -fsSL \
+  https://raw.githubusercontent.com/blaxel-ai/toolkit/main/install.sh \
+  | BINDIR=/usr/local/bin sudo -E sh
+  ```
+- **Blaxel login:** Login to Blaxel:
+  ```bash
+    bl login YOUR-WORKSPACE
+  ```
+
+```bash
+bl create-agent-app myfolder
+cd myfolder
+bl deploy
+```
+
+Also available:
+-  `bl create-mcp-server`
+-  `bl create-job`
+
+
+
+## Contributing
+
+Contributions are welcome! Please feel free to submit a Pull Request.
+
+
+
+## License
+
+This project is licensed under the MIT License - see the LICENSE file for details.