Spaces:
Sleeping
Sleeping
Update cag_service.py
Browse files- cag_service.py +6 -6
cag_service.py
CHANGED
|
@@ -107,12 +107,12 @@ class CAGService:
|
|
| 107 |
|
| 108 |
# Search for similar queries in cache
|
| 109 |
search_result = self.client.query_points(
|
| 110 |
-
|
| 111 |
-
|
| 112 |
-
|
| 113 |
-
|
| 114 |
-
|
| 115 |
-
|
| 116 |
|
| 117 |
if not search_result:
|
| 118 |
return None
|
|
|
|
| 107 |
|
| 108 |
# Search for similar queries in cache
|
| 109 |
search_result = self.client.query_points(
|
| 110 |
+
collection_name=self.cache_collection,
|
| 111 |
+
query=query_embedding.tolist(),
|
| 112 |
+
limit=1,
|
| 113 |
+
score_threshold=self.similarity_threshold,
|
| 114 |
+
with_payload=True
|
| 115 |
+
).points
|
| 116 |
|
| 117 |
if not search_result:
|
| 118 |
return None
|