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 (
{/* Directory Chooser - Always visible */} @@ -72,7 +77,7 @@ export default function BottomMenu({ hasMessages }) { className="flex items-center cursor-pointer" onClick={() => setIsModelMenuOpen(!isModelMenuOpen)} > - {currentModel?.name || 'Select Model'} + {envModelProvider || currentModel?.name || 'Select Model'} {isModelMenuOpen ? ( ) : ( diff --git a/ui/desktop/src/components/settings/models/hardcoded_stuff.tsx b/ui/desktop/src/components/settings/models/hardcoded_stuff.tsx index 282499e782..6bda96bce1 100644 --- a/ui/desktop/src/components/settings/models/hardcoded_stuff.tsx +++ b/ui/desktop/src/components/settings/models/hardcoded_stuff.tsx @@ -48,7 +48,7 @@ export const azure_openai_models = ['gpt-4o']; export const default_models = { openai: 'gpt-4o', anthropic: 'claude-3-5-sonnet-latest', - databricks: 'claude-3-5-sonnet-2', + databricks: 'goose', google: 'gemini-2.0-flash-exp', groq: 'llama-3.3-70b-versatile', openrouter: 'anthropic/claude-3.5-sonnet', diff --git a/ui/desktop/src/main.ts b/ui/desktop/src/main.ts index 3ceaf762bf..fc8b128c7b 100644 --- a/ui/desktop/src/main.ts +++ b/ui/desktop/src/main.ts @@ -73,7 +73,7 @@ const getGooseProvider = () => { //{env-macro-start}// //needed when goose is bundled for a specific provider //{env-macro-end}// - return process.env.GOOSE_PROVIDER; + return [process.env.GOOSE_PROVIDER, process.env.GOOSE_MODEL]; }; const generateSecretKey = () => { @@ -83,8 +83,11 @@ const generateSecretKey = () => { return key; }; +let [provider, model] = getGooseProvider(); + let appConfig = { - GOOSE_PROVIDER: getGooseProvider(), + GOOSE_PROVIDER: provider, + GOOSE_MODEL: model, GOOSE_API_HOST: 'http://127.0.0.1', GOOSE_PORT: 0, GOOSE_WORKING_DIR: '',