Pārlūkot izejas kodu

fix: pgvector metadata filter (#16688)

Signed-off-by: kenwoodjw <blackxin55+@gmail.com>
tags/1.2.0
kenwoodjw pirms 7 mēnešiem
vecāks
revīzija
a113356695
Revīzijas autora e-pasta adrese nav piesaistīta nevienam kontam
1 mainītis faili ar 2 papildinājumiem un 2 dzēšanām
  1. 2
    2
      api/core/rag/datasource/vdb/pgvector/pgvector.py

+ 2
- 2
api/core/rag/datasource/vdb/pgvector/pgvector.py Parādīt failu

@@ -177,7 +177,7 @@ class PGVector(BaseVector):
where_clause = ""
if document_ids_filter:
document_ids = ", ".join(f"'{id}'" for id in document_ids_filter)
where_clause = f" WHERE metadata->>'document_id' in ({document_ids}) "
where_clause = f" WHERE meta->>'document_id' in ({document_ids}) "

with self._get_cursor() as cur:
cur.execute(
@@ -205,7 +205,7 @@ class PGVector(BaseVector):
where_clause = ""
if document_ids_filter:
document_ids = ", ".join(f"'{id}'" for id in document_ids_filter)
where_clause = f" AND metadata->>'document_id' in ({document_ids}) "
where_clause = f" AND meta->>'document_id' in ({document_ids}) "
if self.pg_bigm:
cur.execute("SET pg_bigm.similarity_limit TO 0.000001")
cur.execute(

Notiek ielāde…
Atcelt
Saglabāt