mirror of
https://github.com/aljazceru/goose.git
synced 2025-12-18 14:44:21 +01:00
feat: allow setting ollama host (#874)
This commit is contained in:
@@ -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(
|
||||
|
||||
@@ -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 })),
|
||||
}
|
||||
|
||||
@@ -71,7 +71,7 @@ impl Provider for OllamaProvider {
|
||||
OLLAMA_DOC_URL,
|
||||
vec![ConfigKey::new(
|
||||
"OLLAMA_HOST",
|
||||
false,
|
||||
true,
|
||||
false,
|
||||
Some(OLLAMA_HOST),
|
||||
)],
|
||||
|
||||
Reference in New Issue
Block a user