ai_runs logging, redaction and retention (#60)

Co-authored-by: dev_agent <dev_agent@cubecraftlabs.com>

internal/db/ai_runs.sql.go

@@ -0,0 +1,107 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: ai_runs.sql
+
+package db
+
+import (
+	"context"
+	"database/sql"
+)
+
+const clearSuggestionBatchAIRunsBefore = `-- name: ClearSuggestionBatchAIRunsBefore :execrows
+UPDATE suggestion_batches
+SET ai_run_id = NULL
+WHERE ai_run_id IN (SELECT ai_runs.id FROM ai_runs WHERE ai_runs.created_at < ?1)
+`
+
+// Drops references to the runs created before the cutoff, so the purge
+// can delete them.
+func (q *Queries) ClearSuggestionBatchAIRunsBefore(ctx context.Context, cutoff string) (int64, error) {
+	result, err := q.db.ExecContext(ctx, clearSuggestionBatchAIRunsBefore, cutoff)
+	if err != nil {
+		return 0, err
+	}
+	return result.RowsAffected()
+}
+
+const clearWeekPlanAIRunsBefore = `-- name: ClearWeekPlanAIRunsBefore :execrows
+UPDATE week_plans
+SET ai_run_id = NULL
+WHERE ai_run_id IN (SELECT ai_runs.id FROM ai_runs WHERE ai_runs.created_at < ?1)
+`
+
+// Drops references to the runs created before the cutoff, so the purge
+// can delete them.
+func (q *Queries) ClearWeekPlanAIRunsBefore(ctx context.Context, cutoff string) (int64, error) {
+	result, err := q.db.ExecContext(ctx, clearWeekPlanAIRunsBefore, cutoff)
+	if err != nil {
+		return 0, err
+	}
+	return result.RowsAffected()
+}
+
+const createAIRun = `-- name: CreateAIRun :exec
+INSERT INTO ai_runs (
+	id, skill, skill_version, provider, model, temperature, status,
+	latency_ms, tokens_in, tokens_out, error, request_redacted,
+	response_redacted, created_at, updated_at
+) VALUES (
+	?, ?, ?, ?, ?, ?, ?,
+	?, ?, ?, ?, ?,
+	?, ?, ?
+)
+`
+
+type CreateAIRunParams struct {
+	ID               string
+	Skill            string
+	SkillVersion     string
+	Provider         string
+	Model            string
+	Temperature      float64
+	Status           string
+	LatencyMs        sql.NullInt64
+	TokensIn         sql.NullInt64
+	TokensOut        sql.NullInt64
+	Error            sql.NullString
+	RequestRedacted  string
+	ResponseRedacted sql.NullString
+	CreatedAt        string
+	UpdatedAt        string
+}
+
+func (q *Queries) CreateAIRun(ctx context.Context, arg CreateAIRunParams) error {
+	_, err := q.db.ExecContext(ctx, createAIRun,
+		arg.ID,
+		arg.Skill,
+		arg.SkillVersion,
+		arg.Provider,
+		arg.Model,
+		arg.Temperature,
+		arg.Status,
+		arg.LatencyMs,
+		arg.TokensIn,
+		arg.TokensOut,
+		arg.Error,
+		arg.RequestRedacted,
+		arg.ResponseRedacted,
+		arg.CreatedAt,
+		arg.UpdatedAt,
+	)
+	return err
+}
+
+const deleteAIRunsBefore = `-- name: DeleteAIRunsBefore :execrows
+DELETE FROM ai_runs
+WHERE created_at < ?1
+`
+
+func (q *Queries) DeleteAIRunsBefore(ctx context.Context, cutoff string) (int64, error) {
+	result, err := q.db.ExecContext(ctx, deleteAIRunsBefore, cutoff)
+	if err != nil {
+		return 0, err
+	}
+	return result.RowsAffected()
+}

internal/db/querier.go

@@ -15,11 +15,19 @@ type Querier interface {
 	// Marks the oldest queued job running and returns it. Returns sql.ErrNoRows
 	// when no job is queued.
 	ClaimNextJob(ctx context.Context, now sql.NullString) (Job, error)
+	// Drops references to the runs created before the cutoff, so the purge
+	// can delete them.
+	ClearSuggestionBatchAIRunsBefore(ctx context.Context, cutoff string) (int64, error)
+	// Drops references to the runs created before the cutoff, so the purge
+	// can delete them.
+	ClearWeekPlanAIRunsBefore(ctx context.Context, cutoff string) (int64, error)
+	CreateAIRun(ctx context.Context, arg CreateAIRunParams) error
 	CreateDinner(ctx context.Context, arg CreateDinnerParams) (Dinner, error)
 	CreateJob(ctx context.Context, arg CreateJobParams) error
 	CreatePhoto(ctx context.Context, arg CreatePhotoParams) (Photo, error)
 	CreateRecipeIngredient(ctx context.Context, arg CreateRecipeIngredientParams) error
 	CreateRecipeStep(ctx context.Context, arg CreateRecipeStepParams) error
+	DeleteAIRunsBefore(ctx context.Context, cutoff string) (int64, error)
 	DeleteRecipeIngredients(ctx context.Context, recipeID string) error
 	DeleteRecipeSteps(ctx context.Context, recipeID string) error
 	FailJob(ctx context.Context, arg FailJobParams) (int64, error)