diff --git a/litellm/tests/test_wandb_logger.py b/litellm/tests/test_wandb_logger.py new file mode 100644 index 000000000..3a43754da --- /dev/null +++ b/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() + + + +