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

Use haiku for databricks (#6943)

Browse Source 
Co-authored-by: Douwe Osinga <douwe@squareup.com>
This commit is contained in:
Douwe Osinga
2026-02-04 18:17:59 +01:00
committed by GitHub
parent 5492052f0c
commit 6ae72a6cf0
2 changed files with 19 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose/src/providers/databricks.rs
+1 -1
View File
@@ -36,7 +36,7 @@ const DEFAULT_TIMEOUT_SECS: u64 = 600;
const DATABRICKS_PROVIDER_NAME: &str = "databricks"; const DATABRICKS_PROVIDER_NAME: &str = "databricks";
pub const DATABRICKS_DEFAULT_MODEL: &str = "databricks-claude-sonnet-4"; 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] = &[ pub const DATABRICKS_KNOWN_MODELS: &[&str] = &[
"databricks-claude-sonnet-4-5", "databricks-claude-sonnet-4-5",
"databricks-claude-3-7-sonnet", "databricks-claude-3-7-sonnet",
scripts/test_compaction.sh
+18
View File
@@ -2,6 +2,11 @@
# Compaction smoke test script # Compaction smoke test script
# Tests both manual (trigger prompt) and auto compaction (threshold-based) # 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 if [ -f .env ]; then
export $(grep -v '^#' .env | xargs) export $(grep -v '^#' .env | xargs)
@@ -19,6 +24,19 @@ fi
SCRIPT_DIR=$(pwd) SCRIPT_DIR=$(pwd)
GOOSE_BIN="$SCRIPT_DIR/target/release/goose" 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 # Validation function to check compaction structure in session JSON
validate_compaction() { validate_compaction() {
local session_id=$1 local session_id=$1