fix(proxy_server.py): fix linting errors

This commit is contained in:
Krrish Dholakia 2024-07-11 22:12:33 -07:00
parent 5b40b62079
commit cff66d6151
2 changed files with 17 additions and 3 deletions

View file

@ -1332,6 +1332,9 @@ def client(original_function):
# ADD HIDDEN PARAMS - additional call metadata
if hasattr(result, "_hidden_params"):
result._hidden_params["litellm_call_id"] = getattr(
logging_obj, "litellm_call_id", None
)
result._hidden_params["model_id"] = kwargs.get("model_info", {}).get(
"id", None
)