test(test_proxy.py): fix azure test

This commit is contained in:
Krrish Dholakia 2023-10-20 09:21:51 -07:00
parent 9d1ce893b7
commit 8ad152a7a4
2 changed files with 43 additions and 1 deletions

View file

@ -0,0 +1,41 @@
{
"20231020092117501221": {
"pre_api_call": {
"model": "chatgpt-v-2",
"messages": [
{
"role": "user",
"content": "Hey!"
}
],
"optional_params": {},
"litellm_params": {
"return_async": false,
"api_key": null,
"force_timeout": 600,
"logger_fn": null,
"verbose": false,
"custom_llm_provider": "azure",
"api_base": "https://openai-gpt-4-test-v-1.openai.azure.com/",
"litellm_call_id": "853a40c6-2b0a-41f0-9750-4974d079a6e7",
"model_alias_map": {},
"completion_call_id": null,
"metadata": null,
"stream_response": {}
},
"input": [
{
"role": "user",
"content": "Hey!"
}
],
"api_key": "6314c6dc63f448c9873844297f408c74",
"additional_args": {
"headers": null,
"api_version": "2023-05-15",
"api_base": "https://openai-gpt-4-test-v-1.openai.azure.com/"
},
"log_event_type": "pre_api_call"
}
}
}

View file

@ -20,11 +20,12 @@ def test_azure_call():
"model": "azure/chatgpt-v-2",
"messages": [{"role": "user", "content": "Hey!"}]
}
result = litellm_completion(data=data, user_api_base=os.getenv("AZURE_API_BASE"), type="chat_completion", user_temperature=None, user_max_tokens=None, user_model=None, user_headers=None, user_debug=False)
result = litellm_completion(data=data, user_api_base=os.getenv("AZURE_API_BASE"), type="chat_completion", user_temperature=None, user_max_tokens=None, user_model=None, user_headers=None, user_debug=False, model_router=None)
return result
except Exception as e:
pytest.fail(f"An error occurred: {e}")
# test_azure_call()
## test debug
def test_debug():
try: