Fix : file agent loop problem

This commit is contained in:
martin legrand
2025-03-11 12:44:20 +01:00
parent 479bc2769b
commit cace6b5612
8 changed files with 7 additions and 3 deletions
+3 -1
View File
@@ -3,10 +3,13 @@ from typing import Tuple, Callable
from abc import abstractmethod
import os
import random
import time
from sources.memory import Memory
from sources.utility import pretty_print
random.seed(time.time())
class executorResult:
"""
A class to store the result of a tool execution.
@@ -106,7 +109,6 @@ class Agent():
if speech_module is None:
return
messages = ["Please be patient sir, I am working on it.",
"At it, sir. In the meantime, how about a joke?",
"Computing... I recommand you have a coffee while I work.",
"Hold on, Im crunching numbers.",
"Working on it sir, please let me think."]
-1
View File
@@ -22,7 +22,6 @@ class CasualAgent(Agent):
def process(self, prompt, speech_module) -> str:
complete = False
exec_success = False
self.memory.push('user', prompt)
self.wait_message(speech_module)
+4
View File
@@ -30,6 +30,10 @@ class FileAgent(Agent):
exec_success, _ = self.execute_modules(answer)
answer = self.remove_blocks(answer)
self.last_answer = answer
complete = True
for name, tool in self.tools.items():
if tool.found_executable_blocks():
complete = False # AI read results and continue the conversation
return answer, reasoning
if __name__ == "__main__":