From 93cb18c63349ecdb87bcfa9c8de1dc4847189c15 Mon Sep 17 00:00:00 2001 From: hughcrt Date: Tue, 11 Jun 2024 21:30:11 -0300 Subject: [PATCH] fix: lunary callback tags --- litellm/integrations/lunary.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/litellm/integrations/lunary.py b/litellm/integrations/lunary.py index 141ea64884..096888feb9 100644 --- a/litellm/integrations/lunary.py +++ b/litellm/integrations/lunary.py @@ -105,19 +105,17 @@ class LunaryLogger: end_time=datetime.now(timezone.utc), error=None, ): - # Method definition try: print_verbose(f"Lunary Logging - Logging request for model {model}") litellm_params = kwargs.get("litellm_params", {}) optional_params = kwargs.get("optional_params", {}) - metadata = litellm_params.get("metadata", {}) or {} + metadata = litellm_params.get("metadata", {}) if optional_params: - # merge into extra extra = {**extra, **optional_params} - tags = litellm_params.pop("tags", None) or [] + tags = metadata.get("metadata", None) if extra: extra.pop("extra_body", None)