agentrun-inner-test 0.0.46__py3-none-any.whl → 0.0.56__py3-none-any.whl

agentrun/__init__.py

@@ -18,7 +18,7 @@ Provides simple and easy-to-use APIs for managing AI Agent runtime environments,
 
 from typing import TYPE_CHECKING
 
-__version__ = "0.0.46"
+__version__ = "0.0.56"
 
 # Agent Runtime
 from agentrun.agent_runtime import (

agentrun/agent_runtime/api/control.py

@@ -35,7 +35,7 @@ from alibabacloud_agentrun20250910.models import (
 )
 from alibabacloud_tea_openapi.exceptions._client import ClientException
 from alibabacloud_tea_openapi.exceptions._server import ServerException
-from alibabacloud_tea_util.models import RuntimeOptions
+from darabonba.runtime import RuntimeOptions
 import pydash
 
 from agentrun.utils.config import Config

agentrun/credential/api/control.py

@@ -25,7 +25,7 @@ from alibabacloud_agentrun20250910.models import (
 )
 from alibabacloud_tea_openapi.exceptions._client import ClientException
 from alibabacloud_tea_openapi.exceptions._server import ServerException
-from alibabacloud_tea_util.models import RuntimeOptions
+from darabonba.runtime import RuntimeOptions
 import pydash
 
 from agentrun.utils.config import Config

agentrun/integration/agentscope/__init__.py

@@ -3,10 +3,11 @@
 提供 AgentRun 模型与沙箱工具的 AgentScope 适配入口。 / 提供 AgentRun 模型with沙箱工具的 AgentScope 适配入口。
 """
 
-from .builtin import model, sandbox_toolset, toolset
+from .builtin import knowledgebase_toolset, model, sandbox_toolset, toolset
 
 __all__ = [
     "model",
     "toolset",
     "sandbox_toolset",
+    "knowledgebase_toolset",
 ]

agentrun/integration/agentscope/builtin.py

@@ -8,6 +8,9 @@ from typing import Any, Callable, List, Optional, Union
 
 from typing_extensions import Unpack
 
+from agentrun.integration.builtin import (
+    knowledgebase_toolset as _knowledgebase_toolset,
+)
 from agentrun.integration.builtin import model as _model
 from agentrun.integration.builtin import ModelArgs
 from agentrun.integration.builtin import sandbox_toolset as _sandbox_toolset
@@ -63,3 +66,23 @@ def sandbox_toolset(
         config=config,
         sandbox_idle_timeout_seconds=sandbox_idle_timeout_seconds,
     ).to_agentscope(prefix=prefix)
+
+
+def knowledgebase_toolset(
+    knowledge_base_names: List[str],
+    *,
+    prefix: Optional[str] = None,
+    modify_tool_name: Optional[Callable[[Tool], Tool]] = None,
+    filter_tools_by_name: Optional[Callable[[str], bool]] = None,
+    config: Optional[Config] = None,
+) -> List[Any]:
+    """将知识库检索封装为 AgentScope 工具列表。 / AgentScope Built-in Integration Functions"""
+
+    return _knowledgebase_toolset(
+        knowledge_base_names=knowledge_base_names,
+        config=config,
+    ).to_agentscope(
+        prefix=prefix,
+        modify_tool_name=modify_tool_name,
+        filter_tools_by_name=filter_tools_by_name,
+    )

agentrun/integration/builtin/__init__.py

@@ -4,6 +4,7 @@
 This module provides built-in integration functions for quickly creating models and tools.
 """
 
+from .knowledgebase import knowledgebase_toolset
 from .model import model, ModelArgs
 from .sandbox import sandbox_toolset
 from .toolset import toolset
@@ -13,4 +14,5 @@ __all__ = [
     "ModelArgs",
     "toolset",
     "sandbox_toolset",
+    "knowledgebase_toolset",
 ]

agentrun/integration/builtin/knowledgebase.py

@@ -0,0 +1,137 @@
+"""知识库工具集 / KnowledgeBase ToolSet
+
+提供知识库检索功能的工具集，支持多知识库联合检索。
+Provides toolset for knowledge base retrieval, supporting multi-knowledge-base search.
+"""
+
+from __future__ import annotations
+
+from typing import Any, Dict, List, Optional
+
+from agentrun.integration.utils.tool import CommonToolSet, tool
+from agentrun.knowledgebase import KnowledgeBase
+from agentrun.utils.config import Config
+
+
+class KnowledgeBaseToolSet(CommonToolSet):
+    """知识库工具集 / KnowledgeBase ToolSet
+
+    提供知识库检索功能，支持对多个知识库进行联合检索。
+    Provides knowledge base retrieval capabilities, supporting joint retrieval
+    across multiple knowledge bases.
+
+    使用指南 / Usage Guide:
+    ============================================================
+
+    ## 基本用法 / Basic Usage
+
+    1. **创建工具集 / Create ToolSet**:
+       - 使用 `knowledgebase_toolset` 函数创建工具集实例
+       - Use `knowledgebase_toolset` function to create a toolset instance
+       - 指定要检索的知识库名称列表
+       - Specify the list of knowledge base names to search
+
+    2. **执行检索 / Execute Search**:
+       - 调用 `search_document` 工具进行检索
+       - Call `search_document` tool to perform retrieval
+       - 返回所有指定知识库的检索结果
+       - Returns retrieval results from all specified knowledge bases
+
+    ## 示例 / Examples
+
+    ```python
+    from agentrun.integration.langchain import knowledgebase_toolset
+
+    # 创建工具集 / Create toolset
+    tools = knowledgebase_toolset(
+        knowledge_base_names=["kb-product-docs", "kb-faq"],
+    )
+
+    # 在 Agent 中使用 / Use in Agent
+    agent = create_react_agent(llm, tools)
+    ```
+    """
+
+    def __init__(
+        self,
+        knowledge_base_names: List[str],
+        config: Optional[Config] = None,
+    ) -> None:
+        """初始化知识库工具集 / Initialize KnowledgeBase ToolSet
+
+        Args:
+            knowledge_base_names: 知识库名称列表 / List of knowledge base names
+            config: 配置 / Configuration
+        """
+        super().__init__()
+
+        self.knowledge_base_names = knowledge_base_names
+        self.config = config
+
+    @tool(
+        name="search_document",
+        description=(
+            "Search and retrieve relevant documents from configured knowledge"
+            " bases. Use this tool when you need to find information from the"
+            " knowledge base to answer user questions. Returns relevant"
+            " document chunks with their content and metadata. The search is"
+            " performed across all configured knowledge bases and results are"
+            " grouped by knowledge base name."
+        ),
+    )
+    def search_document(self, query: str) -> Dict[str, Any]:
+        """检索文档 / Search documents
+
+        根据查询文本从配置的知识库中检索相关文档。
+        Retrieves relevant documents from configured knowledge bases based on query text.
+
+        Args:
+            query: 查询文本 / Query text
+
+        Returns:
+            Dict[str, Any]: 检索结果，包含各知识库的检索结果 /
+                           Retrieval results containing results from each knowledge base
+        """
+        return KnowledgeBase.multi_retrieve(
+            query=query,
+            knowledge_base_names=self.knowledge_base_names,
+            config=self.config,
+        )
+
+
+def knowledgebase_toolset(
+    knowledge_base_names: List[str],
+    *,
+    config: Optional[Config] = None,
+) -> KnowledgeBaseToolSet:
+    """创建知识库工具集 / Create KnowledgeBase ToolSet
+
+    将知识库检索功能封装为通用工具集，可转换为各框架支持的格式。
+    Wraps knowledge base retrieval functionality into a common toolset that can be
+    converted to formats supported by various frameworks.
+
+    Args:
+        knowledge_base_names: 知识库名称列表 / List of knowledge base names
+        config: 配置 / Configuration
+
+    Returns:
+        KnowledgeBaseToolSet: 知识库工具集实例 / KnowledgeBase toolset instance
+
+    Example:
+        >>> from agentrun.integration.builtin import knowledgebase_toolset
+        >>>
+        >>> # 创建工具集 / Create toolset
+        >>> kb_tools = knowledgebase_toolset(
+        ...     knowledge_base_names=["kb-docs", "kb-faq"],
+        ... )
+        >>>
+        >>> # 转换为 LangChain 格式 / Convert to LangChain format
+        >>> langchain_tools = kb_tools.to_langchain()
+        >>>
+        >>> # 转换为 Google ADK 格式 / Convert to Google ADK format
+        >>> adk_tools = kb_tools.to_google_adk()
+    """
+    return KnowledgeBaseToolSet(
+        knowledge_base_names=knowledge_base_names,
+        config=config,
+    )

agentrun/integration/crewai/__init__.py

@@ -4,9 +4,10 @@
 CrewAI 与 LangChain 兼容，因此直接复用 LangChain 的转换逻辑。 / CrewAI with LangChain 兼容，因此直接复用 LangChain 的转换逻辑。
 """
 
-from .builtin import model, sandbox_toolset
+from .builtin import knowledgebase_toolset, model, sandbox_toolset
 
 __all__ = [
     "model",
     "sandbox_toolset",
+    "knowledgebase_toolset",
 ]

agentrun/integration/crewai/builtin.py

@@ -8,6 +8,9 @@ from typing import Any, Callable, List, Optional, Union
 
 from typing_extensions import Unpack
 
+from agentrun.integration.builtin import (
+    knowledgebase_toolset as _knowledgebase_toolset,
+)
 from agentrun.integration.builtin import model as _model
 from agentrun.integration.builtin import ModelArgs
 from agentrun.integration.builtin import sandbox_toolset as _sandbox_toolset
@@ -63,3 +66,23 @@ def sandbox_toolset(
         config=config,
         sandbox_idle_timeout_seconds=sandbox_idle_timeout_seconds,
     ).to_crewai(prefix=prefix)
+
+
+def knowledgebase_toolset(
+    knowledge_base_names: List[str],
+    *,
+    prefix: Optional[str] = None,
+    modify_tool_name: Optional[Callable[[Tool], Tool]] = None,
+    filter_tools_by_name: Optional[Callable[[str], bool]] = None,
+    config: Optional[Config] = None,
+) -> List[Any]:
+    """将知识库检索封装为 CrewAI 工具列表。 / CrewAI Built-in Integration Functions"""
+
+    return _knowledgebase_toolset(
+        knowledge_base_names=knowledge_base_names,
+        config=config,
+    ).to_crewai(
+        prefix=prefix,
+        modify_tool_name=modify_tool_name,
+        filter_tools_by_name=filter_tools_by_name,
+    )

agentrun/integration/google_adk/__init__.py

@@ -3,10 +3,11 @@
 提供与 Google Agent Development Kit 的模型与沙箱工具集成。 / 提供with Google Agent Development Kit 的模型with沙箱工具集成。
 """
 
-from .builtin import model, sandbox_toolset, toolset
+from .builtin import knowledgebase_toolset, model, sandbox_toolset, toolset
 
 __all__ = [
     "model",
     "toolset",
     "sandbox_toolset",
+    "knowledgebase_toolset",
 ]

agentrun/integration/google_adk/builtin.py

@@ -8,6 +8,9 @@ from typing import Any, Callable, List, Optional, Union
 
 from typing_extensions import Unpack
 
+from agentrun.integration.builtin import (
+    knowledgebase_toolset as _knowledgebase_toolset,
+)
 from agentrun.integration.builtin import model as _model
 from agentrun.integration.builtin import ModelArgs
 from agentrun.integration.builtin import sandbox_toolset as _sandbox_toolset
@@ -63,3 +66,23 @@ def sandbox_toolset(
         config=config,
         sandbox_idle_timeout_seconds=sandbox_idle_timeout_seconds,
     ).to_google_adk(prefix=prefix)
+
+
+def knowledgebase_toolset(
+    knowledge_base_names: List[str],
+    *,
+    prefix: Optional[str] = None,
+    modify_tool_name: Optional[Callable[[Tool], Tool]] = None,
+    filter_tools_by_name: Optional[Callable[[str], bool]] = None,
+    config: Optional[Config] = None,
+) -> List[Any]:
+    """将知识库检索封装为 Google ADK 工具列表。 / Google ADK Built-in Integration Functions"""
+
+    return _knowledgebase_toolset(
+        knowledge_base_names=knowledge_base_names,
+        config=config,
+    ).to_google_adk(
+        prefix=prefix,
+        modify_tool_name=modify_tool_name,
+        filter_tools_by_name=filter_tools_by_name,
+    )

agentrun/integration/langchain/__init__.py

@@ -20,11 +20,12 @@ from agentrun.integration.langgraph.agent_converter import (
     AgentRunConverter,
 )  # 向后兼容
 
-from .builtin import model, sandbox_toolset, toolset
+from .builtin import knowledgebase_toolset, model, sandbox_toolset, toolset
 
 __all__ = [
     "AgentRunConverter",
     "model",
     "toolset",
     "sandbox_toolset",
+    "knowledgebase_toolset",
 ]

agentrun/integration/langchain/builtin.py

@@ -8,6 +8,9 @@ from typing import Any, Callable, List, Optional, Union
 
 from typing_extensions import Unpack
 
+from agentrun.integration.builtin import (
+    knowledgebase_toolset as _knowledgebase_toolset,
+)
 from agentrun.integration.builtin import model as _model
 from agentrun.integration.builtin import ModelArgs
 from agentrun.integration.builtin import sandbox_toolset as _sandbox_toolset
@@ -69,3 +72,23 @@ def sandbox_toolset(
         modify_tool_name=modify_tool_name,
         filter_tools_by_name=filter_tools_by_name,
     )
+
+
+def knowledgebase_toolset(
+    knowledge_base_names: List[str],
+    *,
+    prefix: Optional[str] = None,
+    modify_tool_name: Optional[Callable[[Tool], Tool]] = None,
+    filter_tools_by_name: Optional[Callable[[str], bool]] = None,
+    config: Optional[Config] = None,
+) -> List[Any]:
+    """将知识库检索封装为 LangChain ``StructuredTool`` 列表。 / LangChain Built-in Integration Functions"""
+
+    return _knowledgebase_toolset(
+        knowledge_base_names=knowledge_base_names,
+        config=config,
+    ).to_langchain(
+        prefix=prefix,
+        modify_tool_name=modify_tool_name,
+        filter_tools_by_name=filter_tools_by_name,
+    )

agentrun/integration/langgraph/__init__.py

@@ -25,11 +25,12 @@
 """
 
 from .agent_converter import AgentRunConverter
-from .builtin import model, sandbox_toolset, toolset
+from .builtin import knowledgebase_toolset, model, sandbox_toolset, toolset
 
 __all__ = [
     "AgentRunConverter",
     "model",
     "toolset",
     "sandbox_toolset",
+    "knowledgebase_toolset",
 ]

agentrun/integration/langgraph/builtin.py

@@ -8,6 +8,9 @@ from typing import Any, Callable, List, Optional, Union
 
 from typing_extensions import Unpack
 
+from agentrun.integration.builtin import (
+    knowledgebase_toolset as _knowledgebase_toolset,
+)
 from agentrun.integration.builtin import model as _model
 from agentrun.integration.builtin import ModelArgs
 from agentrun.integration.builtin import sandbox_toolset as _sandbox_toolset
@@ -63,3 +66,23 @@ def sandbox_toolset(
         config=config,
         sandbox_idle_timeout_seconds=sandbox_idle_timeout_seconds,
     ).to_langgraph(prefix=prefix)
+
+
+def knowledgebase_toolset(
+    knowledge_base_names: List[str],
+    *,
+    prefix: Optional[str] = None,
+    modify_tool_name: Optional[Callable[[Tool], Tool]] = None,
+    filter_tools_by_name: Optional[Callable[[str], bool]] = None,
+    config: Optional[Config] = None,
+) -> List[Any]:
+    """将知识库检索封装为 LangGraph ``StructuredTool`` 列表。 / LangGraph Built-in Integration Functions"""
+
+    return _knowledgebase_toolset(
+        knowledge_base_names=knowledge_base_names,
+        config=config,
+    ).to_langgraph(
+        prefix=prefix,
+        modify_tool_name=modify_tool_name,
+        filter_tools_by_name=filter_tools_by_name,
+    )

agentrun/integration/pydantic_ai/__init__.py

@@ -3,10 +3,11 @@
 提供 AgentRun 模型与沙箱工具的 PydanticAI 适配入口。 / 提供 AgentRun 模型with沙箱工具的 PydanticAI 适配入口。
 """
 
-from .builtin import model, sandbox_toolset, toolset
+from .builtin import knowledgebase_toolset, model, sandbox_toolset, toolset
 
 __all__ = [
     "model",
     "toolset",
     "sandbox_toolset",
+    "knowledgebase_toolset",
 ]

agentrun/integration/pydantic_ai/builtin.py

@@ -8,6 +8,9 @@ from typing import Any, Callable, List, Optional, Union
 
 from typing_extensions import Unpack
 
+from agentrun.integration.builtin import (
+    knowledgebase_toolset as _knowledgebase_toolset,
+)
 from agentrun.integration.builtin import model as _model
 from agentrun.integration.builtin import ModelArgs
 from agentrun.integration.builtin import sandbox_toolset as _sandbox_toolset
@@ -63,3 +66,23 @@ def sandbox_toolset(
         config=config,
         sandbox_idle_timeout_seconds=sandbox_idle_timeout_seconds,
     ).to_pydantic_ai(prefix=prefix)
+
+
+def knowledgebase_toolset(
+    knowledge_base_names: List[str],
+    *,
+    prefix: Optional[str] = None,
+    modify_tool_name: Optional[Callable[[Tool], Tool]] = None,
+    filter_tools_by_name: Optional[Callable[[str], bool]] = None,
+    config: Optional[Config] = None,
+) -> List[Any]:
+    """将知识库检索封装为 PydanticAI 工具列表。 / PydanticAI Built-in Integration Functions"""
+
+    return _knowledgebase_toolset(
+        knowledge_base_names=knowledge_base_names,
+        config=config,
+    ).to_pydantic_ai(
+        prefix=prefix,
+        modify_tool_name=modify_tool_name,
+        filter_tools_by_name=filter_tools_by_name,
+    )

agentrun/knowledgebase/__client_async_template.py

@@ -0,0 +1,173 @@
+"""KnowledgeBase 客户端 / KnowledgeBase Client
+
+此模块提供知识库管理的客户端API。
+This module provides the client API for knowledge base management.
+"""
+
+from typing import Optional
+
+from alibabacloud_agentrun20250910.models import (
+    CreateKnowledgeBaseInput,
+    ListKnowledgeBasesRequest,
+    UpdateKnowledgeBaseInput,
+)
+
+from agentrun.utils.config import Config
+from agentrun.utils.exception import HTTPError
+
+from .api.control import KnowledgeBaseControlAPI
+from .knowledgebase import KnowledgeBase
+from .model import (
+    KnowledgeBaseCreateInput,
+    KnowledgeBaseListInput,
+    KnowledgeBaseListOutput,
+    KnowledgeBaseUpdateInput,
+)
+
+
+class KnowledgeBaseClient:
+    """KnowledgeBase 客户端 / KnowledgeBase Client
+
+    提供知识库的创建、删除、更新和查询功能。
+    Provides create, delete, update and query functions for knowledge bases.
+    """
+
+    def __init__(self, config: Optional[Config] = None):
+        """初始化客户端 / Initialize client
+
+        Args:
+            config: 配置对象,可选 / Configuration object, optional
+        """
+        self.__control_api = KnowledgeBaseControlAPI(config)
+
+    async def create_async(
+        self, input: KnowledgeBaseCreateInput, config: Optional[Config] = None
+    ):
+        """创建知识库(异步) / Create knowledge base asynchronously
+
+        Args:
+            input: 知识库输入参数 / KnowledgeBase input parameters
+            config: 配置对象,可选 / Configuration object, optional
+
+        Returns:
+            KnowledgeBase: 创建的知识库对象 / Created knowledge base object
+
+        Raises:
+            ResourceAlreadyExistError: 资源已存在 / Resource already exists
+            HTTPError: HTTP 请求错误 / HTTP request error
+        """
+        try:
+            result = await self.__control_api.create_knowledge_base_async(
+                CreateKnowledgeBaseInput().from_map(input.model_dump()),
+                config=config,
+            )
+
+            return KnowledgeBase.from_inner_object(result)
+        except HTTPError as e:
+            raise e.to_resource_error(
+                "KnowledgeBase", input.knowledge_base_name
+            ) from e
+
+    async def delete_async(
+        self, knowledge_base_name: str, config: Optional[Config] = None
+    ):
+        """删除知识库（异步）/ Delete knowledge base asynchronously
+
+        Args:
+            knowledge_base_name: 知识库名称 / KnowledgeBase name
+            config: 配置 / Configuration
+
+        Raises:
+            ResourceNotExistError: 知识库不存在 / KnowledgeBase not found
+        """
+        try:
+            result = await self.__control_api.delete_knowledge_base_async(
+                knowledge_base_name, config=config
+            )
+
+            return KnowledgeBase.from_inner_object(result)
+
+        except HTTPError as e:
+            raise e.to_resource_error(
+                "KnowledgeBase", knowledge_base_name
+            ) from e
+
+    async def update_async(
+        self,
+        knowledge_base_name: str,
+        input: KnowledgeBaseUpdateInput,
+        config: Optional[Config] = None,
+    ):
+        """更新知识库（异步）/ Update knowledge base asynchronously
+
+        Args:
+            knowledge_base_name: 知识库名称 / KnowledgeBase name
+            input: 知识库更新输入参数 / KnowledgeBase update input parameters
+            config: 配置 / Configuration
+
+        Returns:
+            KnowledgeBase: 更新后的知识库对象 / Updated knowledge base object
+
+        Raises:
+            ResourceNotExistError: 知识库不存在 / KnowledgeBase not found
+        """
+        try:
+            result = await self.__control_api.update_knowledge_base_async(
+                knowledge_base_name,
+                UpdateKnowledgeBaseInput().from_map(input.model_dump()),
+                config=config,
+            )
+
+            return KnowledgeBase.from_inner_object(result)
+        except HTTPError as e:
+            raise e.to_resource_error(
+                "KnowledgeBase", knowledge_base_name
+            ) from e
+
+    async def get_async(
+        self, knowledge_base_name: str, config: Optional[Config] = None
+    ):
+        """获取知识库（异步）/ Get knowledge base asynchronously
+
+        Args:
+            knowledge_base_name: 知识库名称 / KnowledgeBase name
+            config: 配置 / Configuration
+
+        Returns:
+            KnowledgeBase: 知识库对象 / KnowledgeBase object
+
+        Raises:
+            ResourceNotExistError: 知识库不存在 / KnowledgeBase not found
+        """
+        try:
+            result = await self.__control_api.get_knowledge_base_async(
+                knowledge_base_name, config=config
+            )
+            return KnowledgeBase.from_inner_object(result)
+        except HTTPError as e:
+            raise e.to_resource_error(
+                "KnowledgeBase", knowledge_base_name
+            ) from e
+
+    async def list_async(
+        self,
+        input: Optional[KnowledgeBaseListInput] = None,
+        config: Optional[Config] = None,
+    ):
+        """列出知识库（异步）/ List knowledge bases asynchronously
+
+        Args:
+            input: 分页查询参数 / Pagination query parameters
+            config: 配置 / Configuration
+
+        Returns:
+            List[KnowledgeBaseListOutput]: 知识库列表 / KnowledgeBase list
+        """
+        if input is None:
+            input = KnowledgeBaseListInput()
+
+        results = await self.__control_api.list_knowledge_bases_async(
+            ListKnowledgeBasesRequest().from_map(input.model_dump()),
+            config=config,
+        )
+        return [KnowledgeBaseListOutput.from_inner_object(item) for item in results.items]  # type: ignore