Jeffrey Smith
680ec3b897
All checks were successful
Co-authored-by: Jeffrey Smith <jasafpro@gmail.com> Co-committed-by: Jeffrey Smith <jasafpro@gmail.com>
98 lines
3.8 KiB
Go
98 lines
3.8 KiB
Go
// Package metrics defines all Prometheus metrics for Armature Core.
|
|
// All metrics use the "armature_" prefix. Registered via promauto
|
|
// so they are available on the default registry's /metrics endpoint.
|
|
package metrics
|
|
|
|
import (
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/client_golang/prometheus/promauto"
|
|
)
|
|
|
|
// ── HTTP Request Metrics ────────────────────
|
|
|
|
var (
|
|
HTTPRequestsTotal = promauto.NewCounterVec(prometheus.CounterOpts{
|
|
Name: "armature_http_requests_total",
|
|
Help: "Total HTTP requests by method, path pattern, and status code.",
|
|
}, []string{"method", "path_pattern", "status"})
|
|
|
|
HTTPRequestDuration = promauto.NewHistogramVec(prometheus.HistogramOpts{
|
|
Name: "armature_http_request_duration_seconds",
|
|
Help: "HTTP request latency in seconds.",
|
|
Buckets: []float64{0.01, 0.05, 0.1, 0.25, 0.5, 1, 2.5, 5, 10, 30},
|
|
}, []string{"method", "path_pattern"})
|
|
)
|
|
|
|
// ── WebSocket Metrics ───────────────────────
|
|
|
|
var WebSocketConnections = promauto.NewGauge(prometheus.GaugeOpts{
|
|
Name: "armature_websocket_connections",
|
|
Help: "Current number of active WebSocket connections.",
|
|
})
|
|
|
|
// ── Completion / Token Metrics ──────────────
|
|
|
|
var (
|
|
CompletionTokensTotal = promauto.NewCounterVec(prometheus.CounterOpts{
|
|
Name: "armature_completion_tokens_total",
|
|
Help: "Total tokens processed by direction (prompt|completion) and model.",
|
|
}, []string{"direction", "model_id"})
|
|
|
|
CompletionsTotal = promauto.NewCounterVec(prometheus.CounterOpts{
|
|
Name: "armature_completions_total",
|
|
Help: "Total completion requests by provider, model, and status.",
|
|
}, []string{"provider_config_id", "model_id", "status"})
|
|
|
|
CompletionDuration = promauto.NewHistogramVec(prometheus.HistogramOpts{
|
|
Name: "armature_completion_duration_seconds",
|
|
Help: "Completion request latency (wall time including tool loops).",
|
|
Buckets: []float64{0.1, 0.5, 1, 2, 5, 10, 30, 60, 120},
|
|
}, []string{"provider_config_id", "model_id"})
|
|
)
|
|
|
|
// ── Provider Health ─────────────────────────
|
|
|
|
var ProviderStatus = promauto.NewGaugeVec(prometheus.GaugeOpts{
|
|
Name: "armature_provider_status",
|
|
Help: "Provider health status: 0=unknown, 1=healthy, 2=degraded, 3=down.",
|
|
}, []string{"provider_config_id"})
|
|
|
|
// ── Database Pool Metrics ───────────────────
|
|
|
|
var (
|
|
DBOpenConnections = promauto.NewGauge(prometheus.GaugeOpts{
|
|
Name: "armature_db_open_connections",
|
|
Help: "Number of open database connections.",
|
|
})
|
|
DBInUseConnections = promauto.NewGauge(prometheus.GaugeOpts{
|
|
Name: "armature_db_in_use_connections",
|
|
Help: "Number of database connections currently in use.",
|
|
})
|
|
DBIdleConnections = promauto.NewGauge(prometheus.GaugeOpts{
|
|
Name: "armature_db_idle_connections",
|
|
Help: "Number of idle database connections.",
|
|
})
|
|
DBWaitCount = promauto.NewGauge(prometheus.GaugeOpts{
|
|
Name: "armature_db_wait_count_total",
|
|
Help: "Total number of connections waited for.",
|
|
})
|
|
DBWaitDuration = promauto.NewGauge(prometheus.GaugeOpts{
|
|
Name: "armature_db_wait_duration_seconds_total",
|
|
Help: "Total time blocked waiting for a new connection (seconds).",
|
|
})
|
|
)
|
|
|
|
// ── Task Scheduler Metrics ──────────────────
|
|
|
|
var TaskExecutionsTotal = promauto.NewCounterVec(prometheus.CounterOpts{
|
|
Name: "armature_task_executions_total",
|
|
Help: "Total task executions by status (success|error).",
|
|
}, []string{"status"})
|
|
|
|
// ── Sandbox Metrics ─────────────────────────
|
|
|
|
var SandboxExecutionsTotal = promauto.NewCounterVec(prometheus.CounterOpts{
|
|
Name: "armature_sandbox_executions_total",
|
|
Help: "Total Starlark sandbox executions by entry point and status.",
|
|
}, []string{"entry_point", "status"})
|