Skip to content

Commit 9243669

Browse files
committed
Removing auto_agent related dependencies in retries_tool.py
1 parent e0a7a55 commit 9243669

1 file changed

Lines changed: 0 additions & 8 deletions

File tree

MaxKernel/hitl_agent/tools/retries_tool.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -21,9 +21,7 @@ async def set_max_compilation_retries_fn(retries: int, tool_context: ToolContext
2121

2222
# Update in-memory config variables
2323
import hitl_agent.config as hitl_cfg
24-
import auto_agent.config as auto_cfg
2524
hitl_cfg.MAX_COMPILATION_RETRIES = retries
26-
auto_cfg.MAX_COMPILATION_RETRIES = retries
2725
os.environ["MAX_COMPILATION_RETRIES"] = str(retries)
2826

2927
# Update .env file
@@ -52,12 +50,6 @@ async def set_max_compilation_retries_fn(retries: int, tool_context: ToolContext
5250
except Exception as e:
5351
logging.warning(f"Could not update hitl_loop max_retries: {e}")
5452

55-
try:
56-
from auto_agent.subagents.kernel_writing.agent import kernel_compilation_validation_loop as auto_loop
57-
auto_loop.max_retries = retries
58-
except Exception as e:
59-
logging.warning(f"Could not update auto_loop max_retries: {e}")
60-
6153
return f"Successfully updated maximum compilation retries to: {retries}"
6254

6355

0 commit comments

Comments
 (0)