diff --git a/app.py b/app.py index 671fd1d..94af1f3 100644 --- a/app.py +++ b/app.py @@ -288,10 +288,14 @@ def _configure_all_cli_auth(token): settings = { "env": { - "ANTHROPIC_MODEL": os.environ.get("ANTHROPIC_MODEL", "databricks-claude-sonnet-4-6"), + "ANTHROPIC_MODEL": os.environ.get("ANTHROPIC_MODEL", "databricks-claude-opus-4-6"), "ANTHROPIC_BASE_URL": anthropic_base_url, "ANTHROPIC_AUTH_TOKEN": token, + "ANTHROPIC_DEFAULT_OPUS_MODEL": "databricks-claude-opus-4-6", + "ANTHROPIC_DEFAULT_SONNET_MODEL": "databricks-claude-sonnet-4-6", + "ANTHROPIC_DEFAULT_HAIKU_MODEL": "databricks-claude-haiku-4-5", "ANTHROPIC_CUSTOM_HEADERS": "x-databricks-use-coding-agent-mode: true", + "CLAUDE_CODE_DISABLE_EXPERIMENTAL_BETAS": "1", } } diff --git a/setup_claude.py b/setup_claude.py index d13dfcb..cf7e3c8 100644 --- a/setup_claude.py +++ b/setup_claude.py @@ -32,10 +32,14 @@ settings = { "env": { - "ANTHROPIC_MODEL": os.environ.get("ANTHROPIC_MODEL", "databricks-claude-sonnet-4-6"), + "ANTHROPIC_MODEL": os.environ.get("ANTHROPIC_MODEL", "databricks-claude-opus-4-6"), "ANTHROPIC_BASE_URL": anthropic_base_url, "ANTHROPIC_AUTH_TOKEN": token, - "ANTHROPIC_CUSTOM_HEADERS": "x-databricks-use-coding-agent-mode: true" + "ANTHROPIC_DEFAULT_OPUS_MODEL": "databricks-claude-opus-4-6", + "ANTHROPIC_DEFAULT_SONNET_MODEL": "databricks-claude-sonnet-4-6", + "ANTHROPIC_DEFAULT_HAIKU_MODEL": "databricks-claude-haiku-4-5", + "ANTHROPIC_CUSTOM_HEADERS": "x-databricks-use-coding-agent-mode: true", + "CLAUDE_CODE_DISABLE_EXPERIMENTAL_BETAS": "1", } }