Skip to content

Commit f629b0d

Browse files
committed
tests
1 parent 2c5d1f0 commit f629b0d

File tree

1 file changed

+67
-64
lines changed

1 file changed

+67
-64
lines changed

tests/integrations/openai/test_openai.py

Lines changed: 67 additions & 64 deletions
Original file line numberDiff line numberDiff line change
@@ -1214,74 +1214,77 @@ async def test_error_in_responses_async_api(sentry_init, capture_events):
12141214
)
12151215

12161216

1217-
EXAMPLE_RESPONSES_STREAM = [
1218-
ResponseCreatedEvent(
1219-
sequence_number=1,
1220-
type="response.created",
1221-
response=Response(
1222-
id="chat-id",
1223-
created_at=10000000,
1224-
model="response-model-id",
1225-
object="response",
1226-
output=[],
1227-
parallel_tool_calls=False,
1228-
tool_choice="none",
1229-
tools=[],
1217+
if SKIP_RESPONSES_TESTS:
1218+
EXAMPLE_RESPONSES_STREAM = []
1219+
else:
1220+
EXAMPLE_RESPONSES_STREAM = [
1221+
ResponseCreatedEvent(
1222+
sequence_number=1,
1223+
type="response.created",
1224+
response=Response(
1225+
id="chat-id",
1226+
created_at=10000000,
1227+
model="response-model-id",
1228+
object="response",
1229+
output=[],
1230+
parallel_tool_calls=False,
1231+
tool_choice="none",
1232+
tools=[],
1233+
),
12301234
),
1231-
),
1232-
ResponseTextDeltaEvent(
1233-
item_id="msg_1",
1234-
sequence_number=2,
1235-
type="response.output_text.delta",
1236-
logprobs=[],
1237-
content_index=0,
1238-
output_index=0,
1239-
delta="hel",
1240-
),
1241-
ResponseTextDeltaEvent(
1242-
item_id="msg_1",
1243-
sequence_number=3,
1244-
type="response.output_text.delta",
1245-
logprobs=[],
1246-
content_index=0,
1247-
output_index=0,
1248-
delta="lo ",
1249-
),
1250-
ResponseTextDeltaEvent(
1251-
item_id="msg_1",
1252-
sequence_number=4,
1253-
type="response.output_text.delta",
1254-
logprobs=[],
1255-
content_index=0,
1256-
output_index=0,
1257-
delta="world",
1258-
),
1259-
ResponseCompletedEvent(
1260-
sequence_number=5,
1261-
type="response.completed",
1262-
response=Response(
1263-
id="chat-id",
1264-
created_at=10000000,
1265-
model="response-model-id",
1266-
object="response",
1267-
output=[],
1268-
parallel_tool_calls=False,
1269-
tool_choice="none",
1270-
tools=[],
1271-
usage=ResponseUsage(
1272-
input_tokens=20,
1273-
input_tokens_details=InputTokensDetails(
1274-
cached_tokens=5,
1275-
),
1276-
output_tokens=10,
1277-
output_tokens_details=OutputTokensDetails(
1278-
reasoning_tokens=8,
1235+
ResponseTextDeltaEvent(
1236+
item_id="msg_1",
1237+
sequence_number=2,
1238+
type="response.output_text.delta",
1239+
logprobs=[],
1240+
content_index=0,
1241+
output_index=0,
1242+
delta="hel",
1243+
),
1244+
ResponseTextDeltaEvent(
1245+
item_id="msg_1",
1246+
sequence_number=3,
1247+
type="response.output_text.delta",
1248+
logprobs=[],
1249+
content_index=0,
1250+
output_index=0,
1251+
delta="lo ",
1252+
),
1253+
ResponseTextDeltaEvent(
1254+
item_id="msg_1",
1255+
sequence_number=4,
1256+
type="response.output_text.delta",
1257+
logprobs=[],
1258+
content_index=0,
1259+
output_index=0,
1260+
delta="world",
1261+
),
1262+
ResponseCompletedEvent(
1263+
sequence_number=5,
1264+
type="response.completed",
1265+
response=Response(
1266+
id="chat-id",
1267+
created_at=10000000,
1268+
model="response-model-id",
1269+
object="response",
1270+
output=[],
1271+
parallel_tool_calls=False,
1272+
tool_choice="none",
1273+
tools=[],
1274+
usage=ResponseUsage(
1275+
input_tokens=20,
1276+
input_tokens_details=InputTokensDetails(
1277+
cached_tokens=5,
1278+
),
1279+
output_tokens=10,
1280+
output_tokens_details=OutputTokensDetails(
1281+
reasoning_tokens=8,
1282+
),
1283+
total_tokens=30,
12791284
),
1280-
total_tokens=30,
12811285
),
12821286
),
1283-
),
1284-
]
1287+
]
12851288

12861289

12871290
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)