Sonu Prasad commited on
Commit
e252af5
·
1 Parent(s): 92c7f16

fix: update search API to query_points for qdrant-client 1.9+

Browse files
Files changed (1) hide show
  1. vector_store.py +3 -3
vector_store.py CHANGED
@@ -129,15 +129,15 @@ class QdrantVectorStore:
129
  )]
130
  )
131
 
132
- results = self.client.search(
133
  collection_name=config.QDRANT_COLLECTION,
134
- query_vector=query_embedding.tolist(),
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"],