feat : integrate reasoning view

This commit is contained in:
martin legrand
2025-05-17 22:04:07 +02:00
parent 713c01193f
commit 102dc60efb
6 changed files with 53 additions and 3 deletions
+5
View File
@@ -44,6 +44,7 @@ class Agent():
self.blocks_result = []
self.success = True
self.last_answer = ""
self.last_reasoning = ""
self.status_message = "Haven't started yet"
self.verbose = verbose
self.executor = ThreadPoolExecutor(max_workers=1)
@@ -64,6 +65,10 @@ class Agent():
def get_last_answer(self) -> str:
return self.last_answer
@property
def get_last_reasoning(self) -> str:
return self.last_reasoning
@property
def get_blocks(self) -> list:
return self.blocks_result
+2 -1
View File
@@ -221,8 +221,9 @@ class PlannerAgent(Agent):
agent_prompt = self.make_prompt(task['task'], required_infos)
pretty_print(f"Agent {task['agent']} started working...", color="status")
self.logger.info(f"Agent {task['agent']} started working on {task['task']}.")
answer, _ = await self.agents[task['agent'].lower()].process(agent_prompt, None)
answer, reasoning = await self.agents[task['agent'].lower()].process(agent_prompt, None)
self.last_answer = answer
self.last_reasoning = reasoning
self.blocks_result = self.agents[task['agent'].lower()].blocks_result
agent_answer = self.agents[task['agent'].lower()].raw_answer_blocks(answer)
success = self.agents[task['agent'].lower()].get_success