blaxel 0.2.31__py3-none-any.whl → 0.2.31rc121__py3-none-any.whl

blaxel/telemetry/exporters.py

@@ -3,15 +3,11 @@ 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
 
@@ -37,7 +33,6 @@ 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."""
 
@@ -54,14 +49,12 @@ 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,15 +24,13 @@ 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,10 +7,7 @@ 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,
@@ -142,22 +139,12 @@ 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
 
@@ -189,11 +176,7 @@ 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,7 +40,6 @@ 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):
@@ -53,8 +52,9 @@ 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,7 +8,6 @@ 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
@@ -25,7 +24,9 @@ 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,7 +2,6 @@
 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
@@ -14,27 +13,26 @@ 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
@@ -47,14 +45,12 @@ 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)
 
@@ -66,7 +62,6 @@ 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
@@ -91,7 +86,6 @@ 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.
@@ -103,7 +97,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,12 +52,10 @@ 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.
 
@@ -79,9 +77,7 @@ 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.
 
@@ -99,7 +95,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.31.dist-info → blaxel-0.2.31rc121.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: blaxel
-Version: 0.2.31
+Version: 0.2.31rc121
 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.14,>=3.10
+Requires-Python: >=3.10
 Requires-Dist: attrs>=21.3.0
 Requires-Dist: httpx>=0.27.0
 Requires-Dist: mcp>=1.9.4