Spaces:
Running
Running
Upload src/vector_db.py with huggingface_hub
Browse files- src/vector_db.py +6 -3
src/vector_db.py
CHANGED
|
@@ -185,14 +185,16 @@ class UnifiedQdrant:
|
|
| 185 |
results = self.client.query_points(
|
| 186 |
collection_name=self.collection_name,
|
| 187 |
query=query_vec,
|
| 188 |
-
limit=10
|
|
|
|
| 189 |
).points
|
| 190 |
else:
|
| 191 |
results = self.client.query_points(
|
| 192 |
collection_name=self.collection_name,
|
| 193 |
query=query_vec,
|
| 194 |
shard_key_selector=shard_keys,
|
| 195 |
-
limit=10
|
|
|
|
| 196 |
).points
|
| 197 |
|
| 198 |
return results, search_mode
|
|
@@ -209,7 +211,8 @@ class UnifiedQdrant:
|
|
| 209 |
results = self.client.query_points(
|
| 210 |
collection_name=self.collection_name,
|
| 211 |
query=query_vec,
|
| 212 |
-
limit=10
|
|
|
|
| 213 |
).points
|
| 214 |
return results
|
| 215 |
|
|
|
|
| 185 |
results = self.client.query_points(
|
| 186 |
collection_name=self.collection_name,
|
| 187 |
query=query_vec,
|
| 188 |
+
limit=10,
|
| 189 |
+
with_payload=True
|
| 190 |
).points
|
| 191 |
else:
|
| 192 |
results = self.client.query_points(
|
| 193 |
collection_name=self.collection_name,
|
| 194 |
query=query_vec,
|
| 195 |
shard_key_selector=shard_keys,
|
| 196 |
+
limit=10,
|
| 197 |
+
with_payload=True
|
| 198 |
).points
|
| 199 |
|
| 200 |
return results, search_mode
|
|
|
|
| 211 |
results = self.client.query_points(
|
| 212 |
collection_name=self.collection_name,
|
| 213 |
query=query_vec,
|
| 214 |
+
limit=10,
|
| 215 |
+
with_payload=True
|
| 216 |
).points
|
| 217 |
return results
|
| 218 |
|