fix(main.py): log hidden params for text completion calls

litellm/integrations/braintrust_logging.py

@@ -11,6 +11,7 @@ from typing import Literal, Optional
 
 import dotenv
 import httpx
+from pydantic import BaseModel
 
 import litellm
 from litellm import verbose_logger
@@ -280,22 +281,20 @@ class BraintrustLogger(CustomLogger):
             )  # if litellm_params['metadata'] == None
             metadata = self.add_metadata_from_header(litellm_params, metadata)
             clean_metadata = {}
-            try:
+            new_metadata = {}
-                metadata = copy.deepcopy(
+            for key, value in metadata.items():
-                    metadata
+                if (
-                )  # Avoid modifying the original metadata
+                    isinstance(value, list)
-            except:
+                    or isinstance(value, dict)
-                new_metadata = {}
+                    or isinstance(value, str)
-                for key, value in metadata.items():
+                    or isinstance(value, int)
-                    if (
+                    or isinstance(value, float)
-                        isinstance(value, list)
+                ):
-                        or isinstance(value, dict)
+                    new_metadata[key] = value
-                        or isinstance(value, str)
+                elif isinstance(value, BaseModel):
-                        or isinstance(value, int)
+                    new_metadata[key] = value.model_dump_json()
-                        or isinstance(value, float)
+
-                    ):
+            metadata = new_metadata
-                        new_metadata[key] = copy.deepcopy(value)
-                metadata = new_metadata
 
             tags = []
             if isinstance(metadata, dict):

litellm/llms/predibase.py

@@ -371,6 +371,7 @@ class PredibaseChatCompletion(BaseLLM):
                 response_headers["llm_provider-{}".format(k)] = v
 
         model_response._hidden_params["additional_headers"] = response_headers
+
         return model_response
 
     def completion(

litellm/main.py

@@ -128,6 +128,7 @@ from .types.llms.openai import HttpxBinaryResponseContent
 from .types.utils import (
     AdapterCompletionStreamWrapper,
     ChatCompletionMessageToolCall,
+    HiddenParams,
     all_litellm_params,
 )
 
@@ -3709,6 +3710,9 @@ async def atext_completion(
             text_choices["finish_reason"] = response["choices"][0]["finish_reason"]
             text_completion_response["choices"] = [text_choices]
             text_completion_response["usage"] = response.get("usage", None)
+            text_completion_response._hidden_params = HiddenParams(
+                **response._hidden_params
+            )
             return text_completion_response
     except Exception as e:
         custom_llm_provider = custom_llm_provider or "openai"
@@ -3980,6 +3984,7 @@ def text_completion(
     text_choices["finish_reason"] = response["choices"][0]["finish_reason"]
     text_completion_response["choices"] = [text_choices]
     text_completion_response["usage"] = response.get("usage", None)
+    text_completion_response._hidden_params = HiddenParams(**response._hidden_params)
 
     return text_completion_response
 

litellm/proxy/_new_secret_config.yaml

@@ -1,10 +1,4 @@
 model_list:
-  - model_name: "gpt-4"
+  - model_name: "*"
     litellm_params:
-      model: "gpt-4"
+      model: "*"
-  - model_name: "gpt-4"
-    litellm_params:
-      model: "gpt-4o"
-  - model_name: "gpt-4o-mini"
-    litellm_params:
-      model: "gpt-4o-mini"