This commit is contained in:
parent
03daddd2fa
commit
673308a513
|
|
@ -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;
|
||||
$$;
|
||||
Loading…
Reference in New Issue