send payloads containing processed_frames

roboflow · Jul 18, 2024 · d022f8a · d022f8a
1 parent ea37a2b
commit d022f8a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/inference/usage_tracking/collector.py b/inference/usage_tracking/collector.py
@@ -501,6 +501,9 @@ def _offload_to_api(self, payloads: List[APIKeyUsage]):
         api_keys_failed = set()
         for payload in payloads:
             for api_key, workflow_payloads in payload.items():
+                if any("processed_frames" not in w for w in workflow_payloads.values()):
+                    api_keys_failed.add(api_key)
+                    continue
                 enterprise = any(
                     w.get("enterprise") for w in workflow_payloads.values()
                 )