Merge pull request #65 from AI-Hypercomputer/shangkun-style

style: run format.sh to fix formatting and linter errors

Author: shangkunwang01

MaxKernel/auto_agent/config.py

@@ -15,6 +15,7 @@
 INCLUDE_THOUGHTS = os.environ.get("INCLUDE_THOUGHTS", "true").lower() == "true"
 MAX_COMPILATION_RETRIES = int(os.environ.get("MAX_COMPILATION_RETRIES", "6"))
 
+
 # Set events compaction policy to avoid memory overflow
 def get_compaction_config():
   return EventsCompactionConfig(
@@ -24,6 +25,7 @@ def get_compaction_config():
     overlap_size=0,
   )
 
+
 # Model configuration
 model_config = types.GenerateContentConfig(
   temperature=TEMPERATURE,

MaxKernel/auto_agent/subagents/kernel_writing/agent.py

@@ -13,7 +13,11 @@
   load_kernel_and_plan_to_state,
   load_single_kernel_to_state,
 )
-from auto_agent.config import get_thinking_planner, model_config, MAX_COMPILATION_RETRIES
+from auto_agent.config import (
+  MAX_COMPILATION_RETRIES,
+  get_thinking_planner,
+  model_config,
+)
 from auto_agent.constants import MODEL_NAME
 from auto_agent.custom_types import CustomLlmAgent
 from auto_agent.subagents.kernel_writing.kernel_compilation import (

MaxKernel/hitl_agent/agent.py

@@ -33,8 +33,8 @@
 )
 from hitl_agent.tools.tools import (
   filesystem_tool_r,
-  set_working_directory,
   set_max_compilation_retries,
+  set_working_directory,
 )
 
 # Root orchestration agent

MaxKernel/hitl_agent/subagents/kernel_writing/agent.py

@@ -15,7 +15,11 @@
   load_single_kernel_to_state,
   save_kernel_and_plan_paths,
 )
-from hitl_agent.config import model_config, thinking_planner, MAX_COMPILATION_RETRIES
+from hitl_agent.config import (
+  MAX_COMPILATION_RETRIES,
+  model_config,
+  thinking_planner,
+)
 from hitl_agent.constants import MODEL_NAME
 from hitl_agent.custom_types import CustomLlmAgent
 from hitl_agent.subagents.kernel_writing.kernel_compilation import (

MaxKernel/hitl_agent/tools/filesystem_tools.py

@@ -1,6 +1,7 @@
 """Filesystem tools configuration for the HITL agent."""
 
 import os
+
 from google.adk.tools.mcp_tool.mcp_session_manager import StdioConnectionParams
 from google.adk.tools.mcp_tool.mcp_toolset import MCPToolset
 from mcp import StdioServerParameters

MaxKernel/hitl_agent/tools/retries_tool.py

@@ -2,11 +2,12 @@
 
 import logging
 import os
+
 from google.adk.tools import FunctionTool, ToolContext
 
 
 async def set_max_compilation_retries_fn(
-    retries: int, tool_context: ToolContext, persist: bool = False
+  retries: int, tool_context: ToolContext, persist: bool = False
 ) -> str:
   """Set the maximum number of compilation validation and auto-fixing attempts.
 
@@ -24,6 +25,7 @@ async def set_max_compilation_retries_fn(
 
   # Update in-memory config variables
   import hitl_agent.config as hitl_cfg
+
   hitl_cfg.MAX_COMPILATION_RETRIES = retries
   os.environ["MAX_COMPILATION_RETRIES"] = str(retries)
 
@@ -49,7 +51,10 @@ async def set_max_compilation_retries_fn(
 
   # Update the active validation loops in-memory
   try:
-    from hitl_agent.subagents.kernel_writing.agent import kernel_compilation_validation_loop as hitl_loop
+    from hitl_agent.subagents.kernel_writing.agent import (
+      kernel_compilation_validation_loop as hitl_loop,
+    )
+
     hitl_loop.max_retries = retries
   except Exception as e:
     logging.warning(f"Could not update hitl_loop max_retries: {e}")

MaxKernel/hitl_agent/tools/tools.py

@@ -1,6 +1,7 @@
 """Tool setup for HITL kernel generation agents."""
 
 import logging
+
 from google.adk.models import LlmRequest
 from google.adk.tools import ToolContext
 from google.adk.tools.retrieval.vertex_ai_rag_retrieval import (
@@ -9,10 +10,13 @@
 from vertexai.preview import rag
 
 from hitl_agent.config import RAG_CORPUS
+from hitl_agent.tools.filesystem_tools import (
+  filesystem_tool_r,
+  filesystem_tool_rw,
+)
+from hitl_agent.tools.retries_tool import set_max_compilation_retries
 from hitl_agent.tools.search_api_tool import search_api_tool
 from hitl_agent.tools.workspace_tool import set_working_directory
-from hitl_agent.tools.filesystem_tools import filesystem_tool_r, filesystem_tool_rw
-from hitl_agent.tools.retries_tool import set_max_compilation_retries
 
 
 # Custom VertexAiRagRetrieval that forces function_declarations mode to avoid

MaxKernel/hitl_agent/tools/workspace_tool.py

@@ -2,15 +2,19 @@
 
 import logging
 import os
+
 from google.adk.tools import FunctionTool, ToolContext
 from google.adk.tools.mcp_tool.mcp_session_manager import MCPSessionManager
 
 import hitl_agent.config as config
-from hitl_agent.tools.filesystem_tools import filesystem_tool_r, filesystem_tool_rw
+from hitl_agent.tools.filesystem_tools import (
+  filesystem_tool_r,
+  filesystem_tool_rw,
+)
 
 
 async def set_working_directory_fn(
-    path: str, tool_context: ToolContext, persist: bool = False
+  path: str, tool_context: ToolContext, persist: bool = False
 ) -> str:
   """Set the active workspace/working directory path for the session.
 
@@ -65,10 +69,10 @@ async def set_working_directory_fn(
     server_params_r.args[-1] = abs_path
     await filesystem_tool_r.close()
     filesystem_tool_r._mcp_session_manager = MCPSessionManager(
-        connection_params=filesystem_tool_r._connection_params,
-        errlog=filesystem_tool_r._errlog,
-        sampling_callback=filesystem_tool_r._sampling_callback,
-        sampling_capabilities=filesystem_tool_r._sampling_capabilities,
+      connection_params=filesystem_tool_r._connection_params,
+      errlog=filesystem_tool_r._errlog,
+      sampling_callback=filesystem_tool_r._sampling_callback,
+      sampling_capabilities=filesystem_tool_r._sampling_capabilities,
     )
 
   # Update read-write filesystem tool
@@ -78,10 +82,10 @@ async def set_working_directory_fn(
     server_params_rw.args[-1] = abs_path
     await filesystem_tool_rw.close()
     filesystem_tool_rw._mcp_session_manager = MCPSessionManager(
-        connection_params=filesystem_tool_rw._connection_params,
-        errlog=filesystem_tool_rw._errlog,
-        sampling_callback=filesystem_tool_rw._sampling_callback,
-        sampling_capabilities=filesystem_tool_rw._sampling_capabilities,
+      connection_params=filesystem_tool_rw._connection_params,
+      errlog=filesystem_tool_rw._errlog,
+      sampling_callback=filesystem_tool_rw._sampling_callback,
+      sampling_capabilities=filesystem_tool_rw._sampling_capabilities,
     )
 
   return f"Successfully switched working directory to: {abs_path} (persisted: {persist})"