Fix backwards vavlues in pgvector.py (#515)

zilliztech · Aug 15, 2023 · 5aa7e1e · 5aa7e1e
1 parent e242160
commit 5aa7e1e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gptcache/manager/vector_data/pgvector.py b/gptcache/manager/vector_data/pgvector.py
@@ -149,7 +149,7 @@ def search(self, data: np.ndarray, top_k: int = -1):
             ).order_by(
                 similarity
             ).limit(top_k).all()
-            search_result = [(r[0].id, r[1]) for r in search_result]
+            search_result = [(r[1], r[0].id) for r in search_result]
 
         return search_result