Add initial metric

backend/open_webui/routers/auths.py

@@ -4,6 +4,7 @@ import time
 import datetime
 import logging
 from aiohttp import ClientSession
+from open_webui.utils.telemetry import setup
 
 from open_webui.models.auths import (
     AddUserForm,
@@ -396,6 +397,8 @@ async def signin(request: Request, response: Response, form_data: SigninForm):
         user = Auths.authenticate_user(form_data.email.lower(), form_data.password)
 
     if user:
+        if setup.login_counter is not None:
+            setup.login_counter.add(1, {"method": "regular"})
 
         expires_delta = parse_duration(request.app.state.config.JWT_EXPIRES_IN)
         expires_at = None

backend/open_webui/utils/telemetry/setup.py

@@ -1,23 +1,54 @@
 from fastapi import FastAPI
-from opentelemetry import trace
+from opentelemetry import trace, metrics
 from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
+from opentelemetry.exporter.otlp.proto.grpc.metric_exporter import OTLPMetricExporter
 from opentelemetry.sdk.resources import SERVICE_NAME, Resource
 from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.metrics import MeterProvider
+from opentelemetry.sdk.metrics.export import (
+    PeriodicExportingMetricReader,
+)
 from sqlalchemy import Engine
 
 from open_webui.utils.telemetry.exporters import LazyBatchSpanProcessor
 from open_webui.utils.telemetry.instrumentors import Instrumentor
 from open_webui.env import OTEL_SERVICE_NAME, OTEL_EXPORTER_OTLP_ENDPOINT
 
+meter = None
+login_counter = None
+active_sessions_gauge = None
+
 
 def setup(app: FastAPI, db_engine: Engine):
+    resource = Resource.create(attributes={SERVICE_NAME: OTEL_SERVICE_NAME})
+
     # set up trace
-    trace.set_tracer_provider(
-        TracerProvider(
-            resource=Resource.create(attributes={SERVICE_NAME: OTEL_SERVICE_NAME})
-        )
-    )
+    trace_provider = TracerProvider(resource=resource)
+    trace.set_tracer_provider(trace_provider)
     # otlp export
-    exporter = OTLPSpanExporter(endpoint=OTEL_EXPORTER_OTLP_ENDPOINT)
-    trace.get_tracer_provider().add_span_processor(LazyBatchSpanProcessor(exporter))
+    span_exporter = OTLPSpanExporter(
+        endpoint=OTEL_EXPORTER_OTLP_ENDPOINT, insecure=True
+    )
+    trace_provider.add_span_processor(LazyBatchSpanProcessor(span_exporter))
     Instrumentor(app=app, db_engine=db_engine).instrument()
+
+    # set up metrics
+    metric_exporter = OTLPMetricExporter(
+        endpoint=OTEL_EXPORTER_OTLP_ENDPOINT, insecure=True
+    )
+    reader = PeriodicExportingMetricReader(
+        exporter=metric_exporter, export_interval_millis=5000
+    )
+    meter_provider = MeterProvider(resource=resource, metric_readers=[reader])
+    metrics.set_meter_provider(meter_provider)
+
+    global meter, login_counter, active_sessions_gauge
+    meter = metrics.get_meter(OTEL_SERVICE_NAME)
+
+    login_counter = meter.create_counter(
+        "user_login_total", description="Total number of user logins"
+    )
+
+    active_sessions_gauge = meter.create_gauge(
+        "active.sessions", description="Number of currently active user sessions"
+    )