Skip to content

Commit 8e886f8

Browse files
committed
Rename ModelActivity to ModelInvoker
1 parent 0765007 commit 8e886f8

File tree

2 files changed

+4
-8
lines changed

2 files changed

+4
-8
lines changed

samples-v2/openai_agents/_invoke_model_activity.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -206,11 +206,7 @@ def from_json(cls, json_str: str) -> 'ActivityModelInput':
206206
return cls(**data)
207207

208208

209-
class ModelActivity:
210-
"""Class wrapper for model invocation activities to allow model customization. By default, we use an OpenAIProvider with retries disabled.
211-
Disabling retries in your model of choice is recommended to allow activity retries to define the retry model.
212-
"""
213-
209+
class ModelInvoker:
214210
def __init__(self, model_provider: Optional[ModelProvider] = None):
215211
"""Initialize the activity with a model provider."""
216212
self._model_provider = model_provider or OpenAIProvider(

samples-v2/openai_agents/function_app.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -61,14 +61,14 @@ def basic_hello_world_orchestrator(context):
6161

6262
@app.activity_trigger(input_name="input")
6363
async def invoke_model_activity(input: str):
64-
from _invoke_model_activity import ModelActivity, ActivityModelInput
64+
from _invoke_model_activity import ModelInvoker, ActivityModelInput
6565
from pydantic_core import to_json
6666

6767
# Deserialize input string into ActivityModelInput object
6868
activity_input = ActivityModelInput.from_json(input)
6969

70-
model_activity = ModelActivity()
71-
result = await model_activity.invoke_model_activity(activity_input)
70+
model_invoker = ModelInvoker()
71+
result = await model_invoker.invoke_model_activity(activity_input)
7272

7373
json_obj = ModelResponse.__pydantic_serializer__.to_json(result)
7474
return json_obj.decode()

0 commit comments

Comments
 (0)