diff --git a/litellm/tests/test_datadog.py b/litellm/tests/test_datadog.py new file mode 100644 index 0000000000..789b39fd9d --- /dev/null +++ b/litellm/tests/test_datadog.py @@ -0,0 +1,27 @@ +import sys +import os +import io + +sys.path.insert(0, os.path.abspath("../..")) + +from litellm import completion +import litellm +import pytest + +import time + + +@pytest.mark.skip(reason="beta test - this is a new feature") +def test_datadog_logging(): + try: + litellm.success_callback = ["datadog"] + litellm.set_verbose = True + response = completion( + model="gpt-3.5-turbo", + messages=[{"role": "user", "content": "what llm are u"}], + max_tokens=10, + temperature=0.2, + ) + print(response) + except Exception as e: + print(e)