diff --git a/openai_agents/basic/workflows/local_image_workflow.py b/openai_agents/basic/workflows/local_image_workflow.py index b536f517..9c82a599 100644 --- a/openai_agents/basic/workflows/local_image_workflow.py +++ b/openai_agents/basic/workflows/local_image_workflow.py @@ -1,3 +1,5 @@ +from datetime import timedelta + from agents import Agent, Runner from temporalio import workflow @@ -24,7 +26,7 @@ async def run( b64_image = await workflow.execute_activity( read_image_as_base64, image_path, - start_to_close_timeout=workflow.timedelta(seconds=30), + start_to_close_timeout=timedelta(seconds=30), ) agent = Agent( diff --git a/openai_agents/reasoning_content/workflows/reasoning_content_workflow.py b/openai_agents/reasoning_content/workflows/reasoning_content_workflow.py index 0a9f0a15..2a584a4b 100644 --- a/openai_agents/reasoning_content/workflows/reasoning_content_workflow.py +++ b/openai_agents/reasoning_content/workflows/reasoning_content_workflow.py @@ -1,4 +1,5 @@ from dataclasses import dataclass +from datetime import timedelta from temporalio import workflow @@ -22,7 +23,7 @@ async def run(self, prompt: str, model_name: str | None = None) -> ReasoningResu reasoning_content, regular_content = await workflow.execute_activity( get_reasoning_response, args=[prompt, model_name], - start_to_close_timeout=workflow.timedelta(minutes=5), + start_to_close_timeout=timedelta(minutes=5), ) return ReasoningResult(