diff --git a/tests/contrib/openai_agents/test_openai_streaming.py b/tests/contrib/openai_agents/test_openai_streaming.py index 851dc207d..ab711cd86 100644 --- a/tests/contrib/openai_agents/test_openai_streaming.py +++ b/tests/contrib/openai_agents/test_openai_streaming.py @@ -159,7 +159,9 @@ async def stream_response( input_tokens=10, output_tokens=5, total_tokens=15, - input_tokens_details=InputTokensDetails(cached_tokens=0), + input_tokens_details=InputTokensDetails.model_validate( + {"cached_tokens": 0, "cache_write_tokens": 0} + ), output_tokens_details=OutputTokensDetails(reasoning_tokens=0), ), )