import argparse def get_role_documents(role: str) -> list[str]: if role == 'tier1': return ['conductor/product.md', 'conductor/product-guidelines.md'] elif role == 'tier2': return ['conductor/tech-stack.md', 'conductor/workflow.md'] elif role == 'tier3': return ['conductor/workflow.md'] return [] def create_parser(): parser = argparse.ArgumentParser(description="MMA Execution Script") parser.add_argument( "--role", choices=['tier1', 'tier2', 'tier3', 'tier4'], required=True, help="The tier role to execute" ) parser.add_argument( "prompt", type=str, help="The prompt for the tier" ) return parser def main(): parser = create_parser() args = parser.parse_args() print(f"Role: {args.role}") print(f"Prompt: {args.prompt}") docs = get_role_documents(args.role) print(f"Selected Documents: {docs}") if __name__ == "__main__": main()