From e0608eae72f6cb3abe92e38f95e0402b036afd93 Mon Sep 17 00:00:00 2001
From: hughcrt <chocart.hugues@icloud.com>
Date: Thu, 13 Jun 2024 15:10:24 -0300
Subject: [PATCH] add tests

---
 litellm/integrations/lunary.py |  2 +-
 litellm/tests/test_lunary.py   | 15 ++++++---------
 2 files changed, 7 insertions(+), 10 deletions(-)

diff --git a/litellm/integrations/lunary.py b/litellm/integrations/lunary.py
index f6b025c670e..f9b2f25e704 100644
--- a/litellm/integrations/lunary.py
+++ b/litellm/integrations/lunary.py
@@ -110,7 +110,7 @@ class LunaryLogger:
 
             litellm_params = kwargs.get("litellm_params", {})
             optional_params = kwargs.get("optional_params", {})
-            metadata = litellm_params.get("metadata", {})
+            metadata = litellm_params.get("metadata", {}) or {}
 
             if optional_params:
                 extra = {**extra, **optional_params}
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()