(test) add weights & biases logger

litellm/tests/test_wandb_logger.py

@@ -0,0 +1,55 @@
+import sys
+import os
+import io
+
+sys.path.insert(0, os.path.abspath('../..'))
+
+from litellm import completion
+import litellm
+
+litellm.success_callback = ["wandb"]
+
+litellm.set_verbose = True
+import time
+
+
+
+def test_wandb_logging():
+    try:
+        response = completion(model="claude-instant-1.2",
+                              messages=[{
+                                  "role": "user",
+                                  "content": "Hi 👋 - i'm claude"
+                              }],
+                              max_tokens=10,
+                              temperature=0.2
+                              )
+        print(response)
+    except Exception as e:
+        print(e)
+
+test_wandb_logging()
+
+# def test_langfuse_logging_custom_generation_name():
+#     try:
+#         response = completion(model="gpt-3.5-turbo",
+#                               messages=[{
+#                                   "role": "user",
+#                                   "content": "Hi 👋 - i'm claude"
+#                               }],
+#                               max_tokens=10,
+#                               metadata = {
+#                                 "generation_name": "litellm-ishaan-gen", # set langfuse generation name
+#                                 # custom metadata fields
+#                                 "project": "litellm-proxy" 
+#                               }
+#         )
+#         print(response)
+#     except Exception as e:
+#         print(e)
+
+# test_langfuse_logging_custom_generation_name()
+
+
+
+