浏览代码

fix: Compatible with special characters in pg full-text search. (#8921)

Co-authored-by: Aurelius Huang <cm.huang@aftership.com>
Aurelius Huang 6 月之前
父节点
当前提交
4585cffce1
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      api/core/rag/datasource/vdb/pgvector/pgvector.py

+ 1 - 1
api/core/rag/datasource/vdb/pgvector/pgvector.py

@@ -166,7 +166,7 @@ class PGVector(BaseVector):
 
         with self._get_cursor() as cur:
             cur.execute(
-                f"""SELECT meta, text, ts_rank(to_tsvector(coalesce(text, '')), to_tsquery(%s)) AS score
+                f"""SELECT meta, text, ts_rank(to_tsvector(coalesce(text, '')), plainto_tsquery(%s)) AS score
                 FROM {self.table_name}
                 WHERE to_tsvector(text) @@ plainto_tsquery(%s)
                 ORDER BY score DESC