From 7f7802ec1b1d2ba6697f428adfcdb57b724da2e7 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 25 Oct 2023 11:26:58 -0700 Subject: [PATCH] (test) proxy server update caching tests --- openai_proxy/tests/test_caching.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/openai_proxy/tests/test_caching.py b/openai_proxy/tests/test_caching.py index d6940e1da..5151752ad 100644 --- a/openai_proxy/tests/test_caching.py +++ b/openai_proxy/tests/test_caching.py @@ -3,12 +3,13 @@ openai.api_base = "http://0.0.0.0:8000" dotenv.load_dotenv() openai.api_key = os.getenv("ANTHROPIC_API_KEY") # this gets passed as a header + response1 = openai.ChatCompletion.create( model = "claude-instant-1", messages = [ { "role": "user", - "content": "this is a test message, what model / llm are you" + "content": "write a short poem about litellm" } ], ) @@ -25,7 +26,7 @@ response2 = openai.ChatCompletion.create( messages = [ { "role": "user", - "content": "this is a test message, what model / llm are you" + "content": "write a short poem about litellm" } ], ) @@ -43,7 +44,7 @@ try: messages = [ { "role": "user", - "content": "this is a test message, what model / llm are you" + "content": "write a short poem about litellm" } ], )