feat: models

db/migrations/0001_init.up.sql

@@ -55,6 +55,23 @@ CREATE TABLE IF NOT EXISTS snapshots (
   created_at TIMESTAMPTZ NOT NULL DEFAULT now()
 );
 
+CREATE TABLE IF NOT EXISTS models (
+  id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+  model_id TEXT NOT NULL,
+  name TEXT,
+  base_url TEXT NOT NULL,
+  api_key TEXT NOT NULL,
+  client_type TEXT NOT NULL,
+  dimensions INTEGER,
+  type TEXT NOT NULL DEFAULT 'chat',
+  created_at TIMESTAMPTZ NOT NULL DEFAULT now(),
+  updated_at TIMESTAMPTZ NOT NULL DEFAULT now(),
+  CONSTRAINT models_model_id_unique UNIQUE (model_id),
+  CONSTRAINT models_type_check CHECK (type IN ('chat', 'embedding')),
+  CONSTRAINT models_client_type_check CHECK (client_type IN ('openai', 'anthropic', 'google')),
+  CONSTRAINT models_dimensions_check CHECK (type != 'embedding' OR dimensions IS NOT NULL)
+);
+
 CREATE INDEX IF NOT EXISTS idx_snapshots_container_id ON snapshots(container_id);
 CREATE INDEX IF NOT EXISTS idx_snapshots_parent_id ON snapshots(parent_snapshot_id);
 

db/queries/models.sql

@@ -0,0 +1,71 @@
+-- name: CreateModel :one
+INSERT INTO models (model_id, name, base_url, api_key, client_type, dimensions, type)
+VALUES (
+  sqlc.arg(model_id),
+  sqlc.arg(name),
+  sqlc.arg(base_url),
+  sqlc.arg(api_key),
+  sqlc.arg(client_type),
+  sqlc.arg(dimensions),
+  sqlc.arg(type)
+)
+RETURNING *;
+
+-- name: GetModelByID :one
+SELECT * FROM models WHERE id = sqlc.arg(id);
+
+-- name: GetModelByModelID :one
+SELECT * FROM models WHERE model_id = sqlc.arg(model_id);
+
+-- name: ListModels :many
+SELECT * FROM models
+ORDER BY created_at DESC;
+
+-- name: ListModelsByType :many
+SELECT * FROM models
+WHERE type = sqlc.arg(type)
+ORDER BY created_at DESC;
+
+-- name: ListModelsByClientType :many
+SELECT * FROM models
+WHERE client_type = sqlc.arg(client_type)
+ORDER BY created_at DESC;
+
+-- name: UpdateModel :one
+UPDATE models
+SET
+  name = sqlc.arg(name),
+  base_url = sqlc.arg(base_url),
+  api_key = sqlc.arg(api_key),
+  client_type = sqlc.arg(client_type),
+  dimensions = sqlc.arg(dimensions),
+  type = sqlc.arg(type),
+  updated_at = now()
+WHERE id = sqlc.arg(id)
+RETURNING *;
+
+-- name: UpdateModelByModelID :one
+UPDATE models
+SET
+  name = sqlc.arg(name),
+  base_url = sqlc.arg(base_url),
+  api_key = sqlc.arg(api_key),
+  client_type = sqlc.arg(client_type),
+  dimensions = sqlc.arg(dimensions),
+  type = sqlc.arg(type),
+  updated_at = now()
+WHERE model_id = sqlc.arg(model_id)
+RETURNING *;
+
+-- name: DeleteModel :exec
+DELETE FROM models WHERE id = sqlc.arg(id);
+
+-- name: DeleteModelByModelID :exec
+DELETE FROM models WHERE model_id = sqlc.arg(model_id);
+
+-- name: CountModels :one
+SELECT COUNT(*) FROM models;
+
+-- name: CountModelsByType :one
+SELECT COUNT(*) FROM models WHERE type = sqlc.arg(type);
+