Method restore

src/memory/context.py:957–1002

Restore a specific version of a memory system from history Args: memory_name: Name of the memory system version: Version string to restore auto_initialize: Whether to automatically initialize the restored memory Returns:

(self, memory_name: str, version: str, auto_initialize: bool = True)

                return False
    
    async def restore(self, memory_name: str, version: str, auto_initialize: bool = True) -> Optional[MemoryConfig]:
        """Restore a specific version of a memory system from history
        
        Args:
            memory_name: Name of the memory system
            version: Version string to restore
            auto_initialize: Whether to automatically initialize the restored memory
            
        Returns:
            MemoryConfig of the restored version, or None if not found
        """
        # Look up version from dict-based history (O(1) lookup)
        version_config = None
        if memory_name in self._memory_history_versions:
            version_config = self._memory_history_versions[memory_name].get(version)
        
        if version_config is None:
            logger.warning(f"| ⚠️ Version {version} not found for memory {memory_name}")
            return None
        
        # Create a copy to avoid modifying the history
        restored_config = MemoryConfig(**version_config.model_dump())
        
        # Set as current active config
        self._memory_configs[memory_name] = restored_config
        
        # Update version manager current version
        version_history = await version_manager.get_version_history("memory", memory_name)
        if version_history:
            # Check if version exists in version history, if not register it
            if version not in version_history.versions:
                await version_manager.register_version("memory", memory_name, version)
            version_history.current_version = version
        else:
            # If version history doesn't exist, register the version first
            await version_manager.register_version("memory", memory_name, version)
        
        # Initialize if requested
        if auto_initialize and restored_config.cls is not None:
            await self.build(restored_config)
        
        # Persist to JSON (current_version changes)
        await self.save_to_json()
        
        logger.info(f"| 🔄 Restored memory {memory_name} to version {version}")
        return restored_config
    
    async def get_variables(self, memory_name: Optional[str] = None) -> Dict[str, 'Variable']:
        """Get variables from memory systems, where each memory's code is used as the variable value.