Rename ActivityCallTracker to DurableAIAgentContext

Author: AnatoliB

samples-v2/openai_agents/activity_call_tracker.py renamed to samples-v2/openai_agents/durable_ai_agent_context.py

@@ -2,7 +2,7 @@
 from azure.durable_functions.models.DurableOrchestrationContext import DurableOrchestrationContext
 from yield_exception import YieldException
 
-class ActivityCallTracker:
+class DurableAIAgentContext:
     def __init__(self, context: DurableOrchestrationContext):
         self.context = context
         self.activities_called = 0

samples-v2/openai_agents/durable_decorators.py

@@ -2,25 +2,25 @@
 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 durable_ai_agent_context import DurableAIAgentContext
 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)
+        context = DurableAIAgentContext(context)
+        durable_openai_runner = DurableOpenAIRunner(context=context)
         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 from context.yield_and_clear_tasks()
             yield e.task
         finally:
-            yield from activity_call_tracker.yield_and_clear_tasks()
+            yield from context.yield_and_clear_tasks()
 
     return wrapper

samples-v2/openai_agents/durable_model_stub.py

@@ -8,7 +8,7 @@
     HandoffInput, 
     AgentOutputSchemaInput
 )
-from activity_call_tracker import ActivityCallTracker
+from durable_ai_agent_context import DurableAIAgentContext
 import azure.durable_functions as df
 
 import json
@@ -44,10 +44,10 @@ class _DurableModelStub(Model):
     def __init__(
         self,
         model_name: Optional[str],
-        activity_call_tracker: ActivityCallTracker,
+        context: DurableAIAgentContext,
     ) -> None:
         self.model_name = model_name
-        self.activity_call_tracker = activity_call_tracker
+        self.context = context
 
     async def get_response(
         self,
@@ -131,7 +131,7 @@ def make_tool_info(tool: Tool) -> ToolInput:
 
         activity_input_json = activity_input.to_json()
 
-        response = self.activity_call_tracker.call_activity("invoke_model_activity", activity_input_json)
+        response = self.context.call_activity("invoke_model_activity", activity_input_json)
         json_response = json.loads(response)
         model_response = ModelResponse(**json_response)
         return model_response

samples-v2/openai_agents/durable_openai_runner.py

@@ -1,6 +1,6 @@
 import logging
 from typing import Any
-from activity_call_tracker import ActivityCallTracker
+from durable_ai_agent_context import DurableAIAgentContext
 
 import json
 from dataclasses import replace
@@ -22,9 +22,9 @@
 logger = logging.getLogger(__name__)
 
 class DurableOpenAIRunner:
-    def __init__(self, activity_call_tracker: ActivityCallTracker) -> None:
+    def __init__(self, context: DurableAIAgentContext) -> None:
         self._runner = DEFAULT_AGENT_RUNNER or AgentRunner()
-        self.activity_call_tracker = activity_call_tracker
+        self.context = context
 
     def run_sync(
         self,
@@ -57,7 +57,7 @@ def run_sync(
             run_config,
             model = _DurableModelStub(
                 model_name = model_name,
-                activity_call_tracker = self.activity_call_tracker,
+                context = self.context,
             ),
         )