Method restore

src/environment/context.py:1126–1171

Restore a specific version of an environment from history Args: env_name: Name of the environment version: Version string to restore auto_initialize: Whether to automatically initialize the restored environment Returns:

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

    
    
    async def restore(self, env_name: str, version: str, auto_initialize: bool = True) -> Optional[EnvironmentConfig]:
        """Restore a specific version of an environment from history
        
        Args:
            env_name: Name of the environment
            version: Version string to restore
            auto_initialize: Whether to automatically initialize the restored environment
            
        Returns:
            EnvironmentConfig of the restored version, or None if not found
        """
        # Look up version from dict-based history (O(1) lookup)
        version_config = None
        if env_name in self._environment_history_versions:
            version_config = self._environment_history_versions[env_name].get(version)
        
        if version_config is None:
            logger.warning(f"| ⚠️ Version {version} not found for environment {env_name}")
            return None
        
        # Create a copy to avoid modifying the history
        restored_config = EnvironmentConfig(**version_config.model_dump())
        
        # Set as current active config
        self._environment_configs[env_name] = restored_config
        
        # Update version manager current version
        version_history = await version_manager.get_version_history("environment", env_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("environment", env_name, version)
            version_history.current_version = version
        else:
            # If version history doesn't exist, register the version first
            await version_manager.register_version("environment", env_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 environment {env_name} to version {version}")
        return restored_config
    
    async def save_contract(self, env_names: Optional[List[str]] = None):
        """Save the contract for an environment"""