feat: add configurable event compaction support

Author: shangkunwang01

MaxKernel/auto_agent/agent.py

@@ -5,6 +5,7 @@
 """
 
 from google.adk.apps.app import App, EventsCompactionConfig
+
 from auto_agent.subagents.autotuning.agent import autotune_agent
 from auto_agent.subagents.kernel_writing import (
   implement_kernel_agent,
@@ -30,10 +31,13 @@
   max_iterations=5,
 )
 
-compaction_config = EventsCompactionConfig(
-  token_threshold=200000,
-  event_retention_size=100,
-)
+if EVENTS_COMPACTION:
+  compaction_config = EventsCompactionConfig(
+    token_threshold=200000,
+    event_retention_size=100,
+  )
+else:
+  compaction_config = None
 
 app = App(
   name="auto_agent",
@@ -42,4 +46,3 @@
 )
 
 __all__ = ["root_agent", "app"]
-

MaxKernel/auto_agent/config.py

@@ -5,17 +5,20 @@
 from google.adk.planners import BuiltInPlanner
 from google.genai import types
 
-from auto_agent.constants import TOP_K, TOP_P
+from auto_agent.constants import EVENTS_COMPACTION, TEMPERATURE, TOP_K, TOP_P
 
 # Environment variables
 WORKDIR = os.environ.get("WORKDIR", os.path.dirname(os.path.abspath(__file__)))
 TPU_VERSION = os.environ.get("TPU_VERSION", "")
 RAG_CORPUS = os.environ.get("RAG_CORPUS", "")
 INCLUDE_THOUGHTS = os.environ.get("INCLUDE_THOUGHTS", "true").lower() == "true"
 
+# Events Compaction Configuration
+EVENTS_COMPACTION = EVENTS_COMPACTION
+
 # Model configuration
 model_config = types.GenerateContentConfig(
-  temperature=0.5,
+  temperature=TEMPERATURE,
   top_p=TOP_P,
   top_k=TOP_K,
 )

MaxKernel/auto_agent/constants.py

@@ -1,10 +1,11 @@
 MODEL_NAME = "gemini-3.1-pro-preview"
 MAX_ITERATIONS = 5
 LLM_GEN_RETRY_COUNT = 3
-TEMPERATURE = 0.1
+TEMPERATURE = 0.5
 TOP_P = 0.9
 TOP_K = 5
 REQUEST_TIMEOUT = 3600 * 3
 TPU_SERVER_PORT = 5463
 CPU_SERVER_PORT = 5464
 EVAL_SERVER_PORT = 1245
+EVENTS_COMPACTION = True