From b2d861fa11c871713b98d7bde2c5d6effa6a8597 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 13 Feb 2024 07:02:58 -0800 Subject: [PATCH] (fix) langfuse logging --- litellm/integrations/langfuse.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/integrations/langfuse.py b/litellm/integrations/langfuse.py index 5a57d13597c..b548f22bc4a 100644 --- a/litellm/integrations/langfuse.py +++ b/litellm/integrations/langfuse.py @@ -231,7 +231,7 @@ class LangFuseLogger: print_verbose(f"Langfuse Layer Logging - logging to langfuse v2 ") if supports_tags: - metadata_tags = metadata["tags"] + metadata_tags = metadata.get("tags", []) tags = metadata_tags generation_name = metadata.get("generation_name", None) @@ -278,7 +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),