tenseleyflow/loader / 312824d

Browse files

Adopt runtime context for turn iteration

Authored by espadonne
SHA
312824db47022049740c943132d095e91881e8fa
Parents
0b4b303
Tree
c64fc2c

2 changed files

StatusFile+-
M src/loader/runtime/conversation.py 1 1
M src/loader/runtime/turn_iteration.py 4 3
src/loader/runtime/conversation.pymodified
@@ -93,7 +93,7 @@ class ConversationRuntime:
9393
             turn_completion=self.turn_completion,
9494
         )
9595
         self.turn_iteration = TurnIterationController(
96
-            agent,
96
+            self.context,
9797
             phase_tracker=self.phase_tracker,
9898
             turn_requester=AssistantTurnRequester(self.context, self.tracer),
9999
             repairer=self.repairer,
src/loader/runtime/turn_iteration.pymodified
@@ -8,6 +8,7 @@ from enum import StrEnum
88
99
 from ..llm.base import Message, Role
1010
 from .assistant_turns import AssistantTurnRequester
11
+from .context import RuntimeContext
1112
 from .dod import DefinitionOfDone
1213
 from .events import AgentEvent, TurnSummary
1314
 from .executor import ToolExecutor
@@ -53,14 +54,14 @@ class TurnIterationController:
5354
5455
     def __init__(
5556
         self,
56
-        agent,
57
+        context: RuntimeContext,
5758
         *,
5859
         phase_tracker: TurnPhaseTracker,
5960
         turn_requester: AssistantTurnRequester,
6061
         repairer: ResponseRepairer,
6162
         response_router: AssistantResponseRouter,
6263
     ) -> None:
63
-        self.agent = agent
64
+        self.context = context
6465
         self.phase_tracker = phase_tracker
6566
         self.turn_requester = turn_requester
6667
         self.repairer = repairer
@@ -219,7 +220,7 @@ class TurnIterationController:
219220
             max_empty_retries=max_empty_retries,
220221
         )
221222
         if empty_decision.should_continue and empty_decision.retry_message:
222
-            self.agent.session.append(
223
+            self.context.session.append(
223224
                 Message(
224225
                     role=Role.USER,
225226
                     content=empty_decision.retry_message,