Skip to content

Commit 298245d

Browse files
alexander-alderman-webbalexander-alderman-webb
committed
restore token numbers
1 parent a6c52d6 commit 298245d

1 file changed

Lines changed: 39 additions & 39 deletions

File tree

‎tests/integrations/openai_agents/test_openai_agents.py‎

Lines changed: 39 additions & 39 deletions
Original file line numberDiff line numberDiff line change
@@ -981,11 +981,11 @@ async def test_handoff_span(sentry_init, capture_events, get_model_response):
981981
input_tokens_details=InputTokensDetails(
982982
cached_tokens=0,
983983
),
984-
output_tokens=5,
984+
output_tokens=20,
985985
output_tokens_details=OutputTokensDetails(
986-
reasoning_tokens=0,
986+
reasoning_tokens=5,
987987
),
988-
total_tokens=15,
988+
total_tokens=30,
989989
),
990990
)
991991
)
@@ -1015,15 +1015,15 @@ async def test_handoff_span(sentry_init, capture_events, get_model_response):
10151015
model="gpt-4",
10161016
object="response",
10171017
usage=ResponseUsage(
1018-
input_tokens=15,
1018+
input_tokens=10,
10191019
input_tokens_details=InputTokensDetails(
10201020
cached_tokens=0,
10211021
),
1022-
output_tokens=10,
1022+
output_tokens=20,
10231023
output_tokens_details=OutputTokensDetails(
1024-
reasoning_tokens=0,
1024+
reasoning_tokens=5,
10251025
),
1026-
total_tokens=25,
1026+
total_tokens=30,
10271027
),
10281028
)
10291029
)
@@ -1108,11 +1108,11 @@ async def test_max_turns_before_handoff_span(
11081108
input_tokens_details=InputTokensDetails(
11091109
cached_tokens=0,
11101110
),
1111-
output_tokens=5,
1111+
output_tokens=20,
11121112
output_tokens_details=OutputTokensDetails(
1113-
reasoning_tokens=0,
1113+
reasoning_tokens=5,
11141114
),
1115-
total_tokens=15,
1115+
total_tokens=30,
11161116
),
11171117
)
11181118
)
@@ -1142,15 +1142,15 @@ async def test_max_turns_before_handoff_span(
11421142
model="gpt-4",
11431143
object="response",
11441144
usage=ResponseUsage(
1145-
input_tokens=15,
1145+
input_tokens=10,
11461146
input_tokens_details=InputTokensDetails(
11471147
cached_tokens=0,
11481148
),
1149-
output_tokens=10,
1149+
output_tokens=20,
11501150
output_tokens_details=OutputTokensDetails(
1151-
reasoning_tokens=0,
1151+
reasoning_tokens=5,
11521152
),
1153-
total_tokens=25,
1153+
total_tokens=30,
11541154
),
11551155
)
11561156
)
@@ -1901,15 +1901,15 @@ async def test_mcp_tool_execution_spans(
19011901
model="gpt-4.1-2025-04-14",
19021902
object="response",
19031903
usage=ResponseUsage(
1904-
input_tokens=10,
1904+
input_tokens=15,
19051905
input_tokens_details=InputTokensDetails(
19061906
cached_tokens=0,
19071907
),
1908-
output_tokens=20,
1908+
output_tokens=10,
19091909
output_tokens_details=OutputTokensDetails(
1910-
reasoning_tokens=5,
1910+
reasoning_tokens=0,
19111911
),
1912-
total_tokens=30,
1912+
total_tokens=25,
19131913
),
19141914
)
19151915
)
@@ -2031,15 +2031,15 @@ async def test_mcp_tool_execution_with_error(
20312031
model="gpt-4.1-2025-04-14",
20322032
object="response",
20332033
usage=ResponseUsage(
2034-
input_tokens=10,
2034+
input_tokens=15,
20352035
input_tokens_details=InputTokensDetails(
20362036
cached_tokens=0,
20372037
),
2038-
output_tokens=20,
2038+
output_tokens=10,
20392039
output_tokens_details=OutputTokensDetails(
2040-
reasoning_tokens=5,
2040+
reasoning_tokens=0,
20412041
),
2042-
total_tokens=30,
2042+
total_tokens=25,
20432043
),
20442044
)
20452045
)
@@ -2159,15 +2159,15 @@ async def test_mcp_tool_execution_without_pii(
21592159
model="gpt-4.1-2025-04-14",
21602160
object="response",
21612161
usage=ResponseUsage(
2162-
input_tokens=10,
2162+
input_tokens=15,
21632163
input_tokens_details=InputTokensDetails(
21642164
cached_tokens=0,
21652165
),
2166-
output_tokens=20,
2166+
output_tokens=10,
21672167
output_tokens_details=OutputTokensDetails(
21682168
reasoning_tokens=5,
21692169
),
2170-
total_tokens=30,
2170+
total_tokens=25,
21712171
),
21722172
)
21732173
)
@@ -2345,11 +2345,11 @@ def failing_tool(message: str) -> str:
23452345
usage=ResponseUsage(
23462346
input_tokens=10,
23472347
input_tokens_details=InputTokensDetails(
2348-
cached_tokens=5,
2348+
cached_tokens=0,
23492349
),
23502350
output_tokens=5,
23512351
output_tokens_details=OutputTokensDetails(
2352-
reasoning_tokens=3,
2352+
reasoning_tokens=0,
23532353
),
23542354
total_tokens=15,
23552355
),
@@ -2381,15 +2381,15 @@ def failing_tool(message: str) -> str:
23812381
model="gpt-4-0613",
23822382
object="response",
23832383
usage=ResponseUsage(
2384-
input_tokens=20,
2384+
input_tokens=15,
23852385
input_tokens_details=InputTokensDetails(
23862386
cached_tokens=0,
23872387
),
2388-
output_tokens=15,
2388+
output_tokens=10,
23892389
output_tokens_details=OutputTokensDetails(
23902390
reasoning_tokens=0,
23912391
),
2392-
total_tokens=35,
2392+
total_tokens=25,
23932393
),
23942394
)
23952395
)
@@ -2686,11 +2686,11 @@ def calculator(a: int, b: int) -> int:
26862686
usage=ResponseUsage(
26872687
input_tokens=10,
26882688
input_tokens_details=InputTokensDetails(
2689-
cached_tokens=5,
2689+
cached_tokens=0,
26902690
),
26912691
output_tokens=5,
26922692
output_tokens_details=OutputTokensDetails(
2693-
reasoning_tokens=3,
2693+
reasoning_tokens=0,
26942694
),
26952695
total_tokens=15,
26962696
),
@@ -2724,11 +2724,11 @@ def calculator(a: int, b: int) -> int:
27242724
usage=ResponseUsage(
27252725
input_tokens=20,
27262726
input_tokens_details=InputTokensDetails(
2727-
cached_tokens=0,
2727+
cached_tokens=5,
27282728
),
27292729
output_tokens=15,
27302730
output_tokens_details=OutputTokensDetails(
2731-
reasoning_tokens=0,
2731+
reasoning_tokens=3,
27322732
),
27332733
total_tokens=35,
27342734
),
@@ -3279,15 +3279,15 @@ def simple_tool(message: str) -> str:
32793279
model="gpt-4",
32803280
object="response",
32813281
usage=ResponseUsage(
3282-
input_tokens=20,
3282+
input_tokens=10,
32833283
input_tokens_details=InputTokensDetails(
3284-
cached_tokens=5,
3284+
cached_tokens=0,
32853285
),
3286-
output_tokens=10,
3286+
output_tokens=5,
32873287
output_tokens_details=OutputTokensDetails(
3288-
reasoning_tokens=8,
3288+
reasoning_tokens=0,
32893289
),
3290-
total_tokens=30,
3290+
total_tokens=15,
32913291
),
32923292
)
32933293
)

0 commit comments

Comments
 (0)