| 58 | |
| 59 | |
| 60 | class DummyUIManager: |
| 61 | def __init__(self): |
| 62 | self.printed_calls = [] |
| 63 | self.is_streaming_response = False |
| 64 | self.interrupt_requested = False |
| 65 | self.verbose_mode = False |
| 66 | self.console = MagicMock() |
| 67 | self._start_calls = [] |
| 68 | self._finish_calls = [] |
| 69 | |
| 70 | def print_tool_call(self, tool_name, raw_arguments): |
| 71 | self.printed_calls.append((tool_name, raw_arguments)) |
| 72 | |
| 73 | async def finish_tool_execution(self, result=None, success=True): |
| 74 | self._finish_calls.append((result, success)) |
| 75 | |
| 76 | async def do_confirm_tool_execution(self, tool_name, arguments): |
| 77 | return True |
| 78 | |
| 79 | async def start_tool_execution(self, tool_name, arguments): |
| 80 | self._start_calls.append((tool_name, arguments)) |
| 81 | |
| 82 | def finish_tool_calls(self): |
| 83 | pass |
| 84 | |
| 85 | |
| 86 | class DummyUIManagerNoFinish: |
no outgoing calls