Skip to content

Commit 2216f9d

Browse files
authored
feat: add toggles for skills checklist, false by default (#519)
* feat: add toggles for skills checklist, false by default * feat: add toggles for skills checklist, false by default
1 parent b7fca4d commit 2216f9d

File tree

1 file changed

+16
-13
lines changed

1 file changed

+16
-13
lines changed

veadk/agent.py

Lines changed: 16 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)