beamlit 0.0.26rc22__py3-none-any.whl → 0.0.27rc23__py3-none-any.whl

beamlit/common/instrumentation.py

@@ -0,0 +1,101 @@
+from typing import Any
+
+from fastapi import FastAPI
+from opentelemetry import metrics, trace
+from opentelemetry.exporter.otlp.proto.grpc.metric_exporter import (
+    OTLPMetricExporter,
+)
+from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import (
+    OTLPSpanExporter,
+)
+from opentelemetry.instrumentation.fastapi import (  # type: ignore
+    FastAPIInstrumentor,
+)
+from opentelemetry.instrumentation.httpx import (  # type: ignore
+    HTTPXClientInstrumentor,
+)
+from opentelemetry.instrumentation.logging import (  # type: ignore
+    LoggingInstrumentor,
+)
+from opentelemetry.sdk.metrics import MeterProvider
+from opentelemetry.sdk.metrics.export import (
+    PeriodicExportingMetricReader,
+)
+from opentelemetry.sdk.resources import Resource
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import BatchSpanProcessor
+from typing_extensions import Dict
+
+from .settings import get_settings
+
+tracer: trace.Tracer | None = None
+meter: metrics.Meter | 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 get_resource_attributes() -> Dict[str, Any]:
+    resources = Resource.create()
+    resources_dict: 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
+
+
+def get_metrics_exporter() -> OTLPMetricExporter:
+    return OTLPMetricExporter()
+
+
+def get_span_exporter() -> OTLPSpanExporter:
+    return OTLPSpanExporter()
+
+
+def instrument_app(app: FastAPI):
+    global tracer
+    global meter
+    settings = get_settings()
+    if settings is None:
+        raise Exception("Settings are not initialized")
+    resource = Resource.create(
+        {
+            "service.name": settings.name,
+            "service.namespace": settings.workspace,
+            "service.workspace": settings.workspace,
+        }
+    )
+    # Set up the TracerProvider
+    trace_provider = TracerProvider(resource=resource)
+    span_processor = BatchSpanProcessor(get_span_exporter())
+    trace_provider.add_span_processor(span_processor)
+    trace.set_tracer_provider(trace_provider)
+    tracer = trace_provider.get_tracer(__name__)
+
+    metrics_exporter = PeriodicExportingMetricReader(get_metrics_exporter())
+    meter_provider = MeterProvider(
+        resource=resource, metric_readers=[metrics_exporter]
+    )
+    metrics.set_meter_provider(meter_provider)
+    meter = meter_provider.get_meter(__name__)
+
+    FastAPIInstrumentor.instrument_app(  # type: ignore
+        app=app, tracer_provider=trace_provider, meter_provider=meter_provider
+    )
+    HTTPXClientInstrumentor().instrument(meter_provider=meter_provider)  # type: ignore
+    LoggingInstrumentor(tracer_provider=trace_provider).instrument(
+        set_logging_format=True
+    )

beamlit/serve/app.py

@@ -8,8 +8,15 @@ from uuid import uuid4
 
 from asgi_correlation_id import CorrelationIdMiddleware
 from beamlit.common.settings import get_settings, init
+from beamlit.common.instrumentation import (
+    get_metrics_exporter,
+    get_resource_attributes,
+    get_span_exporter,
+    instrument_app,
+)
 from fastapi import FastAPI, Request, Response
 from fastapi.responses import JSONResponse
+from traceloop.sdk import Traceloop
 
 from .middlewares import AccessLogMiddleware, AddProcessTimeHeader
 
@@ -33,6 +40,14 @@ logger.info(
     f" on {settings.server.host}:{settings.server.port}"
 )
 
+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)
 app.add_middleware(
     CorrelationIdMiddleware,
@@ -41,7 +56,7 @@ app.add_middleware(
 )
 app.add_middleware(AddProcessTimeHeader)
 app.add_middleware(AccessLogMiddleware)
-
+instrument_app(app)
 
 @app.get("/health")
 async def health():

{beamlit-0.0.26rc22.dist-info → beamlit-0.0.27rc23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beamlit
-Version: 0.0.26rc22
+Version: 0.0.27rc23
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 Requires-Python: >=3.12
@@ -13,11 +13,18 @@ Requires-Dist: langchain-core<0.4.0,>=0.3.13
 Requires-Dist: langchain-openai<0.3.0,>=0.2.4
 Requires-Dist: langgraph<0.3.0,>=0.2.40
 Requires-Dist: mcp>=1.1.2
+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
 Requires-Dist: python-dateutil>=2.8.0
 Requires-Dist: pyyaml<6.1.0,>=6.0.2
 Requires-Dist: requests<2.33.0,>=2.32.3
+Requires-Dist: traceloop-sdk>=0.33.12
 Description-Content-Type: text/markdown
 
 # beamlit

{beamlit-0.0.26rc22.dist-info → beamlit-0.0.27rc23.dist-info}/RECORD

@@ -122,6 +122,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=yDoMJDKj-xjTGl7U1YI59KpWxiOV65HSiUulgO8xdTA,277
 beamlit/common/generate.py,sha256=LtdCju_QayRS4lZrrb_0VHqWWvTcv4Mbf-iV1TB_Qko,7522
+beamlit/common/instrumentation.py,sha256=MsBDfFcMYqGDiHHj4j5hLHE4EWxZExkhmCeFS3SKzJY,3181
 beamlit/common/logger.py,sha256=VFRbaZh93n8ZGugeeYKe88IP2nI3g2JNa7XN4j8wVJE,1116
 beamlit/common/secrets.py,sha256=sid81bOe3LflkMKDHwBsBs9nIju8bp5-v9qU9gkyNMc,212
 beamlit/common/settings.py,sha256=igwhNY6DPm0tiSW-_2fCkllHdTyXVthVAT6p3kggJxo,5435
@@ -238,10 +239,10 @@ beamlit/models/websocket_channel.py,sha256=jg3vN7yS_oOIwGtndtIUr1LsyEA58RXLXahqS
 beamlit/models/workspace.py,sha256=l__bIpbA4oJvxXo7UbEoCcqkvu9MiNt5aXXpZ3bgwHg,4309
 beamlit/models/workspace_labels.py,sha256=WbnUY6eCTkUNdY7hhhSF-KQCl8fWFfkCf7hzCTiNp4A,1246
 beamlit/models/workspace_user.py,sha256=70CcifQWYbeWG7TDui4pblTzUe5sVK0AS19vNCzKE8g,3423
-beamlit/serve/app.py,sha256=4mtrrYW32ywFIapL26Uvsp3AnKcEpa8G-juJy_0A5Qo,2618
+beamlit/serve/app.py,sha256=OpwPjRdyHZK6J-ziPwhiRDGGa2mvCrFVcBFE6alJVOM,3071
 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.26rc22.dist-info/METADATA,sha256=k7j9e8a8SMtmeRn0f66x1fFVX3aGx62hDYQtLeCfu2c,2053
-beamlit-0.0.26rc22.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.26rc22.dist-info/RECORD,,
+beamlit-0.0.27rc23.dist-info/METADATA,sha256=wZtsDtJuxfbhkS9ATsx7Ed6Eowpg7I9AJGtHNwseexs,2405
+beamlit-0.0.27rc23.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.27rc23.dist-info/RECORD,,