AdarshDRC commited on
Commit
c56ede7
·
verified ·
1 Parent(s): 0928262

Update src/services/db_client.py

Browse files
Files changed (1) hide show
  1. src/services/db_client.py +2 -2
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=1000, include_metadata=True)
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": 1000, "include_metadata": True}
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