fix: add query_logo_id field in /ann/search response

openfoodfacts · Dec 23, 2022 · 698025c · 698025c
1 parent b8f2d04
commit 698025c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/robotoff/app/api.py b/robotoff/app/api.py
@@ -907,7 +907,7 @@ def on_get(
             )
             if not response["hits"]["hits"]:
                 # We don't have any embedding indexed
-                resp.media = {"results": [], "count": 0}
+                resp.media = {"results": [], "count": 0, "query_logo_id": None}
                 return
             logo_id = int(response["hits"]["hits"][0]["_id"])
 
@@ -923,7 +923,11 @@ def on_get(
             if item[0] != logo_id
         ][:count]
         results = [{"logo_id": item[0], "distance": item[1]} for item in raw_results]
-        resp.media = {"results": results, "count": len(results)}
+        resp.media = {
+            "results": results,
+            "count": len(results),
+            "query_logo_id": logo_id,
+        }
 
 
 class WebhookProductResource: