fix put litellm prefix in generation name

2026-07-13 09:06:18 +00:00 · 2024-05-29 18:40:53 -07:00
parent 67f1f374ec
commit 8c6a19d3ab
1 changed files with 3 additions and 4 deletions
@@ -459,10 +459,9 @@ class LangFuseLogger:
                # If using litellm proxy user `key_alias` if not None
                # 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)
-                generation_name = (
-                    _user_api_key_alias
-                    or f"litellm-{kwargs.get('call_type', 'completion')}"
-                )
+                generation_name = f"litellm-{kwargs.get('call_type', 'completion')}"
+                if _user_api_key_alias is not None:
+                    generation_name = f"litellm:{_user_api_key_alias}"

            if response_obj is not None and "system_fingerprint" in response_obj:
                system_fingerprint = response_obj.get("system_fingerprint", None)