feat : enhance planner

This commit is contained in:
martin legrand 2025-04-25 21:15:49 +02:00
parent a11445e7c0
commit 309a481a69
3 changed files with 10 additions and 2 deletions

2
api.py
View File

@ -140,7 +140,7 @@ async def get_latest_answer():
"answer": interaction.current_agent.last_answer,
"agent_name": interaction.current_agent.agent_name if interaction.current_agent else "None",
"success": interaction.current_agent.success,
"blocks": {f'{i}': block.jsonify() for i, block in enumerate(interaction.current_agent.get_blocks_result())} if interaction.current_agent else {},
"blocks": {f'{i}': block.jsonify() for i, block in enumerate(interaction.get_last_blocks_result())} if interaction.current_agent else {},
"status": interaction.current_agent.get_status_message if interaction.current_agent else "No status available",
"uid": uid
}

View File

@ -242,5 +242,4 @@ class Agent():
self.memory.push('user', feedback)
if save_path != None:
tool.save_block(blocks, save_path)
self.blocks_result = self.blocks_result
return True, feedback

View File

@ -69,6 +69,15 @@ class Interaction:
break
return ai_name
def get_last_blocks_result(self) -> List[Dict]:
"""Get the last blocks result."""
if self.current_agent is None:
return []
blks = []
for agent in self.agents:
blks.extend(agent.get_blocks_result())
return blks
def load_last_session(self):
"""Recover the last session."""
for agent in self.agents: