Extract invoke_model_activity to a separate file

Author: AnatoliB

samples-v2/openai_agents/function_app.py

@@ -1,11 +1,14 @@
 import azure.functions as func
-from agents import ModelResponse
 from model_invoker import ModelInvoker, ActivityModelInput
 from durable_decorators import durable_openai_agent_orchestrator
+from model_activity import create_invoke_model_activity
 
 
 app = func.FunctionApp(http_auth_level=func.AuthLevel.FUNCTION)
 
+# TODO: Can we hide this line?
+create_invoke_model_activity(app)
+
 
 @app.route(route="orchestrators/{functionName}")
 @app.durable_client_input(client_name="client")
@@ -22,14 +25,3 @@ def basic_hello_world_orchestrator(context):
     from basic.hello_world import main
     result = main()
     return result
-
-
-@app.activity_trigger(input_name="input")
-async def invoke_model_activity(input: str):
-    activity_input = ActivityModelInput.from_json(input)
-    
-    model_invoker = ModelInvoker()
-    result = await model_invoker.invoke_model_activity(activity_input)
-
-    json_obj = ModelResponse.__pydantic_serializer__.to_json(result)
-    return json_obj.decode()

samples-v2/openai_agents/model_activity.py

@@ -0,0 +1,19 @@
+import azure.functions as func
+from agents import ModelResponse
+from model_invoker import ModelInvoker, ActivityModelInput
+
+
+def create_invoke_model_activity(app: func.FunctionApp):
+    """Create and register the invoke_model_activity function with the provided FunctionApp."""
+    
+    @app.activity_trigger(input_name="input")
+    async def invoke_model_activity(input: str):
+        activity_input = ActivityModelInput.from_json(input)
+        
+        model_invoker = ModelInvoker()
+        result = await model_invoker.invoke_model_activity(activity_input)
+
+        json_obj = ModelResponse.__pydantic_serializer__.to_json(result)
+        return json_obj.decode()
+    
+    return invoke_model_activity