Skip to content

Commit 351e93d

Browse files
committed
Linting
1 parent ed7b9f5 commit 351e93d

File tree

2 files changed

+10
-3
lines changed

2 files changed

+10
-3
lines changed

temporalio/contrib/openai_agents/_invoke_model_activity.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -214,15 +214,21 @@ def make_tool(tool: ToolInput) -> Tool:
214214
raise e
215215
if should_retry_header == "false":
216216
raise ApplicationError(
217-
"Non retryable OpenAI error", non_retryable=True, next_retry_delay=retry_after
217+
"Non retryable OpenAI error",
218+
non_retryable=True,
219+
next_retry_delay=retry_after,
218220
) from e
219221

220222
# Specifically retryable status codes
221223
if e.response.status_code in [408, 409, 429, 500]:
222224
raise ApplicationError(
223-
"Retryable OpenAI status code", non_retryable=False, next_retry_delay=retry_after
225+
"Retryable OpenAI status code",
226+
non_retryable=False,
227+
next_retry_delay=retry_after,
224228
) from e
225229

226230
raise ApplicationError(
227-
"Non retryable OpenAI status code", non_retryable=True, next_retry_delay=retry_after
231+
"Non retryable OpenAI status code",
232+
non_retryable=True,
233+
next_retry_delay=retry_after,
228234
) from e

tests/contrib/openai_agents/test_openai.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1825,6 +1825,7 @@ def status_error(status: int):
18251825
assert event.activity_task_started_event_attributes.attempt == 1
18261826
assert found
18271827

1828+
18281829
async def test_exception_handling(client: Client):
18291830
new_config = client.config()
18301831
new_config["data_converter"] = pydantic_data_converter

0 commit comments

Comments
 (0)