unique_toolkit 2026.24.0.dev6__py3-none-any.whl → 2026.24.0.dev7__py3-none-any.whl

unique_toolkit/monitoring/gunicorn.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+
+import os
+
+
+def child_exit(server, worker) -> None:
+    """Gunicorn child_exit hook: clean up dead worker's per-PID metric files.
+
+    Wire this up in gunicorn.conf.py:
+
+        from unique_toolkit.monitoring.gunicorn import child_exit  # noqa: F401
+
+    Gunicorn calls any function named child_exit(server, worker) in the
+    config file whenever a worker process exits. Without this, dead workers'
+    .db files accumulate in PROMETHEUS_MULTIPROC_DIR and inflate aggregated
+    metric values on every future scrape.
+
+    No-ops when PROMETHEUS_MULTIPROC_DIR is not set (single-process mode).
+    """
+    if not os.environ.get("PROMETHEUS_MULTIPROC_DIR"):
+        return
+    from prometheus_client import multiprocess  # pyright: ignore[reportMissingImports]
+
+    multiprocess.mark_process_dead(worker.pid)

unique_toolkit/monitoring/middleware.py

@@ -11,22 +11,23 @@ from prometheus_client import (  # pyright: ignore[reportMissingImports]
 from .registry import REGISTRY
 
 _http_requests_total = Counter(
-    "http_requests_total",
+    "python_http_requests_total",
     "Total HTTP requests",
     ["method", "path", "status_code"],
     registry=REGISTRY,
 )
 _http_request_duration_seconds = Histogram(
-    "http_request_duration_seconds",
+    "python_http_request_duration_seconds",
     "HTTP request duration",
     ["method", "path"],
     registry=REGISTRY,
 )
 _http_requests_in_progress = Gauge(
-    "http_requests_in_progress",
+    "python_http_requests_in_progress",
     "In-flight HTTP requests",
     ["method"],
     registry=REGISTRY,
+    multiprocess_mode="livesum",
 )
 
 _DEFAULT_EXCLUDED_PATHS = frozenset({"/health", "/metrics", "/"})

unique_toolkit/monitoring/registry.py

@@ -2,9 +2,11 @@ from __future__ import annotations
 
 import asyncio
 import functools
+import os
 import time
+import warnings
 from contextlib import contextmanager
-from typing import Any, Callable
+from typing import Any, Callable, Literal
 
 from prometheus_client import (  # pyright: ignore[reportMissingImports]
     CollectorRegistry,
@@ -18,7 +20,36 @@ REGISTRY = CollectorRegistry()
 
 
 def get_metrics() -> bytes:
-    """Returns all registered metrics in Prometheus text format."""
+    """Returns all registered metrics in Prometheus text format.
+
+    In multiprocess mode (PROMETHEUS_MULTIPROC_DIR is set), aggregates
+    per-PID .db files from all Gunicorn workers via MultiProcessCollector.
+    In single-process mode, returns metrics from the in-process REGISTRY.
+
+    NOTE: PROMETHEUS_MULTIPROC_DIR must be set in the shell/entrypoint
+    *before* Python starts. Setting it from Python after prometheus_client
+    has been imported has no effect — metrics will silently bypass per-PID
+    files and this function will emit a warning.
+    """
+    if os.environ.get("PROMETHEUS_MULTIPROC_DIR"):
+        from prometheus_client import (  # pyright: ignore[reportMissingImports]
+            multiprocess,
+            values,
+        )
+
+        if not getattr(values.ValueClass, "_multiprocess", False):
+            warnings.warn(
+                "PROMETHEUS_MULTIPROC_DIR is set but prometheus_client was imported "
+                "before the env var — metrics will silently bypass per-PID files. "
+                "Set PROMETHEUS_MULTIPROC_DIR in the shell/entrypoint before launching Python.",
+                RuntimeWarning,
+                stacklevel=2,
+            )
+
+        registry = CollectorRegistry()
+        multiprocess.MultiProcessCollector(registry)
+        return generate_latest(registry)
+
     return generate_latest(REGISTRY)
 
 
@@ -43,8 +74,31 @@ class MetricNamespace:
     def counter(self, name: str, description: str, labels: list[str]) -> Counter:
         return Counter(f"{self._prefix}_{name}", description, labels, registry=REGISTRY)
 
-    def gauge(self, name: str, description: str, labels: list[str]) -> Gauge:
-        return Gauge(f"{self._prefix}_{name}", description, labels, registry=REGISTRY)
+    def gauge(
+        self,
+        name: str,
+        description: str,
+        labels: list[str],
+        multiprocess_mode: Literal[
+            "all",
+            "liveall",
+            "min",
+            "livemin",
+            "max",
+            "livemax",
+            "sum",
+            "livesum",
+            "mostrecent",
+            "livemostrecent",
+        ] = "all",
+    ) -> Gauge:
+        return Gauge(
+            f"{self._prefix}_{name}",
+            description,
+            labels,
+            registry=REGISTRY,
+            multiprocess_mode=multiprocess_mode,
+        )
 
 
 @contextmanager

{unique_toolkit-2026.24.0.dev6.dist-info → unique_toolkit-2026.24.0.dev7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.24.0.dev6
+Version: 2026.24.0.dev7
 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>

{unique_toolkit-2026.24.0.dev6.dist-info → unique_toolkit-2026.24.0.dev7.dist-info}/RECORD

@@ -376,8 +376,9 @@ unique_toolkit/language_model/settings.py,sha256=xkoMv3ATb2ft5V3MA8JvqW-EO8AnZ5t
 unique_toolkit/language_model/stream_transform.py,sha256=iNNGpLY_7G6Bpp9Y4KtbIkQm3WB4yZt75BXcqx0DVkI,2775
 unique_toolkit/language_model/utils.py,sha256=bPQ4l6_YO71w-zaIPanUUmtbXC1_hCvLK0tAFc3VCRc,1902
 unique_toolkit/monitoring/__init__.py,sha256=z1gkOCLxGcg936MFuLSeWFa0tREcBFSUbV09NxilBw4,1065
-unique_toolkit/monitoring/middleware.py,sha256=QSOcRwaByN4t6gWY5-IUgW2JXoN7Yl-0JR5kSmOjxM8,2877
-unique_toolkit/monitoring/registry.py,sha256=8VG1NkfzrDiMfPr7KsX7uWEJBK3rNSRWOB8HNQ5B9jY,3333
+unique_toolkit/monitoring/gunicorn.py,sha256=urTFFS5zYHKoxb1TEQHcLqb2byi-aYgFRHGwgYRzYYI,844
+unique_toolkit/monitoring/middleware.py,sha256=JNh5v_INAkexuLtXhCG9n5n2t6Pnf8ZdPTnimAVS0_s,2931
+unique_toolkit/monitoring/registry.py,sha256=8r9jqaQoRHZveGRZe_GQvBzvQpmUV4cpw8Ej1ssDIdE,5056
 unique_toolkit/protocols/support.py,sha256=x5edzbBmKAUuvm3oLFh52gkgQsqR5LU7M0t8RHK0DZk,5161
 unique_toolkit/services/__init__.py,sha256=b9Wl7Wvz1JmZDbDexIZzBoXZ0wscXuczGAH8LoL-Qak,572
 unique_toolkit/services/chat_service.py,sha256=FPY-GTFAkD0C9pYfPfkfAvyIKnL9DX9iDk75gDeghCc,95209
@@ -391,6 +392,6 @@ unique_toolkit/short_term_memory/service.py,sha256=Ncw0_0CNM6kmzJSLVJGG5GcKpBduI
 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.24.0.dev6.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-unique_toolkit-2026.24.0.dev6.dist-info/METADATA,sha256=9iqeteqQZ9CeoDz6t3gI0L0YdK6AnpC376d7cqfdps4,7392
-unique_toolkit-2026.24.0.dev6.dist-info/RECORD,,
+unique_toolkit-2026.24.0.dev7.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+unique_toolkit-2026.24.0.dev7.dist-info/METADATA,sha256=ThX3w1DhE99W9OR-nA3Kmd-hNVmqxdGU2xtsV9mUvqg,7392
+unique_toolkit-2026.24.0.dev7.dist-info/RECORD,,