Rename public methods to private in DurableAIAgentContext

Author: AnatoliB

samples-v2/openai_agents/durable_ai_agent_context.py

@@ -17,7 +17,7 @@ def __init__(self, context: DurableOrchestrationContext):
         self._activities_called = 0
         self._tasks_to_yield = []
 
-    def get_activity_call_result(self, activity_name, input: str):
+    def _get_activity_call_result(self, activity_name, input: str):
         task = self._context.call_activity(activity_name, input)
 
         self._activities_called += 1
@@ -46,7 +46,7 @@ def set_custom_status(self, status: str):
     def wait_for_external_event(self, event_name: str):
         return self._context.wait_for_external_event(event_name)
 
-    def yield_and_clear_tasks(self):
+    def _yield_and_clear_tasks(self):
         """Yield all accumulated tasks and clear the tasks list."""
         for task in self._tasks_to_yield:
             yield task
@@ -71,7 +71,7 @@ def activity_as_tool(
         activity_name = activity_func._function._name
 
         async def run_activity(ctx: RunContextWrapper[Any], input: str) -> Any:
-            result = self.get_activity_call_result(activity_name, input)
+            result = self._get_activity_call_result(activity_name, input)
             return result
 
         schema = function_schema(

samples-v2/openai_agents/durable_decorators.py

@@ -77,11 +77,11 @@ def wrapper(durable_orchestration_context: DurableOrchestrationContext):
             try:
                 # prime the subiterator
                 value = next(gen)
-                yield from durable_ai_agent_context.yield_and_clear_tasks()
+                yield from durable_ai_agent_context._yield_and_clear_tasks()
                 while True:
                     try:
                         # send whatever was sent into us down to the subgenerator
-                        yield from durable_ai_agent_context.yield_and_clear_tasks()
+                        yield from durable_ai_agent_context._yield_and_clear_tasks()
                         sent = yield value
                     except GeneratorExit:
                         # ensure the subgenerator is closed
@@ -101,19 +101,19 @@ def wrapper(durable_orchestration_context: DurableOrchestrationContext):
                         else:
                             value = next(gen)
             except StopIteration as e:
-                yield from durable_ai_agent_context.yield_and_clear_tasks()
+                yield from durable_ai_agent_context._yield_and_clear_tasks()
                 return e.value
             except YieldException as e:
-                yield from durable_ai_agent_context.yield_and_clear_tasks()
+                yield from durable_ai_agent_context._yield_and_clear_tasks()
                 yield e.task
         else:
             try:
                 result = func(durable_ai_agent_context)
                 return result
             except YieldException as e:
-                yield from durable_ai_agent_context.yield_and_clear_tasks()
+                yield from durable_ai_agent_context._yield_and_clear_tasks()
                 yield e.task
             finally:
-                yield from durable_ai_agent_context.yield_and_clear_tasks()
+                yield from durable_ai_agent_context._yield_and_clear_tasks()
 
     return wrapper

samples-v2/openai_agents/durable_model_stub.py

@@ -131,7 +131,7 @@ def make_tool_info(tool: Tool) -> ToolInput:
 
         activity_input_json = activity_input.to_json()
 
-        response = self.context.get_activity_call_result("invoke_model_activity", activity_input_json)
+        response = self.context._get_activity_call_result("invoke_model_activity", activity_input_json)
         json_response = json.loads(response)
         model_response = ModelResponse(**json_response)
         return model_response