Skip to content

Commit 926e3f7

Browse files
committed
Remove 'activity' from the model activity name
1 parent 5080b68 commit 926e3f7

File tree

3 files changed

+9
-2
lines changed

3 files changed

+9
-2
lines changed

temporalio/contrib/openai_agents/_invoke_model_activity.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,11 @@ 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+
161166
@activity.defn
162167
@_auto_heartbeater
163168
async def invoke_model_activity(self, input: ActivityModelInput) -> ModelResponse:

temporalio/contrib/openai_agents/_temporal_model_stub.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,7 @@ def make_tool_info(tool: Tool) -> ToolInput:
153153
summary = None
154154

155155
return await workflow.execute_activity_method(
156-
ModelActivity.invoke_model_activity,
156+
ModelActivity.invoke_model,
157157
activity_input,
158158
summary=summary,
159159
task_queue=self.model_params.task_queue,

temporalio/contrib/openai_agents/_temporal_openai_agents.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -281,8 +281,10 @@ 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)
284285
config["activities"] = list(config.get("activities") or []) + [
285-
ModelActivity(self._model_provider).invoke_model_activity
286+
model_activity.invoke_model_activity,
287+
model_activity.invoke_model,
286288
]
287289
return self.next_worker_plugin.configure_worker(config)
288290

0 commit comments

Comments
 (0)