Skip to content

Commit f9ae167

Browse files
committed
Add invoke_model_activity
1 parent acc41aa commit f9ae167

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

samples-v2/openai_agents/durable_model_stub.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
from __future__ import annotations
2+
from _invoke_model_activity import ActivityModelInput, ModelTracingInput
23
import azure.durable_functions as df
34

45
import logging
@@ -119,7 +120,7 @@ def make_tool_info(tool: Tool) -> ToolInput:
119120
)
120121

121122
activity_output =self.durable_orchestration_context.call_activity(
122-
"InvokeModelActivity",
123+
"invoke_model_activity",
123124
activity_input
124125
)
125126

samples-v2/openai_agents/function_app.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
from typing import Any
12
import azure.functions as func
23
import logging
34

@@ -22,3 +23,10 @@ def basic_hello_world_orchestrator(context):
2223
from basic.hello_world import main
2324
result = main()
2425
return result
26+
27+
@app.activity_trigger(input_name="input")
28+
def invoke_model_activity(input: Any):
29+
# Instantiate ModelActivity
30+
from _invoke_model_activity import ModelActivity, ActivityModelInput
31+
model_activity = ModelActivity()
32+
return model_activity.invoke_model_activity(input)

0 commit comments

Comments
 (0)