fix redis cache testing and add new test for azure

This commit is contained in:
ishaan-jaff 2023-09-11 08:01:00 -07:00
parent 3da6e68166
commit 1909444285
2 changed files with 14 additions and 0 deletions

View file

@ -6,6 +6,7 @@ tiktoken
importlib_metadata
baseten
cohere
redis
replicate
anthropic
gptcache

View file

@ -379,6 +379,19 @@ def test_completion_azure():
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# new azure test for using litellm. vars
# def test_completion_azure():
# try:
# print("azure gpt-3.5 test\n\n")
# response = completion(
# model="azure/chatgpt-v-2",
# messages=messages,
# )
# # Add any assertions here to check the response
# print(response)
# except Exception as e:
# pytest.fail(f"Error occurred: {e}")
def test_completion_azure_deployment_id():
try: