diff --git a/sources/agents/planner_agent.py b/sources/agents/planner_agent.py index 9f300e3..21f99ab 100644 --- a/sources/agents/planner_agent.py +++ b/sources/agents/planner_agent.py @@ -23,9 +23,9 @@ class PlannerAgent(Agent): "web": BrowserAgent(name, "prompts/base/browser_agent.txt", provider, verbose=False, browser=browser) } self.role = { - "en": "Research, setup and code", - "fr": "Recherche, configuration et codage", - "zh": "研究,设置和编码", + "en": "Complex Task", + "fr": "Tache complexe", + "zh": "复杂任务", } self.type = "planner_agent" diff --git a/sources/interaction.py b/sources/interaction.py index c5820bc..7bf148f 100644 --- a/sources/interaction.py +++ b/sources/interaction.py @@ -58,6 +58,8 @@ class Interaction: def load_last_session(self): """Recover the last session.""" for agent in self.agents: + if agent.type == "planner_agent": + continue agent.memory.load_memory(agent.type) def save_session(self): diff --git a/sources/speech_to_text.py b/sources/speech_to_text.py index 5b8b9ce..c4c7d4b 100644 --- a/sources/speech_to_text.py +++ b/sources/speech_to_text.py @@ -97,7 +97,7 @@ class Transcript: return "cuda:0" else: return "cpu" - + def remove_hallucinations(self, text: str) -> str: """Remove model hallucinations from the text.""" # TODO find a better way to do this