phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 10:14:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(test) proxy - azure/chat/completion test

Browse source
This commit is contained in:
ishaan-jaff 2023-11-23 20:56:40 -08:00
parent ca852e1dcd
commit 39072bd196
1 changed files with 52 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

52
litellm/tests/test_proxy_server.py
View file

@ -46,3 +46,55 @@ def test_chat_completion():
# Run the test
test_chat_completion()
def test_chat_completion_azure():
try:
# Your test data
test_data = {
"model": "azure/chatgpt-v-2",
"messages": [
{
"role": "user",
"content": "hi"
},
],
"max_tokens": 10,
}
print("testing proxy server with Azure Request")
response = client.post("/v1/chat/completions", json=test_data)
assert response.status_code == 200
result = response.json()
print(f"Received response: {result}")
except Exception as e:
pytest.fail("LiteLLM Proxy test failed. Exception", e)
# Run the test
test_chat_completion()
# def test_embedding():
# try:
# # Your test data
# test_data = {
# "model": "",
# "messages": [
# {
# "role": "user",
# "content": "hi"
# },
# ],
# "max_tokens": 10,
# }
# print("testing proxy server with OpenAI embedding")
# response = client.post("/v1/embeddings", json=test_data)
# assert response.status_code == 200
# result = response.json()
# print(f"Received response: {result}")
# except Exception as e:
# pytest.fail("LiteLLM Proxy test failed. Exception", e)
# # Run the test
# test_embedding()