From 480dd63f885db0815ea21c40eebfa3527e01a961 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 31 Jul 2023 18:47:50 -0700 Subject: [PATCH] better testing --- litellm/tests/test_bad_params.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/litellm/tests/test_bad_params.py b/litellm/tests/test_bad_params.py index 95f669594..dd1e8d509 100644 --- a/litellm/tests/test_bad_params.py +++ b/litellm/tests/test_bad_params.py @@ -9,17 +9,20 @@ import traceback current_dir = os.path.dirname(os.path.abspath(__file__)) # Get the parent directory by joining the current directory with '..' -parent_dir = os.path.join(current_dir, '..') +parent_dir = os.path.join(current_dir, '../..') # Add the parent directory to the system path sys.path.append(parent_dir) -import main -from main import embedding, completion, set_verbose -main.success_callback = ["posthog"] -main.failure_callback = ["slack", "sentry", "posthog"] -set_verbose(True) +import litellm +from litellm import embedding, completion + + + +litellm.success_callback = ["posthog"] +litellm.failure_callback = ["slack", "sentry", "posthog"] + user_message = "Hello, how are you?" messages = [{ "content": user_message,"role": "user"}]