File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed
temporalio/contrib/openai_agents Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff 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 :
You can’t perform that action at this time.
0 commit comments