diff --git a/src/control_backend/agents/bdi/text_belief_extractor_agent.py b/src/control_backend/agents/bdi/text_belief_extractor_agent.py index 7e3570f..feabf40 100644 --- a/src/control_backend/agents/bdi/text_belief_extractor_agent.py +++ b/src/control_backend/agents/bdi/text_belief_extractor_agent.py @@ -116,9 +116,19 @@ class TextBeliefExtractorAgent(BaseAgent): self._handle_beliefs_message(msg) case "goals": self._handle_goals_message(msg) + case "conversation_history": + if msg.body == "reset": + self._reset() case _: self.logger.warning("Received unexpected message from %s", msg.sender) + def _reset(self): + self.conversation = ChatHistory(messages=[]) + self.belief_inferrer.available_beliefs.clear() + self._current_beliefs = BeliefState() + self.goal_inferrer.goals.clear() + self._current_goal_completions = {} + def _handle_beliefs_message(self, msg: InternalMessage): try: belief_list = BeliefList.model_validate_json(msg.body)