feat: support vanna tools

pyproject.toml

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

veadk/community/langchain_ai/middlewares/save_session.py

@@ -18,6 +18,7 @@
 from langchain.agents.middleware import after_agent
 from langchain_core.messages.ai import AIMessage
 from langchain_core.messages.human import HumanMessage
+from langchain_core.messages.tool import ToolMessage
 from langgraph.runtime import Runtime
 
 from veadk.community.langchain_ai.store.memory.viking_memory import (
@@ -46,15 +47,22 @@ def save_session(state: AgentState, runtime: Runtime) -> None:
 
     events = {}
     for message in messages:
-        print(type(message))
+        event = None
+
         if isinstance(message, HumanMessage):
-            event = {"role": "user", "parts": [{"text": message.content}]}
+            if message.content and message.content.strip():
+                event = {"role": "user", "parts": [{"text": message.content}]}
 
         elif isinstance(message, AIMessage):
-            event = {"role": "assistant", "parts": [{"text": message.content}]}
-        else:
-            ...
+            if message.content and message.content.strip():
+                event = {"role": "assistant", "parts": [{"text": message.content}]}
+
+        elif isinstance(message, ToolMessage):
+            # ToolMessage is not saved to memory store, skip it
+            continue
 
-        events[message.id] = event
+        if event:
+            events[message.id] = event
 
-    store.put(namespace=(app_name, user_id), key=session_id, value=events)
+    if events:
+        store.put(namespace=(app_name, user_id), key=session_id, value=events)

veadk/community/langchain_ai/store/memory/viking_memory.py

@@ -48,10 +48,11 @@ def batch(self, ops: Iterable[Op]) -> list[Result]:
         for op in ops:
             if isinstance(op, PutOp):
                 self._apply_put_op(op)
+                results.append(None)
             elif isinstance(op, GetOp):
-                self._apply_get_op(op)
+                results.append(self._apply_get_op(op))
             elif isinstance(op, SearchOp):
-                results.extend(self._apply_search_op(op))
+                results.append(self._apply_search_op(op))
             # elif isinstance(op, ListNamespacesOp):
             #     self._apply_list_namespaces_op(op)
             else:
@@ -98,5 +99,7 @@ def _apply_search_op(self, op: SearchOp):
         if not query:
             return []
 
-        value = self._backend.search_memory(user_id=user_id, query=query, top_k=1)
+        value = self._backend.search_memory(
+            user_id=user_id, query=query, top_k=op.limit or 1
+        )
         return value

veadk/skills/check_skills_callback.py

@@ -166,7 +166,11 @@ def reload_skills_from_config(skills_config: List[str]) -> Dict[str, Skill]:
                 for skill_dir in path.iterdir():
                     if skill_dir.is_dir():
                         skill = load_skill_from_directory(skill_dir)
-                        all_skills[skill.name] = skill
+                        # Only add skill if it loaded successfully
+                        if skill is not None:
+                            all_skills[skill.name] = skill
+                        else:
+                            logger.warning(f"Skipped failed skill from {skill_dir}")
             except Exception as e:
                 logger.error(
                     f"Failed to reload skills from local directory {path}: {e}"

veadk/skills/utils.py

@@ -85,39 +85,41 @@ def init_skill_check_list(
     return init_skill_check_list
 
 
-def load_skill_from_directory(skill_directory: Path) -> Skill:
+def load_skill_from_directory(skill_directory: Path) -> Optional[Skill]:
     logger.info(f"Load skill from {skill_directory}")
     skill_readme = skill_directory / "SKILL.md"
     if not skill_readme.exists():
         logger.error(f"Skill '{skill_directory}' has no SKILL.md file.")
-        raise ValueError(f"Skill '{skill_directory}' has no SKILL.md file")
+        return None
+
+    try:
+        skill = frontmatter.load(str(skill_readme))
 
-    skill = frontmatter.load(str(skill_readme))
+        skill_name = skill.get("name", "")
+        skill_description = skill.get("description", "")
+        checklist = skill.get("checklist", [])
 
-    skill_name = skill.get("name", "")
-    skill_description = skill.get("description", "")
-    checklist = skill.get("checklist", [])
+        if not skill_name or not skill_description:
+            logger.error(
+                f"Skill {skill_readme} is missing name or description. Please check the SKILL.md file."
+            )
+            return None
 
-    if not skill_name or not skill_description:
-        logger.error(
-            f"Skill {skill_readme} is missing name or description. Please check the SKILL.md file."
+        logger.info(
+            f"Successfully loaded skill {skill_name} locally from {skill_readme}, name={skill_name}, description={skill_description}"
         )
-        raise ValueError(
-            f"Skill {skill_readme} is missing name or description. Please check the SKILL.md file."
+        if checklist:
+            logger.info(f"Skill {skill_name} checklist: {checklist}")
+
+        return Skill(
+            name=skill_name,  # type: ignore
+            description=skill_description,  # type: ignore
+            path=str(skill_directory),
+            checklist=checklist,
         )
-
-    logger.info(
-        f"Successfully loaded skill {skill_name} locally from {skill_readme}, name={skill_name}, description={skill_description}"
-    )
-    if checklist:
-        logger.info(f"Skill {skill_name} checklist: {checklist}")
-
-    return Skill(
-        name=skill_name,  # type: ignore
-        description=skill_description,  # type: ignore
-        path=str(skill_directory),
-        checklist=checklist,
-    )
+    except Exception as e:
+        logger.error(f"Failed to load skill from {skill_directory}: {e}")
+        return None
 
 
 def load_skills_from_directory(skills_directory: Path) -> list[Skill]:
@@ -126,7 +128,8 @@ def load_skills_from_directory(skills_directory: Path) -> list[Skill]:
     for skill_directory in skills_directory.iterdir():
         if skill_directory.is_dir():
             skill = load_skill_from_directory(skill_directory)
-            skills.append(skill)
+            if skill is not None:
+                skills.append(skill)
     return skills