phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Add test_openai_deployments_model_chat_completions_azure

Browse source
This commit is contained in:
Marc Abramowitz 2024-05-02 10:27:32 -07:00
parent dd166680d1
commit 152b5c8ceb
1 changed files with 28 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

28
litellm/tests/test_proxy_server.py
View file

@ -104,6 +104,34 @@ def test_chat_completion_azure(client_no_auth):
# test_chat_completion_azure()
def test_openai_deployments_model_chat_completions_azure(client_no_auth):
global headers
try:
# Your test data
test_data = {
"model": "azure/chatgpt-v-2",
"messages": [
{"role": "user", "content": "write 1 sentence poem"},
],
"max_tokens": 10,
}
url = "/openai/deployments/azure/chatgpt-v-2/chat/completions"
print(f"testing proxy server with Azure Request {url}")
response = client_no_auth.post(url, json=test_data)
assert response.status_code == 200
result = response.json()
print(f"Received response: {result}")
assert len(result["choices"][0]["message"]["content"]) > 0
except Exception as e:
pytest.fail(f"LiteLLM Proxy test failed. Exception - {str(e)}")
# Run the test
# test_openai_deployments_model_chat_completions_azure()
### EMBEDDING
def test_embedding(client_no_auth):
global headers