auto setup

This commit is contained in:
Michael Neale
2025-09-15 13:19:02 -07:00
parent cedd565177
commit 8e821d0a5b
4 changed files with 102 additions and 0 deletions

View File

@@ -11,3 +11,4 @@ anyhow = { workspace = true }
thiserror = { workspace = true }
toml = "0.8"
shellexpand = "3.0"
dirs = "5.0"

View File

@@ -71,6 +71,50 @@ impl Default for Config {
impl Config {
pub fn load(config_path: Option<&str>) -> Result<Self> {
// Check if any config file exists
let config_exists = if let Some(path) = config_path {
Path::new(path).exists()
} else {
// Check default locations
let default_paths = [
"./g3.toml",
"~/.config/g3/config.toml",
"~/.g3.toml",
];
default_paths.iter().any(|path| {
let expanded_path = shellexpand::tilde(path);
Path::new(expanded_path.as_ref()).exists()
})
};
// If no config exists, create and save a default Qwen config
if !config_exists {
let qwen_config = Self::default_qwen_config();
// Save to default location
let config_dir = dirs::home_dir()
.map(|mut path| {
path.push(".config");
path.push("g3");
path
})
.unwrap_or_else(|| std::path::PathBuf::from("."));
// Create directory if it doesn't exist
std::fs::create_dir_all(&config_dir).ok();
let config_file = config_dir.join("config.toml");
if let Err(e) = qwen_config.save(config_file.to_str().unwrap()) {
eprintln!("Warning: Could not save default config: {}", e);
} else {
println!("Created default Qwen configuration at: {}", config_file.display());
}
return Ok(qwen_config);
}
// Existing config loading logic
let mut settings = config::Config::builder();
// Load default configuration
@@ -108,6 +152,30 @@ impl Config {
Ok(config)
}
fn default_qwen_config() -> Self {
Self {
providers: ProvidersConfig {
openai: None,
anthropic: None,
embedded: Some(EmbeddedConfig {
model_path: "~/.cache/g3/models/qwen2.5-7b-instruct-q3_k_m.gguf".to_string(),
model_type: "qwen".to_string(),
context_length: Some(32768), // Qwen2.5 supports 32k context
max_tokens: Some(2048),
temperature: Some(0.1),
gpu_layers: Some(32),
threads: Some(8),
}),
default_provider: "embedded".to_string(),
},
agent: AgentConfig {
max_context_length: 8192,
enable_streaming: true,
timeout_seconds: 60,
},
}
}
pub fn save(&self, path: &str) -> Result<()> {
let toml_string = toml::to_string_pretty(self)?;
std::fs::write(path, toml_string)?;