Skip to content

Commit 3d67afd

Browse files
committed
use actual classes
1 parent a078464 commit 3d67afd

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

tests/integrations/openai_agents/test_openai_agents.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,11 @@
1919
ResponseFunctionToolCall,
2020
)
2121

22+
from openai.types.responses.response_usage import (
23+
InputTokensDetails,
24+
OutputTokensDetails,
25+
)
26+
2227
test_run_config = agents.RunConfig(tracing_disabled=True)
2328

2429

@@ -29,8 +34,8 @@ def mock_usage():
2934
input_tokens=10,
3035
output_tokens=20,
3136
total_tokens=30,
32-
input_tokens_details={"cached_tokens": 0},
33-
output_tokens_details={"reasoning_tokens": 5},
37+
input_tokens_details=InputTokensDetails(cached_tokens=0),
38+
output_tokens_details=OutputTokensDetails(reasoning_tokens=5),
3439
)
3540

3641

0 commit comments

Comments
 (0)