This commit is contained in:
Zvonimir Sabljic
2024-07-24 08:51:10 -07:00
parent 8393eb46cd
commit 0ebe872250
2 changed files with 5 additions and 3 deletions

View File

@@ -105,7 +105,7 @@ class BugHunter(BaseAgent):
"fix_attempted": False
}]
if False and hunt_conclusion.conclusion == magic_words.PROBLEM_IDENTIFIED:
if hunt_conclusion.conclusion == magic_words.PROBLEM_IDENTIFIED:
# if no need for logs, implement iteration same as before
self.next_state.current_iteration["status"] = IterationStatus.AWAITING_BUG_FIX
await self.send_message("The bug is found - I'm attempting to fix it.")

View File

@@ -33,7 +33,7 @@ class Troubleshooter(IterationPromptMixin, BaseAgent):
agent_type = "troubleshooter"
display_name = "Troubleshooter"
async def run(self):
async def run(self) -> AgentResponse:
if self.current_state.unfinished_iterations:
if self.current_state.current_iteration.get("status") == IterationStatus.FIND_SOLUTION:
return await self.propose_solution()
@@ -98,7 +98,9 @@ class Troubleshooter(IterationPromptMixin, BaseAgent):
await self.trace_loop("loop-feedback")
elif bug_report is not None:
iteration_status = IterationStatus.HUNTING_FOR_BUG
elif change_description is not None:
else:
# should be - elif change_description is not None: - but to prevent bugs with the extension
# this might be caused if we show the input field instead of buttons
iteration_status = IterationStatus.FIND_SOLUTION