phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-05 21:26:48 -07:00
parent 818a70838d
commit 7bf1b4d661
4 changed files with 23 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

31
litellm/integrations/braintrust_logging.py
View file

@ -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:
metadata = copy.deepcopy(
metadata
) # Avoid modifying the original metadata
except:
new_metadata = {}
for key, value in metadata.items():
if (
isinstance(value, list)
or isinstance(value, dict)
or isinstance(value, str)
or isinstance(value, int)
or isinstance(value, float)
):
new_metadata[key] = copy.deepcopy(value)
metadata = new_metadata
new_metadata = {}
for key, value in metadata.items():
if (
isinstance(value, list)
or isinstance(value, dict)
or isinstance(value, str)
or isinstance(value, int)
or isinstance(value, float)
):
new_metadata[key] = value
elif isinstance(value, BaseModel):
new_metadata[key] = value.model_dump_json()
metadata = new_metadata
tags = []
if isinstance(metadata, dict):