src/writing_studio/utils/metrics.py

"""Prometheus metrics for monitoring."""

from prometheus_client import Counter, Gauge, Histogram, Summary

# Request metrics
request_count = Counter(
    "writing_studio_requests_total",
    "Total number of analysis requests",
    ["status"],
)

request_duration = Histogram(
    "writing_studio_request_duration_seconds",
    "Request duration in seconds",
    ["operation"],
)

# Model metrics
model_load_duration = Histogram(
    "writing_studio_model_load_duration_seconds",
    "Model loading duration in seconds",
)

generation_duration = Summary(
    "writing_studio_generation_duration_seconds",
    "Text generation duration in seconds",
)

# Cache metrics
cache_hits = Counter(
    "writing_studio_cache_hits_total",
    "Total number of cache hits",
)

cache_misses = Counter(
    "writing_studio_cache_misses_total",
    "Total number of cache misses",
)

cache_size = Gauge(
    "writing_studio_cache_size",
    "Current cache size",
)

# Error metrics
error_count = Counter(
    "writing_studio_errors_total",
    "Total number of errors",
    ["error_type"],
)

# System metrics
active_requests = Gauge(
    "writing_studio_active_requests",
    "Number of active requests",
)