feat : integrate reasoning view

sources/agents/agent.py

@@ -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

sources/agents/planner_agent.py

@@ -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

sources/interaction.py

@@ -22,6 +22,7 @@ class Interaction:
         self.current_agent = None
         self.last_query = None
         self.last_answer = None
+        self.last_reasoning = None
         self.agents = agents
         self.tts_enabled = tts_enabled
         self.stt_enabled = stt_enabled
@@ -158,7 +159,7 @@ class Interaction:
         tmp = self.last_answer
         self.current_agent = agent
         self.is_generating = True
-        self.last_answer, _ = await agent.process(self.last_query, self.speech)
+        self.last_answer, self.last_reasoning = await agent.process(self.last_query, self.speech)
         self.is_generating = False
         if push_last_agent_memory:
             self.current_agent.memory.push('user', self.last_query)

sources/schemas.py

@@ -19,6 +19,7 @@ class QueryRequest(BaseModel):
 class QueryResponse(BaseModel):
     done: str
     answer: str
+    reasoning: str
     agent_name: str
     success: str
     blocks: dict
@@ -32,6 +33,7 @@ class QueryResponse(BaseModel):
         return {
             "done": self.done,
             "answer": self.answer,
+            "reasoning": self.reasoning,
             "agent_name": self.agent_name,
             "success": self.success,
             "blocks": self.blocks,