feat: DIA-1573: Track predictions made with prompts (#257)

Co-authored-by: hakan458 <[email protected]>
HumanSignal · Nov 18, 2024 · cc42eb3 · cc42eb3
1 parent 3784b83
commit cc42eb3
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/server/handlers/result_handlers.py b/server/handlers/result_handlers.py
@@ -208,18 +208,19 @@ def __call__(self, result_batch: list[Dict]):
         # coerce back to dicts for sending
         result_batch = [record.dict() for record in result_batch]
         if result_batch:
-            logger.info(f"LSEHandler sending {len(result_batch)} predictions to LSE")
+            num_predictions = len(result_batch)
+            logger.info(f"LSEHandler sending {num_predictions} predictions to LSE")
             self.client.make_request(
                 "POST",
-                "/api/model-run/batch-predictions",
+                f"/api/model-run/batch-predictions?num_predictions={num_predictions}",
                 data=json.dumps(
                     {
                         "modelrun_id": self.modelrun_id,
                         "results": result_batch,
                     }
                 ),
             )
-            logger.info(f"LSEHandler sent {len(result_batch)} predictions to LSE")
+            logger.info(f"LSEHandler sent {num_predictions} predictions to LSE")
         else:
             logger.error(
                 f"No valid results to send to LSE for modelrun_id {self.modelrun_id}"