Skip to content

Commit bf34389

Browse files
committed
Fix tests
1 parent c6b9e3a commit bf34389

File tree

1 file changed

+37
-34
lines changed

1 file changed

+37
-34
lines changed

tests/integrations/openai/test_openai.py

Lines changed: 37 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -62,41 +62,44 @@ async def __call__(self, *args, **kwargs):
6262
)
6363

6464

65-
EXAMPLE_RESPONSE = Response(
66-
id="chat-id",
67-
output=[
68-
ResponseOutputMessage(
69-
id="message-id",
70-
content=[
71-
ResponseOutputText(
72-
annotations=[],
73-
text="the model response",
74-
type="output_text",
75-
),
76-
],
77-
role="assistant",
78-
status="completed",
79-
type="message",
80-
),
81-
],
82-
parallel_tool_calls=False,
83-
tool_choice="none",
84-
tools=[],
85-
created_at=10000000,
86-
model="model-id",
87-
object="response",
88-
usage=ResponseUsage(
89-
input_tokens=20,
90-
input_tokens_details=InputTokensDetails(
91-
cached_tokens=5,
92-
),
93-
output_tokens=10,
94-
output_tokens_details=OutputTokensDetails(
95-
reasoning_tokens=8,
65+
if SKIP_API_TESTS:
66+
EXAMPLE_RESPONSE = None
67+
else:
68+
EXAMPLE_RESPONSE = Response(
69+
id="chat-id",
70+
output=[
71+
ResponseOutputMessage(
72+
id="message-id",
73+
content=[
74+
ResponseOutputText(
75+
annotations=[],
76+
text="the model response",
77+
type="output_text",
78+
),
79+
],
80+
role="assistant",
81+
status="completed",
82+
type="message",
83+
),
84+
],
85+
parallel_tool_calls=False,
86+
tool_choice="none",
87+
tools=[],
88+
created_at=10000000,
89+
model="model-id",
90+
object="response",
91+
usage=ResponseUsage(
92+
input_tokens=20,
93+
input_tokens_details=InputTokensDetails(
94+
cached_tokens=5,
95+
),
96+
output_tokens=10,
97+
output_tokens_details=OutputTokensDetails(
98+
reasoning_tokens=8,
99+
),
100+
total_tokens=30,
96101
),
97-
total_tokens=30,
98-
),
99-
)
102+
)
100103

101104

102105
async def async_iterator(values):

0 commit comments

Comments
 (0)