veadk-python 0.5.24__tar.gz → 0.5.25__tar.gz

{veadk_python-0.5.24/veadk_python.egg-info → veadk_python-0.5.25}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veadk-python
-Version: 0.5.24
+Version: 0.5.25
 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.24 → veadk_python-0.5.25}/pyproject.toml

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

{veadk_python-0.5.24 → veadk_python-0.5.25}/veadk/community/langchain_ai/middlewares/save_session.py

@@ -18,6 +18,7 @@ from langchain.agents import AgentState
 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_python-0.5.24 → veadk_python-0.5.25}/veadk/community/langchain_ai/store/memory/viking_memory.py

@@ -48,10 +48,11 @@ class VikingMemoryStore(BaseStore):
         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 @@ class VikingMemoryStore(BaseStore):
         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_python-0.5.24 → veadk_python-0.5.25}/veadk/version.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-VERSION = "0.5.24"
+VERSION = "0.5.25"

{veadk_python-0.5.24 → veadk_python-0.5.25/veadk_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veadk-python
-Version: 0.5.24
+Version: 0.5.25
 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