Add durable_openai_agent_orchestrator decorator

Author: AnatoliB

samples-v2/openai_agents/function_app.py

@@ -6,10 +6,32 @@
 from activity_call_tracker import ActivityCallTracker
 from model_invoker import ModelInvoker, ActivityModelInput
 import event_loop
+from functools import wraps
 
 
 app = func.FunctionApp(http_auth_level=func.AuthLevel.FUNCTION)
 
+
+def durable_openai_agent_orchestrator(func):
+    @wraps(func)
+    def wrapper(context):
+        event_loop.ensure_event_loop()
+        activity_call_tracker = ActivityCallTracker(context)
+        durable_openai_runner = DurableOpenAIRunner(activity_call_tracker=activity_call_tracker)
+        set_default_agent_runner(durable_openai_runner)
+
+        try:
+            result = func(context)
+            return result
+        except YieldException as e:
+            yield from activity_call_tracker.yield_and_clear_tasks()
+            yield e.task
+        finally:
+            yield from activity_call_tracker.yield_and_clear_tasks()
+    
+    return wrapper
+
+
 @app.route(route="orchestrators/{functionName}")
 @app.durable_client_input(client_name="client")
 async def orchestration_starter(req: func.HttpRequest, client):
@@ -20,21 +42,11 @@ async def orchestration_starter(req: func.HttpRequest, client):
 
 
 @app.orchestration_trigger(context_name="context")
+@durable_openai_agent_orchestrator
 def basic_hello_world_orchestrator(context):
-    event_loop.ensure_event_loop()
-    activity_call_tracker = ActivityCallTracker(context)
-    durable_openai_runner = DurableOpenAIRunner(activity_call_tracker=activity_call_tracker)
-    set_default_agent_runner(durable_openai_runner)
-
-    try:
-        from basic.hello_world import main
-        result = main()
-        return result
-    except YieldException as e:
-        yield from activity_call_tracker.yield_and_clear_tasks()
-        yield e.task
-    finally:
-        yield from activity_call_tracker.yield_and_clear_tasks()
+    from basic.hello_world import main
+    result = main()
+    return result
 
 
 @app.activity_trigger(input_name="input")