ed/manual_slop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(mma): Fix tier 2 model name to valid preview model and adjust tests

Browse Source
This commit is contained in:
Edward R. Gonzalez
2026-02-25 19:57:42 -05:00
parent 9da08e9c42
commit 794cc2a7f2
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/mma_exec.py
View File

@@ -7,7 +7,7 @@ def get_model_for_role(role: str) -> str:
if role == 'tier1-orchestrator' or role == 'tier1': if role == 'tier1-orchestrator' or role == 'tier1':
return 'gemini-3.1-pro-preview' return 'gemini-3.1-pro-preview'
elif role == 'tier2-tech-lead' or role == 'tier2': elif role == 'tier2-tech-lead' or role == 'tier2':
return 'gemini-3.0-flash-preview' return 'gemini-3-flash-preview'
else: else:
return 'gemini-2.5-flash-lite' return 'gemini-2.5-flash-lite'

4
tests/test_mma_exec.py
View File

@@ -42,7 +42,7 @@ def test_get_role_documents():
def test_get_model_for_role(): def test_get_model_for_role():
"""Test that get_model_for_role returns the correct model for each 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('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('tier3-worker') == 'gemini-2.5-flash-lite'
assert get_model_for_role('tier4-qa') == '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." prompt = "Write a unit test."
docs = ["file1.py", "docs/spec.md"] 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" expected_model = "gemini-2.5-flash-lite"
mock_stdout = "Mocked AI Response" mock_stdout = "Mocked AI Response"