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