unique_toolkit 2026.32.0.dev4__py3-none-any.whl → 2026.32.0.dev5__py3-none-any.whl

unique_toolkit/monitoring/middleware.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import time
+from collections.abc import Sequence
 
 from prometheus_client import (  # pyright: ignore[reportMissingImports]
     Counter,
@@ -16,12 +17,6 @@ _http_requests_total = Counter(
     ["method", "path", "status_code"],
     registry=REGISTRY,
 )
-_http_request_duration_seconds = Histogram(
-    "python_http_request_duration_seconds",
-    "HTTP request duration",
-    ["method", "path"],
-    registry=REGISTRY,
-)
 _http_requests_in_progress = Gauge(
     "python_http_requests_in_progress",
     "In-flight HTTP requests",
@@ -30,6 +25,35 @@ _http_requests_in_progress = Gauge(
     multiprocess_mode="livesum",
 )
 
+# Created lazily on first MetricsMiddleware construction so services can choose the
+# bucket layout. Prometheus allows a metric name to be registered only once per
+# registry, so the buckets passed to the first middleware win for the process.
+_http_request_duration_seconds: Histogram | None = None
+
+
+def _get_or_create_duration_histogram(
+    buckets: Sequence[float] | None,
+) -> Histogram:
+    global _http_request_duration_seconds
+    if _http_request_duration_seconds is None:
+        if buckets is None:
+            _http_request_duration_seconds = Histogram(
+                "python_http_request_duration_seconds",
+                "HTTP request duration",
+                ["method", "path"],
+                registry=REGISTRY,
+            )
+        else:
+            _http_request_duration_seconds = Histogram(
+                "python_http_request_duration_seconds",
+                "HTTP request duration",
+                ["method", "path"],
+                registry=REGISTRY,
+                buckets=tuple(buckets),
+            )
+    return _http_request_duration_seconds
+
+
 _DEFAULT_EXCLUDED_PATHS = frozenset({"/health", "/metrics", "/"})
 
 
@@ -51,13 +75,29 @@ class MetricsMiddleware:
 
         # Quart
         app.asgi_app = MetricsMiddleware(app.asgi_app)
+
+    ``duration_buckets`` overrides the latency histogram buckets (seconds). Without
+    it, prometheus_client's defaults apply and their top finite bucket is 10s —
+    histogram_quantile cannot resolve past the largest finite bucket, so services
+    with slower endpoints should pass buckets covering their full latency range::
+
+        app.add_middleware(MetricsMiddleware, duration_buckets=(1, 5, 30, 60, 120))
+
+    The histogram is registered once per process; the buckets of the first
+    constructed middleware apply.
     """
 
-    def __init__(self, app, excluded_paths: set[str] | None = None):
+    def __init__(
+        self,
+        app,
+        excluded_paths: set[str] | None = None,
+        duration_buckets: Sequence[float] | None = None,
+    ):
         self.app = app
         self._excluded_paths = (
             excluded_paths if excluded_paths is not None else _DEFAULT_EXCLUDED_PATHS
         )
+        self._duration = _get_or_create_duration_histogram(duration_buckets)
 
     async def __call__(self, scope, receive, send):
         # Pass through non-HTTP (WebSocket, lifespan)
@@ -87,9 +127,7 @@ class MetricsMiddleware:
         finally:
             duration = time.perf_counter() - start
             _http_requests_in_progress.labels(method=method).dec()
-            _http_request_duration_seconds.labels(method=method, path=path).observe(
-                duration
-            )
+            self._duration.labels(method=method, path=path).observe(duration)
             _http_requests_total.labels(
                 method=method, path=path, status_code=status_code
             ).inc()

{unique_toolkit-2026.32.0.dev4.dist-info → unique_toolkit-2026.32.0.dev5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.32.0.dev4
+Version: 2026.32.0.dev5
 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>
@@ -16,7 +16,7 @@ Requires-Dist: sseclient>=0.0.27,<0.0.28
 Requires-Dist: openai>=1.105.0,<3
 Requires-Dist: platformdirs>=4.0.0,<5
 Requires-Dist: pillow>=12.1.1
-Requires-Dist: unique-sdk>=2026.32.0.dev5,<2026.32.0rc0
+Requires-Dist: unique-sdk>=2026.32.0.dev8,<2026.32.0rc0
 Requires-Dist: jambo>=0.1.2,<0.2
 Requires-Dist: jinja2>=3.1.6,<4
 Requires-Dist: docxtpl>=0.20.1,<0.21

{unique_toolkit-2026.32.0.dev4.dist-info → unique_toolkit-2026.32.0.dev5.dist-info}/RECORD

@@ -403,7 +403,7 @@ unique_toolkit/language_model/stream_transform.py,sha256=iNNGpLY_7G6Bpp9Y4KtbIkQ
 unique_toolkit/language_model/utils.py,sha256=bPQ4l6_YO71w-zaIPanUUmtbXC1_hCvLK0tAFc3VCRc,1902
 unique_toolkit/monitoring/__init__.py,sha256=z1gkOCLxGcg936MFuLSeWFa0tREcBFSUbV09NxilBw4,1065
 unique_toolkit/monitoring/gunicorn.py,sha256=urTFFS5zYHKoxb1TEQHcLqb2byi-aYgFRHGwgYRzYYI,844
-unique_toolkit/monitoring/middleware.py,sha256=JNh5v_INAkexuLtXhCG9n5n2t6Pnf8ZdPTnimAVS0_s,2931
+unique_toolkit/monitoring/middleware.py,sha256=MNsUuKgt74mSnUwttXTB_AxX546Pz2zKoAP9OewVrwQ,4531
 unique_toolkit/monitoring/registry.py,sha256=De2Bx5QBBRJQY2MM3xtntiJ2JU2FoKkVT7rNMG7ewgA,5874
 unique_toolkit/protocols/support.py,sha256=x5edzbBmKAUuvm3oLFh52gkgQsqR5LU7M0t8RHK0DZk,5161
 unique_toolkit/services/__init__.py,sha256=b9Wl7Wvz1JmZDbDexIZzBoXZ0wscXuczGAH8LoL-Qak,572
@@ -418,6 +418,6 @@ unique_toolkit/short_term_memory/service.py,sha256=2JMBEBn6GoyK-a0xvdcWpPft-3C74
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=vRe0aMJWuVIK-VsqomgxQImzke7MDmwgSxxJfYCa7zw,1386
 unique_toolkit/test_utilities/events.py,sha256=FLrhMh0A_mQutAUjoD0FdMcVQ4Gq31r35OVGzRjvLro,6760
-unique_toolkit-2026.32.0.dev4.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-unique_toolkit-2026.32.0.dev4.dist-info/METADATA,sha256=INpTCsVe9pxypJm__jhb4ItWNy616eN71Bn04P9Y6WU,7465
-unique_toolkit-2026.32.0.dev4.dist-info/RECORD,,
+unique_toolkit-2026.32.0.dev5.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+unique_toolkit-2026.32.0.dev5.dist-info/METADATA,sha256=1bEknHHM6C2M0YX9ZPdElqUVEOA1AnI_u-_zmXVKrQM,7465
+unique_toolkit-2026.32.0.dev5.dist-info/RECORD,,