LLM/pentestagent
1
0
Fork 0
mirror of https://github.com/GH05TCREW/pentestagent.git synced 2026-03-07 14:23:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed broken code for assist mode when changing the code for agent mode for tool result visualization.

Browse Source
This commit is contained in:
famez
2026-02-16 14:28:21 +01:00
parent 9e7e84c03c
commit 0e7bd4f087
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pentestagent/interface/tui.py
View File

@@ -3186,6 +3186,7 @@ Be concise. Use the actual data from notes."""
self._set_status("thinking", "assist")
try:
last_tool_message: ToolMessage
async for response in self.agent.assist(message):
if self._should_stop:
self._add_system("[!] Stopped by user")
@@ -3207,7 +3208,7 @@ Be concise. Use the actual data from notes."""
if call.name == "finish":
continue # Skip - summary shown as final message
args_str = str(call.arguments)
self._add_tool(call.name, args_str)
last_tool_message = self._add_tool(call.name, args_str)
# Show tool results (displayed after execution completes)
# Skip 'finish' tool - its result is shown as the final summary
@@ -3216,11 +3217,11 @@ Be concise. Use the actual data from notes."""
if result.tool_name == "finish":
continue # Skip - summary shown separately
if result.success:
self._add_tool_result(
self._add_tool_result(last_tool_message,
result.tool_name, result.result or "Done"
)
else:
self._add_tool_result(
self._add_tool_result(last_tool_message,
result.tool_name, f"Error: {result.error}"
)