From fd9bddc71a8d1408f4ce149d296b6d5b65520d8e Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 12 Jan 2024 17:05:51 -0800
Subject: [PATCH] (v0)

---
 litellm/main.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/litellm/main.py b/litellm/main.py
index 99e5590327..df570bb36e 100644
--- 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(