Spaces:
Runtime error
Runtime error
Sonu Prasad commited on
Commit ·
e252af5
1
Parent(s): 92c7f16
fix: update search API to query_points for qdrant-client 1.9+
Browse files- vector_store.py +3 -3
vector_store.py
CHANGED
|
@@ -129,15 +129,15 @@ class QdrantVectorStore:
|
|
| 129 |
)]
|
| 130 |
)
|
| 131 |
|
| 132 |
-
results = self.client.
|
| 133 |
collection_name=config.QDRANT_COLLECTION,
|
| 134 |
-
|
| 135 |
query_filter=filter_condition,
|
| 136 |
limit=top_k
|
| 137 |
)
|
| 138 |
|
| 139 |
search_results = []
|
| 140 |
-
for i, hit in enumerate(results):
|
| 141 |
chunk = DocumentChunk(
|
| 142 |
chunk_id=hit.payload["chunk_id"],
|
| 143 |
paper_id=hit.payload["paper_id"],
|
|
|
|
| 129 |
)]
|
| 130 |
)
|
| 131 |
|
| 132 |
+
results = self.client.query_points(
|
| 133 |
collection_name=config.QDRANT_COLLECTION,
|
| 134 |
+
query=query_embedding.tolist(),
|
| 135 |
query_filter=filter_condition,
|
| 136 |
limit=top_k
|
| 137 |
)
|
| 138 |
|
| 139 |
search_results = []
|
| 140 |
+
for i, hit in enumerate(results.points):
|
| 141 |
chunk = DocumentChunk(
|
| 142 |
chunk_id=hit.payload["chunk_id"],
|
| 143 |
paper_id=hit.payload["paper_id"],
|