diff --git a/test/components/converters/test_json.py b/test/components/converters/test_json.py index a9eec33012f..9262964ac0d 100644 --- a/test/components/converters/test_json.py +++ b/test/components/converters/test_json.py @@ -234,7 +234,7 @@ def test_run_with_bad_filter(tmpdir, caplog): assert len(records) == 1 assert ( records[0].msg - == f'Failed to extract text from {test_file}. Skipping it. Error: Cannot index array with string "motivation"' + == f'Failed to extract text from {test_file}. Skipping it. Error: Cannot index array with string ("motivation")' ) assert result == {"documents": []} diff --git a/test/components/generators/chat/test_openai.py b/test/components/generators/chat/test_openai.py index af36840b726..6d1f88b04b5 100644 --- a/test/components/generators/chat/test_openai.py +++ b/test/components/generators/chat/test_openai.py @@ -1445,7 +1445,7 @@ def chat_completion_chunks(): completion_tokens_details=CompletionTokensDetails( accepted_prediction_tokens=0, audio_tokens=0, reasoning_tokens=0, rejected_prediction_tokens=0 ), - prompt_tokens_details=PromptTokensDetails(audio_tokens=0, cached_tokens=0), + prompt_tokens_details=PromptTokensDetails(audio_tokens=0, cached_tokens=0, cache_write_tokens=0), ), ), ] @@ -1652,7 +1652,7 @@ def streaming_chunks(): "reasoning_tokens": 0, "rejected_prediction_tokens": 0, }, - "prompt_tokens_details": {"audio_tokens": 0, "cached_tokens": 0}, + "prompt_tokens_details": {"audio_tokens": 0, "cached_tokens": 0, "cache_write_tokens": 0}, }, }, ), @@ -1870,7 +1870,7 @@ def test_handle_stream_response(self, chat_completion_chunks, chat_completion_ch "reasoning_tokens": 0, "rejected_prediction_tokens": 0, }, - "prompt_tokens_details": {"audio_tokens": 0, "cached_tokens": 0}, + "prompt_tokens_details": {"audio_tokens": 0, "cached_tokens": 0, "cache_write_tokens": 0}, } def test_convert_usage_chunk_to_streaming_chunk(self): diff --git a/test/components/generators/chat/test_openai_responses_conversion.py b/test/components/generators/chat/test_openai_responses_conversion.py index 1eba04068ae..90baedfccac 100644 --- a/test/components/generators/chat/test_openai_responses_conversion.py +++ b/test/components/generators/chat/test_openai_responses_conversion.py @@ -517,7 +517,7 @@ def test_convert_only_text(self): truncation="disabled", usage=ResponseUsage( input_tokens=15, - input_tokens_details=InputTokensDetails(cached_tokens=0), + input_tokens_details=InputTokensDetails(cached_tokens=0, cache_write_tokens=0), output_tokens=77, output_tokens_details=OutputTokensDetails(reasoning_tokens=64), total_tokens=92, @@ -863,7 +863,7 @@ def test_convert_only_text(self): "truncation": "disabled", "usage": { "input_tokens": 15, - "input_tokens_details": {"cached_tokens": 0}, + "input_tokens_details": {"cached_tokens": 0, "cache_write_tokens": 0}, "output_tokens": 77, "output_tokens_details": {"reasoning_tokens": 64}, "total_tokens": 92, @@ -1005,7 +1005,7 @@ def test_convert_only_function_call(self): reasoning=Reasoning(effort="medium", generate_summary=None, summary=None), usage=ResponseUsage( input_tokens=62, - input_tokens_details=InputTokensDetails(cached_tokens=0), + input_tokens_details=InputTokensDetails(cached_tokens=0, cache_write_tokens=0), output_tokens=83, output_tokens_details=OutputTokensDetails(reasoning_tokens=64), total_tokens=145, @@ -1207,7 +1207,7 @@ def test_convert_only_function_call(self): "reasoning": {"effort": "medium", "generate_summary": None, "summary": None}, "usage": { "input_tokens": 62, - "input_tokens_details": {"cached_tokens": 0}, + "input_tokens_details": {"cached_tokens": 0, "cache_write_tokens": 0}, "output_tokens": 83, "output_tokens_details": {"reasoning_tokens": 64}, "total_tokens": 145, diff --git a/test/components/generators/conftest.py b/test/components/generators/conftest.py index 55a44d3ef4a..7e9b1e9aea8 100644 --- a/test/components/generators/conftest.py +++ b/test/components/generators/conftest.py @@ -243,7 +243,7 @@ def openai_mock_responses(): reasoning=Reasoning(effort="low", generate_summary=None, summary="auto"), usage=ResponseUsage( input_tokens=11, - input_tokens_details=InputTokensDetails(cached_tokens=0), + input_tokens_details=InputTokensDetails(cached_tokens=0, cache_write_tokens=0), output_tokens=13, output_tokens_details=OutputTokensDetails(reasoning_tokens=0), total_tokens=24, @@ -309,7 +309,7 @@ def openai_mock_async_responses(): reasoning=Reasoning(effort="low", generate_summary=None, summary="auto"), usage=ResponseUsage( input_tokens=11, - input_tokens_details=InputTokensDetails(cached_tokens=0), + input_tokens_details=InputTokensDetails(cached_tokens=0, cache_write_tokens=0), output_tokens=13, output_tokens_details=OutputTokensDetails(reasoning_tokens=0), total_tokens=24,