Jesse_Chen
50 lines
1.4 KiB
PL/PgSQL
50 lines
1.4 KiB
PL/PgSQL
begin;
|
|
|
|
create or replace function public.default_rectification_chat_session_model()
|
|
returns trigger
|
|
language plpgsql
|
|
security definer
|
|
set search_path = ''
|
|
as $$
|
|
begin
|
|
if new.session_type = 'birth_time_rectification' and new.model_id is null then
|
|
select c.model_id into new.model_id
|
|
from public.model_config_versions v
|
|
join public.model_configs c on c.id = v.config_id
|
|
join public.model_providers p on p.id = v.provider_id
|
|
where v.status = 'published'
|
|
and v.enabled
|
|
and v.is_default
|
|
and p.enabled
|
|
limit 1;
|
|
end if;
|
|
return new;
|
|
end
|
|
$$;
|
|
|
|
revoke all on function public.default_rectification_chat_session_model() from public, anon, authenticated;
|
|
|
|
drop trigger if exists chat_sessions_default_rectification_model on public.chat_sessions;
|
|
create trigger chat_sessions_default_rectification_model
|
|
before insert or update of session_type, model_id on public.chat_sessions
|
|
for each row execute function public.default_rectification_chat_session_model();
|
|
|
|
with default_model as (
|
|
select c.model_id
|
|
from public.model_config_versions v
|
|
join public.model_configs c on c.id = v.config_id
|
|
join public.model_providers p on p.id = v.provider_id
|
|
where v.status = 'published'
|
|
and v.enabled
|
|
and v.is_default
|
|
and p.enabled
|
|
limit 1
|
|
)
|
|
update public.chat_sessions s
|
|
set model_id = default_model.model_id
|
|
from default_model
|
|
where s.session_type = 'birth_time_rectification'
|
|
and s.model_id is null;
|
|
|
|
commit;
|