refactor: wrap event compaction into a function

Author: shangkunwang01

MaxKernel/auto_agent/agent.py

@@ -4,8 +4,10 @@
 for the human-in-the-loop kernel generation process.
 """
 
-from google.adk.apps.app import App, EventsCompactionConfig
+from google.adk.apps.app import App
 
+from auto_agent.config import get_compaction_config
+from auto_agent.constants import EVENTS_COMPACTION
 from auto_agent.subagents.autotuning.agent import autotune_agent
 from auto_agent.subagents.kernel_writing import (
   implement_kernel_agent,
@@ -32,12 +34,7 @@
 )
 
 if EVENTS_COMPACTION:
-  compaction_config = EventsCompactionConfig(
-    token_threshold=200000,
-    event_retention_size=100,
-    compaction_interval=1,
-    overlap_size=0,
-  )
+  compaction_config = get_compaction_config()
 else:
   compaction_config = None
 

MaxKernel/auto_agent/config.py

@@ -2,19 +2,26 @@
 
 import os
 
+from google.adk.apps.app import EventsCompactionConfig
 from google.adk.planners import BuiltInPlanner
 from google.genai import types
 
-from auto_agent.constants import EVENTS_COMPACTION, TEMPERATURE, TOP_K, TOP_P
+from auto_agent.constants import 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
+# Set events compaction policy to avoid memory overflow
+def get_compaction_config():
+  return EventsCompactionConfig(
+    token_threshold=200000,
+    event_retention_size=100,
+    compaction_interval=1,
+    overlap_size=0,
+  )
 
 # Model configuration
 model_config = types.GenerateContentConfig(