Extract yield_and_clear_tasks method

Author: AnatoliB

samples-v2/openai_agents/activity_call_tracker.py

@@ -25,3 +25,9 @@ def call_activity(self, activity_name, input: str):
             result_json = completed_tasks[self.activities_called - 1].Result
             result = json.loads(result_json)
             return result
+
+    def yield_and_clear_tasks(self):
+        """Yield all accumulated tasks and clear the tasks list."""
+        for task in self.tasks_to_yield:
+            yield task
+        self.tasks_to_yield.clear()

samples-v2/openai_agents/function_app.py

@@ -31,14 +31,10 @@ def basic_hello_world_orchestrator(context):
         result = main()
         return result
     except YieldException as e:
-        for task in activity_call_tracker.tasks_to_yield:
-            yield task
-        activity_call_tracker.tasks_to_yield.clear()
+        yield from activity_call_tracker.yield_and_clear_tasks()
         yield e.task
     finally:
-        for task in activity_call_tracker.tasks_to_yield:
-            yield task
-        activity_call_tracker.tasks_to_yield.clear()
+        yield from activity_call_tracker.yield_and_clear_tasks()
 
 
 @app.activity_trigger(input_name="input")