From 0ebc7955da67b488877cab7929019d40688fb3ed Mon Sep 17 00:00:00 2001 From: sigoden Date: Mon, 25 Mar 2024 11:13:54 +0800 Subject: refactor: improve creating config for openai-compatible client (#374) --- src/config/mod.rs | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'src/config/mod.rs') diff --git a/src/config/mod.rs b/src/config/mod.rs index b6bc730..bcebdd0 100644 --- a/src/config/mod.rs +++ b/src/config/mod.rs @@ -1046,8 +1046,9 @@ fn create_config_file(config_path: &Path) -> Result<()> { let client = Select::new("Platform:", list_client_types()).prompt()?; let mut config = serde_json::json!({}); - config["model"] = client.into(); - config[CLIENTS_FIELD] = create_client_config(client)?; + let (model, clients_config) = create_client_config(client)?; + config["model"] = model.into(); + config[CLIENTS_FIELD] = clients_config; let config_data = serde_yaml::to_string(&config).with_context(|| "Failed to create config")?; -- cgit v1.2.3