108 lines
2.7 KiB
Go
108 lines
2.7 KiB
Go
// 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()
|
|
}
|