veadk-python 0.5.31__tar.gz → 0.5.34__tar.gz

{veadk_python-0.5.31 → veadk_python-0.5.34}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veadk-python
-Version: 0.5.31
+Version: 0.5.34
 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
@@ -209,9 +209,9 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: pydantic-settings==2.10.1
-Requires-Dist: a2a-sdk>=0.3.7
+Requires-Dist: a2a-sdk==0.3.7
 Requires-Dist: deprecated==1.2.18
-Requires-Dist: google-adk>=1.19.0
+Requires-Dist: google-adk==1.19.0
 Requires-Dist: litellm<=1.82.6,>=1.74.3
 Requires-Dist: loguru==0.7.3
 Requires-Dist: opentelemetry-exporter-otlp==1.37.0
@@ -243,6 +243,7 @@ Requires-Dist: llama-index-llms-openai-like==0.5.1; extra == "extensions"
 Requires-Dist: llama-index-vector-stores-redis>=0.6.1; extra == "extensions"
 Requires-Dist: llama-index-vector-stores-opensearch==0.6.1; extra == "extensions"
 Requires-Dist: opensearch-py==2.8.0; extra == "extensions"
+Requires-Dist: lark-oapi; extra == "extensions"
 Provides-Extra: database
 Requires-Dist: redis>=5.0; extra == "database"
 Requires-Dist: pymysql>=1.1.1; extra == "database"
@@ -343,6 +344,21 @@ res = asyncio.run(agent.run("hello!"))
 print(res)
 ```
 
+## Feishu bot channel
+
+VeADK now provides `veadk.extensions.FeishuChannelExtension` for bridging a Feishu bot with a `Runner`. It maps `union_id` to `user_id`, and `thread_id` / `chat_id` to `session_id`, so VeADK memory and tracing can work directly in Feishu conversations.
+
+```python
+from veadk import Agent, Runner
+from veadk.extensions import FeishuChannelExtension
+
+agent = Agent()
+runner = Runner(agent=agent, app_name="feishu_demo")
+channel = FeishuChannelExtension(runner=runner)
+```
+
+Configure credentials with `TOOL_FEISHU_CHANNEL_APP_ID` and `TOOL_FEISHU_CHANNEL_APP_SECRET`, or in `config.yaml` under `tool.feishu_channel`.
+
 ## Command line tools
 
 VeADK provides several useful command line tools for faster deployment and optimization, such as:
@@ -365,6 +381,11 @@ Before commit or push your changes, please make sure the unittests are passed ,o
 pytest -n 16
 ```
 
+## Security and privacy
+
+This project takes security seriously.
+For vulnerability reporting and supported versions, see [SECURITY.md](SECURITY.md)
+
 ## Contact with us
 
 Join our discussion group by scanning the QR code below:

{veadk_python-0.5.31 → veadk_python-0.5.34}/README.md

@@ -80,6 +80,21 @@ res = asyncio.run(agent.run("hello!"))
 print(res)
 ```
 
+## Feishu bot channel
+
+VeADK now provides `veadk.extensions.FeishuChannelExtension` for bridging a Feishu bot with a `Runner`. It maps `union_id` to `user_id`, and `thread_id` / `chat_id` to `session_id`, so VeADK memory and tracing can work directly in Feishu conversations.
+
+```python
+from veadk import Agent, Runner
+from veadk.extensions import FeishuChannelExtension
+
+agent = Agent()
+runner = Runner(agent=agent, app_name="feishu_demo")
+channel = FeishuChannelExtension(runner=runner)
+```
+
+Configure credentials with `TOOL_FEISHU_CHANNEL_APP_ID` and `TOOL_FEISHU_CHANNEL_APP_SECRET`, or in `config.yaml` under `tool.feishu_channel`.
+
 ## Command line tools
 
 VeADK provides several useful command line tools for faster deployment and optimization, such as:
@@ -102,6 +117,11 @@ Before commit or push your changes, please make sure the unittests are passed ,o
 pytest -n 16
 ```
 
+## Security and privacy
+
+This project takes security seriously.
+For vulnerability reporting and supported versions, see [SECURITY.md](SECURITY.md)
+
 ## Contact with us
 
 Join our discussion group by scanning the QR code below:

