@@ -360,8 +360,8 @@ def _runtime_checks(self, break_point: Optional[AgentBreakpoint], snapshot: Opti
360
360
if break_point and isinstance (break_point .break_point , ToolBreakpoint ):
361
361
_validate_tool_breakpoint_is_valid (agent_breakpoint = break_point , tools = self .tools )
362
362
363
+ @staticmethod
363
364
def _check_chat_generator_breakpoint (
364
- self ,
365
365
execution_context : _ExecutionContext ,
366
366
break_point : Optional [AgentBreakpoint ],
367
367
parent_snapshot : Optional [PipelineSnapshot ],
@@ -398,8 +398,8 @@ def _check_chat_generator_breakpoint(
398
398
)
399
399
_trigger_chat_generator_breakpoint (agent_snapshot = agent_snapshot , parent_snapshot = parent_snapshot )
400
400
401
+ @staticmethod
401
402
def _check_tool_invoker_breakpoint (
402
- self ,
403
403
execution_context : _ExecutionContext ,
404
404
break_point : Optional [AgentBreakpoint ],
405
405
parent_snapshot : Optional [PipelineSnapshot ],
@@ -490,7 +490,7 @@ def run(
490
490
491
491
while exe_context .counter < self .max_agent_steps :
492
492
# Handle breakpoint and ChatGenerator call
493
- self ._check_chat_generator_breakpoint (
493
+ Agent ._check_chat_generator_breakpoint (
494
494
execution_context = exe_context , break_point = break_point , parent_snapshot = parent_snapshot
495
495
)
496
496
# We skip the chat generator when restarting from a snapshot from a ToolBreakpoint
@@ -515,7 +515,7 @@ def run(
515
515
break
516
516
517
517
# Handle breakpoint and ToolInvoker call
518
- self ._check_tool_invoker_breakpoint (
518
+ Agent ._check_tool_invoker_breakpoint (
519
519
execution_context = exe_context , break_point = break_point , parent_snapshot = parent_snapshot
520
520
)
521
521
# We only send the messages from the LLM to the tool invoker
0 commit comments