Changeset 0.10.0 (#56)

server/database/migrations/004_roles_usage.sql

@@ -0,0 +1,64 @@
+-- 004_roles_usage.sql
+-- v0.10.0: Model Roles + Usage Tracking
+--
+-- New tables: usage_log, model_pricing
+-- Seed: model_roles in global_settings
+
+-- ── Model roles seed ──────────────────────────
+-- Uses existing global_settings table. No new tables for roles.
+-- Team overrides go in teams.settings JSONB (existing column).
+
+INSERT INTO global_settings (key, value) VALUES
+    ('model_roles', '{
+        "utility":    { "primary": null, "fallback": null },
+        "embedding":  { "primary": null, "fallback": null },
+        "generation": { "primary": null, "fallback": null }
+    }'::jsonb)
+ON CONFLICT (key) DO NOTHING;
+
+-- ── Usage tracking ────────────────────────────
+-- Every completion (streaming and non-streaming) logs token counts and cost.
+-- Cost is calculated at insert time from model_pricing.
+-- provider_scope is denormalized for efficient admin filtering (excludes BYOK).
+
+CREATE TABLE IF NOT EXISTS usage_log (
+    id                    UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+    channel_id            UUID REFERENCES channels(id) ON DELETE SET NULL,
+    user_id               UUID NOT NULL REFERENCES users(id) ON DELETE CASCADE,
+    provider_config_id    UUID REFERENCES provider_configs(id) ON DELETE SET NULL,
+    provider_scope        TEXT NOT NULL DEFAULT 'global',
+    model_id              TEXT NOT NULL,
+    role                  TEXT,  -- NULL = user chat, 'utility', 'embedding', 'generation'
+    prompt_tokens         INT NOT NULL DEFAULT 0,
+    completion_tokens     INT NOT NULL DEFAULT 0,
+    cache_creation_tokens INT NOT NULL DEFAULT 0,
+    cache_read_tokens     INT NOT NULL DEFAULT 0,
+    cost_input            NUMERIC(12,6),
+    cost_output           NUMERIC(12,6),
+    created_at            TIMESTAMPTZ NOT NULL DEFAULT NOW()
+);
+
+CREATE INDEX IF NOT EXISTS idx_usage_log_user     ON usage_log(user_id);
+CREATE INDEX IF NOT EXISTS idx_usage_log_created  ON usage_log(created_at);
+CREATE INDEX IF NOT EXISTS idx_usage_log_provider ON usage_log(provider_config_id);
+CREATE INDEX IF NOT EXISTS idx_usage_log_model    ON usage_log(model_id);
+CREATE INDEX IF NOT EXISTS idx_usage_log_scope    ON usage_log(provider_scope);
+
+-- ── Model pricing ─────────────────────────────
+-- Two sources: 'catalog' (auto-populated from provider API sync) and
+-- 'manual' (admin override). Manual entries are never overwritten by sync.
+
+CREATE TABLE IF NOT EXISTS model_pricing (
+    id                  UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+    provider_config_id  UUID NOT NULL REFERENCES provider_configs(id) ON DELETE CASCADE,
+    model_id            TEXT NOT NULL,
+    input_per_m         NUMERIC(10,6),
+    output_per_m        NUMERIC(10,6),
+    cache_create_per_m  NUMERIC(10,6),
+    cache_read_per_m    NUMERIC(10,6),
+    currency            TEXT NOT NULL DEFAULT 'USD',
+    source              TEXT NOT NULL DEFAULT 'manual',
+    updated_at          TIMESTAMPTZ DEFAULT NOW(),
+    updated_by          UUID REFERENCES users(id),
+    UNIQUE(provider_config_id, model_id)
+);

server/database/testhelper.go

@@ -158,6 +158,8 @@ func TruncateAll(t *testing.T) {
 	}
 	// Order matters due to foreign keys — truncate with CASCADE
 	tables := []string{
+		"usage_log",
+		"model_pricing",
 		"notes",
 		"audit_log",
 		"channel_cursors",
@@ -178,6 +180,22 @@ func TruncateAll(t *testing.T) {
 	for _, table := range tables {
 		DB.Exec(fmt.Sprintf("TRUNCATE TABLE %s CASCADE", table))
 	}
+
+	// Re-seed global_settings — CASCADE from users wipes rows with updated_by FK.
+	// Re-run the seed SQL from migrations 001 + 004.
+	DB.Exec(`
+		INSERT INTO global_settings (key, value) VALUES
+			('registration', '{"enabled": true}'::jsonb),
+			('site', '{"name": "Chat Switchboard", "tagline": "Multi-Model AI Chat"}'::jsonb),
+			('banner', '{"enabled": false, "text": "", "position": "both", "bg": "#007a33", "fg": "#ffffff"}'::jsonb),
+			('banner_presets', '{}'::jsonb),
+			('model_roles', '{
+				"utility":    { "primary": null, "fallback": null },
+				"embedding":  { "primary": null, "fallback": null },
+				"generation": { "primary": null, "fallback": null }
+			}'::jsonb)
+		ON CONFLICT (key) DO NOTHING
+	`)
 }
 
 // SeedTestUser creates a test user and returns the user ID.