Skip to content

Commit 98a820a

Browse files
committed
feat: add toggles for skills checklist, false by default
1 parent b7fca4d commit 98a820a

1 file changed

Lines changed: 15 additions & 13 deletions

File tree

veadk/agent.py

Lines changed: 15 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,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

Comments
 (0)