Skip to content

Commit

Permalink
formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
grzegorz-roboflow committed Aug 22, 2024
1 parent 6cf6af3 commit bcf8bc3
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion inference/usage_tracking/collector.py
Original file line number Diff line number Diff line change
Expand Up @@ -599,7 +599,12 @@ def _extract_usage_params_from_func_kwargs(
usage_api_key = func_kwargs["api_key"]
elif _self and hasattr(_self, "api_key") and _self.api_key:
usage_api_key = _self.api_key
elif "kwargs" in func_kwargs and isinstance(func_kwargs["kwargs"], dict) and "api_key" in func_kwargs["kwargs"] and func_kwargs["kwargs"]["api_key"]:
elif (
"kwargs" in func_kwargs
and isinstance(func_kwargs["kwargs"], dict)
and "api_key" in func_kwargs["kwargs"]
and func_kwargs["kwargs"]["api_key"]
):
usage_api_key = func_kwargs["kwargs"]["api_key"]
else:
logger.debug("Could not obtain API key from func kwargs")
Expand Down

0 comments on commit bcf8bc3

Please sign in to comment.