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
|
importlib_metadata
|
||||||
baseten
|
baseten
|
||||||
cohere
|
cohere
|
||||||
|
redis
|
||||||
replicate
|
replicate
|
||||||
anthropic
|
anthropic
|
||||||
gptcache
|
gptcache
|
||||||
|
|
|
@ -379,6 +379,19 @@ def test_completion_azure():
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
pytest.fail(f"Error occurred: {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():
|
def test_completion_azure_deployment_id():
|
||||||
try:
|
try:
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue