diff --git a/tests/conftest.py b/tests/conftest.py index 9c687027b6..1f05f0eda6 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -1432,6 +1432,7 @@ def nonstreaming_responses_model_response(): input_tokens=10, input_tokens_details=openai.types.responses.response_usage.InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=20, output_tokens_details=openai.types.responses.response_usage.OutputTokensDetails( diff --git a/tests/integrations/langchain/test_langchain.py b/tests/integrations/langchain/test_langchain.py index f46cec9652..9e956f3d3b 100644 --- a/tests/integrations/langchain/test_langchain.py +++ b/tests/integrations/langchain/test_langchain.py @@ -768,6 +768,7 @@ def test_tool_execution_span( input_tokens=142, input_tokens_details=InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=50, output_tokens_details=OutputTokensDetails( @@ -779,6 +780,7 @@ def test_tool_execution_span( input_tokens=89, input_tokens_details=InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=28, output_tokens_details=OutputTokensDetails( diff --git a/tests/integrations/openai/test_openai.py b/tests/integrations/openai/test_openai.py index 8182c09d12..69ca86f533 100644 --- a/tests/integrations/openai/test_openai.py +++ b/tests/integrations/openai/test_openai.py @@ -96,6 +96,7 @@ async def __call__(self, *args, **kwargs): input_tokens=20, input_tokens_details=InputTokensDetails( cached_tokens=5, + cache_write_tokens=0, ), output_tokens=10, output_tokens_details=OutputTokensDetails( @@ -5166,6 +5167,7 @@ async def test_error_in_responses_async_api( input_tokens=20, input_tokens_details=InputTokensDetails( cached_tokens=5, + cache_write_tokens=0, ), output_tokens=10, output_tokens_details=OutputTokensDetails( diff --git a/tests/integrations/openai_agents/test_openai_agents.py b/tests/integrations/openai_agents/test_openai_agents.py index a68cb96b74..fdb323f2c4 100644 --- a/tests/integrations/openai_agents/test_openai_agents.py +++ b/tests/integrations/openai_agents/test_openai_agents.py @@ -92,6 +92,7 @@ def sync_event_loop(): input_tokens=20, input_tokens_details=InputTokensDetails( cached_tokens=5, + cache_write_tokens=0, ), output_tokens=10, output_tokens_details=OutputTokensDetails( @@ -109,7 +110,10 @@ def mock_usage(): input_tokens=10, output_tokens=20, total_tokens=30, - input_tokens_details=InputTokensDetails(cached_tokens=0), + input_tokens_details=InputTokensDetails( + cached_tokens=0, + cache_write_tokens=0, + ), output_tokens_details=OutputTokensDetails(reasoning_tokens=5), ) @@ -1536,6 +1540,7 @@ async def test_handoff_span( input_tokens=10, input_tokens_details=InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=20, output_tokens_details=OutputTokensDetails( @@ -1575,6 +1580,7 @@ async def test_handoff_span( input_tokens=10, input_tokens_details=InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=20, output_tokens_details=OutputTokensDetails( @@ -1713,6 +1719,7 @@ async def test_max_turns_before_handoff_span( input_tokens=10, input_tokens_details=InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=20, output_tokens_details=OutputTokensDetails( @@ -1752,6 +1759,7 @@ async def test_max_turns_before_handoff_span( input_tokens=10, input_tokens_details=InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=20, output_tokens_details=OutputTokensDetails( @@ -1876,6 +1884,7 @@ def simple_test_tool(message: str) -> str: input_tokens=10, input_tokens_details=InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=5, output_tokens_details=OutputTokensDetails( @@ -1887,6 +1896,7 @@ def simple_test_tool(message: str) -> str: input_tokens=15, input_tokens_details=InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=10, output_tokens_details=OutputTokensDetails( @@ -2664,6 +2674,7 @@ async def test_hosted_mcp_tool_propagation_header_streamed( input_tokens=20, input_tokens_details=InputTokensDetails( cached_tokens=5, + cache_write_tokens=0, ), output_tokens=10, output_tokens_details=OutputTokensDetails( @@ -3607,6 +3618,7 @@ def failing_tool(message: str) -> str: input_tokens=10, input_tokens_details=InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=5, output_tokens_details=OutputTokensDetails( @@ -3618,6 +3630,7 @@ def failing_tool(message: str) -> str: input_tokens=15, input_tokens_details=InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=10, output_tokens_details=OutputTokensDetails( @@ -3778,6 +3791,7 @@ async def test_invoke_agent_span_includes_usage_data( input_tokens=10, input_tokens_details=InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=20, output_tokens_details=OutputTokensDetails( @@ -3921,6 +3935,7 @@ async def test_ai_client_span_includes_response_model( input_tokens=10, input_tokens_details=InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=20, output_tokens_details=OutputTokensDetails( @@ -4052,6 +4067,7 @@ async def test_ai_client_span_response_model_with_chat_completions( input_tokens=15, input_tokens_details=InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=25, output_tokens_details=OutputTokensDetails( @@ -4177,6 +4193,7 @@ def calculator(a: int, b: int) -> int: input_tokens=10, input_tokens_details=InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=5, output_tokens_details=OutputTokensDetails( @@ -4216,6 +4233,7 @@ def calculator(a: int, b: int) -> int: input_tokens=20, input_tokens_details=InputTokensDetails( cached_tokens=5, + cache_write_tokens=0, ), output_tokens=15, output_tokens_details=OutputTokensDetails( @@ -4391,6 +4409,7 @@ async def test_invoke_agent_span_includes_response_model( input_tokens=10, input_tokens_details=InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=20, output_tokens_details=OutputTokensDetails( @@ -4539,6 +4558,7 @@ def calculator(a: int, b: int) -> int: input_tokens=10, input_tokens_details=InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=5, output_tokens_details=OutputTokensDetails( @@ -4578,6 +4598,7 @@ def calculator(a: int, b: int) -> int: input_tokens=20, input_tokens_details=InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=15, output_tokens_details=OutputTokensDetails( @@ -4922,6 +4943,7 @@ async def test_streaming_ttft_on_chat_span( input_tokens=20, input_tokens_details=InputTokensDetails( cached_tokens=5, + cache_write_tokens=0, ), output_tokens=10, output_tokens_details=OutputTokensDetails( @@ -5183,6 +5205,7 @@ def simple_tool(message: str) -> str: input_tokens=10, input_tokens_details=InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=5, output_tokens_details=OutputTokensDetails( @@ -5222,6 +5245,7 @@ def simple_tool(message: str) -> str: input_tokens=20, input_tokens_details=InputTokensDetails( cached_tokens=5, + cache_write_tokens=0, ), output_tokens=10, output_tokens_details=OutputTokensDetails( @@ -5627,6 +5651,7 @@ async def test_runner_run_streamed_with_starting_agent_kwarg( input_tokens=10, input_tokens_details=InputTokensDetails( cached_tokens=0, + cache_write_tokens=0, ), output_tokens=20, output_tokens_details=OutputTokensDetails(