From a113356695339d7e22d54b4f7ff1a26bca838a39 Mon Sep 17 00:00:00 2001 From: kenwoodjw Date: Tue, 25 Mar 2025 11:34:33 +0800 Subject: [PATCH] fix: pgvector metadata filter (#16688) Signed-off-by: kenwoodjw --- api/core/rag/datasource/vdb/pgvector/pgvector.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/core/rag/datasource/vdb/pgvector/pgvector.py b/api/core/rag/datasource/vdb/pgvector/pgvector.py index 783ad93b0c..7c0d36ebe0 100644 --- a/api/core/rag/datasource/vdb/pgvector/pgvector.py +++ b/api/core/rag/datasource/vdb/pgvector/pgvector.py @@ -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(