Lunary: Fix tool calling

litellm/tests/test_lunary.py

@@ -11,7 +11,6 @@ litellm.failure_callback = ["lunary"]
 litellm.success_callback = ["lunary"]
 litellm.set_verbose = True
 
-
 def test_lunary_logging():
     try:
         response = completion(
@@ -59,9 +58,46 @@ def test_lunary_logging_with_metadata():
     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?"}]
+    tools = [
+        {
+            "type": "function",
+            "function": {
+                "name": "get_current_weather",
+                "description": "Get the current weather in a given location",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "location": {
+                            "type": "string",
+                            "description": "The city and state, e.g. San Francisco, CA",
+                        },
+                        "unit": {"type": "string", "enum": ["celsius", "fahrenheit"]},
+                    },
+                    "required": ["location"],
+                },
+            },
+        }
+    ]
+
+    response = litellm.completion(
+        model="gpt-3.5-turbo-1106",
+        messages=messages,
+        tools=tools,
+        tool_choice="auto",  # auto is default, but we'll be explicit
+    )
+    
+    response_message = response.choices[0].message
+    print("\nLLM Response:\n", response.choices[0].message)
+
+
+#test_lunary_with_tools()
 
 def test_lunary_logging_with_streaming_and_metadata():
     try: