forked from phoenix/litellm-mirror
fix redis cache testing and add new test for azure
This commit is contained in:
parent
3da6e68166
commit
1909444285
2 changed files with 14 additions and 0 deletions
|
@ -6,6 +6,7 @@ tiktoken
|
|||
importlib_metadata
|
||||
baseten
|
||||
cohere
|
||||
redis
|
||||
replicate
|
||||
anthropic
|
||||
gptcache
|
||||
|
|
|
@ -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:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue