Skip to content

Commit 9352321

Browse files
committed
Don't try to have both
1 parent 6109d64 commit 9352321

File tree

2 files changed

+2
-9
lines changed

2 files changed

+2
-9
lines changed

temporalio/contrib/openai_agents/_invoke_model_activity.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -158,14 +158,9 @@ def __init__(self, model_provider: Optional[ModelProvider] = None):
158158
openai_client=AsyncOpenAI(max_retries=0)
159159
)
160160

161-
@activity.defn
162-
async def invoke_model(self, input: ActivityModelInput) -> ModelResponse:
163-
"""Activity that invokes a model with the given input."""
164-
return await self.invoke_model_activity(input)
165-
166161
@activity.defn
167162
@_auto_heartbeater
168-
async def invoke_model_activity(self, input: ActivityModelInput) -> ModelResponse:
163+
async def invoke_model(self, input: ActivityModelInput) -> ModelResponse:
169164
"""Activity that invokes a model with the given input."""
170165
model = self._model_provider.get_model(input.get("model_name"))
171166

temporalio/contrib/openai_agents/_temporal_openai_agents.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -281,10 +281,8 @@ def configure_worker(self, config: WorkerConfig) -> WorkerConfig:
281281
config["interceptors"] = list(config.get("interceptors") or []) + [
282282
OpenAIAgentsTracingInterceptor()
283283
]
284-
model_activity = ModelActivity(self._model_provider)
285284
config["activities"] = list(config.get("activities") or []) + [
286-
model_activity.invoke_model_activity,
287-
model_activity.invoke_model,
285+
ModelActivity(self._model_provider).invoke_model,
288286
]
289287
return self.next_worker_plugin.configure_worker(config)
290288

0 commit comments

Comments
 (0)