veadk-python 0.5.21__tar.gz → 0.5.23__tar.gz

{veadk_python-0.5.21/veadk_python.egg-info → veadk_python-0.5.23}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veadk-python
-Version: 0.5.21
+Version: 0.5.23
 Summary: Volcengine agent development kit, integrations with Volcengine cloud services.
 Author-email: Yaozheng Fang <fangyozheng@gmail.com>, Guodong Li <cu.eric.lee@gmail.com>, Zhi Han <sliverydayday@gmail.com>, Meng Wang <mengwangwm@gmail.com>
 License:                                  Apache License

{veadk_python-0.5.21 → veadk_python-0.5.23}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "veadk-python"
-version = "0.5.21"
+version = "0.5.23"
 description = "Volcengine agent development kit, integrations with Volcengine cloud services."
 readme = "README.md"
 requires-python = ">=3.10"

{veadk_python-0.5.21 → veadk_python-0.5.23}/veadk/agent.py

@@ -86,6 +86,7 @@ class Agent(LlmAgent):
         skills (list[str]): List of skills that equip the agent with specific capabilities.
         example_store (Optional[BaseExampleProvider]): Example store for providing example Q/A.
         enable_shadowchar (bool): Whether to enable shadow character for the agent.
+        enable_dynamic_load_skills (bool): Whether to enable dynamic loading of skills.
     """
 
     model_config = ConfigDict(arbitrary_types_allowed=True, extra="allow")
@@ -156,6 +157,10 @@ class Agent(LlmAgent):
 
     enable_dataset_gen: bool = False
 
+    enable_dynamic_load_skills: bool = False
+    enable_skills_checklist: bool = False
+    _skills_with_checklist: Dict[str, Any] = {}
+
     def model_post_init(self, __context: Any) -> None:
         super().model_post_init(None)  # for sub_agents init
 
@@ -227,6 +232,8 @@ class Agent(LlmAgent):
 
         self._prepare_tracers()
 
+        self._validate_tool_dependencies()
+
         if self.knowledgebase:
             from veadk.tools.builtin_tools.load_knowledgebase import (
                 LoadKnowledgebaseTool,
@@ -300,6 +307,23 @@ class Agent(LlmAgent):
 
         if self.skills:
             self.load_skills()
+            if self.enable_skills_checklist:
+                logger.info("Skills checklist enabled")
+                from veadk.skills.utils import create_init_skill_check_list_callback
+
+                init_callback = create_init_skill_check_list_callback(
+                    self._skills_with_checklist
+                )
+                if self.before_tool_callback:
+                    if isinstance(self.before_tool_callback, list):
+                        self.before_tool_callback.append(init_callback)
+                    else:
+                        self.before_tool_callback = [
+                            self.before_tool_callback,
+                            init_callback,
+                        ]
+                else:
+                    self.before_tool_callback = init_callback
 
         if self.example_store:
             from google.adk.tools.example_tool import ExampleTool
@@ -347,13 +371,14 @@ class Agent(LlmAgent):
         from pathlib import Path
 
         from veadk.skills.skill import Skill
+        from veadk.skills.check_skills_callback import check_skills
         from veadk.skills.utils import (
             load_skills_from_cloud,
             load_skills_from_directory,
         )
         from veadk.tools.skills_tools.skills_toolset import SkillsToolset
 
-        skills: Dict[str, Skill] = {}
+        self.skills_dict: Dict[str, Skill] = {}
 
         # Determine skills_mode if not set
         if not self.skills_mode:
@@ -428,17 +453,28 @@ class Agent(LlmAgent):
             path = Path(item)
             if path.exists() and path.is_dir():
                 for skill in load_skills_from_directory(path):
-                    skills[skill.name] = skill
+                    self.skills_dict[skill.name] = skill
             else:
                 for skill in load_skills_from_cloud(item):
-                    skills[skill.name] = skill
-        if skills:
+                    self.skills_dict[skill.name] = skill
+        if self.skills_dict:
             self.instruction += "\nYou have the following skills:\n"
 
-            for skill in skills.values():
+            self._skills_with_checklist = self.skills_dict
+
+            has_checklist = False
+            for skill in self.skills_dict.values():
                 self.instruction += (
                     f"- name: {skill.name}\n- description: {skill.description}\n\n"
                 )
+                if skill.checklist:
+                    has_checklist = True
+
+            if has_checklist:
+                self.instruction += (
+                    "Some skills have a checklist that you must complete step by step. "
+                    "Use the `update_check_list` tool to mark each item as completed.\n\n"
+                )
 
             if self.skills_mode not in [
                 "skills_sandbox",
@@ -459,10 +495,52 @@ class Agent(LlmAgent):
                     "You can use the skills by calling the `skills_tool` tool.\n\n"
                 )
 
-            self.tools.append(SkillsToolset(skills, self.skills_mode))
         else:
             logger.warning("No skills loaded.")
 
+        self.tools.append(SkillsToolset(self.skills_dict, self.skills_mode))
+
+        if self.enable_dynamic_load_skills:
+            if self.before_agent_callback:
+                if isinstance(self.before_agent_callback, list):
+                    self.before_agent_callback.append(check_skills)
+                else:
+                    self.before_agent_callback = [
+                        self.before_agent_callback,
+                        check_skills,
+                    ]
+            else:
+                self.before_agent_callback = check_skills
+
+    def _validate_tool_dependencies(self):
+        tool_names = set()
+        for tool in self.tools:
+            if hasattr(tool, "__name__"):
+                tool_names.add(tool.__name__)
+            elif hasattr(tool, "name"):
+                tool_names.add(tool.name)
+
+        has_video_generate = "video_generate" in tool_names
+        has_video_task_query = "video_task_query" in tool_names
+
+        if has_video_generate and not has_video_task_query:
+            from veadk.tools.builtin_tools.video_generate import video_task_query
+
+            logger.warning(
+                "video_generate tool is mounted but video_task_query is not. "
+                "video_task_query is required for querying video generation status. "
+                "Automatically adding video_task_query to tools."
+            )
+            self.tools.append(video_task_query)
+        elif has_video_task_query and not has_video_generate:
+            from veadk.tools.builtin_tools.video_generate import video_generate
+
+            logger.warning(
+                "video_task_query tool is mounted but video_generate is not. "
+                "Automatically adding video_generate to tools."
+            )
+            self.tools.append(video_generate)
+
     def _prepare_tracers(self):
         enable_apmplus_tracer = os.getenv("ENABLE_APMPLUS", "false").lower() == "true"
         enable_cozeloop_tracer = os.getenv("ENABLE_COZELOOP", "false").lower() == "true"

{veadk_python-0.5.21 → veadk_python-0.5.23}/veadk/consts.py

@@ -65,10 +65,10 @@ DEFAULT_COZELOOP_SPACE_NAME = "VeADK Space"
 DEFAULT_IMAGE_EDIT_MODEL_NAME = "doubao-seededit-3-0-i2i-250628"
 DEFAULT_IMAGE_EDIT_MODEL_API_BASE = "https://ark.cn-beijing.volces.com/api/v3/"
 
-DEFAULT_VIDEO_MODEL_NAME = "doubao-seedance-1-5-pro-251215"
+DEFAULT_VIDEO_MODEL_NAME = "doubao-seedance-2-0-260128"
 DEFAULT_VIDEO_MODEL_API_BASE = "https://ark.cn-beijing.volces.com/api/v3/"
 
-DEFAULT_IMAGE_GENERATE_MODEL_NAME = "doubao-seedream-4-5-251128"
+DEFAULT_IMAGE_GENERATE_MODEL_NAME = "doubao-seedream-5-0-260128"
 DEFAULT_IMAGE_GENERATE_MODEL_API_BASE = "https://ark.cn-beijing.volces.com/api/v3/"
 
 VEFAAS_IAM_CRIDENTIAL_PATH = "/var/run/secrets/iam/credential"