Reimplement the function retrieve_by_ids

hv0905 · Dec 31, 2023 · 382f3a7 · 382f3a7
1 parent 6c318e8
commit 382f3a7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/app/Services/vector_db_context.py b/app/Services/vector_db_context.py
@@ -60,6 +60,11 @@ async def retrieve_by_ids(self, image_id: list[str], with_vectors=False) -> list
                                              ids=image_id,
                                              with_payload=True,
                                              with_vectors=with_vectors)
+        if len(image_id) != len(result):
+            logger.error("{} points not exist.", len(image_id) - len(result))
+            result_point_ids = {t.id for t in result}
+            missing_point_ids = set(image_id) - result_point_ids
+            raise PointNotFoundError(str(missing_point_ids))
         return self._get_img_data_from_points(result)
 
     async def validate_ids(self, image_id: list[str]) -> list[str]: