diff --git a/MaxKernel/auto_agent/config.py b/MaxKernel/auto_agent/config.py index 66e8138..21c356a 100644 --- a/MaxKernel/auto_agent/config.py +++ b/MaxKernel/auto_agent/config.py @@ -3,11 +3,12 @@ import os from google.adk.apps.app import EventsCompactionConfig +from google.adk.apps.llm_event_summarizer import LlmEventSummarizer from google.adk.planners import BuiltInPlanner from google.genai import types -from auto_agent.custom_types import TimeoutGemini + from auto_agent.constants import MODEL_NAME, TEMPERATURE, TOP_K, TOP_P -from google.adk.apps.llm_event_summarizer import LlmEventSummarizer +from auto_agent.custom_types import TimeoutGemini # Environment variables WORKDIR = os.environ.get("WORKDIR", os.path.dirname(os.path.abspath(__file__))) @@ -19,6 +20,7 @@ compaction_model = TimeoutGemini(model=MODEL_NAME) + # Set events compaction policy to avoid memory overflow def get_compaction_config(): return EventsCompactionConfig( @@ -26,7 +28,7 @@ def get_compaction_config(): event_retention_size=5, compaction_interval=0, overlap_size=0, - summarizer=LlmEventSummarizer(llm=compaction_model) + summarizer=LlmEventSummarizer(llm=compaction_model), ) diff --git a/MaxKernel/evaluation/jax_kernel_evaluator.py b/MaxKernel/evaluation/jax_kernel_evaluator.py index fdb2798..9acd341 100644 --- a/MaxKernel/evaluation/jax_kernel_evaluator.py +++ b/MaxKernel/evaluation/jax_kernel_evaluator.py @@ -428,12 +428,14 @@ def _build_task_json( rtol: Evaluator level fallback for rtol. """ effective_atol = ( - atol if atol is not None else - (task.atol if task.atol is not None else DEFAULT_ATOL) + atol + if atol is not None + else (task.atol if task.atol is not None else DEFAULT_ATOL) ) effective_rtol = ( - rtol if rtol is not None else - (task.rtol if task.rtol is not None else DEFAULT_RTOL) + rtol + if rtol is not None + else (task.rtol if task.rtol is not None else DEFAULT_RTOL) ) task_info = {