diff --git a/scripts/mma_exec.py b/scripts/mma_exec.py index d4256a7..d59c8e6 100644 --- a/scripts/mma_exec.py +++ b/scripts/mma_exec.py @@ -68,13 +68,17 @@ 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-flash-preview' + return 'gemini-3.1-pro-preview' + # return 'gemini-3-flash-preview' elif role == 'tier3-worker' or role == 'tier3': - return 'gemini-3-flash-preview' + # return 'gemini-3-flash-preview' + return 'gemini-3.1-pro-preview' elif role == 'tier4-qa' or role == 'tier4': - return 'gemini-2.5-flash-lite' + # return 'gemini-2.5-flash-lite' + return 'gemini-3.1-pro-preview' else: - return 'gemini-2.5-flash-lite' + # return 'gemini-2.5-flash-lite' + return 'gemini-3.1-pro-preview' def get_role_documents(role: str) -> list[str]: if role == 'tier1-orchestrator' or role == 'tier1':