test: update test
This commit is contained in:
parent
ba6819dc18
commit
44566977f1
@ -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"
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user