Spaces:
Sleeping
Sleeping
Update src/services/db_client.py
Browse files
src/services/db_client.py
CHANGED
|
@@ -104,7 +104,7 @@ def delete_and_recreate_indexes(pc: Pinecone):
|
|
| 104 |
ensure_indexes(pc)
|
| 105 |
|
| 106 |
def search_faces(idx, vec: List[float], det_score: float) -> Dict[str, Any]:
|
| 107 |
-
res = idx.query(vector=vec, top_k=
|
| 108 |
image_map = {}
|
| 109 |
for match in res.get("matches", []):
|
| 110 |
meta = match.get("metadata", {})
|
|
@@ -123,7 +123,7 @@ def search_faces(idx, vec: List[float], det_score: float) -> Dict[str, Any]:
|
|
| 123 |
import numpy as np
|
| 124 |
|
| 125 |
def search_objects(idx, vec: List[float], filter_dict: dict = None) -> List[Dict[str, Any]]:
|
| 126 |
-
query_kwargs = {"vector": vec, "top_k":
|
| 127 |
if filter_dict:
|
| 128 |
query_kwargs["filter"] = filter_dict
|
| 129 |
|
|
|
|
| 104 |
ensure_indexes(pc)
|
| 105 |
|
| 106 |
def search_faces(idx, vec: List[float], det_score: float) -> Dict[str, Any]:
|
| 107 |
+
res = idx.query(vector=vec, top_k=50, include_metadata=True)
|
| 108 |
image_map = {}
|
| 109 |
for match in res.get("matches", []):
|
| 110 |
meta = match.get("metadata", {})
|
|
|
|
| 123 |
import numpy as np
|
| 124 |
|
| 125 |
def search_objects(idx, vec: List[float], filter_dict: dict = None) -> List[Dict[str, Any]]:
|
| 126 |
+
query_kwargs = {"vector": vec, "top_k": 50, "include_metadata": True}
|
| 127 |
if filter_dict:
|
| 128 |
query_kwargs["filter"] = filter_dict
|
| 129 |
|