Struct ConfigGenerator 

pub struct ConfigGenerator { /* private fields */ }

Modular configuration generator for auto-detecting model parameters

Implementations

impl ConfigGenerator

pub fn new() -> Result<Self>

Create a new config generator with all modules initialized

pub fn generate_config_from_directory_enhanced( &self, model_dir: &Path, model_id: &str, model_type: &str, ) -> Result<ModelConfig>

Generate a config from a downloaded model directory using enhanced metadata-driven detection

This function inspects .mlpackage files in a directory and generates a complete ModelConfig with proper shapes and component configurations, using metadata-driven component role detection to support both unified and split FFN architectures.

pub fn generate_config_from_directory( &self, model_dir: &Path, model_id: &str, model_type: &str, ) -> Result<ModelConfig>

Generate a config from a downloaded model directory (legacy method)

This function inspects .mlpackage files in a directory and generates a complete ModelConfig with proper shapes and component configurations.

pub fn load_cached_config(&self, model_id: &str) -> Result<Option<ModelConfig>>

Load a cached configuration if available

pub fn has_cached_config(&self, model_id: &str) -> bool

Check if a cached configuration exists

pub fn clear_cached_config(&self, model_id: &str) -> Result<()>

Clear cached configuration for a model

impl ConfigGenerator

pub fn find_mlpackage_files(&self, model_dir: &Path) -> Result<Vec<PathBuf>>

Find all .mlpackage files in a directory (legacy interface)

pub fn infer_component_name_from_file(&self, package_path: &Path) -> String

Infer component name from package filename (legacy interface)

pub fn analyze_mlpackage(&self, package_path: &Path) -> Result<ComponentConfig>

Analyze a single .mlpackage file (legacy interface)

pub fn extract_function_based_components( &self, package_path: &Path, _base_config: &ComponentConfig, ) -> Result<Option<HashMap<String, ComponentConfig>>>

Extract function-based components (legacy interface)

pub fn parse_tensor_configs_from_schema( &self, schema: &[Value], ) -> Result<HashMap<String, TensorConfig>>

Parse tensor configurations from schema (legacy interface)

pub fn compute_shape_info_generic( &self, components: &HashMap<String, ComponentConfig>, ) -> Result<ShapeConfig>

Compute shape info (legacy interface) Returns an error if components have insufficient tensor metadata

pub fn generate_naming_config_generic( &self, packages: &[PathBuf], ) -> NamingConfig

Generate naming config (legacy interface)

pub fn determine_execution_mode_generic( &self, components: &HashMap<String, ComponentConfig>, ) -> String

Determine execution mode (legacy interface)

pub fn cache_generated_config( &self, model_id: &str, config: &ModelConfig, ) -> Result<()>

Cache generated config (legacy interface)