refactor: move client_type key from provider to model

This commit is contained in:
Acbox
2026-02-18 18:30:27 +08:00
parent 77e9f585a1
commit d6c47472b2
43 changed files with 552 additions and 1015 deletions
+8 -18
View File
@@ -1,8 +1,7 @@
-- name: CreateLlmProvider :one
INSERT INTO llm_providers (name, client_type, base_url, api_key, metadata)
INSERT INTO llm_providers (name, base_url, api_key, metadata)
VALUES (
sqlc.arg(name),
sqlc.arg(client_type),
sqlc.arg(base_url),
sqlc.arg(api_key),
sqlc.arg(metadata)
@@ -19,16 +18,10 @@ SELECT * FROM llm_providers WHERE name = sqlc.arg(name);
SELECT * FROM llm_providers
ORDER BY created_at DESC;
-- name: ListLlmProvidersByClientType :many
SELECT * FROM llm_providers
WHERE client_type = sqlc.arg(client_type)
ORDER BY created_at DESC;
-- name: UpdateLlmProvider :one
UPDATE llm_providers
SET
name = sqlc.arg(name),
client_type = sqlc.arg(client_type),
base_url = sqlc.arg(base_url),
api_key = sqlc.arg(api_key),
metadata = sqlc.arg(metadata),
@@ -42,15 +35,13 @@ DELETE FROM llm_providers WHERE id = sqlc.arg(id);
-- name: CountLlmProviders :one
SELECT COUNT(*) FROM llm_providers;
-- name: CountLlmProvidersByClientType :one
SELECT COUNT(*) FROM llm_providers WHERE client_type = sqlc.arg(client_type);
-- name: CreateModel :one
INSERT INTO models (model_id, name, llm_provider_id, dimensions, input_modalities, type)
INSERT INTO models (model_id, name, llm_provider_id, client_type, dimensions, input_modalities, type)
VALUES (
sqlc.arg(model_id),
sqlc.arg(name),
sqlc.arg(llm_provider_id),
sqlc.narg(client_type),
sqlc.arg(dimensions),
sqlc.arg(input_modalities),
sqlc.arg(type)
@@ -73,10 +64,9 @@ WHERE type = sqlc.arg(type)
ORDER BY created_at DESC;
-- name: ListModelsByClientType :many
SELECT m.* FROM models AS m
JOIN llm_providers AS p ON p.id = m.llm_provider_id
WHERE p.client_type = sqlc.arg(client_type)
ORDER BY m.created_at DESC;
SELECT * FROM models
WHERE client_type = sqlc.arg(client_type)
ORDER BY created_at DESC;
-- name: ListModelsByProviderID :many
SELECT * FROM models
@@ -94,6 +84,7 @@ UPDATE models
SET
name = sqlc.arg(name),
llm_provider_id = sqlc.arg(llm_provider_id),
client_type = sqlc.narg(client_type),
dimensions = sqlc.arg(dimensions),
input_modalities = sqlc.arg(input_modalities),
type = sqlc.arg(type),
@@ -107,6 +98,7 @@ SET
model_id = sqlc.arg(new_model_id),
name = sqlc.arg(name),
llm_provider_id = sqlc.arg(llm_provider_id),
client_type = sqlc.narg(client_type),
dimensions = sqlc.arg(dimensions),
input_modalities = sqlc.arg(input_modalities),
type = sqlc.arg(type),
@@ -141,5 +133,3 @@ RETURNING *;
SELECT * FROM model_variants
WHERE model_uuid = sqlc.arg(model_uuid)
ORDER BY weight DESC, created_at DESC;