mirror of
https://github.com/memohai/Memoh.git
synced 2026-04-27 07:16:19 +09:00
refactor: move client_type key from provider to model
This commit is contained in:
+8
-18
@@ -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;
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user