Skip to content

Commit 460496e

Browse files
feat: configure context compression for auto agent
1 parent 6900faf commit 460496e

1 file changed

Lines changed: 14 additions & 1 deletion

File tree

MaxKernel/auto_agent/agent.py

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
for the human-in-the-loop kernel generation process.
55
"""
66

7+
from google.adk.apps.app import App, EventsCompactionConfig
78
from auto_agent.subagents.autotuning.agent import autotune_agent
89
from auto_agent.subagents.kernel_writing import (
910
implement_kernel_agent,
@@ -29,4 +30,16 @@
2930
max_iterations=5,
3031
)
3132

32-
__all__ = ["root_agent"]
33+
compaction_config = EventsCompactionConfig(
34+
token_threshold=200000,
35+
event_retention_size=100,
36+
)
37+
38+
app = App(
39+
name="auto_agent",
40+
root_agent=root_agent,
41+
events_compaction_config=compaction_config,
42+
)
43+
44+
__all__ = ["root_agent", "app"]
45+

0 commit comments

Comments
 (0)