ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test azure openai fix

Browse Source
This commit is contained in:
Ishaan Jaffer 2025-09-27 10:17:16 -07:00
parent d5c716fab9
commit f2f8391384
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/llm_translation/test_azure_openai.py
View File

@ -339,7 +339,7 @@ def test_azure_gpt_4o_with_tool_call_and_response_format(api_version):
with patch.object(client.chat.completions.with_raw_response, "create") as mock_post:
response = litellm.completion(
model="azure/gpt-4o-new-test",
model="azure/gpt-4.1-nano",
messages=[
{
"role": "system",
@ -474,7 +474,7 @@ def test_azure_max_retries_0(
try:
completion(
model="azure/gpt-4o-new-test",
model="azure/gpt-4.1-nano",
messages=[{"role": "user", "content": "Hello world"}],
max_retries=max_retries,
stream=stream,
@ -502,7 +502,7 @@ async def test_async_azure_max_retries_0(
try:
await acompletion(
model="azure/gpt-4o-new-test",
model="azure/gpt-4.1-nano",
messages=[{"role": "user", "content": "Hello world"}],
max_retries=max_retries,
stream=stream,
@ -565,7 +565,7 @@ async def test_azure_embedding_max_retries_0(
from litellm import aembedding, embedding
args = {
"model": "azure/azure-embedding-model",
"model": "text-embedding-ada-002",
"input": "Hello world",
"max_retries": max_retries,
}
@ -598,7 +598,7 @@ def test_azure_safety_result():
litellm._turn_on_debug()
response = completion(
model="azure/gpt-4o-new-test",
model="azure/gpt-4.1-nano",
messages=[{"role": "user", "content": "Hello world"}],
)
print(f"response: {response}")