From dbb2939b98f9b5a69ca005731702402a60d54767 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Thu, 15 Feb 2024 21:21:30 -0800 Subject: [PATCH] feat(langfuse.py): support langfuse prompt management --- litellm/integrations/langfuse.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/litellm/integrations/langfuse.py b/litellm/integrations/langfuse.py index b548f22bc4a..1432cac05c5 100644 --- a/litellm/integrations/langfuse.py +++ b/litellm/integrations/langfuse.py @@ -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 ") @@ -293,6 +294,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