blaxel 0.2.31rc120__py3-none-any.whl → 0.2.32__py3-none-any.whl

blaxel/telemetry/exporters.py

@@ -3,11 +3,15 @@ from __future__ import annotations
 from typing import Callable, Dict, Sequence
 
 try:
-    from opentelemetry.exporter.otlp.proto.http._log_exporter import OTLPLogExporter
+    from opentelemetry.exporter.otlp.proto.http._log_exporter import (
+        OTLPLogExporter,
+    )
     from opentelemetry.exporter.otlp.proto.http.metric_exporter import (
         OTLPMetricExporter,
     )
-    from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
+    from opentelemetry.exporter.otlp.proto.http.trace_exporter import (
+        OTLPSpanExporter,
+    )
     from opentelemetry.sdk._logs import LogData
     from opentelemetry.sdk.metrics.export import MetricExportResult, MetricsData
 
@@ -33,6 +37,7 @@ class DynamicHeadersSpanExporter(OTLPSpanExporter):  # type: ignore[misc]
         self._session.headers.update(self._get_headers())
         return super().export(spans)
 
+
 class DynamicHeadersMetricExporter(OTLPMetricExporter):  # type: ignore[misc]
     """Metric exporter with dynamic headers."""
 
@@ -49,12 +54,14 @@ class DynamicHeadersMetricExporter(OTLPMetricExporter):  # type: ignore[misc]
         self._session.headers.update(self._get_headers())
         return super().export(metrics_data, timeout_millis, **kwargs)
 
+
 class DynamicHeadersLogExporter(OTLPLogExporter):  # type: ignore[misc]
     """Log exporter with dynamic headers."""
+
     def __init__(self, get_headers: Callable[[], Dict[str, str]]):
         self._get_headers = get_headers
         super().__init__()
 
     def export(self, batch: Sequence[LogData]):  # type: ignore[reportUnknownReturnType]
         self._session.headers.update(self._get_headers())
-        return super().export(batch)
+        return super().export(batch)

blaxel/telemetry/instrumentation/blaxel_langgraph.py

