Method add_single

ms_agent/memory/default_memory.py:209–272

(self,
                         messages: List[Message],
                         user_id: Optional[int] = None,
                         agent_id: Optional[int] = None,
                         run_id: Optional[int] = None,
                         memory_type: Optional[str] = None,
                         msg_id: Optional[int] = None)

            idx += 1

    async def add_single(self,
                         messages: List[Message],
                         user_id: Optional[int] = None,
                         agent_id: Optional[int] = None,
                         run_id: Optional[int] = None,
                         memory_type: Optional[str] = None,
                         msg_id: Optional[int] = None) -> None:
        messages_dict = []
        for message in messages:
            if isinstance(message, Message):
                messages_dict.append(message.to_dict_clean())
            else:
                messages_dict.append(message)
        async with self._lock:
            if msg_id is None:
                self.max_msg_id += 1
                msg_id = self.max_msg_id
            else:
                self.max_msg_id = max(msg_id, self.max_msg_id)
            self.cache_messages[msg_id] = messages, self._hash_block(messages)

            try:
                self.memory.add(
                    messages_dict,
                    user_id=user_id or self.user_id,
                    agent_id=agent_id or self.agent_id,
                    run_id=run_id or self.run_id,
                    memory_type=memory_type)
                logger.info('Add memory success.')
            except Exception as e:
                logger.warning(f'Failed to add memory: {e}')

            if self.history_mode == 'overwrite':
                res = self.memory.get_all(
                    user_id=user_id or self.user_id,
                    agent_id=agent_id or self.agent_id,
                    run_id=run_id or self.run_id)  # sorted
                res = [(item['id'], item['memory']) for item in res['results']]
                if len(res):
                    logger.info('All memory info:')
                for item in res:
                    logger.info(item[1])
                valids = []
                unmatched = []
                for id, memory in res:
                    matched = False
                    for item in self.memory_snapshot:
                        if id == item.memory_id:
                            if item.value == memory and item.valid:
                                matched = True
                                valids.append(id)
                                break
                            else:
                                if item.valid:
                                    item.disable(msg_id)
                    if not matched:
                        unmatched.append((id, memory))
                for item in self.memory_snapshot: