unique-toolkit 2026.34.0.dev7__tar.gz → 2026.34.0.dev8__tar.gz

{unique_toolkit-2026.34.0.dev7 → unique_toolkit-2026.34.0.dev8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.34.0.dev7
+Version: 2026.34.0.dev8
 Summary: 
 Author: Cedric Klinkert, Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri, Ejiroghene Oki-Peter, Rami Azouz, Ricardo Visini, Sanzio Monti
 Author-email: Cedric Klinkert <cedric.klinkert@unique.ch>, Martin Fadler <martin.fadler@unique.ch>, Sadique Sheik <sadique@unique.ch>, Fabian Schläpfer <fabian@unique.ch>, Pascal Hauri <pascal@unique.ch>, Ejiroghene Oki-Peter <oki-peter.ejiroghene@unique.ai>, Rami Azouz <rami.ext@unique.ch>, Ricardo Visini <ricardo.visini@unique.ch>, Sanzio Monti <sanzio.monti@unique.ai>
@@ -44,6 +44,8 @@ Requires-Dist: langchain-core>=1.0.0,<2 ; extra == 'langchain'
 Requires-Dist: prometheus-client>=0.24.1,<1 ; extra == 'monitoring'
 Requires-Dist: opentelemetry-sdk>=1.27,<2 ; extra == 'otel'
 Requires-Dist: opentelemetry-exporter-otlp-proto-http>=1.27,<2 ; extra == 'otel'
+Requires-Dist: opentelemetry-instrumentation-fastapi>=0.65b0,<1 ; extra == 'otel'
+Requires-Dist: opentelemetry-instrumentation-requests>=0.65b0,<1 ; extra == 'otel'
 Requires-Python: >=3.12
 Provides-Extra: fastapi
 Provides-Extra: langchain

{unique_toolkit-2026.34.0.dev7 → unique_toolkit-2026.34.0.dev8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "2026.34.0.dev7"
+version = "2026.34.0.dev8"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"
@@ -66,6 +66,8 @@ monitoring = [
 otel = [
     "opentelemetry-sdk>=1.27,<2",
     "opentelemetry-exporter-otlp-proto-http>=1.27,<2",
+    "opentelemetry-instrumentation-fastapi>=0.65b0,<1",
+    "opentelemetry-instrumentation-requests>=0.65b0,<1",
 ]
 [build-system]
 requires = ["uv_build>=0.8.14,<0.9.0"]
@@ -79,6 +81,8 @@ dev = [
     "types-cachetools>=5.0,<6",
     "opentelemetry-sdk>=1.27,<2",
     "opentelemetry-exporter-otlp-proto-http>=1.27,<2",
+    "opentelemetry-instrumentation-fastapi>=0.65b0,<1",
+    "opentelemetry-instrumentation-requests>=0.65b0,<1",
 ]
 
 [tool.uv]
@@ -132,6 +136,8 @@ DEP002 = [
     "uvicorn",
     "opentelemetry-sdk",
     "opentelemetry-exporter-otlp-proto-http",
+    "opentelemetry-instrumentation-fastapi",
+    "opentelemetry-instrumentation-requests",
 ]
 DEP004 = ["langchain_openai", "langchain_core"]
 

{unique_toolkit-2026.34.0.dev7 → unique_toolkit-2026.34.0.dev8}/unique_toolkit/monitoring/__init__.py

@@ -4,6 +4,8 @@ from unique_toolkit.monitoring.tracing import (
     TracingSettings,
     configure_tracing,
     inject_trace_headers,
+    instrument_fastapi_app,
+    instrument_requests,
 )
 
 __all__ = [
@@ -12,6 +14,8 @@ __all__ = [
     "TracingSettings",
     "configure_tracing",
     "inject_trace_headers",
+    "instrument_fastapi_app",
+    "instrument_requests",
 ]
 
 _MONITORING_AVAILABLE = False

{unique_toolkit-2026.34.0.dev7 → unique_toolkit-2026.34.0.dev8}/unique_toolkit/monitoring/tracing.py

@@ -9,7 +9,10 @@ from typing import TYPE_CHECKING, ClassVar, Self, TypeAlias, cast
 from pydantic import AliasChoices, Field
 from pydantic_settings import BaseSettings, SettingsConfigDict
 
+from unique_toolkit.app.find_env_file import find_env_file
+
 if TYPE_CHECKING:
+    from fastapi import FastAPI  # pyright: ignore[reportMissingImports]
     from opentelemetry.trace.propagation.tracecontext import (
         TraceContextTextMapPropagator,
     )
@@ -25,6 +28,10 @@ _OTEL_EXTRA_MESSAGE = (
     "OpenTelemetry tracing requires the unique_toolkit[otel] extra. "
     "Install it with: uv add 'unique-toolkit[otel]'"
 )
+_FASTAPI_EXTRA_MESSAGE = (
+    "FastAPI instrumentation requires the unique_toolkit[fastapi] extra. "
+    "Install it with: uv add 'unique-toolkit[fastapi]'"
+)
 
 
 class TraceExporter(StrEnum):
@@ -42,6 +49,8 @@ class TracingSettings(BaseSettings):
         env_prefix="OTEL_",
         env_ignore_empty=True,
         extra="ignore",
+        env_file=find_env_file(".env", required=False),
+        env_file_encoding="utf-8",
     )
 
     traces_exporter: TraceExporter | None = None
@@ -90,6 +99,11 @@ def _missing_otel_extra() -> ImportError:
     return ImportError(_OTEL_EXTRA_MESSAGE)
 
 
+def _missing_fastapi_extra() -> ImportError:
+    """Create a consistent error for an unavailable optional FastAPI dependency."""
+    return ImportError(_FASTAPI_EXTRA_MESSAGE)
+
+
 def _trace_context_propagator() -> TraceContextTextMapPropagator:
     """Return a W3C trace-context propagator without forwarding baggage."""
     try:
@@ -165,9 +179,12 @@ def configure_tracing(
 
     try:
         from opentelemetry import trace
+        from opentelemetry.baggage.propagation import W3CBaggagePropagator
         from opentelemetry.exporter.otlp.proto.http.trace_exporter import (
             OTLPSpanExporter,
         )
+        from opentelemetry.propagate import set_global_textmap
+        from opentelemetry.propagators.composite import CompositePropagator
         from opentelemetry.sdk.resources import Resource
         from opentelemetry.sdk.trace import TracerProvider
         from opentelemetry.sdk.trace.export import (
@@ -176,6 +193,9 @@ def configure_tracing(
             SimpleSpanProcessor,
         )
         from opentelemetry.trace import ProxyTracerProvider
+        from opentelemetry.trace.propagation.tracecontext import (
+            TraceContextTextMapPropagator,
+        )
     except ImportError as error:
         raise _missing_otel_extra() from error
 
@@ -196,6 +216,54 @@ def configure_tracing(
     if exporter is TraceExporter.CONSOLE:
         provider.add_span_processor(SimpleSpanProcessor(ConsoleSpanExporter()))
     else:
-        provider.add_span_processor(BatchSpanProcessor(OTLPSpanExporter()))
+        # Sizing pinned to match the Node services' instrumentation.ts.
+        provider.add_span_processor(
+            BatchSpanProcessor(
+                OTLPSpanExporter(),
+                max_queue_size=2048,
+                max_export_batch_size=512,
+                schedule_delay_millis=5000,
+            )
+        )
     trace.set_tracer_provider(provider)
+    # Trace-context + baggage, the shared subset of the Node services' propagator.
+    set_global_textmap(
+        CompositePropagator([TraceContextTextMapPropagator(), W3CBaggagePropagator()])
+    )
     return True
+
+
+def instrument_fastapi_app(app: FastAPI, *, excluded_urls: str | None = None) -> None:
+    """Trace inbound requests for an already-created FastAPI app instance.
+
+    Uses ``FastAPIInstrumentor.instrument_app`` rather than the global
+    ``FastAPIInstrumentor().instrument()``: the global form swaps out the
+    ``fastapi.FastAPI`` class, so it only takes effect for app objects created
+    afterward. Instrumenting the instance has no such ordering requirement.
+    """
+    try:
+        import fastapi  # noqa: F401  # pyright: ignore[reportMissingImports, reportUnusedImport]
+    except ImportError as error:
+        # opentelemetry-instrumentation-fastapi imports fastapi internally, so without
+        # this explicit check its own ImportError would be misreported below as a
+        # missing otel extra when fastapi itself is what's actually missing.
+        raise _missing_fastapi_extra() from error
+
+    try:
+        from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
+    except ImportError as error:
+        raise _missing_otel_extra() from error
+    FastAPIInstrumentor.instrument_app(app, excluded_urls=excluded_urls)
+
+
+def instrument_requests() -> None:
+    """Trace outbound calls made with the ``requests`` library.
+
+    Patches ``requests`` at module level, so callers that already imported it
+    are still covered.
+    """
+    try:
+        from opentelemetry.instrumentation.requests import RequestsInstrumentor
+    except ImportError as error:
+        raise _missing_otel_extra() from error
+    RequestsInstrumentor().instrument()