scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

Address PR suggestions

Browse source
This commit is contained in:
Javier Garcia 2026-03-03 09:26:00 +01:00
parent 4558df5b63
commit 0f78462cc4
1 changed files with 25 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

41
litellm/integrations/braintrust_logging.py
View file

@ -50,6 +50,25 @@ class BraintrustLogger(CustomLogger):
)
self.global_braintrust_sync_http_handler = HTTPHandler()
def _build_prompt_metadata(
self, dynamic_metadata: dict, project_id: Optional[str]
) -> Optional[dict]:
"""
Build the prompt metadata dict from dynamic_metadata.
Returns None if neither prompt_id nor prompt_slug are present.
Only includes fields that are not None.
"""
prompt_id = dynamic_metadata.get("prompt_id")
prompt_slug = dynamic_metadata.get("prompt_slug")
if prompt_id is None and prompt_slug is None:
return None
prompt_data: dict = {"project_id": project_id}
if prompt_id is not None:
prompt_data["id"] = prompt_id
if prompt_slug is not None:
prompt_data["slug"] = prompt_slug
return prompt_data
def validate_environment(self, api_key: Optional[str]):
"""
Expects
@ -187,14 +206,9 @@ class BraintrustLogger(CustomLogger):
): # support logging dynamic metadata to braintrust
standard_logging_object[key] = value
prompt_id = dynamic_metadata.get("prompt_id")
prompt_slug = dynamic_metadata.get("prompt_slug")
if prompt_id is not None or prompt_slug is not None:
standard_logging_object["prompt"] = {
"id": prompt_id,
"project_id": project_id,
"slug": prompt_slug,
}
prompt_metadata = self._build_prompt_metadata(dynamic_metadata, project_id)
if prompt_metadata is not None:
standard_logging_object["prompt"] = prompt_metadata
cost = kwargs.get("response_cost", None)
@ -330,14 +344,9 @@ class BraintrustLogger(CustomLogger):
): # support logging dynamic metadata to braintrust
standard_logging_object[key] = value
prompt_id = dynamic_metadata.get("prompt_id")
prompt_slug = dynamic_metadata.get("prompt_slug")
if prompt_id is not None or prompt_slug is not None:
standard_logging_object["prompt"] = {
"id": prompt_id,
"project_id": project_id,
"slug": prompt_slug,
}
prompt_metadata = self._build_prompt_metadata(dynamic_metadata, project_id)
if prompt_metadata is not None:
standard_logging_object["prompt"] = prompt_metadata
cost = kwargs.get("response_cost", None)