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

fix put litellm prefix in generation name

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-29 18:40:53 -07:00
parent 67f1f374ec
commit 8c6a19d3ab
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/integrations/langfuse.py
View file

@ -459,10 +459,9 @@ class LangFuseLogger:
# If using litellm proxy user `key_alias` if not None # If using litellm proxy user `key_alias` if not None
# If `key_alias` is None, just log `litellm-{call_type}` as the generation name # If `key_alias` is None, just log `litellm-{call_type}` as the generation name
_user_api_key_alias = clean_metadata.get("user_api_key_alias", None) _user_api_key_alias = clean_metadata.get("user_api_key_alias", None)
generation_name = ( generation_name = f"litellm-{kwargs.get('call_type', 'completion')}"
_user_api_key_alias if _user_api_key_alias is not None:
or f"litellm-{kwargs.get('call_type', 'completion')}" generation_name = f"litellm:{_user_api_key_alias}"
)
if response_obj is not None and "system_fingerprint" in response_obj: if response_obj is not None and "system_fingerprint" in response_obj:
system_fingerprint = response_obj.get("system_fingerprint", None) system_fingerprint = response_obj.get("system_fingerprint", None)