feat: allow setting ollama host (#874)

This commit is contained in:
Yingjie He
2025-01-29 11:34:41 -08:00
committed by GitHub
parent c59bf888e8
commit 129e4a9e44
12 changed files with 40 additions and 12 deletions

View File

@@ -266,7 +266,8 @@ pub async fn configure_provider_dialog() -> Result<bool, Box<dyn Error>> {
let spin = spinner();
spin.start("Checking your configuration...");
let model_config = goose::model::ModelConfig::new(model.clone());
// Use max tokens to speed up the provider test.
let model_config = goose::model::ModelConfig::new(model.clone()).with_max_tokens(Some(10));
let provider = create(provider_name, model_config)?;
let message = Message::user().with_text(

View File

@@ -13,9 +13,11 @@ struct SecretResponse {
}
#[derive(Deserialize)]
#[serde(rename_all = "camelCase")]
struct SecretRequest {
key: String,
value: String,
is_secret: bool,
}
async fn store_secret(
@@ -33,7 +35,13 @@ async fn store_secret(
return Err(StatusCode::UNAUTHORIZED);
}
match Config::global().set_secret(&request.key, Value::String(request.value)) {
let config = Config::global();
let result = if request.is_secret {
config.set_secret(&request.key, Value::String(request.value))
} else {
config.set(&request.key, Value::String(request.value))
};
match result {
Ok(_) => Ok(Json(SecretResponse { error: false })),
Err(_) => Ok(Json(SecretResponse { error: true })),
}

View File

@@ -71,7 +71,7 @@ impl Provider for OllamaProvider {
OLLAMA_DOC_URL,
vec![ConfigKey::new(
"OLLAMA_HOST",
false,
true,
false,
Some(OLLAMA_HOST),
)],