Skip to content

Commit 6204fa6

Browse files
refactor: changed all the thinking level back to high
1 parent d899438 commit 6204fa6

3 files changed

Lines changed: 4 additions & 4 deletions

File tree

MaxKernel/auto_agent/subagents/kernel_writing/agent.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -312,7 +312,7 @@ async def _run_async_impl(
312312
name="AddDebugStatementsAgent",
313313
model=MODEL_NAME,
314314
generate_content_config=model_config,
315-
planner=get_thinking_planner("medium"),
315+
planner=get_thinking_planner("high"),
316316
instruction=add_debug_statements.PROMPT,
317317
description="Adds strategic debugging statements to diagnose persistent compilation issues.",
318318
tools=[filesystem_tool_r, write_optimized_kernel_tool],
@@ -324,7 +324,7 @@ async def _run_async_impl(
324324
name="CleanupDebugStatementsAgent",
325325
model=MODEL_NAME,
326326
generate_content_config=model_config,
327-
planner=get_thinking_planner("low"),
327+
planner=get_thinking_planner("high"),
328328
instruction=cleanup_debug_statements.PROMPT,
329329
description="Removes debugging statements from successfully compiled kernel.",
330330
tools=[filesystem_tool_r, write_optimized_kernel_tool],

MaxKernel/auto_agent/subagents/profiling/agent.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@
3232
name="GenerateProfilingScriptAgent",
3333
model=MODEL_NAME,
3434
generate_content_config=model_config,
35-
planner=get_thinking_planner("medium"),
35+
planner=get_thinking_planner("high"),
3636
instruction=gen_profiling_script.PROMPT,
3737
description="Generates a profiling script to identify performance bottlenecks in the kernel code and writes it to a file.",
3838
tools=[filesystem_tool_r, write_profiling_script_tool],

MaxKernel/auto_agent/subagents/testing/agent.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -953,7 +953,7 @@ async def _run_async_impl(
953953
name="SummarizeTestResultsAgent",
954954
model=MODEL_NAME,
955955
generate_content_config=model_config,
956-
planner=get_thinking_planner("medium"),
956+
planner=get_thinking_planner("high"),
957957
instruction=summarize_test_results_prompt.PROMPT,
958958
description="Analyzes pytest test results and provides recommendations.",
959959
tools=(

0 commit comments

Comments
 (0)