Include resource_details.billable in workflows usage (#776)

Co-authored-by: Grzegorz Klimaszewski <[email protected]>
roboflow · Nov 5, 2024 · 34171b9 · 34171b9
1 parent 894dfac
commit 34171b9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/inference/usage_tracking/collector.py b/inference/usage_tracking/collector.py
@@ -530,9 +530,10 @@ def _extract_usage_params_from_func_kwargs(
                     logger.debug(
                         "Got non-dict workflow JSON, '%s'", workflow.workflow_json
                     )
-            resource_details = UsageCollector._resource_details_from_workflow_json(
+            new_resource_details = UsageCollector._resource_details_from_workflow_json(
                 workflow_json=workflow_json,
             )
+            resource_details.update(new_resource_details)
             resource_details["is_preview"] = usage_workflow_preview
             resource_id = usage_workflow_id
             if not resource_id and resource_details: