diff --git a/scripts/mma_exec.py b/scripts/mma_exec.py index db672a5..b4444bb 100644 --- a/scripts/mma_exec.py +++ b/scripts/mma_exec.py @@ -7,7 +7,7 @@ def get_model_for_role(role: str) -> str: if role == 'tier1-orchestrator' or role == 'tier1': return 'gemini-3.1-pro-preview' elif role == 'tier2-tech-lead' or role == 'tier2': - return 'gemini-3.0-flash-preview' + return 'gemini-3-flash-preview' else: return 'gemini-2.5-flash-lite' diff --git a/tests/test_mma_exec.py b/tests/test_mma_exec.py index f20f024..23e7115 100644 --- a/tests/test_mma_exec.py +++ b/tests/test_mma_exec.py @@ -42,7 +42,7 @@ def test_get_role_documents(): def test_get_model_for_role(): """Test that get_model_for_role returns the correct model for each role.""" assert get_model_for_role('tier1-orchestrator') == 'gemini-3.1-pro-preview' - assert get_model_for_role('tier2-tech-lead') == 'gemini-3.0-flash-preview' + assert get_model_for_role('tier2-tech-lead') == 'gemini-3-flash-preview' assert get_model_for_role('tier3-worker') == 'gemini-2.5-flash-lite' assert get_model_for_role('tier4-qa') == 'gemini-2.5-flash-lite' @@ -55,7 +55,7 @@ def test_execute_agent(): prompt = "Write a unit test." docs = ["file1.py", "docs/spec.md"] - expected_gemini_arg = "Activate the mma-tier3-worker skill. Write a unit test. @file1.py @docs/spec.md" + expected_gemini_arg = "Use the mma-tier3-worker skill. Write a unit test. @file1.py @docs/spec.md" expected_model = "gemini-2.5-flash-lite" mock_stdout = "Mocked AI Response"