Zep Memory Assistant - AI Agent with Human-Like Memory

class ZepConversableAgent(ConversableAgent):
    """Custom AutoGen agent with Zep memory integration."""

    def __init__(self, name: str, system_message: str, llm_config: dict,
                 zep_session_id: str, zep_client: Zep, min_fact_rating: float):
        super().__init__(name=name, system_message=system_message, llm_config=llm_config)
        self.zep_session_id = zep_session_id
        self.zep_client = zep_client
        self.original_system_message = system_message
        self.register_hook('process_message_before_send', self._zep_persist_assistant_messages)

    def _zep_fetch_and_update_system_message(self):
        """Fetch facts and inject into system prompt."""
        memory = self.zep_client.memory.get(self.zep_session_id, min_rating=self.min_fact_rating)
        context = memory.context or 'No specific facts recalled.'
        self.update_system_message(
            self.original_system_message + f'\n\nRelevant facts:\n{context}'
        )