Move durable_openai_agent_orchestrator to a separate file

Author: AnatoliB

samples-v2/openai_agents/durable_decorators.py

@@ -0,0 +1,26 @@
+from functools import wraps
+from agents.run import set_default_agent_runner
+from durable_openai_runner import DurableOpenAIRunner
+from yield_exception import YieldException
+from activity_call_tracker import ActivityCallTracker
+import event_loop
+
+
+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

samples-v2/openai_agents/function_app.py

@@ -1,37 +1,12 @@
 import azure.functions as func
 from agents import ModelResponse
-from agents.run import set_default_agent_runner
-from durable_openai_runner import DurableOpenAIRunner
-from yield_exception import YieldException
-from activity_call_tracker import ActivityCallTracker
 from model_invoker import ModelInvoker, ActivityModelInput
-import event_loop
-from functools import wraps
+from durable_decorators import durable_openai_agent_orchestrator
 
 
 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):