veadk-python 0.5.3__tar.gz → 0.5.4__tar.gz

{veadk_python-0.5.3/veadk_python.egg-info → veadk_python-0.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veadk-python
-Version: 0.5.3
+Version: 0.5.4
 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
@@ -217,7 +217,6 @@ Requires-Dist: loguru==0.7.3
 Requires-Dist: opentelemetry-exporter-otlp==1.37.0
 Requires-Dist: opentelemetry-instrumentation-logging>=0.56b0
 Requires-Dist: wrapt==1.17.2
-Requires-Dist: openai<1.100
 Requires-Dist: volcengine-python-sdk>=5.0.1
 Requires-Dist: volcengine>=1.0.193
 Requires-Dist: agent-pilot-sdk==0.1.2

{veadk_python-0.5.3 → veadk_python-0.5.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "veadk-python"
-version = "0.5.3"
+version = "0.5.4"
 description = "Volcengine agent development kit, integrations with Volcengine cloud services."
 readme = "README.md"
 requires-python = ">=3.10"
@@ -21,7 +21,6 @@ dependencies = [
     "opentelemetry-exporter-otlp==1.37.0",
     "opentelemetry-instrumentation-logging>=0.56b0",
     "wrapt==1.17.2", # For patching built-in functions
-    "openai<1.100", # For fix https://github.com/BerriAI/litellm/issues/13710
     "volcengine-python-sdk>=5.0.1", # For Volcengine API
     "volcengine>=1.0.193", # For Volcengine sign
     "agent-pilot-sdk==0.1.2", # Prompt optimization by Volcengine AgentPilot/PromptPilot toolkits

{veadk_python-0.5.3 → veadk_python-0.5.4}/veadk/cli/cli_create.py

@@ -147,6 +147,10 @@ def create(agent_name: str, ark_api_key: str) -> None:
     """
     if not agent_name:
         agent_name = click.prompt("Enter the agent name")
+
+    if "-" in agent_name:
+        raise ValueError("Agent name cannot contain '-'. Use '_' instead.")
+
     if not ark_api_key:
         ark_api_key = _prompt_for_ark_api_key()
 

{veadk_python-0.5.3 → veadk_python-0.5.4}/veadk/cli/cli_web.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import logging
 from functools import wraps
 
 import click
@@ -139,5 +140,14 @@ def web(ctx, *args, **kwargs) -> None:
     # from Google ADK and Litellm
     if "--log_level" not in extra_args:
         extra_args.extend(["--log_level", "ERROR"])
+        logging.basicConfig(level=logging.ERROR, force=True)
+
+    if "--log_level" in extra_args:
+        logging.basicConfig(
+            level=getattr(
+                logging, extra_args[extra_args.index("--log_level") + 1].upper()
+            ),
+            force=True,
+        )
 
     cli_web.main(args=extra_args, standalone_mode=False)

veadk_python-0.5.4/veadk/community/langchain_ai/middlewares/save_session.py

@@ -0,0 +1,60 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import annotations
+
+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 langgraph.runtime import Runtime
+
+from veadk.community.langchain_ai.store.memory.viking_memory import (
+    VikingMemoryStore,
+)
+from veadk.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+
+@after_agent
+def save_session(state: AgentState, runtime: Runtime) -> None:
+    """Save the session to the memory store."""
+    store: VikingMemoryStore | None = runtime.store
+    if not store:
+        return
+
+    app_name = store.index
+    user_id = runtime.context.user_id
+    session_id = runtime.context.session_id
+
+    messages = state.get("messages", [])
+    logger.debug(
+        f"Save session {session_id} for user {user_id} with {len(messages)} messages. messages={messages}"
+    )
+
+    events = {}
+    for message in messages:
+        print(type(message))
+        if isinstance(message, HumanMessage):
+            event = {"role": "user", "parts": [{"text": message.content}]}
+
+        elif isinstance(message, AIMessage):
+            event = {"role": "assistant", "parts": [{"text": message.content}]}
+        else:
+            ...
+
+        events[message.id] = event
+
+    store.put(namespace=(app_name, user_id), key=session_id, value=events)

veadk_python-0.5.4/veadk/community/langchain_ai/models/ark_model.py

@@ -0,0 +1,34 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import annotations
+
+from langchain_core.utils.utils import (
+    from_env,
+    secret_from_env,
+)
+from langchain_openai import ChatOpenAI
+
+
+class ArkChatModel(ChatOpenAI):
+    def __init__(self, model: str, **kwargs):
+        super().__init__(
+            model=model,
+            api_key=secret_from_env("MODEL_AGENT_API_KEY")(),
+            base_url=from_env(
+                "MODEL_AGENT_API_BASE",
+                default="https://ark.cn-beijing.volces.com/api/v3",
+            )(),
+            **kwargs,
+        )

veadk_python-0.5.4/veadk/community/langchain_ai/store/memory/viking_memory.py

@@ -0,0 +1,102 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import annotations
+
+import json
+from collections.abc import Iterable
+
+from langgraph.store.base import (
+    BaseStore,
+    GetOp,
+    ListNamespacesOp,
+    Op,
+    PutOp,
+    Result,
+    SearchOp,
+)
+
+from veadk.memory.long_term_memory_backends.vikingdb_memory_backend import (
+    VikingDBLTMBackend,
+)
+
+
+# mem0
+# viking db
+class VikingMemoryStore(BaseStore):
+    def __init__(self, index: str):
+        # index = (index, user_id)
+        # key = session_id
+        self.index = index
+        self._backend = VikingDBLTMBackend(index=index)
+
+    def batch(self, ops: Iterable[Op]) -> list[Result]:
+        # The batch/abatch methods are treated as internal.
+        # Users should access via put/search/get/list_namespaces/etc.
+        results = []
+        for op in ops:
+            if isinstance(op, PutOp):
+                self._apply_put_op(op)
+            elif isinstance(op, GetOp):
+                self._apply_get_op(op)
+            elif isinstance(op, SearchOp):
+                results.extend(self._apply_search_op(op))
+            # elif isinstance(op, ListNamespacesOp):
+            #     self._apply_list_namespaces_op(op)
+            else:
+                raise ValueError(f"Unknown op type: {type(op)}")
+
+        return results
+
+    def abatch(
+        self, ops: Iterable[GetOp | SearchOp | PutOp | ListNamespacesOp]
+    ) -> list[Result]: ...
+
+    def _apply_put_op(self, op: PutOp) -> None:
+        index, user_id = op.namespace
+        session_id = op.key
+
+        assert index == self._backend.index, (
+            "index must be the same as the backend index"
+        )
+
+        value = op.value
+
+        event_strings = []
+
+        for _, event in value.items():
+            event_strings.append(json.dumps(event))
+
+        if self._backend.save_memory(
+            user_id=user_id,
+            session_id=session_id,
+            event_strings=event_strings,
+        ):
+            return None
+
+    def _apply_get_op(self, op: GetOp):
+        return ["Not implemented"]
+
+    def _apply_search_op(self, op: SearchOp):
+        index, user_id = op.namespace_prefix
+        assert index == self._backend.index, (
+            "index must be the same as the backend index"
+        )
+
+        query = op.query
+        if not query:
+            return []
+
+        value = self._backend.search_memory(user_id=user_id, query=query, top_k=1)
+        return value

veadk_python-0.5.4/veadk/community/langchain_ai/tools/load_knowledgebase.py

@@ -0,0 +1,34 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from langchain.tools import ToolRuntime, tool
+
+from veadk.knowledgebase import KnowledgeBase
+from veadk.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+
+@tool
+def load_knowledgebase(query: str, runtime: ToolRuntime) -> list[str]:
+    """Load knowledge base for the current user.
+
+    Args:
+        query: The query to search for in the knowledge base.
+    """
+    knowledgeabse: KnowledgeBase = runtime.context.knowledgebase  # type: ignore
+
+    results = knowledgeabse.search(query)
+
+    return [result.content for result in results]

veadk_python-0.5.4/veadk/community/langchain_ai/tools/load_memory.py

@@ -0,0 +1,43 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+from langchain.tools import ToolRuntime, tool
+
+from veadk.community.langchain_ai.store.memory.viking_memory import (
+    VikingMemoryStore,
+)
+from veadk.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+
+@tool
+def load_memory(query: str, runtime: ToolRuntime) -> list[str]:
+    """Load memories for the current user across all history sessions.
+
+    Args:
+        query: The query to search for in the memory.
+    """
+    store: VikingMemoryStore | None = runtime.store  # type: ignore
+    if not store:
+        return ["Long-term memory store is not initialized."]
+
+    app_name = store.index
+    user_id = runtime.context.user_id
+
+    logger.info(f"Load memory for user {user_id} with query {query}")
+    response = store.search((app_name, user_id), query=query)
+
+    return response

{veadk_python-0.5.3 → veadk_python-0.5.4}/veadk/memory/long_term_memory_backends/vikingdb_memory_backend.py

@@ -143,9 +143,14 @@ class VikingDBLTMBackend(BaseLongTermMemoryBackend):
         )
 
     @override
-    def save_memory(self, user_id: str, event_strings: list[str], **kwargs) -> bool:
+    def save_memory(
+        self,
+        user_id: str,
+        event_strings: list[str],
+        **kwargs,
+    ) -> bool:
         assistant_id = kwargs.get("assistant_id", "assistant")
-        session_id = str(uuid.uuid1())
+        session_id = kwargs.get("session_id", str(uuid.uuid1()))
         messages = []
         for raw_events in event_strings:
             event = json.loads(raw_events)
@@ -161,7 +166,7 @@ class VikingDBLTMBackend(BaseLongTermMemoryBackend):
         }
 
         logger.debug(
-            f"Request for add {len(messages)} memory to VikingDB: collection_name={self.index}, metadata={metadata}, session_id={session_id}"
+            f"Request for add {len(messages)} memory to VikingDB: collection_name={self.index}, metadata={metadata}, session_id={session_id}, messages={messages}"
         )
 
         client = self._get_sdk_client()

veadk_python-0.5.4/veadk/tools/builtin_tools/__init__.py

@@ -0,0 +1,13 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.

veadk_python-0.5.4/veadk/tools/mcp_tool/__init__.py

@@ -0,0 +1,13 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.

veadk_python-0.5.4/veadk/tools/sandbox/__init__.py

@@ -0,0 +1,13 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.

veadk_python-0.5.4/veadk/tracing/__init__.py

@@ -0,0 +1,13 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.

veadk_python-0.5.4/veadk/tracing/telemetry/__init__.py

@@ -0,0 +1,13 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.

veadk_python-0.5.4/veadk/tracing/telemetry/exporters/__init__.py

@@ -0,0 +1,13 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.

veadk_python-0.5.4/veadk/utils/__init__.py

@@ -0,0 +1,13 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.

{veadk_python-0.5.3 → veadk_python-0.5.4}/veadk/version.py

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

{veadk_python-0.5.3 → veadk_python-0.5.4/veadk_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veadk-python
-Version: 0.5.3
+Version: 0.5.4
 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
@@ -217,7 +217,6 @@ Requires-Dist: loguru==0.7.3
 Requires-Dist: opentelemetry-exporter-otlp==1.37.0
 Requires-Dist: opentelemetry-instrumentation-logging>=0.56b0
 Requires-Dist: wrapt==1.17.2
-Requires-Dist: openai<1.100
 Requires-Dist: volcengine-python-sdk>=5.0.1
 Requires-Dist: volcengine>=1.0.193
 Requires-Dist: agent-pilot-sdk==0.1.2

{veadk_python-0.5.3 → veadk_python-0.5.4}/veadk_python.egg-info/SOURCES.txt

@@ -86,6 +86,18 @@ veadk/cli/templates/rl/lightning/data/demo_train.parquet
 veadk/cloud/__init__.py
 veadk/cloud/cloud_agent_engine.py
 veadk/cloud/cloud_app.py
+veadk/community/langchain_ai/checkpoint/__init__.py
+veadk/community/langchain_ai/checkpoint/memory/__init__.py
+veadk/community/langchain_ai/middlewares/__init__.py
+veadk/community/langchain_ai/middlewares/save_session.py
+veadk/community/langchain_ai/models/__init__.py
+veadk/community/langchain_ai/models/ark_model.py
+veadk/community/langchain_ai/store/__init__.py
+veadk/community/langchain_ai/store/memory/__init__.py
+veadk/community/langchain_ai/store/memory/viking_memory.py
+veadk/community/langchain_ai/tools/__init__.py
+veadk/community/langchain_ai/tools/load_knowledgebase.py
+veadk/community/langchain_ai/tools/load_memory.py
 veadk/configs/__init__.py
 veadk/configs/auth_configs.py
 veadk/configs/database_configs.py

{veadk_python-0.5.3 → veadk_python-0.5.4}/veadk_python.egg-info/requires.txt

@@ -7,7 +7,6 @@ loguru==0.7.3
 opentelemetry-exporter-otlp==1.37.0
 opentelemetry-instrumentation-logging>=0.56b0
 wrapt==1.17.2
-openai<1.100
 volcengine-python-sdk>=5.0.1
 volcengine>=1.0.193
 agent-pilot-sdk==0.1.2