From c1800354e94e47937d558d20b63a7af748457b56 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 18 Mar 2024 16:22:59 -0700 Subject: [PATCH] (test) dd test --- litellm/tests/test_datadog.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 litellm/tests/test_datadog.py 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)