tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-05 12:23:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add trace_name in langfuse logging v2

Browse Source
This commit is contained in:
Andrea Ponti
2024-03-27 16:58:08 +01:00
parent 90b859ebcb
commit 25bacfacdc
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/integrations/langfuse.py
+10 -5
View File
@@ -246,13 +246,13 @@ class LangFuseLogger:
metadata_tags = metadata.get("tags", [])
tags = metadata_tags
generation_name = metadata.get("generation_name", None)
if generation_name is None:
# just log `litellm-{call_type}` as the generation name
generation_name = f"litellm-{kwargs.get('call_type', 'completion')}"
trace_name = metadata.get("trace_name", None)
if trace_name is None:
# just log `litellm-{call_type}` as the trace name
trace_name = f"litellm-{kwargs.get('call_type', 'completion')}"
trace_params = {
"name": generation_name,
"name": trace_name,
"input": input,
"user_id": metadata.get("trace_user_id", user_id),
"id": metadata.get("trace_id", None),
@@ -311,6 +311,11 @@ class LangFuseLogger:
"completion_tokens": response_obj["usage"]["completion_tokens"],
"total_cost": cost if supports_costs else None,
}
generation_name = metadata.get("generation_name", None)
if generation_name is None:
# just log `litellm-{call_type}` as the generation name
generation_name = f"litellm-{kwargs.get('call_type', 'completion')}"
generation_params = {
"name": generation_name,
"id": metadata.get("generation_id", generation_id),