From 5aa7e1e5a0e4e7780b3dde461497b30c76d8ae37 Mon Sep 17 00:00:00 2001
From: Rayce Rossum <rayce.rossum@gmail.com>
Date: Mon, 14 Aug 2023 19:21:22 -0700
Subject: [PATCH] Fix backwards vavlues in pgvector.py (#515)
---
gptcache/manager/vector_data/pgvector.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/gptcache/manager/vector_data/pgvector.py b/gptcache/manager/vector_data/pgvector.py
index 95bf9ffd..55ab0c99 100644
--- 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