dearwolf/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-06-29 22:20:43 +00:00
Code Issues Projects Releases Packages Wiki Activity

Write a config file only if there is one specified

Browse source
This commit is contained in:
mudler 2023-06-24 00:53:53 +02:00
parent d9a1fafffe
commit e103fa4a87
1 changed files with 28 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

52
pkg/gallery/models.go
View file

@ -219,33 +219,37 @@ func InstallModel(basePath, nameOverride string, config *Config, configOverrides
return err return err
} }
configFilePath := filepath.Join(basePath, name+".yaml") // write config file
if len(configOverrides) != 0 || len(config.ConfigFile) != 0 {
configFilePath := filepath.Join(basePath, name+".yaml")
// Read and update config file as map[string]interface{} // Read and update config file as map[string]interface{}
configMap := make(map[string]interface{}) configMap := make(map[string]interface{})
err = yaml.Unmarshal([]byte(config.ConfigFile), &configMap) err = yaml.Unmarshal([]byte(config.ConfigFile), &configMap)
if err != nil { if err != nil {
return fmt.Errorf("failed to unmarshal config YAML: %v", err) return fmt.Errorf("failed to unmarshal config YAML: %v", err)
}
configMap["name"] = name
if err := mergo.Merge(&configMap, configOverrides, mergo.WithOverride); err != nil {
return err
}
// Write updated config file
updatedConfigYAML, err := yaml.Marshal(configMap)
if err != nil {
return fmt.Errorf("failed to marshal updated config YAML: %v", err)
}
err = os.WriteFile(configFilePath, updatedConfigYAML, 0644)
if err != nil {
return fmt.Errorf("failed to write updated config file: %v", err)
}
log.Debug().Msgf("Written config file %s", configFilePath)
} }
configMap["name"] = name
if err := mergo.Merge(&configMap, configOverrides, mergo.WithOverride); err != nil {
return err
}
// Write updated config file
updatedConfigYAML, err := yaml.Marshal(configMap)
if err != nil {
return fmt.Errorf("failed to marshal updated config YAML: %v", err)
}
err = os.WriteFile(configFilePath, updatedConfigYAML, 0644)
if err != nil {
return fmt.Errorf("failed to write updated config file: %v", err)
}
log.Debug().Msgf("Written config file %s", configFilePath)
return nil return nil
} }