Skip to content

Commit d5cb1a5

Browse files
committed
Rename private variables
1 parent 3b74f99 commit d5cb1a5

File tree

1 file changed

+15
-15
lines changed

1 file changed

+15
-15
lines changed

samples-v2/openai_agents/durable_ai_agent_context.py

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -13,44 +13,44 @@
1313

1414
class DurableAIAgentContext:
1515
def __init__(self, context: DurableOrchestrationContext):
16-
self.context = context
17-
self.activities_called = 0
18-
self.tasks_to_yield = []
16+
self._context = context
17+
self._activities_called = 0
18+
self._tasks_to_yield = []
1919

2020
def get_activity_call_result(self, activity_name, input: str):
21-
task = self.context.call_activity(activity_name, input)
21+
task = self._context.call_activity(activity_name, input)
2222

23-
self.activities_called += 1
23+
self._activities_called += 1
2424

25-
histories = self.context.histories
25+
histories = self._context.histories
2626
completed_tasks = [entry for entry in histories if entry.event_type == 5]
27-
if len(completed_tasks) < self.activities_called:
27+
if len(completed_tasks) < self._activities_called:
2828
# yield immediately
2929
raise YieldException(task)
3030
else:
3131
# yield later
32-
self.tasks_to_yield.append(task)
32+
self._tasks_to_yield.append(task)
3333

34-
result_json = completed_tasks[self.activities_called - 1].Result
34+
result_json = completed_tasks[self._activities_called - 1].Result
3535
result = json.loads(result_json)
3636
return result
3737

3838
def call_activity(self, activity_name, input: str):
39-
task = self.context.call_activity(activity_name, input)
40-
self.activities_called += 1
39+
task = self._context.call_activity(activity_name, input)
40+
self._activities_called += 1
4141
return task
4242

4343
def set_custom_status(self, status: str):
44-
self.context.set_custom_status(status)
44+
self._context.set_custom_status(status)
4545

4646
def wait_for_external_event(self, event_name: str):
47-
return self.context.wait_for_external_event(event_name)
47+
return self._context.wait_for_external_event(event_name)
4848

4949
def yield_and_clear_tasks(self):
5050
"""Yield all accumulated tasks and clear the tasks list."""
51-
for task in self.tasks_to_yield:
51+
for task in self._tasks_to_yield:
5252
yield task
53-
self.tasks_to_yield.clear()
53+
self._tasks_to_yield.clear()
5454

5555
def activity_as_tool(
5656
self,

0 commit comments

Comments
 (0)