diff --git a/crates/goose-cli/src/test_helpers.rs b/crates/goose-cli/src/test_helpers.rs index d115420040..fa1804ccf7 100644 --- a/crates/goose-cli/src/test_helpers.rs +++ b/crates/goose-cli/src/test_helpers.rs @@ -60,7 +60,7 @@ fn setup_profile(temp_dir_path: &Path, profile_string: Option<&str>) { "profile_items": { "default": { "provider": "databricks", - "model": "claude-3-5-sonnet-2", + "model": "goose", "additional_extensions": [] } } diff --git a/crates/goose-server/src/routes/providers_and_keys.json b/crates/goose-server/src/routes/providers_and_keys.json index ddcaf52532..e56f9f8b42 100644 --- a/crates/goose-server/src/routes/providers_and_keys.json +++ b/crates/goose-server/src/routes/providers_and_keys.json @@ -14,7 +14,7 @@ "databricks": { "name": "Databricks", "description": "Connect to LLMs via Databricks", - "models": ["claude-3-5-sonnet-2"], + "models": ["goose"], "required_keys": ["DATABRICKS_HOST"] }, "google": { diff --git a/ui/desktop/src/ChatWindow.tsx b/ui/desktop/src/ChatWindow.tsx index 4820b88463..06cc1ea4fd 100644 --- a/ui/desktop/src/ChatWindow.tsx +++ b/ui/desktop/src/ChatWindow.tsx @@ -363,6 +363,14 @@ export default function ChatWindow() { useEffect(() => { const setupStoredProvider = async () => { const config = window.electron.getConfig(); + + if (config.GOOSE_PROVIDER && config.GOOSE_MODEL) { + window.electron.logInfo( + 'Initializing system with environment: GOOSE_MODEL and GOOSE_PROVIDER as priority.' + ); + await initializeSystem(config.GOOSE_PROVIDER, config.GOOSE_MODEL); + return; + } const storedProvider = getStoredProvider(config); const storedModel = getStoredModel(); if (storedProvider) { diff --git a/ui/desktop/src/components/BottomMenu.tsx b/ui/desktop/src/components/BottomMenu.tsx index 0d2efc213b..99eae44e84 100644 --- a/ui/desktop/src/components/BottomMenu.tsx +++ b/ui/desktop/src/components/BottomMenu.tsx @@ -47,6 +47,11 @@ export default function BottomMenu({ hasMessages }) { }; }, [isModelMenuOpen]); + let envModelProvider = null; + if (window.electron.getConfig().GOOSE_MODEL && window.electron.getConfig().GOOSE_PROVIDER) { + envModelProvider = `${window.electron.getConfig().GOOSE_MODEL} - ${window.electron.getConfig().GOOSE_PROVIDER}`; + } + return (