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

support langfuse tags feature

Browse source
This commit is contained in:
deena 2024-02-12 19:49:11 +02:00
parent 6b00bee6f4
commit b4ea3ff908
3 changed files with 749 additions and 630 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/integrations/langfuse.py
View file

@ -230,6 +230,10 @@ class LangFuseLogger:
print_verbose(f"Langfuse Layer Logging - logging to langfuse v2 ")
if supports_tags:
metadata_tags = metadata["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
@ -274,8 +278,7 @@ class LangFuseLogger:
"prompt_tokens": response_obj["usage"]["prompt_tokens"],
"completion_tokens": response_obj["usage"]["completion_tokens"],
"total_cost": cost if supports_costs else None,
}
}
generation_params = {
"name": generation_name,
"id": metadata.get("generation_id", generation_id),