Fixup merge.

Author: philliphoff

azure/durable_functions/decorators/durable_app.py

@@ -303,7 +303,7 @@ def generator_wrapper_wrapper(func):
             @wraps(func)
             def generator_wrapper(context):
                 return durable_openai_agent_orchestrator_generator(
-                    func, context, model_retry_options activity_name
+                    func, context, model_retry_options, activity_name
                 )
 
             return generator_wrapper

azure/durable_functions/openai_agents/context.py

@@ -19,10 +19,12 @@ def __init__(
         context: DurableOrchestrationContext,
         task_tracker: TaskTracker,
         model_retry_options: Optional[RetryOptions],
+        activity_name: str,
     ):
         self._context = context
         self._task_tracker = task_tracker
         self._model_retry_options = model_retry_options
+        self.activity_name = activity_name
 
     def call_activity(self, activity_name, input: str):
         """Call an activity function and record the activity call."""

azure/durable_functions/openai_agents/orchestrator_generator.py

@@ -34,7 +34,8 @@ def durable_openai_agent_orchestrator_generator(
     durable_ai_agent_context = DurableAIAgentContext(
         durable_orchestration_context, task_tracker, model_retry_options, activity_name
     )
-    durable_openai_runner = DurableOpenAIRunner(context=durable_ai_agent_context)
+    durable_openai_runner = DurableOpenAIRunner(
+        context=durable_ai_agent_context, activity_name=activity_name)
     set_default_agent_runner(durable_openai_runner)
 
     func_with_context = partial(func, durable_ai_agent_context)