Skip to content

Commit 984a50d

Browse files
committed
Delay openai client creation
1 parent aeac7e2 commit 984a50d

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

temporalio/contrib/openai_agents/_invoke_model_activity.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -155,14 +155,16 @@ class ModelActivity:
155155

156156
def __init__(self, model_provider: Optional[ModelProvider] = None):
157157
"""Initialize the activity with a model provider."""
158-
self._model_provider = model_provider or OpenAIProvider(
159-
openai_client=AsyncOpenAI(max_retries=0)
160-
)
158+
self._model_provider = model_provider
161159

162160
@activity.defn
163161
@_auto_heartbeater
164162
async def invoke_model_activity(self, input: ActivityModelInput) -> ModelResponse:
165163
"""Activity that invokes a model with the given input."""
164+
if not self._model_provider:
165+
self._model_provider = OpenAIProvider(
166+
openai_client=AsyncOpenAI(max_retries=0)
167+
)
166168
model = self._model_provider.get_model(input.get("model_name"))
167169

168170
async def empty_on_invoke_tool(ctx: RunContextWrapper[Any], input: str) -> str:

0 commit comments

Comments
 (0)