(v0)

2024-01-12 17:05:51 -08:00 · 2024-01-12 17:05:51 -08:00 · fd9bddc71a
commit fd9bddc71a
parent 0ae8ab1b13
1 changed files with 1 additions and 0 deletions
--- a/litellm/main.py
+++ b/litellm/main.py
@ -575,6 +575,7 @@ def completion(
            api_base=api_base,
            api_key=api_key,
        )
+        model_response._hidden_params["custom_llm_provider"] = custom_llm_provider
        ### REGISTER CUSTOM MODEL PRICING -- IF GIVEN ###
        if input_cost_per_token is not None and output_cost_per_token is not None:
            litellm.register_model(