tkmind/tkmind_go
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

fix: unblock acp via databricks (#5562)

Browse Source
This commit is contained in:
Alex Hancock
2025-11-04 11:40:05 -05:00
committed by GitHub
parent d4a7bd39be
commit b5eaa6a3e4
1 changed files with 18 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose/src/providers/databricks.rs
+18 -22
View File
@@ -152,28 +152,24 @@ impl DatabricksProvider {
}; };
// Check if the default fast model exists in the workspace // Check if the default fast model exists in the workspace
let model_with_fast = tokio::task::block_in_place(|| { let model_with_fast = if let Ok(Some(models)) = provider.fetch_supported_models().await {
tokio::runtime::Handle::current().block_on(async { if models.contains(&DATABRICKS_DEFAULT_FAST_MODEL.to_string()) {
if let Ok(Some(models)) = provider.fetch_supported_models().await { tracing::debug!(
if models.contains(&DATABRICKS_DEFAULT_FAST_MODEL.to_string()) { "Found {} in Databricks workspace, setting as fast model",
tracing::debug!( DATABRICKS_DEFAULT_FAST_MODEL
"Found {} in Databricks workspace, setting as fast model", );
DATABRICKS_DEFAULT_FAST_MODEL model.with_fast(DATABRICKS_DEFAULT_FAST_MODEL.to_string())
); } else {
model.with_fast(DATABRICKS_DEFAULT_FAST_MODEL.to_string()) tracing::debug!(
} else { "{} not found in Databricks workspace, not setting fast model",
tracing::debug!( DATABRICKS_DEFAULT_FAST_MODEL
"{} not found in Databricks workspace, not setting fast model", );
DATABRICKS_DEFAULT_FAST_MODEL model
); }
model } else {
} tracing::debug!("Could not fetch Databricks models, not setting fast model");
} else { model
tracing::debug!("Could not fetch Databricks models, not setting fast model"); };
model
}
})
});
provider.model = model_with_fast; provider.model = model_with_fast;
Ok(provider) Ok(provider)