Skip to content

Commit d4cf33b

Browse files
aarondr77aarondr77
authored
Merge pull request #2171 from mito-ds/remove-sonnet
mito-ai: remove sonnet
2 parents f0971fa + 6ace2d5 commit d4cf33b

File tree

14 files changed

+24
-46
lines changed

14 files changed

+24
-46
lines changed

‎mito-ai/mito_ai/tests/providers/test_providers.py‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def reset_env_vars(monkeypatch: pytest.MonkeyPatch) -> None:
5959
"name": "claude",
6060
"env_vars": {"ANTHROPIC_API_KEY": "claude-key"},
6161
"constants": {"ANTHROPIC_API_KEY": "claude-key", "OPENAI_API_KEY": None},
62-
"model": "claude-sonnet-4-5-20250929",
62+
"model": "claude-haiku-4-5-20251001",
6363
"mock_patch": "mito_ai.provider_manager.AnthropicClient",
6464
"mock_method": "request_completions",
6565
"provider_name": "Claude",
@@ -143,7 +143,7 @@ async def test_completion_request(
143143
"name": "claude",
144144
"env_vars": {"ANTHROPIC_API_KEY": "claude-key"},
145145
"constants": {"ANTHROPIC_API_KEY": "claude-key", "OPENAI_API_KEY": None},
146-
"model": "claude-sonnet-4-5-20250929",
146+
"model": "claude-haiku-4-5-20251001",
147147
"mock_patch": "mito_ai.provider_manager.AnthropicClient",
148148
"mock_method": "stream_completions",
149149
"provider_name": "Claude",
@@ -235,7 +235,7 @@ def test_claude_error_handling(monkeypatch: pytest.MonkeyPatch, provider_config:
235235

236236
mock_client = MagicMock()
237237
mock_client.capabilities = AICapabilities(
238-
configuration={"model": "claude-sonnet-4-5-20250929"},
238+
configuration={"model": "claude-haiku-4-5-20251001"},
239239
provider="Claude",
240240
type="ai_capabilities"
241241
)
@@ -258,7 +258,7 @@ def test_claude_error_handling(monkeypatch: pytest.MonkeyPatch, provider_config:
258258
},
259259
{
260260
"name": "claude_fallback",
261-
"model": "claude-sonnet-4-5-20250929",
261+
"model": "claude-haiku-4-5-20251001",
262262
"mock_function": "mito_ai.anthropic_client.get_anthropic_completion_from_mito_server",
263263
"provider_name": "Claude",
264264
"key_type": "claude"
@@ -315,7 +315,7 @@ async def test_mito_server_fallback_completion_request(
315315
},
316316
{
317317
"name": "claude_fallback",
318-
"model": "claude-sonnet-4-5-20250929",
318+
"model": "claude-haiku-4-5-20251001",
319319
"mock_function": "mito_ai.anthropic_client.stream_anthropic_completion_from_mito_server",
320320
"provider_name": "Claude",
321321
"key_type": "claude"

‎mito-ai/mito_ai/utils/anthropic_utils.py‎

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,7 @@
1717
ANTHROPIC_TIMEOUT = 60
1818
max_retries = 1
1919

20-
FAST_ANTHROPIC_MODEL = "claude-haiku-4-5-20251001" # This should be in sync with ModelSelector.tsx
21-
LARGE_CONTEXT_MODEL = "claude-sonnet-4-5-20250929" # This should be in sync with ModelSelector.tsx
20+
LARGE_CONTEXT_MODEL = "claude-sonnet-4-5-20250929"
2221
EXTENDED_CONTEXT_BETA = "context-1m-2025-08-07" # Beta feature for extended context window support
2322

2423
def does_message_exceed_max_tokens(system: Union[str, List[TextBlockParam], anthropic.Omit], messages: List[MessageParam]) -> bool:

‎mito-ai/mito_ai/utils/model_utils.py‎

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
# Model ordering: [fastest, ..., slowest] for each provider
1010
ANTHROPIC_MODEL_ORDER = [
1111
"claude-haiku-4-5-20251001", # Fastest
12-
"claude-sonnet-4-5-20250929", # Slower
1312
]
1413

1514
OPENAI_MODEL_ORDER = [
@@ -27,7 +26,6 @@
2726
STANDARD_MODELS = [
2827
"gpt-4.1",
2928
"gpt-5.2",
30-
"claude-sonnet-4-5-20250929",
3129
"claude-haiku-4-5-20251001",
3230
"gemini-3-flash-preview",
3331
"gemini-3-pro-preview",

‎mito-ai/src/Extensions/AiChat/ChatTaskpane.tsx‎

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1248,6 +1248,4 @@ const ChatTaskpane: React.FC<IChatTaskpaneProps> = ({
12481248
);
12491249
};
12501250

1251-
export default ChatTaskpane;
1252-
1253-
1251+
export default ChatTaskpane;

‎mito-ai/src/components/ModelSelector.tsx‎

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,6 @@ import NucleausIcon from '../icons/NucleausIcon';
1010
import BrainIcon from '../icons/BrainIcon';
1111
import LightningIcon from '../icons/LightningIcon';
1212
import {
13-
CLAUDE_SONNET_DISPLAY_NAME,
14-
CLAUDE_SONNET_MODEL_NAME,
1513
CLAUDE_HAIKU_DISPLAY_NAME,
1614
CLAUDE_HAIKU_MODEL_NAME,
1715
GPT_4_1_DISPLAY_NAME,
@@ -90,16 +88,6 @@ const MODEL_MAPPINGS: ModelMapping[] = [
9088
speed: 'Fast',
9189
complexityHandling: 'Medium'
9290
},
93-
{
94-
displayName: CLAUDE_SONNET_DISPLAY_NAME,
95-
fullName: CLAUDE_SONNET_MODEL_NAME,
96-
type: 'smart',
97-
goodFor: [...GOOD_FOR_SMART],
98-
provider: 'Anthropic',
99-
tokenLimit: '1M',
100-
speed: 'Medium',
101-
complexityHandling: 'High'
102-
},
10391
{
10492
displayName: GEMINI_3_FLASH_DISPLAY_NAME,
10593
fullName: GEMINI_3_FLASH_MODEL_NAME,

‎mito-ai/src/tests/AiChat/ModelSelector.test.tsx‎

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@ import {
1212
GPT_4_1_DISPLAY_NAME,
1313
GPT_4_1_MODEL_NAME,
1414
GPT_5_2_MODEL_NAME,
15-
CLAUDE_SONNET_MODEL_NAME,
1615
CLAUDE_HAIKU_MODEL_NAME,
1716
GEMINI_3_FLASH_MODEL_NAME,
1817
GEMINI_3_PRO_MODEL_NAME,
@@ -43,7 +42,6 @@ describe('ModelSelector', () => {
4342
models: [
4443
GPT_4_1_MODEL_NAME,
4544
GPT_5_2_MODEL_NAME,
46-
CLAUDE_SONNET_MODEL_NAME,
4745
CLAUDE_HAIKU_MODEL_NAME,
4846
GEMINI_3_FLASH_MODEL_NAME,
4947
GEMINI_3_PRO_MODEL_NAME,
@@ -82,9 +80,9 @@ describe('ModelSelector', () => {
8280
});
8381

8482
it('loads saved model from localStorage on mount', async () => {
85-
// Set up localStorage with a saved model
83+
// Set up localStorage with a saved model (must be in available models)
8684
const savedConfig = {
87-
model: 'claude-sonnet-4-5-20250929'
85+
model: CLAUDE_HAIKU_MODEL_NAME
8886
};
8987
localStorage.setItem('llmModelConfig', JSON.stringify(savedConfig));
9088

@@ -103,7 +101,6 @@ describe('ModelSelector', () => {
103101
models: [
104102
GPT_4_1_MODEL_NAME,
105103
GPT_5_2_MODEL_NAME,
106-
CLAUDE_SONNET_MODEL_NAME,
107104
CLAUDE_HAIKU_MODEL_NAME,
108105
GEMINI_3_FLASH_MODEL_NAME,
109106
GEMINI_3_PRO_MODEL_NAME,

‎mito-ai/src/utils/models.ts‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,6 @@ export async function getAvailableModels(): Promise<string[]> {
4040
return [
4141
GPT_4_1_MODEL_NAME,
4242
GPT_5_2_MODEL_NAME,
43-
CLAUDE_SONNET_MODEL_NAME,
4443
CLAUDE_HAIKU_MODEL_NAME,
4544
GEMINI_3_FLASH_MODEL_NAME,
4645
GEMINI_3_PRO_MODEL_NAME,

‎tests/llm_providers_tests/llm_providers.spec.ts‎

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,6 @@ import {
2929
GPT_4_1_MODEL_NAME,
3030
GPT_5_2_DISPLAY_NAME,
3131
GPT_5_2_MODEL_NAME,
32-
CLAUDE_SONNET_DISPLAY_NAME,
33-
CLAUDE_SONNET_MODEL_NAME,
3432
CLAUDE_HAIKU_DISPLAY_NAME,
3533
CLAUDE_HAIKU_MODEL_NAME,
3634
GEMINI_3_FLASH_DISPLAY_NAME,
@@ -43,7 +41,6 @@ import {
4341
const modelConfigs = [
4442
{ name: GPT_4_1_DISPLAY_NAME, provider: GPT_4_1_MODEL_NAME },
4543
{ name: GPT_5_2_DISPLAY_NAME, provider: GPT_5_2_MODEL_NAME },
46-
{ name: CLAUDE_SONNET_DISPLAY_NAME, provider: CLAUDE_SONNET_MODEL_NAME },
4744
{ name: CLAUDE_HAIKU_DISPLAY_NAME, provider: CLAUDE_HAIKU_MODEL_NAME },
4845
{ name: GEMINI_3_FLASH_DISPLAY_NAME, provider: GEMINI_3_FLASH_MODEL_NAME },
4946
{ name: GEMINI_3_PRO_DISPLAY_NAME, provider: GEMINI_3_PRO_MODEL_NAME },

‎tests/mitoai_ui_tests/agent.spec.ts‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,9 @@ import {
1919
clickAcceptButton,
2020
clickDenyButton
2121
} from './utils';
22-
import { CLAUDE_SONNET_DISPLAY_NAME } from '../../mito-ai/src/utils/models';
22+
import { CLAUDE_HAIKU_DISPLAY_NAME } from '../../mito-ai/src/utils/models';
2323

24-
const MODEL = CLAUDE_SONNET_DISPLAY_NAME;
24+
const MODEL = CLAUDE_HAIKU_DISPLAY_NAME;
2525

2626
test.describe.parallel("Agent mode basic functionality", () => {
2727

‎tests/mitoai_ui_tests/backgroundAgent.spec.ts‎

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,9 +18,9 @@ import {
1818
waitForAgentToFinish,
1919
startNewMitoAIChat
2020
} from './utils';
21-
import { CLAUDE_SONNET_DISPLAY_NAME } from '../../mito-ai/src/utils/models';
21+
import { CLAUDE_HAIKU_DISPLAY_NAME } from '../../mito-ai/src/utils/models';
2222

23-
const MODEL = CLAUDE_SONNET_DISPLAY_NAME;
23+
const MODEL = CLAUDE_HAIKU_DISPLAY_NAME;
2424

2525
test.describe.parallel("Background Agent functionality", () => {
2626

@@ -100,7 +100,9 @@ test.describe.parallel("Background Agent functionality", () => {
100100

101101
// Verify the agent worked in the original notebook
102102
const firstCodeCell = await getCodeFromCell(page, 0);
103-
expect(firstCodeCell).toContain('meta_stock_prices.csv');
103+
const secondCodeCell = await getCodeFromCell(page, 1);
104+
const combinedCode = `${firstCodeCell}\n${secondCodeCell}`;
105+
expect(combinedCode).toContain('meta_stock_prices.csv');
104106

105107
// Check that the agent never used the run_all_cells tool
106108
// by looking at the content in the chat taskpane

0 commit comments

Comments
 (0)