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

Merge pull request #1995 from BerriAI/litellm_langfuse_prompts

Browse source 
feat(langfuse.py): support langfuse prompt management
This commit is contained in:
Krish Dholakia 2024-02-15 23:57:00 -08:00 • committed by GitHub
commit c996b2e358
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/integrations/langfuse.py
View file

@ -226,6 +226,7 @@ class LangFuseLogger:
try:
tags = []
supports_tags = Version(langfuse.version.__version__) >= Version("2.6.3")
supports_prompt = Version(langfuse.version.__version__) >= Version("2.7.3")
supports_costs = Version(langfuse.version.__version__) >= Version("2.7.3")
print_verbose(f"Langfuse Layer Logging - logging to langfuse v2 ")
@ -294,6 +295,9 @@ class LangFuseLogger:
"level": level,
}
if supports_prompt:
generation_params["prompt"] = metadata.get("prompt", None)
if output is not None and isinstance(output, str) and level == "ERROR":
generation_params["statusMessage"] = output