mirror of
https://github.com/memohai/Memoh.git
synced 2026-04-25 07:00:48 +09:00
fix: use bot model configs first
This commit is contained in:
@@ -19,6 +19,18 @@ SELECT bot_id, max_context_load_time, language, allow_guest
|
||||
FROM bot_settings
|
||||
WHERE bot_id = $1;
|
||||
|
||||
-- name: GetBotModelConfigByBotID :one
|
||||
SELECT
|
||||
bot_model_configs.bot_id,
|
||||
chat_models.model_id AS chat_model_id,
|
||||
memory_models.model_id AS memory_model_id,
|
||||
embedding_models.model_id AS embedding_model_id
|
||||
FROM bot_model_configs
|
||||
LEFT JOIN models AS chat_models ON chat_models.id = bot_model_configs.chat_model_id
|
||||
LEFT JOIN models AS memory_models ON memory_models.id = bot_model_configs.memory_model_id
|
||||
LEFT JOIN models AS embedding_models ON embedding_models.id = bot_model_configs.embedding_model_id
|
||||
WHERE bot_model_configs.bot_id = $1;
|
||||
|
||||
-- name: UpsertBotSettings :one
|
||||
INSERT INTO bot_settings (bot_id, max_context_load_time, language, allow_guest)
|
||||
VALUES ($1, $2, $3, $4)
|
||||
@@ -28,6 +40,15 @@ ON CONFLICT (bot_id) DO UPDATE SET
|
||||
allow_guest = EXCLUDED.allow_guest
|
||||
RETURNING bot_id, max_context_load_time, language, allow_guest;
|
||||
|
||||
-- name: UpsertBotModelConfig :one
|
||||
INSERT INTO bot_model_configs (bot_id, chat_model_id, memory_model_id, embedding_model_id)
|
||||
VALUES ($1, $2, $3, $4)
|
||||
ON CONFLICT (bot_id) DO UPDATE SET
|
||||
chat_model_id = COALESCE(EXCLUDED.chat_model_id, bot_model_configs.chat_model_id),
|
||||
memory_model_id = COALESCE(EXCLUDED.memory_model_id, bot_model_configs.memory_model_id),
|
||||
embedding_model_id = COALESCE(EXCLUDED.embedding_model_id, bot_model_configs.embedding_model_id)
|
||||
RETURNING bot_id, chat_model_id, memory_model_id, embedding_model_id;
|
||||
|
||||
-- name: DeleteSettingsByBotID :exec
|
||||
DELETE FROM bot_settings
|
||||
WHERE bot_id = $1;
|
||||
|
||||
Reference in New Issue
Block a user