diff --git a/crates/goose/src/providers/databricks.rs b/crates/goose/src/providers/databricks.rs index 16fced6ac2..b1b2edc669 100644 --- a/crates/goose/src/providers/databricks.rs +++ b/crates/goose/src/providers/databricks.rs @@ -36,7 +36,7 @@ const DEFAULT_TIMEOUT_SECS: u64 = 600; const DATABRICKS_PROVIDER_NAME: &str = "databricks"; pub const DATABRICKS_DEFAULT_MODEL: &str = "databricks-claude-sonnet-4"; -const DATABRICKS_DEFAULT_FAST_MODEL: &str = "gemini-2-5-flash"; +const DATABRICKS_DEFAULT_FAST_MODEL: &str = "databricks-claude-haiku-4-5"; pub const DATABRICKS_KNOWN_MODELS: &[&str] = &[ "databricks-claude-sonnet-4-5", "databricks-claude-3-7-sonnet", diff --git a/scripts/test_compaction.sh b/scripts/test_compaction.sh index baa3b7ca96..876ee51421 100755 --- a/scripts/test_compaction.sh +++ b/scripts/test_compaction.sh @@ -2,6 +2,11 @@ # Compaction smoke test script # Tests both manual (trigger prompt) and auto compaction (threshold-based) +# +# Environment variable overrides: +# COMPACTION_PROVIDER - Override the provider for tests 1 & 2 (default: use system default) +# COMPACTION_MODEL - Override the model for tests 1 & 2 (default: use system default) +# SKIP_BUILD - Skip cargo build if set if [ -f .env ]; then export $(grep -v '^#' .env | xargs) @@ -19,6 +24,19 @@ fi SCRIPT_DIR=$(pwd) GOOSE_BIN="$SCRIPT_DIR/target/release/goose" +# Apply provider/model overrides if set +if [ -n "$COMPACTION_PROVIDER" ]; then + echo "Using override provider: $COMPACTION_PROVIDER" + export GOOSE_PROVIDER="$COMPACTION_PROVIDER" +fi +if [ -n "$COMPACTION_MODEL" ]; then + echo "Using override model: $COMPACTION_MODEL" + export GOOSE_MODEL="$COMPACTION_MODEL" +fi +if [ -n "$COMPACTION_PROVIDER" ] || [ -n "$COMPACTION_MODEL" ]; then + echo "" +fi + # Validation function to check compaction structure in session JSON validate_compaction() { local session_id=$1