scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 124

test: update test

Browse source
This commit is contained in:
Krrish Dholakia 2025-09-09 20:18:03 -07:00
parent ba6819dc18
commit 44566977f1
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/llm_translation/test_openai.py
View file

@ -451,7 +451,7 @@ class TestOpenAIGPT4OAudioTranscription(BaseLLMAudioTranscriptionTest):
def get_base_audio_transcription_call_args(self) -> dict:
return {
"model": "openai/gpt-4o-transcribe",
"response_format": "verbose_json",
# "response_format": "verbose_json",
"timestamp_granularities": ["word"],
}
@ -657,6 +657,7 @@ def test_openai_tool_calling():
response = litellm.completion(**completion_params)
@pytest.mark.asyncio
async def test_openai_gpt5_reasoning():
response = await litellm.acompletion(
@ -667,11 +668,12 @@ async def test_openai_gpt5_reasoning():
print("response: ", response)
assert response.choices[0].message.content is not None
@pytest.mark.asyncio
async def test_openai_safety_identifier_parameter():
"""Test that safety_identifier parameter is correctly passed to the OpenAI API."""
from openai import AsyncOpenAI
litellm.set_verbose = True
client = AsyncOpenAI(api_key="fake-api-key")
@ -700,7 +702,7 @@ async def test_openai_safety_identifier_parameter():
def test_openai_safety_identifier_parameter_sync():
"""Test that safety_identifier parameter is correctly passed to the OpenAI API."""
from openai import OpenAI
litellm.set_verbose = True
client = OpenAI(api_key="fake-api-key")
@ -724,4 +726,3 @@ def test_openai_safety_identifier_parameter_sync():
assert "safety_identifier" in request_body
# Verify safety_identifier is correctly sent to the API
assert request_body["safety_identifier"] == "user_code_123456"