{veadk_python-0.5.31 → veadk_python-0.5.34}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "veadk-python"
-version = "0.5.31"
+version = "0.5.34"
 description = "Volcengine agent development kit, integrations with Volcengine cloud services."
 readme = "README.md"
 requires-python = ">=3.10"
@@ -13,9 +13,9 @@ authors = [
 ]
 dependencies = [
     "pydantic-settings==2.10.1", # Config management
-    "a2a-sdk>=0.3.7", # For Google Agent2Agent protocol
+    "a2a-sdk==0.3.7", # For Google Agent2Agent protocol
     "deprecated==1.2.18",
-    "google-adk>=1.19.0", # For basic agent architecture
+    "google-adk==1.19.0", # For basic agent architecture
     "litellm>=1.74.3,<=1.82.6", # For model inference
     "loguru==0.7.3", # For better logging
     "opentelemetry-exporter-otlp==1.37.0",
@@ -53,6 +53,7 @@ extensions = [
     "llama-index-vector-stores-redis>=0.6.1",               # For Redis database
     "llama-index-vector-stores-opensearch==0.6.1",          # For Opensearch database
     "opensearch-py==2.8.0",
+    "lark-oapi",
 ]
 database = [
     "redis>=5.0",                   # For Redis database
@@ -97,4 +98,4 @@ include-package-data = true
 exclude = [
     "veadk/integrations/ve_faas/template/*",
     "veadk/integrations/ve_faas/web_template/*"
-]
+]

{veadk_python-0.5.31 → veadk_python-0.5.34}/tests/test_agent.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 import os
-from unittest.mock import Mock, patch
+from unittest.mock import Mock, PropertyMock, patch
 
 from google.adk.agents.llm_agent import LlmAgent
 from google.adk.models.lite_llm import LiteLlm
@@ -76,19 +76,32 @@ def test_agent():
 
 @patch.dict("os.environ", {"MODEL_AGENT_API_KEY": "mock_api_key"})
 def test_agent_default_values():
-    agent = Agent()
-
-    assert agent.name == DEFAULT_AGENT_NAME
-
-    assert agent.model_name == DEFAULT_MODEL_AGENT_NAME
-    assert agent.model_provider == DEFAULT_MODEL_AGENT_PROVIDER
-    assert agent.model_api_base == DEFAULT_MODEL_AGENT_API_BASE
-
-    assert agent.tools == []
-    assert agent.sub_agents == []
-    assert agent.knowledgebase is None
-    assert agent.long_term_memory is None
-    # assert agent.tracers == []
+    with (
+        patch("veadk.agent.settings.model.name", new=DEFAULT_MODEL_AGENT_NAME),
+        patch("veadk.agent.settings.model.provider", new=DEFAULT_MODEL_AGENT_PROVIDER),
+        patch(
+            "veadk.agent.settings.model.api_base",
+            new=DEFAULT_MODEL_AGENT_API_BASE,
+        ),
+        patch(
+            "veadk.configs.model_configs.ModelConfig.api_key",
+            new_callable=PropertyMock,
+            return_value="mock_api_key",
+        ),
+    ):
+        agent = Agent()
+
+        assert agent.name == DEFAULT_AGENT_NAME
+
+        assert agent.model_name == DEFAULT_MODEL_AGENT_NAME
+        assert agent.model_provider == DEFAULT_MODEL_AGENT_PROVIDER
+        assert agent.model_api_base == DEFAULT_MODEL_AGENT_API_BASE
+
+        assert agent.tools == []
+        assert agent.sub_agents == []
+        assert agent.knowledgebase is None
+        assert agent.long_term_memory is None
+        # assert agent.tracers == []
 
 
 @patch.dict("os.environ", {"MODEL_AGENT_API_KEY": "mock_api_key"})

veadk_python-0.5.34/tests/test_feishu_channel_extension.py

@@ -0,0 +1,219 @@
+# 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 types import SimpleNamespace
+
+import pytest
+
+from veadk.extensions.feishu_channel import FeishuChannelExtension
+
+
+@pytest.fixture
+def anyio_backend():
+    return "asyncio"
+
+
+class FakeChannel:
+    def __init__(self):
+        self.handlers = {}
+        self.sent_messages = []
+
+    def on(self, event_name, handler):
+        self.handlers[event_name] = handler
+
+    async def send(self, chat_id, body, options=None):
+        self.sent_messages.append((chat_id, body, options))
+
+
+class FakeStreamController:
+    def __init__(self):
+        self.chunks = []
+
+    async def append(self, chunk):
+        self.chunks.append(chunk)
+
+
+class FakeStreamChannel(FakeChannel):
+    def __init__(self):
+        super().__init__()
+        self.stream_calls = []
+
+    async def stream(self, chat_id, spec, options=None):
+        controller = FakeStreamController()
+        await spec["markdown"](controller)
+        self.stream_calls.append((chat_id, controller.chunks, options))
+
+
+class FakeRunner:
+    def __init__(self):
+        self.calls = []
+
+    async def run(self, messages, user_id="", session_id="", **kwargs):
+        self.calls.append(
+            {
+                "messages": messages,
+                "user_id": user_id,
+                "session_id": session_id,
+            }
+        )
+        return f"echo:{messages}"
+
+
+class FakeStreamingMemory:
+    def __init__(self):
+        self.sessions = []
+        self.session_service = object()
+
+    async def create_session(self, app_name, user_id, session_id):
+        self.sessions.append(
+            {"app_name": app_name, "user_id": user_id, "session_id": session_id}
+        )
+        return True
+
+
+class FakeStreamingRunner:
+    def __init__(self):
+        self.app_name = "stream_app"
+        self.short_term_memory = FakeStreamingMemory()
+        self.run_async_calls = []
+
+    async def run_async(self, user_id, session_id, new_message, run_config=None):
+        self.run_async_calls.append(
+            {
+                "user_id": user_id,
+                "session_id": session_id,
+                "new_message": new_message,
+                "run_config": run_config,
+            }
+        )
+        yield SimpleNamespace(
+            content=SimpleNamespace(
+                parts=[
+                    SimpleNamespace(text="hel", thought=False),
+                    SimpleNamespace(text="thinking", thought=True),
+                ]
+            )
+        )
+        yield SimpleNamespace(
+            content=SimpleNamespace(parts=[SimpleNamespace(text="lo", thought=False)])
+        )
+
+
+def build_message(**overrides):
+    message = SimpleNamespace(
+        id="om_001",
+        message_id="om_001",
+        chat_id="oc_chat",
+        chat_type="p2p",
+        thread_id="",
+        reply_to_message_id="",
+        content_text="你好",
+        sender_id="ou_sender",
+        sender=SimpleNamespace(
+            union_id="on_union",
+            open_id="ou_sender",
+            user_id="u_sender",
+        ),
+        conversation=SimpleNamespace(
+            chat_id="oc_chat",
+            chat_type="p2p",
+            thread_id="",
+        ),
+        reply=SimpleNamespace(message_id=""),
+    )
+    for key, value in overrides.items():
+        setattr(message, key, value)
+    return message
+
+
+@pytest.mark.anyio
+async def test_extension_uses_union_id_and_thread_id():
+    runner = FakeRunner()
+    channel = FakeChannel()
+    extension = FeishuChannelExtension(runner=runner, channel=channel)
+
+    message = build_message(
+        thread_id="thread_1",
+        conversation=SimpleNamespace(
+            chat_id="oc_chat",
+            chat_type="group",
+            thread_id="thread_1",
+        ),
+    )
+
+    await extension._on_message(message)
+
+    assert runner.calls == [
+        {
+            "messages": "你好",
+            "user_id": "on_union",
+            "session_id": "thread_1",
+        }
+    ]
+    assert channel.sent_messages == [
+        ("oc_chat", {"text": "echo:你好"}, {"reply_to": "om_001"})
+    ]
+
+
+@pytest.mark.anyio
+async def test_extension_falls_back_to_chat_id_when_thread_missing():
+    runner = FakeRunner()
+    channel = FakeChannel()
+    extension = FeishuChannelExtension(runner=runner, channel=channel)
+
+    message = build_message(
+        sender=SimpleNamespace(union_id="", open_id="ou_fallback", user_id="u_sender")
+    )
+
+    await extension._on_message(message)
+
+    assert runner.calls[0]["user_id"] == "ou_fallback"
+    assert runner.calls[0]["session_id"] == "oc_chat"
+
+
+@pytest.mark.anyio
+async def test_extension_ignores_empty_message_by_default():
+    runner = FakeRunner()
+    channel = FakeChannel()
+    extension = FeishuChannelExtension(runner=runner, channel=channel)
+
+    message = build_message(content_text="   ")
+
+    await extension._on_message(message)
+
+    assert runner.calls == []
+    assert channel.sent_messages == []
+
+
+@pytest.mark.anyio
+async def test_extension_streaming_uses_markdown_producer_controller():
+    runner = FakeStreamingRunner()
+    channel = FakeStreamChannel()
+    extension = FeishuChannelExtension(
+        runner=runner,
+        channel=channel,
+        streaming=True,
+    )
+
+    await extension._on_message(build_message())
+
+    assert runner.short_term_memory.sessions == [
+        {
+            "app_name": "stream_app",
+            "user_id": "on_union",
+            "session_id": "oc_chat",
+        }
+    ]
+    assert len(runner.run_async_calls) == 1
+    assert channel.stream_calls == [("oc_chat", ["hel", "lo"], {"reply_to": "om_001"})]

veadk_python-0.5.34/veadk/community/langchain_ai/tools/execute_skills.py

@@ -0,0 +1,100 @@
+# 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 typing import List, Optional
+
+from langchain.tools import ToolRuntime, tool
+
+from veadk.tools.builtin_tools._agentkit import (
+    get_agentkit_account_id,
+    resolve_agentkit_tool_id,
+)
+from veadk.tools.builtin_tools.run_sandbox_agent import (
+    _format_execution_result,
+    _build_agent_command,
+    _build_agent_runner_code,
+)
+from veadk.utils.logger import get_logger
+from veadk.tools.builtin_tools._agentkit import (
+    get_agentkit_endpoint_config,
+    invoke_agentkit_run_code,
+)
+
+logger = get_logger(__name__)
+
+
+@tool
+def execute_skills(
+    workflow_prompt: str,
+    runtime: ToolRuntime,
+    skills: Optional[List[str]] = None,
+    timeout: int = 900,
+) -> str:
+    """execute skills in a code sandbox and return the output.
+    For C++ code, don't execute it directly, compile and execute via Python; write sources and object files to /tmp.
+
+    Args:
+        workflow_prompt (str): instruction of workflow
+        skills (Optional[List[str]]): The skills will be invoked
+        timeout (int, optional): The timeout in seconds for the code execution, less than or equal to 900. Defaults to 900.
+
+    Returns:
+        str: The output of the code execution.
+    """
+
+    tool_id = resolve_agentkit_tool_id("AGENTKIT_TOOL_ID_SKILLS")
+    service, region, host, _ = get_agentkit_endpoint_config()
+    logger.debug(f"tools endpoint: {host}")
+
+    session_id = runtime.session_id  # type: ignore
+    agent_name = runtime.context.agent_name  # type: ignore
+    user_id = runtime.context.user_id  # type: ignore
+    tool_user_session_id = agent_name + "_" + user_id + "_" + session_id
+    logger.debug(f"tool_user_session_id: {tool_user_session_id}")
+
+    logger.debug(
+        f"Execute skills in session_id={session_id}, tool_id={tool_id}, host={host}, service={service}, region={region}, timeout={timeout}"
+    )
+
+    cmd = _build_agent_command(workflow_prompt=workflow_prompt, skills=skills)
+    try:
+        account_id = get_agentkit_account_id()
+    except KeyError as e:
+        logger.error(f"Error occurred while getting account id: {e}")
+        return {"error": str(e)}
+
+    env_vars = {"TOOL_USER_SESSION_ID": tool_user_session_id}
+    if account_id:
+        env_vars["TOS_SKILLS_DIR"] = f"tos://agentkit-platform-{account_id}/skills/"
+
+    code = _build_agent_runner_code(
+        cmd=cmd,
+        timeout=timeout,
+        env_vars=env_vars,
+    )
+
+    res = invoke_agentkit_run_code(
+        tool_id=tool_id,
+        tool_user_session_id=tool_user_session_id,
+        code=code,
+        timeout=timeout,
+        kernel_name="python3",
+    )
+    logger.debug(f"Invoke run code response: {res}")
+
+    try:
+        return _format_execution_result(res["Result"]["Result"])
+    except KeyError as e:
+        logger.error(f"Error occurred while running code: {e}, response is {res}")
+        return res

{veadk_python-0.5.31 → veadk_python-0.5.34}/veadk/community/langchain_ai/tools/run_code.py

@@ -12,15 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import json
-import os
-
 from langchain.tools import ToolRuntime, tool
 
-from veadk.auth.veauth.utils import get_credential_from_vefaas_iam
-from veadk.config import getenv
+from veadk.tools.builtin_tools._agentkit import (
+    get_agentkit_endpoint_config,
+    invoke_agentkit_run_code,
+    resolve_agentkit_tool_id,
+)
 from veadk.utils.logger import get_logger
-from veadk.utils.volcengine_sign import ve_request
 
 logger = get_logger(__name__)
 
@@ -39,18 +38,8 @@ def run_code(code: str, language: str, runtime: ToolRuntime, timeout: int = 30)
         str: The output of the code execution.
     """
 
-    tool_id = getenv("AGENTKIT_TOOL_ID")
-
-    service = getenv(
-        "AGENTKIT_TOOL_SERVICE_CODE", "agentkit"
-    )  # temporary service for code run tool
-    region = getenv("AGENTKIT_TOOL_REGION", "cn-beijing")
-    host = getenv(
-        "AGENTKIT_TOOL_HOST", service + "." + region + ".volces.com"
-    )  # temporary host for code run tool
-    scheme = os.getenv("AGENTKIT_TOOL_SCHEME", "https").lower()
-    if scheme not in {"http", "https"}:
-        scheme = "https"
+    tool_id = resolve_agentkit_tool_id("AGENTKIT_TOOL_ID_SCRIPT")
+    service, region, host, _ = get_agentkit_endpoint_config()
     logger.debug(f"tools endpoint: {host}")
 
     session_id = runtime.context.session_id  # type: ignore
@@ -64,44 +53,12 @@ def run_code(code: str, language: str, runtime: ToolRuntime, timeout: int = 30)
         f"Running code in language: {language}, session_id={session_id}, code={code}, tool_id={tool_id}, host={host}, service={service}, region={region}, timeout={timeout}"
     )
 
-    header = {}
-
-    logger.debug("Get AK/SK from tool context failed.")
-    ak = os.getenv("VOLCENGINE_ACCESS_KEY")
-    sk = os.getenv("VOLCENGINE_SECRET_KEY")
-    if not (ak and sk):
-        logger.debug(
-            "Get AK/SK from environment variables failed. Try to use credential from Iam."
-        )
-        credential = get_credential_from_vefaas_iam()
-        ak = credential.access_key_id
-        sk = credential.secret_access_key
-        header = {"X-Security-Token": credential.session_token}
-    else:
-        logger.debug("Successfully get AK/SK from environment variables.")
-
-    res = ve_request(
-        request_body={
-            "ToolId": tool_id,
-            "UserSessionId": tool_user_session_id,
-            "OperationType": "RunCode",
-            "OperationPayload": json.dumps(
-                {
-                    "code": code,
-                    "timeout": timeout,
-                    "kernel_name": language,
-                }
-            ),
-        },
-        action="InvokeTool",
-        ak=ak,
-        sk=sk,
-        service=service,
-        version="2025-10-30",
-        region=region,
-        host=host,
-        header=header,
-        scheme=scheme,  # type: ignore
+    res = invoke_agentkit_run_code(
+        tool_id=tool_id,
+        tool_user_session_id=tool_user_session_id,
+        code=code,
+        timeout=timeout,
+        kernel_name=language,
     )
     logger.debug(f"Invoke run code response: {res}")
 

veadk_python-0.5.34/veadk/extensions/__init__.py

@@ -0,0 +1,17 @@
+# 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 veadk.extensions.feishu_channel import FeishuChannelExtension
+
+__all__ = ["FeishuChannelExtension"]