fixing exception handling test

This commit is contained in:
Krrish Dholakia 2023-08-12 16:49:55 -07:00
parent ffde46c1a1
commit 05aa99fde3

View file

@ -37,7 +37,7 @@ def test_context_window(model):
try: try:
azure = model == "chatgpt-test" azure = model == "chatgpt-test"
print(f"model: {model}") print(f"model: {model}")
response = completion(model=model, messages=messages, custom_llm_provider=custom_llm_provider, logger_fn=logging_fn) response = completion(model=model, messages=messages, custom_llm_provider="azure", logger_fn=logging_fn)
print(f"response: {response}") print(f"response: {response}")
except InvalidRequestError: except InvalidRequestError:
print("InvalidRequestError") print("InvalidRequestError")