diff --git a/litellm/integrations/langfuse.py b/litellm/integrations/langfuse.py index 7ce843339a6..c60f33d034d 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 ") @@ -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