@@ -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,23 @@ 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+ logger .info ("Skills checklist enabled" )
310+ from veadk .skills .utils import create_init_skill_check_list_callback
308311
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 )
312+ init_callback = create_init_skill_check_list_callback (
313+ self ._skills_with_checklist
314+ )
315+ if self .before_tool_callback :
316+ if isinstance (self .before_tool_callback , list ):
317+ self .before_tool_callback .append (init_callback )
318+ else :
319+ self .before_tool_callback = [
320+ self .before_tool_callback ,
321+ init_callback ,
322+ ]
315323 else :
316- self .before_tool_callback = [
317- self .before_tool_callback ,
318- init_callback ,
319- ]
320- else :
321- self .before_tool_callback = init_callback
324+ self .before_tool_callback = init_callback
322325
323326 if self .example_store :
324327 from google .adk .tools .example_tool import ExampleTool
0 commit comments