@@ -24,13 +24,15 @@ class BlaxelLanggraphInstrumentor(BaseInstrumentor):
 
         # Restore original methods
         if hasattr(
-            blaxel.langgraph.custom.gemini.GeminiRestClient, "_blaxel_original_generate_content"
+            blaxel.langgraph.custom.gemini.GeminiRestClient,
+            "_blaxel_original_generate_content",
         ):
             blaxel.langgraph.custom.gemini.GeminiRestClient.generate_content = (
                 blaxel.langgraph.custom.gemini.GeminiRestClient._blaxel_original_generate_content
             )
             delattr(
-                blaxel.langgraph.custom.gemini.GeminiRestClient, "_blaxel_original_generate_content"
+                blaxel.langgraph.custom.gemini.GeminiRestClient,
+                "_blaxel_original_generate_content",
             )
 
         if hasattr(

blaxel/telemetry/instrumentation/blaxel_langgraph_gemini.py

@@ -7,7 +7,10 @@ from typing import Collection
 
 from opentelemetry import context as context_api
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
-from opentelemetry.instrumentation.utils import _SUPPRESS_INSTRUMENTATION_KEY, unwrap
+from opentelemetry.instrumentation.utils import (
+    _SUPPRESS_INSTRUMENTATION_KEY,
+    unwrap,
+)
 from opentelemetry.semconv_ai import (
     SUPPRESS_LANGUAGE_MODEL_INSTRUMENTATION_KEY,
     LLMRequestTypeValues,
@@ -139,12 +142,22 @@ def _set_input_attributes(span, args, kwargs, llm_model):
     _set_span_attribute(span, SpanAttributes.LLM_REQUEST_MODEL, llm_model)
     _set_span_attribute(span, SpanAttributes.LLM_REQUEST_TEMPERATURE, kwargs.get("temperature"))
     _set_span_attribute(
-        span, SpanAttributes.LLM_REQUEST_MAX_TOKENS, kwargs.get("max_output_tokens")
+        span,
+        SpanAttributes.LLM_REQUEST_MAX_TOKENS,
+        kwargs.get("max_output_tokens"),
     )
     _set_span_attribute(span, SpanAttributes.LLM_REQUEST_TOP_P, kwargs.get("top_p"))
     _set_span_attribute(span, SpanAttributes.LLM_TOP_K, kwargs.get("top_k"))
-    _set_span_attribute(span, SpanAttributes.LLM_PRESENCE_PENALTY, kwargs.get("presence_penalty"))
-    _set_span_attribute(span, SpanAttributes.LLM_FREQUENCY_PENALTY, kwargs.get("frequency_penalty"))
+    _set_span_attribute(
+        span,
+        SpanAttributes.LLM_PRESENCE_PENALTY,
+        kwargs.get("presence_penalty"),
+    )
+    _set_span_attribute(
+        span,
+        SpanAttributes.LLM_FREQUENCY_PENALTY,
+        kwargs.get("frequency_penalty"),
+    )
 
     return
 
@@ -176,7 +189,11 @@ def _set_response_attributes(span, response, llm_model):
                 _set_span_attribute(span, f"{prefix}.content", item.text)
                 _set_span_attribute(span, f"{prefix}.role", "assistant")
         elif isinstance(response.text, str):
-            _set_span_attribute(span, f"{SpanAttributes.LLM_COMPLETIONS}.0.content", response.text)
+            _set_span_attribute(
+                span,
+                f"{SpanAttributes.LLM_COMPLETIONS}.0.content",
+                response.text,
+            )
             _set_span_attribute(span, f"{SpanAttributes.LLM_COMPLETIONS}.0.role", "assistant")
     else:
         if isinstance(response, list):

blaxel/telemetry/instrumentation/utils.py

@@ -40,6 +40,7 @@ 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):
@@ -52,9 +53,8 @@ class JSONEncoder(json.JSONEncoder):
 
 
 def should_send_prompts():
-    return (
-        os.getenv("TRACELOOP_TRACE_CONTENT") or "true"
-    ).lower() == "true"
+    return (os.getenv("TRACELOOP_TRACE_CONTENT") or "true").lower() == "true"
+
 
 @dont_throw
 def process_request(span, args, kwargs):

blaxel/telemetry/log/log.py

@@ -8,6 +8,7 @@ from opentelemetry.sdk._logs._internal.export import LogExporter
 
 _logger = logging.getLogger(__name__)
 
+
 class AsyncLogRecordProcessor(LogRecordProcessor):
     """This is an implementation of LogRecordProcessor which passes
     received logs in the export-friendly LogData representation to the
@@ -24,9 +25,7 @@ class AsyncLogRecordProcessor(LogRecordProcessor):
     def _start_worker(self):
         """Start the background worker thread."""
         self._worker_thread = threading.Thread(
-            target=self._process_logs,
-            daemon=True,
-            name="LogExporterWorker"
+            target=self._process_logs, daemon=True, name="LogExporterWorker"
         )
         self._worker_thread.start()
 

blaxel/telemetry/log/logger.py

@@ -2,6 +2,7 @@
 This module provides a custom colored formatter for logging and an initialization function
 to set up logging configurations for Blaxel applications.
 """
+
 import json
 import logging
 import os
@@ -13,26 +14,27 @@ class JsonFormatter(logging.Formatter):
     """
     A logger compatible with standard json logging.
     """
+
     def __init__(self):
         super().__init__()
-        self.trace_id_name = os.environ.get('BL_LOGGER_TRACE_ID', 'trace_id')
-        self.span_id_name = os.environ.get('BL_LOGGER_SPAN_ID', 'span_id')
-        self.labels_name = os.environ.get('BL_LOGGER_LABELS', 'labels')
-        self.trace_id_prefix = os.environ.get('BL_LOGGER_TRACE_ID_PREFIX', '')
-        self.span_id_prefix = os.environ.get('BL_LOGGER_SPAN_ID_PREFIX', '')
-        self.task_index = os.environ.get('BL_TASK_KEY', 'TASK_INDEX')
-        self.task_prefix = os.environ.get('BL_TASK_PREFIX', '')
-        self.execution_key = os.environ.get('BL_EXECUTION_KEY', 'BL_EXECUTION_ID')
-        self.execution_prefix = os.environ.get('BL_EXECUTION_PREFIX', '')
+        self.trace_id_name = os.environ.get("BL_LOGGER_TRACE_ID", "trace_id")
+        self.span_id_name = os.environ.get("BL_LOGGER_SPAN_ID", "span_id")
+        self.labels_name = os.environ.get("BL_LOGGER_LABELS", "labels")
+        self.trace_id_prefix = os.environ.get("BL_LOGGER_TRACE_ID_PREFIX", "")
+        self.span_id_prefix = os.environ.get("BL_LOGGER_SPAN_ID_PREFIX", "")
+        self.task_index = os.environ.get("BL_TASK_KEY", "TASK_INDEX")
+        self.task_prefix = os.environ.get("BL_TASK_PREFIX", "")
+        self.execution_key = os.environ.get("BL_EXECUTION_KEY", "BL_EXECUTION_ID")
+        self.execution_prefix = os.environ.get("BL_EXECUTION_PREFIX", "")
 
     def format(self, record):
         """
         Formats the log record by converting it to a JSON object with trace context and environment variables.
         """
         log_entry = {
-            'message': record.getMessage(),
-            'severity': record.levelname,
-            self.labels_name: {}
+            "message": record.getMessage(),
+            "severity": record.levelname,
+            self.labels_name: {},
         }
 
         # Add trace context if available
@@ -45,12 +47,14 @@ class JsonFormatter(logging.Formatter):
         # Add task ID if available
         task_id = os.environ.get(self.task_index)
         if task_id:
-            log_entry[self.labels_name]['blaxel-task'] = f"{self.task_prefix}{task_id}"
+            log_entry[self.labels_name]["blaxel-task"] = f"{self.task_prefix}{task_id}"
 
         # Add execution ID if available
         execution_id = os.environ.get(self.execution_key)
         if execution_id:
-            log_entry[self.labels_name]['blaxel-execution'] = f"{self.execution_prefix}{execution_id.split('-')[-1]}"
+            log_entry[self.labels_name]["blaxel-execution"] = (
+                f"{self.execution_prefix}{execution_id.split('-')[-1]}"
+            )
 
         return json.dumps(log_entry)
 
@@ -62,6 +66,7 @@ class ColoredFormatter(logging.Formatter):
     Attributes:
         COLORS (dict): A mapping of log level names to their corresponding ANSI color codes.
     """
+
     COLORS = {
         "DEBUG": "\033[1;36m",  # Cyan
         "INFO": "\033[1;32m",  # Green
@@ -86,6 +91,7 @@ class ColoredFormatter(logging.Formatter):
         record.levelname = f"{color}{record.levelname}\033[0m:{tab}"
         return super().format(record)
 
+
 def init_logger(log_level: str):
     """
     Initializes the logging configuration for Blaxel.
@@ -97,7 +103,7 @@ def init_logger(log_level: str):
         log_level (str): The logging level to set (e.g., "DEBUG", "INFO").
     """
     # Disable urllib3 logging
-    logging.getLogger('urllib3').setLevel(logging.CRITICAL)
+    logging.getLogger("urllib3").setLevel(logging.CRITICAL)
     logging.getLogger("httpx").setLevel(logging.CRITICAL)
     handler = logging.StreamHandler()
 

blaxel/telemetry/span.py

@@ -12,7 +12,7 @@ from opentelemetry.trace import Span, Tracer
 
 from blaxel.core import settings
 
-T = TypeVar('T')
+T = TypeVar("T")
 
 
 class DefaultAttributesSpanProcessor(SpanProcessor):
@@ -21,7 +21,7 @@ class DefaultAttributesSpanProcessor(SpanProcessor):
     def __init__(self, default_attributes: Dict[str, str]):
         self.default_attributes = default_attributes
 
-    def on_start(self, span: SdkSpan, parent_context = None) -> None:
+    def on_start(self, span: SdkSpan, parent_context=None) -> None:
         """Add default attributes to the span when it starts."""
         for key, value in self.default_attributes.items():
             span.set_attribute(key, value)
@@ -52,10 +52,12 @@ class SpanManager:
             "blaxel.environment": settings.env,
             "workload.id": settings.name,
             "workload.type": f"{settings.type}s",
-            "workspace": settings.workspace
+            "workspace": settings.workspace,
         }
 
-    def create_active_span(self, name: str, attributes: Dict[str, Any], parent: Span | None = None) -> ContextManager[Span]:
+    def create_active_span(
+        self, name: str, attributes: Dict[str, Any], parent: Span | None = None
+    ) -> ContextManager[Span]:
         """
         Creates an active span and executes the provided function within its context.
 
@@ -77,7 +79,9 @@ class SpanManager:
             context = trace.set_span_in_context(parent)
         return self.tracer.start_as_current_span(name, attributes=attributes, context=context)
 
-    def create_span(self, name: str, attributes: Dict[str, Any], parent: Span | None = None) -> Span:
+    def create_span(
+        self, name: str, attributes: Dict[str, Any], parent: Span | None = None
+    ) -> Span:
         """
         Creates a new span without making it active.
 
@@ -95,7 +99,7 @@ class SpanManager:
             "blaxel.environment": settings.env,
             "workload.id": settings.name,
             "workload.type": f"{settings.type}s",
-            "workspace": settings.workspace
+            "workspace": settings.workspace,
         }
 
         context = None

{blaxel-0.2.31rc120.dist-info → blaxel-0.2.32.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: blaxel
-Version: 0.2.31rc120
+Version: 0.2.32
 Summary: Blaxel - AI development platform SDK
 Project-URL: Homepage, https://blaxel.ai
 Project-URL: Documentation, https://docs.blaxel.ai
@@ -8,7 +8,7 @@ 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-Python: <3.14,>=3.10
 Requires-Dist: attrs>=21.3.0
 Requires-Dist: httpx>=0.27.0
 Requires-Dist: mcp>=1.9.4