fix: remove unused model name from args in factory functions for subagents

Author: shangkunwang01

MaxKernel/auto_agent/subagents/autotuning/agent.py

@@ -161,7 +161,7 @@ async def _run_async_impl(
       )
 
 
-def create_autotune_runner(model_name: str = MODEL_NAME) -> AutotuneRunner:
+def create_autotune_runner() -> AutotuneRunner:
   return AutotuneRunner(
     name="AutotuneRunner",
     output_key="autotune_results",

MaxKernel/auto_agent/subagents/kernel_writing/agent.py

@@ -369,9 +369,7 @@ def create_cleanup_debug_statements_agent(
 cleanup_debug_statements_agent = create_cleanup_debug_statements_agent()
 
 
-def create_kernel_compilation_checker_for_validation(
-  model_name: str = MODEL_NAME,
-) -> KernelCompilationChecker:
+def create_kernel_compilation_checker_for_validation() -> KernelCompilationChecker:
   return KernelCompilationChecker(
     name="KernelCompilationCheckerForValidation",
     input_key="kernel_code",
@@ -390,9 +388,7 @@ def create_kernel_compilation_validation_loop(
 ) -> KernelCompilationValidationLoop:
   return KernelCompilationValidationLoop(
     name="KernelCompilationValidationLoop",
-    compilation_checker=create_kernel_compilation_checker_for_validation(
-      model_name
-    ),
+    compilation_checker=create_kernel_compilation_checker_for_validation(),
     fix_agent=create_fix_kernel_compilation_agent(model_name),
     debug_agent=create_add_debug_statements_agent(model_name),
     max_retries=6,

MaxKernel/auto_agent/subagents/profiling/agent.py

@@ -66,7 +66,7 @@ def create_read_profiling_script_agent(
 
 
 # Profiling execution agent
-def create_eval_profile_agent(model_name: str = MODEL_NAME) -> KernelProfiler:
+def create_eval_profile_agent() -> KernelProfiler:
   return KernelProfiler(
     name="ProfileEvalAgent",
     input_key="profiling_script",
@@ -159,7 +159,7 @@ def create_profile_agent(model_name: str = MODEL_NAME) -> SequentialAgent:
     sub_agents=[
       create_generate_profiling_script_agent(model_name),
       create_read_profiling_script_agent(model_name),
-      create_eval_profile_agent(model_name),
+      create_eval_profile_agent(),
       create_summarize_profile_agent(model_name),
     ],
     description="Profiles the Pallas kernel to identify performance bottlenecks.",

MaxKernel/auto_agent/subagents/testing/agent.py

@@ -899,9 +899,7 @@ def create_generate_test_file_agent(
 
 
 # Validation agents
-def create_syntax_validation_agent(
-  model_name: str = MODEL_NAME,
-) -> SyntaxValidationAgent:
+def create_syntax_validation_agent() -> SyntaxValidationAgent:
   return SyntaxValidationAgent(
     name="SyntaxValidationAgent",
     input_key="test_file_path",
@@ -912,9 +910,7 @@ def create_syntax_validation_agent(
 syntax_validation_agent = create_syntax_validation_agent()
 
 
-def create_import_validation_agent(
-  model_name: str = MODEL_NAME,
-) -> ImportValidationAgent:
+def create_import_validation_agent() -> ImportValidationAgent:
   return ImportValidationAgent(
     name="ImportValidationAgent",
     input_key="test_file_path",
@@ -925,9 +921,7 @@ def create_import_validation_agent(
 import_validation_agent = create_import_validation_agent()
 
 
-def create_structure_validation_agent(
-  model_name: str = MODEL_NAME,
-) -> TestStructureValidationAgent:
+def create_structure_validation_agent() -> TestStructureValidationAgent:
   return TestStructureValidationAgent(
     name="TestStructureValidationAgent",
     input_key="test_file_path",
@@ -938,9 +932,7 @@ def create_structure_validation_agent(
 structure_validation_agent = create_structure_validation_agent()
 
 
-def create_mock_execution_validation_agent(
-  model_name: str = MODEL_NAME,
-) -> MockTestExecutionAgent:
+def create_mock_execution_validation_agent() -> MockTestExecutionAgent:
   return MockTestExecutionAgent(
     name="MockTestExecutionAgent",
     input_key="test_file_path",
@@ -975,10 +967,10 @@ def create_validation_loop_agent(
 ) -> TestValidationLoopAgent:
   return TestValidationLoopAgent(
     name="TestValidationLoopAgent",
-    syntax_agent=create_syntax_validation_agent(model_name),
-    import_agent=create_import_validation_agent(model_name),
-    structure_agent=create_structure_validation_agent(model_name),
-    mock_execution_agent=create_mock_execution_validation_agent(model_name),
+    syntax_agent=create_syntax_validation_agent(),
+    import_agent=create_import_validation_agent(),
+    structure_agent=create_structure_validation_agent(),
+    mock_execution_agent=create_mock_execution_validation_agent(),
     fix_agent=create_fix_test_script_agent(model_name),
     max_retries=6,
   )
@@ -1006,7 +998,7 @@ def create_validated_test_generation_agent(
 
 
 # Test execution agents
-def create_run_tests_agent(model_name: str = MODEL_NAME) -> TestRunner:
+def create_run_tests_agent() -> TestRunner:
   return TestRunner(
     name="RunTestsAgent",
     input_key="test_file_path",
@@ -1044,7 +1036,7 @@ def create_unified_test_agent(model_name: str = MODEL_NAME) -> SequentialAgent:
   return SequentialAgent(
     name="UnifiedTestAgent",
     sub_agents=[
-      create_run_tests_agent(model_name),
+      create_run_tests_agent(),
       create_summarize_test_results_agent(model_name),
     ],
     description="Executes the generated pytest test file and provides a comprehensive summary.",