beamlit 0.0.34rc55__py3-none-any.whl → 0.0.34rc56__py3-none-any.whl

beamlit/common/instrumentation.py

@@ -15,6 +15,9 @@ from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import (
 )
 from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
 from opentelemetry.instrumentation.httpx import HTTPXClientInstrumentor
+from opentelemetry.instrumentation.system_metrics import (
+    SystemMetricsInstrumentor,
+)
 from opentelemetry.metrics import NoOpMeterProvider
 from opentelemetry.sdk._logs import LoggerProvider, LoggingHandler
 from opentelemetry.sdk._logs.export import BatchLogRecordProcessor
@@ -26,6 +29,8 @@ from opentelemetry.sdk.trace.export import BatchSpanProcessor
 from opentelemetry.trace import NoOpTracerProvider
 from typing_extensions import Dict
 
+from beamlit.authentication import get_authentication_headers
+
 from .settings import get_settings
 
 tracer: trace.Tracer | None = None
@@ -33,16 +38,13 @@ meter: metrics.Meter | None = None
 logger: LoggerProvider | None = None
 
 
-def get_tracer() -> trace.Tracer:
-    if tracer is None:
-        raise Exception("Tracer is not initialized")
-    return tracer
-
-
-def get_meter() -> metrics.Meter:
-    if meter is None:
-        raise Exception("Meter is not initialized")
-    return meter
+def auth_headers() -> Dict[str, str]:
+    settings = get_settings()
+    headers = get_authentication_headers(settings)
+    return {
+        "x-beamlit-authorization": headers.get("X-Beamlit-Authorization", ""),
+        "x-beamlit-workspace": headers.get("X-Beamlit-Workspace", ""),
+    }
 
 
 def get_logger() -> LoggerProvider:
@@ -57,8 +59,6 @@ def get_resource_attributes() -> Dict[str, Any]:
     for key in resources.attributes:
         resources_dict[key] = resources.attributes[key]
     settings = get_settings()
-    if settings is None:
-        raise Exception("Settings are not initialized")
     resources_dict["workspace"] = settings.workspace
     resources_dict["service.name"] = settings.name
     return resources_dict
@@ -66,33 +66,29 @@ def get_resource_attributes() -> Dict[str, Any]:
 
 def get_metrics_exporter() -> OTLPMetricExporter | None:
     settings = get_settings()
-    if settings is None:
-        raise Exception("Settings are not initialized")
     if not settings.enable_opentelemetry:
-        # Return None or a NoOpExporter equivalent
         return None
-    return OTLPMetricExporter()
+    return OTLPMetricExporter(headers=auth_headers())
 
 
 def get_span_exporter() -> OTLPSpanExporter | None:
     settings = get_settings()
     if not settings.enable_opentelemetry:
         return None
-    return OTLPSpanExporter()
+    return OTLPSpanExporter(headers=auth_headers())
 
 
 def get_log_exporter() -> OTLPLogExporter | None:
     settings = get_settings()
     if not settings.enable_opentelemetry:
         return None
-    return OTLPLogExporter()
+    return OTLPLogExporter(headers=auth_headers())
 
 
 def instrument_app(app: FastAPI):
     global tracer
     global meter
     settings = get_settings()
-
     if not settings.enable_opentelemetry:
         # Use NoOp implementations to stub tracing and metrics
         trace.set_tracer_provider(NoOpTracerProvider())
@@ -147,6 +143,7 @@ def instrument_app(app: FastAPI):
     # Only instrument the app when OpenTelemetry is enabled
     FastAPIInstrumentor.instrument_app(app)
     HTTPXClientInstrumentor().instrument()
+    SystemMetricsInstrumentor().instrument()
 
 
 def shutdown_instrumentation():

{beamlit-0.0.34rc55.dist-info → beamlit-0.0.34rc56.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beamlit
-Version: 0.0.34rc55
+Version: 0.0.34rc56
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 Requires-Python: >=3.12
@@ -17,6 +17,7 @@ Requires-Dist: opentelemetry-api>=1.28.2
 Requires-Dist: opentelemetry-exporter-otlp>=1.28.2
 Requires-Dist: opentelemetry-instrumentation-fastapi>=0.49b2
 Requires-Dist: opentelemetry-instrumentation-httpx>=0.49b2
+Requires-Dist: opentelemetry-instrumentation-system-metrics>=0.50b0
 Requires-Dist: opentelemetry-sdk>=1.28.2
 Requires-Dist: pydantic-settings<2.7.0,>=2.6.1
 Requires-Dist: pydantic<2.11.0,>=2.10.3

{beamlit-0.0.34rc55.dist-info → beamlit-0.0.34rc56.dist-info}/RECORD

@@ -130,7 +130,7 @@ beamlit/authentication/credentials.py,sha256=p_1xenabCbQuRz7BiFk7oTK4uCxAt_zoyku
 beamlit/authentication/device_mode.py,sha256=tmr22gllKOZwBRub_QjF5pYa425x-nE8tQNpZ_EGR6g,3644
 beamlit/common/__init__.py,sha256=saX5X3hRCJ9erSlXuSkZ2VGgquvpgdcofAU_9sM4bCE,354
 beamlit/common/error.py,sha256=f9oJDFxhoHK-vpjxBgEp0NwWIk0N_THPemUI7uQxVzU,270
-beamlit/common/instrumentation.py,sha256=bW3UN0Ca24zPFZSYjqWkrFfBVDp_yL10T3jVD403KvI,5386
+beamlit/common/instrumentation.py,sha256=8aR5XLLq22_vScJlDqQVxvqFG-0K4ymqJ98DUp7Lb74,5471
 beamlit/common/logger.py,sha256=nN_dSOl4bs13QU3Rod-w3e3jYOnlSrHx3_bs-ACY6Aw,1115
 beamlit/common/secrets.py,sha256=sid81bOe3LflkMKDHwBsBs9nIju8bp5-v9qU9gkyNMc,212
 beamlit/common/settings.py,sha256=b2rvby-ufG3M0AB1ReoWFM-1EzF1LaE-gbokO9HvQDI,3810
@@ -256,6 +256,6 @@ beamlit/serve/app.py,sha256=gYQvUK_S7g0Em-idND8HrVDqgg5LlIemheSGlX2Jj8U,3638
 beamlit/serve/middlewares/__init__.py,sha256=1dVmnOmhAQWvWktqHkKSIX-YoF6fmMU8xkUQuhg_rJU,148
 beamlit/serve/middlewares/accesslog.py,sha256=Mu4T4_9OvHybjA0ApzZFpgi2C8f3X1NbUk-76v634XM,631
 beamlit/serve/middlewares/processtime.py,sha256=lDAaIasZ4bwvN-HKHvZpaD9r-yrkVNZYx4abvbjbrCg,411
-beamlit-0.0.34rc55.dist-info/METADATA,sha256=4yvNNr_yX2lRYsd14PFw8dB6ykRdp1BIBe0lKOeckM4,2344
-beamlit-0.0.34rc55.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.34rc55.dist-info/RECORD,,
+beamlit-0.0.34rc56.dist-info/METADATA,sha256=b7oOq6dqYesq2q5ivAj5ckFcfIZ_KLdOW_HUokE0Lpc,2412
+beamlit-0.0.34rc56.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.34rc56.dist-info/RECORD,,