add pgvecto_rs support and upgrade SQLAlchemy (#3833)

2024-04-29 11:58:17 +08:00 · 2024-04-29 11:58:17 +08:00 · 3e9dbe3e0a
commit 3e9dbe3e0a
parent 975b2fb79e
26 changed files with 584 additions and 220 deletions
--- a/api/core/rag/datasource/vdb/relyt/relyt_vector.py
+++ b/api/core/rag/datasource/vdb/relyt/relyt_vector.py
@ -235,7 +235,7 @@ class RelytVector(BaseVector):
        docs = []
        for document, score in results:
            score_threshold = kwargs.get('score_threshold') if kwargs.get('score_threshold') else 0.0
-            if score > score_threshold:
+            if 1 - score > score_threshold:
                docs.append(document)
        return docs