This commit is contained in:
hailin 2025-05-29 02:04:47 +08:00
parent 03daddd2fa
commit 673308a513
1 changed files with 7 additions and 11 deletions

View File

@ -149,7 +149,7 @@ $$;
-- $$;
create function match_file_items_bge_m3 (
query_embedding "extensions"."vector"(1024),
query_embedding vector(1024),
match_count int DEFAULT null,
file_ids UUID[] DEFAULT null
) returns table (
@ -163,23 +163,19 @@ language plpgsql
as $$
#variable_conflict use_column
begin
-- 显式设置 schema 搜索路径
SET LOCAL search_path = extensions, public;
return query
select
id,
file_id,
content,
tokens,
1 - (
file_items.bge_m3_embedding::"extensions"."vector"(1024)
<=>
query_embedding::"extensions"."vector"(1024)
) as similarity
1 - (file_items.bge_m3_embedding <=> query_embedding) as similarity
from file_items
where file_id = ANY(file_ids)
order by
file_items.bge_m3_embedding::"extensions"."vector"(1024)
<=>
query_embedding::"extensions"."vector"(1024)
order by file_items.bge_m3_embedding <=> query_embedding
limit match_count;
end;
$$;