fix: bugs in traceloop integration

litellm/tests/test_traceloop.py

@@ -0,0 +1,57 @@
+import litellm
+from litellm import completion
+from traceloop.sdk import Traceloop
+
+Traceloop.init(app_name="test_traceloop", disable_batch=True)
+litellm.success_callback = ["traceloop"]
+
+
+def test_traceloop_logging():
+    try:
+        response = completion(
+            model="claude-instant-1.2",
+            messages=[
+                {"role": "user", "content": "Tell me a joke about OpenTelemetry"}
+            ],
+            max_tokens=10,
+            temperature=0.2,
+        )
+        print(response)
+    except Exception as e:
+        print(e)
+
+
+test_traceloop_logging()
+
+
+def test_traceloop_tracing_function_calling():
+    function1 = [
+        {
+            "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"],
+            },
+        }
+    ]
+    try:
+        response = completion(
+            model="gpt-3.5-turbo",
+            messages=[{"role": "user", "content": "what's the weather in boston"}],
+            temperature=0.1,
+            functions=function1,
+        )
+        print(response)
+    except Exception as e:
+        print(e)
+
+
+test_traceloop_tracing_function_calling()