@@ -158,6 +158,7 @@ class Agent(LlmAgent):
158158 enable_dataset_gen : bool = False
159159
160160 enable_dynamic_load_skills : bool = False
161+ enable_skills_checklist : bool = False
161162 _skills_with_checklist : Dict [str , Any ] = {}
162163
163164 def model_post_init (self , __context : Any ) -> None :
@@ -304,21 +305,22 @@ def model_post_init(self, __context: Any) -> None:
304305
305306 if self .skills :
306307 self .load_skills ()
307- from veadk .skills .utils import create_init_skill_check_list_callback
308+ if self .enable_skills_checklist :
309+ from veadk .skills .utils import create_init_skill_check_list_callback
308310
309- init_callback = create_init_skill_check_list_callback (
310- self ._skills_with_checklist
311- )
312- if self .before_tool_callback :
313- if isinstance (self .before_tool_callback , list ):
314- self .before_tool_callback .append (init_callback )
311+ init_callback = create_init_skill_check_list_callback (
312+ self ._skills_with_checklist
313+ )
314+ if self .before_tool_callback :
315+ if isinstance (self .before_tool_callback , list ):
316+ self .before_tool_callback .append (init_callback )
317+ else :
318+ self .before_tool_callback = [
319+ self .before_tool_callback ,
320+ init_callback ,
321+ ]
315322 else :
316- self .before_tool_callback = [
317- self .before_tool_callback ,
318- init_callback ,
319- ]
320- else :
321- self .before_tool_callback = init_callback
323+ self .before_tool_callback = init_callback
322324
323325 if self .example_store :
324326 from google .adk .tools .example_tool import ExampleTool
0 commit comments