diff --git a/litellm/integrations/lunary.py b/litellm/integrations/lunary.py index 141ea64884c..f9b2f25e704 100644 --- a/litellm/integrations/lunary.py +++ b/litellm/integrations/lunary.py @@ -105,7 +105,6 @@ class LunaryLogger: end_time=datetime.now(timezone.utc), error=None, ): - # Method definition try: print_verbose(f"Lunary Logging - Logging request for model {model}") @@ -114,10 +113,9 @@ class LunaryLogger: metadata = litellm_params.get("metadata", {}) or {} if optional_params: - # merge into extra extra = {**extra, **optional_params} - tags = litellm_params.pop("tags", None) or [] + tags = metadata.get("tags", None) if extra: extra.pop("extra_body", None) diff --git a/litellm/tests/test_lunary.py b/litellm/tests/test_lunary.py index c9a8afd57f1..cd068d9900f 100644 --- a/litellm/tests/test_lunary.py +++ b/litellm/tests/test_lunary.py @@ -24,8 +24,7 @@ def test_lunary_logging(): except Exception as e: print(e) - -# test_lunary_logging() +test_lunary_logging() def test_lunary_template(): @@ -38,8 +37,7 @@ def test_lunary_template(): except Exception as e: print(e) - -# test_lunary_template() +test_lunary_template() def test_lunary_logging_with_metadata(): @@ -52,16 +50,16 @@ def test_lunary_logging_with_metadata(): metadata={ "run_name": "litellmRUN", "project_name": "litellm-completion", + "tags": ["tag1", "tag2"] }, ) print(response) except Exception as e: print(e) -#test_lunary_logging_with_metadata() +test_lunary_logging_with_metadata() def test_lunary_with_tools(): - import litellm messages = [{"role": "user", "content": "What's the weather like in San Francisco, Tokyo, and Paris?"}] @@ -97,7 +95,7 @@ def test_lunary_with_tools(): print("\nLLM Response:\n", response.choices[0].message) -#test_lunary_with_tools() +test_lunary_with_tools() def test_lunary_logging_with_streaming_and_metadata(): try: @@ -117,5 +115,4 @@ def test_lunary_logging_with_streaming_and_metadata(): except Exception as e: print(e) - -# test_lunary_logging_with_streaming_and_metadata() +test_lunary_logging_with_streaming_and_metadata()