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

test fixes

Browse Source
This commit is contained in:
Ishaan Jaffer 2025-10-25 11:20:09 -07:00
parent c0555c84c0
commit 3dffb6bd40
2 changed files with 4 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/llm_translation/test_azure_ai.py
View File

@ -300,16 +300,16 @@ async def test_azure_ai_request_format():
@pytest.mark.asyncio
@pytest.mark.parametrize("model", ["azure/gpt5_series/gpt-5", "azure/gpt-5"])
@pytest.mark.parametrize("model", ["azure/gpt5_series/gpt-5-mini", "azure/gpt-5-mini"])
async def test_azure_gpt5_reasoning(model):
litellm._turn_on_debug()
response = await litellm.acompletion(
model="azure/gpt5_series/gpt-5",
model=model,
messages=[{"role": "user", "content": "What is the capital of France?"}],
reasoning_effort="minimal",
max_tokens=10,
api_base=os.getenv("AZURE_GPT5_API_BASE"),
api_key=os.getenv("AZURE_GPT5_API_KEY"),
api_base=os.getenv("AZURE_API_BASE"),
api_key=os.getenv("AZURE_API_KEY"),
)
print("response: ", response)
assert response.choices[0].message.content is not None

17
tests/llm_translation/test_azure_openai.py
View File

@ -630,20 +630,3 @@ def test_azure_openai_responses_bridge():
== "test-azure-computer-use-preview"
)
assert mock_responses.call_args.kwargs["custom_llm_provider"] == "azure"
def test_azure_openai_gpt_5_responses_api():
try:
from litellm import responses
litellm._turn_on_debug()
response = responses(
model="azure/gpt-5",
input="Hi good morning",
api_key=os.getenv("AZURE_GPT5_API_KEY"),
api_base=os.getenv("AZURE_GPT5_API_BASE"),
)
print(f"response: {response}")
except litellm.RateLimitError:
pytest.skip("Skipping test due to RateLimitError")