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