use keyring for databricks host

block · Nov 28, 2024 · 46afb7d · 46afb7d
1 parent 3d3f2bd
commit 46afb7d
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/crates/goose-cli/src/profile.rs b/crates/goose-cli/src/profile.rs
@@ -140,11 +140,9 @@ pub fn set_provider_config(provider_name: &str, model: String) -> ProviderConfig
             })
         }
         PROVIDER_DATABRICKS => {
-            let host = get_env_value_or_input(
-                "DATABRICKS_HOST",
-                "Please enter your Databricks host:",
-                false,
-            );
+            let host = get_or_set_key(provider_name, "DATABRICKS_HOST")
+                .expect("Failed to get databricks host");
+
             ProviderConfig::Databricks(DatabricksProviderConfig {
                 host: host.clone(),
                 // TODO revisit configuration