scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-04 02:31:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 69

test(responses): use real provider config in spend regression

Browse source
This commit is contained in:
Gianpietro Dal Zio 2026-09-29 17:51:10 +02:00
parent 2fe85623d1
commit 58d153c66a
No known key found for this signature in database
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/unit/responses/test_streaming_iterator.py
View file

@ -279,6 +279,7 @@ async def test_completed_event_is_unwrapped_only_for_non_streaming_callers(strea
@pytest.mark.asyncio
@pytest.mark.parametrize("stream", [False, True])
async def test_completed_response_builds_real_spend_log_payload(stream: bool):
from litellm.llms.openai.responses.transformation import OpenAIResponsesAPIConfig
from litellm.proxy.spend_tracking.spend_tracking_utils import get_logging_payload
from litellm.types.llms.openai import ResponseAPIUsage
@ -327,7 +328,7 @@ async def test_completed_response_builds_real_spend_log_payload(stream: bool):
iterator: Final = ResponsesAPIStreamingIterator(
response=httpx.Response(200),
model=model,
responses_api_provider_config=Mock(spec=BaseResponsesAPIConfig),
responses_api_provider_config=OpenAIResponsesAPIConfig(),
logging_obj=logging_obj,
)
iterator.completed_response = ResponseCompletedEvent(