polymech/zeroclaw
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(onboard): correct custom model rerun command

Browse Source
This commit is contained in:
argenis de la rosa 2026-03-11 00:56:19 -04:00 committed by Argenis
parent 330229273e
commit 44ccbe6f5a
1 changed files with 42 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
src/onboard/wizard.rs
View File

@ -2939,6 +2939,21 @@ fn provider_supports_device_flow(provider_name: &str) -> bool {
)
}
fn onboarding_custom_model_rerun_command(config: &Config, provider: &str) -> Option<String> {
let model = config.default_model.as_deref()?.trim();
if model.is_empty() {
return None;
}
if model == default_model_for_provider(provider) {
return None;
}
Some(format!(
"zeroclaw onboard --api-key \"sk-...\" --provider {provider} --model {model}"
))
}
fn prompt_allowed_domains_for_tool(tool_name: &str) -> Result<Vec<String>> {
let prompt = format!(
" {}.allowed_domains (comma-separated, '*' allows all)",
@ -5915,6 +5930,14 @@ fn print_summary(config: &Config) {
" {}",
style(format!("export {env_var}=\"sk-...\"")).yellow()
);
if let Some(command) = onboarding_custom_model_rerun_command(config, provider) {
println!();
println!(
" {} Or re-run with your custom model:",
style(format!("{step}.")).cyan().bold()
);
println!(" {}", style(command).yellow());
}
}
println!();
step += 1;
@ -7328,6 +7351,25 @@ mod tests {
assert!(!provider_supports_device_flow("openrouter"));
}
#[test]
fn onboarding_custom_model_rerun_command_only_for_non_default_models() {
let mut config = Config::default();
config.default_model = Some("openrouter/auto".to_string());
assert_eq!(
onboarding_custom_model_rerun_command(&config, "openrouter"),
Some(
"zeroclaw onboard --api-key \"sk-...\" --provider openrouter --model openrouter/auto"
.to_string()
)
);
config.default_model = Some(default_model_for_provider("openrouter"));
assert_eq!(
onboarding_custom_model_rerun_command(&config, "openrouter"),
None
);
}
#[test]
fn local_provider_choices_include_sglang() {
let choices = local_provider_choices();