beamlit 0.0.34rc53__tar.gz → 0.0.34rc54__tar.gz

{beamlit-0.0.34rc53 → beamlit-0.0.34rc54}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beamlit
-Version: 0.0.34rc53
+Version: 0.0.34rc54
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 Requires-Python: >=3.12
@@ -17,7 +17,6 @@ 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-logging>=0.49b2
 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.34rc53 → beamlit-0.0.34rc54}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "beamlit"
-version = "0.0.34.pre-53"
+version = "0.0.34.pre-54"
 description = "Add your description here"
 readme = "README.md"
 authors = [
@@ -32,7 +32,6 @@ dependencies = [
     "opentelemetry-api>=1.28.2",
     "opentelemetry-instrumentation-fastapi>=0.49b2",
     "opentelemetry-instrumentation-httpx>=0.49b2",
-    "opentelemetry-instrumentation-logging>=0.49b2"
 ]
 [dependency-groups]
 dev = [

{beamlit-0.0.34rc53 → beamlit-0.0.34rc54}/src/beamlit/common/instrumentation.py

@@ -10,7 +10,6 @@ from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import (
 )
 from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
 from opentelemetry.instrumentation.httpx import HTTPXClientInstrumentor
-from opentelemetry.instrumentation.logging import LoggingInstrumentor
 from opentelemetry.metrics import NoOpMeterProvider
 from opentelemetry.sdk.metrics import MeterProvider
 from opentelemetry.sdk.metrics.export import PeriodicExportingMetricReader
@@ -112,12 +111,17 @@ def instrument_app(app: FastAPI):
         meter = meter_provider.get_meter(__name__)
     else:
         meter = metrics.get_meter(__name__)
-
     # Only instrument the app when OpenTelemetry is enabled
-    FastAPIInstrumentor.instrument_app(
-        app=app, tracer_provider=trace.get_tracer_provider(), meter_provider=metrics.get_meter_provider()
-    )
-    HTTPXClientInstrumentor().instrument(meter_provider=metrics.get_meter_provider())
-    LoggingInstrumentor(tracer_provider=trace.get_tracer_provider()).instrument(
-        set_logging_format=True
-    )
+    FastAPIInstrumentor.instrument_app(app)
+    HTTPXClientInstrumentor().instrument()
+
+
+def shutdown_instrumentation():
+    if tracer is not None:
+        trace_provider = trace.get_tracer_provider()
+        if isinstance(trace_provider, TracerProvider):
+            trace_provider.shutdown()
+    if meter is not None:
+        meter_provider = metrics.get_meter_provider()
+        if isinstance(meter_provider, MeterProvider):
+            meter_provider.shutdown()

{beamlit-0.0.34rc53 → beamlit-0.0.34rc54}/src/beamlit/serve/app.py

@@ -3,6 +3,7 @@ import importlib
 import os
 import sys
 import traceback
+from contextlib import asynccontextmanager
 from logging import getLogger
 from uuid import uuid4
 
@@ -13,9 +14,9 @@ from traceloop.sdk import Traceloop
 
 from beamlit.common import HTTPError, get_settings, init
 from beamlit.common.instrumentation import (
-    get_metrics_exporter,
     get_resource_attributes,
     get_span_exporter,
+    shutdown_instrumentation,
     instrument_app,
 )
 
@@ -41,16 +42,14 @@ logger.info(
     f" on {settings.server.host}:{settings.server.port}"
 )
 
-if settings.enable_opentelemetry:
-    Traceloop.init(
-        app_name=settings.name,
-        exporter=get_span_exporter(),
-        metrics_exporter=get_metrics_exporter(),
-        resource_attributes=get_resource_attributes(),
-        should_enrich_metrics=os.getenv("ENRICHED_METRICS", "false") == "true",
-    )
 
-app = FastAPI(docs_url=None, redoc_url=None)
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    yield
+    shutdown_instrumentation()
+
+
+app = FastAPI(docs_url=None, redoc_url=None, lifespan=lifespan)
 app.add_middleware(
     CorrelationIdMiddleware,
     header_name="x-beamlit-request-id",
@@ -59,6 +58,13 @@ app.add_middleware(
 app.add_middleware(AddProcessTimeHeader)
 app.add_middleware(AccessLogMiddleware)
 instrument_app(app)
+if settings.enable_opentelemetry:
+    Traceloop.init(
+        app_name=settings.name,
+        exporter=get_span_exporter(),
+        resource_attributes=get_resource_attributes(),
+        should_enrich_metrics=os.getenv("ENRICHED_METRICS", "false") == "true",
+    )
 
 
 @app.get("/health")

{beamlit-0.0.34rc53 → beamlit-0.0.34rc54}/uv.lock

@@ -172,7 +172,6 @@ dependencies = [
     { name = "opentelemetry-exporter-otlp" },
     { name = "opentelemetry-instrumentation-fastapi" },
     { name = "opentelemetry-instrumentation-httpx" },
-    { name = "opentelemetry-instrumentation-logging" },
     { name = "opentelemetry-sdk" },
     { name = "pydantic" },
     { name = "pydantic-settings" },
@@ -182,7 +181,7 @@ dependencies = [
     { name = "traceloop-sdk" },
 ]
 
-[package.dependency-groups]
+[package.dev-dependencies]
 dev = [
     { name = "ruff" },
 ]
@@ -202,7 +201,6 @@ requires-dist = [
     { name = "opentelemetry-exporter-otlp", specifier = ">=1.28.2" },
     { name = "opentelemetry-instrumentation-fastapi", specifier = ">=0.49b2" },
     { name = "opentelemetry-instrumentation-httpx", specifier = ">=0.49b2" },
-    { name = "opentelemetry-instrumentation-logging", specifier = ">=0.49b2" },
     { name = "opentelemetry-sdk", specifier = ">=1.28.2" },
     { name = "pydantic", specifier = ">=2.10.3,<2.11.0" },
     { name = "pydantic-settings", specifier = ">=2.6.1,<2.7.0" },
@@ -212,7 +210,7 @@ requires-dist = [
     { name = "traceloop-sdk", specifier = ">=0.33.12" },
 ]
 
-[package.metadata.dependency-groups]
+[package.metadata.requires-dev]
 dev = [{ name = "ruff", specifier = ">=0.8.2" }]
 
 [[package]]