agentrun-inner-test 0.0.40__tar.gz → 0.0.64__tar.gz

{agentrun_inner_test-0.0.40/agentrun_inner_test.egg-info → agentrun_inner_test-0.0.64}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentrun-inner-test
-Version: 0.0.40
+Version: 0.0.64
 Summary: Alibaba Cloud Agent Run SDK
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
@@ -13,8 +13,10 @@ Requires-Dist: typing-extensions>=4.15.0
 Requires-Dist: litellm>=1.79.3
 Requires-Dist: alibabacloud-devs20230714>=2.4.1
 Requires-Dist: pydash>=8.0.5
-Requires-Dist: alibabacloud-agentrun20250910>=4.0.3
+Requires-Dist: alibabacloud-agentrun20250910>=5.2.0
 Requires-Dist: alibabacloud_tea_openapi>=0.4.2
+Requires-Dist: alibabacloud_bailian20231229>=2.6.2
+Requires-Dist: agentrun-mem0ai>=0.0.6
 Provides-Extra: server
 Requires-Dist: fastapi>=0.104.0; extra == "server"
 Requires-Dist: uvicorn>=0.24.0; extra == "server"

{agentrun_inner_test-0.0.40 → agentrun_inner_test-0.0.64}/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.40"
+__version__ = "0.0.64"
 
 # Agent Runtime
 from agentrun.agent_runtime import (
@@ -55,6 +55,22 @@ from agentrun.credential import (
     CredentialUpdateInput,
     RelatedResource,
 )
+# Memory Collection
+from agentrun.memory_collection import (
+    EmbedderConfig,
+    EmbedderConfigConfig,
+    LLMConfig,
+    LLMConfigConfig,
+    MemoryCollection,
+    MemoryCollectionClient,
+    MemoryCollectionCreateInput,
+    MemoryCollectionListInput,
+    MemoryCollectionListOutput,
+    MemoryCollectionUpdateInput,
+    NetworkConfiguration,
+    VectorStoreConfig,
+    VectorStoreConfigConfig,
+)
 # Model Service
 from agentrun.model import (
     BackendType,
@@ -173,6 +189,23 @@ __all__ = [
     "CredentialCreateInput",
     "CredentialUpdateInput",
     "CredentialListInput",
+    ######## Memory Collection ########
+    # base
+    "MemoryCollection",
+    "MemoryCollectionClient",
+    # inner model
+    "EmbedderConfig",
+    "EmbedderConfigConfig",
+    "LLMConfig",
+    "LLMConfigConfig",
+    "NetworkConfiguration",
+    "VectorStoreConfig",
+    "VectorStoreConfigConfig",
+    # api model
+    "MemoryCollectionCreateInput",
+    "MemoryCollectionUpdateInput",
+    "MemoryCollectionListInput",
+    "MemoryCollectionListOutput",
     ######## Model ########
     # base
     "ModelClient",

{agentrun_inner_test-0.0.40 → agentrun_inner_test-0.0.64}/agentrun/agent_runtime/__endpoint_async_template.py

@@ -18,6 +18,7 @@ from agentrun.agent_runtime.model import (
     AgentRuntimeEndpointUpdateInput,
 )
 from agentrun.utils.config import Config
+from agentrun.utils.model import PageableInput
 from agentrun.utils.resource import ResourceBase
 
 
@@ -160,6 +161,45 @@ class AgentRuntimeEndpoint(
             config=config,
         )
 
+    @classmethod
+    async def _list_page_async(
+        cls,
+        page_input: PageableInput,
+        config: Optional[Config] = None,
+        **kwargs,
+    ) -> List["AgentRuntimeEndpoint"]:
+        """分页列出端点 / List endpoints by page
+
+        此方法是 ResourceBase 要求实现的抽象方法，用于支持分页查询。
+        This method is an abstract method required by ResourceBase to support pagination.
+
+        Args:
+            page_input: 分页参数 / Pagination parameters
+            config: 配置对象,可选 / Configuration object, optional
+            **kwargs: 其他参数，必须包含 agent_runtime_id / Other parameters, must include agent_runtime_id
+
+        Returns:
+            List[AgentRuntimeEndpoint]: 端点对象列表 / List of endpoint objects
+
+        Raises:
+            ValueError: 当 agent_runtime_id 未提供时 / When agent_runtime_id is not provided
+            HTTPError: HTTP 请求错误 / HTTP request error
+        """
+        agent_runtime_id = kwargs.get("agent_runtime_id")
+        if not agent_runtime_id:
+            raise ValueError(
+                "agent_runtime_id is required for listing endpoints"
+            )
+
+        return await cls.__get_client().list_endpoints_async(
+            agent_runtime_id,
+            AgentRuntimeEndpointListInput(
+                page_number=page_input.page_number,
+                page_size=page_input.page_size,
+            ),
+            config=config,
+        )
+
     @classmethod
     async def list_by_id_async(
         cls, agent_runtime_id: str, config: Optional[Config] = None

{agentrun_inner_test-0.0.40 → agentrun_inner_test-0.0.64}/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_inner_test-0.0.40 → agentrun_inner_test-0.0.64}/agentrun/agent_runtime/endpoint.py

@@ -28,6 +28,7 @@ from agentrun.agent_runtime.model import (
     AgentRuntimeEndpointUpdateInput,
 )
 from agentrun.utils.config import Config
+from agentrun.utils.model import PageableInput
 from agentrun.utils.resource import ResourceBase
 
 
@@ -286,6 +287,84 @@ class AgentRuntimeEndpoint(
             config=config,
         )
 
+    @classmethod
+    async def _list_page_async(
+        cls,
+        page_input: PageableInput,
+        config: Optional[Config] = None,
+        **kwargs,
+    ) -> List["AgentRuntimeEndpoint"]:
+        """分页列出端点 / List endpoints by page
+
+        此方法是 ResourceBase 要求实现的抽象方法，用于支持分页查询。
+        This method is an abstract method required by ResourceBase to support pagination.
+
+        Args:
+            page_input: 分页参数 / Pagination parameters
+            config: 配置对象,可选 / Configuration object, optional
+            **kwargs: 其他参数，必须包含 agent_runtime_id / Other parameters, must include agent_runtime_id
+
+        Returns:
+            List[AgentRuntimeEndpoint]: 端点对象列表 / List of endpoint objects
+
+        Raises:
+            ValueError: 当 agent_runtime_id 未提供时 / When agent_runtime_id is not provided
+            HTTPError: HTTP 请求错误 / HTTP request error
+        """
+        agent_runtime_id = kwargs.get("agent_runtime_id")
+        if not agent_runtime_id:
+            raise ValueError(
+                "agent_runtime_id is required for listing endpoints"
+            )
+
+        return await cls.__get_client().list_endpoints_async(
+            agent_runtime_id,
+            AgentRuntimeEndpointListInput(
+                page_number=page_input.page_number,
+                page_size=page_input.page_size,
+            ),
+            config=config,
+        )
+
+    @classmethod
+    def _list_page(
+        cls,
+        page_input: PageableInput,
+        config: Optional[Config] = None,
+        **kwargs,
+    ) -> List["AgentRuntimeEndpoint"]:
+        """分页列出端点 / List endpoints by page
+
+        此方法是 ResourceBase 要求实现的抽象方法，用于支持分页查询。
+        This method is an abstract method required by ResourceBase to support pagination.
+
+        Args:
+            page_input: 分页参数 / Pagination parameters
+            config: 配置对象,可选 / Configuration object, optional
+            **kwargs: 其他参数，必须包含 agent_runtime_id / Other parameters, must include agent_runtime_id
+
+        Returns:
+            List[AgentRuntimeEndpoint]: 端点对象列表 / List of endpoint objects
+
+        Raises:
+            ValueError: 当 agent_runtime_id 未提供时 / When agent_runtime_id is not provided
+            HTTPError: HTTP 请求错误 / HTTP request error
+        """
+        agent_runtime_id = kwargs.get("agent_runtime_id")
+        if not agent_runtime_id:
+            raise ValueError(
+                "agent_runtime_id is required for listing endpoints"
+            )
+
+        return cls.__get_client().list_endpoints(
+            agent_runtime_id,
+            AgentRuntimeEndpointListInput(
+                page_number=page_input.page_number,
+                page_size=page_input.page_size,
+            ),
+            config=config,
+        )
+
     @classmethod
     async def list_by_id_async(
         cls, agent_runtime_id: str, config: Optional[Config] = None

{agentrun_inner_test-0.0.40 → agentrun_inner_test-0.0.64}/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_inner_test-0.0.40 → agentrun_inner_test-0.0.64}/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_inner_test-0.0.40 → agentrun_inner_test-0.0.64}/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_inner_test-0.0.40 → agentrun_inner_test-0.0.64}/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_inner_test-0.0.64/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_inner_test-0.0.40 → agentrun_inner_test-0.0.64}/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_inner_test-0.0.40 → agentrun_inner_test-0.0.64}/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_inner_test-0.0.40 → agentrun_inner_test-0.0.64}/agentrun/integration/crewai/model_adapter.py

@@ -17,11 +17,15 @@ class CrewAIModelAdapter(ModelAdapter):
         from crewai import LLM
 
         info = common_model.get_model_info()  # 确保模型可用
+
+        # 注意：不在此处设置 stream_options，因为：
+        # 1. CrewAI 内部决定是否使用流式请求
+        # 2. 在非流式请求中传递 stream_options 不符合 OpenAI API 规范
+        # 3. CrewAI 会自行处理 usage 信息
         return LLM(
             api_key=info.api_key,
             model=f"{info.provider or 'openai'}/{info.model}",
             base_url=info.base_url,
             default_headers=info.headers,
-            stream_options={"include_usage": True},
             # async_client=AsyncClient(headers=info.headers),
         )

{agentrun_inner_test-0.0.40 → agentrun_inner_test-0.0.64}/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_inner_test-0.0.40 → agentrun_inner_test-0.0.64}/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_inner_test-0.0.40 → agentrun_inner_test-0.0.64}/agentrun/integration/google_adk/model_adapter.py

@@ -34,10 +34,13 @@ class GoogleADKModelAdapter(ModelAdapter):
 
         info = common_model.get_model_info()
 
+        # 注意：不在此处设置 stream_options，因为：
+        # 1. Google ADK 内部决定是否使用流式请求
+        # 2. 在非流式请求中传递 stream_options 不符合 OpenAI API 规范
+        # 3. Google ADK 会自行处理 usage 信息
         return LiteLlm(
             model=f"{info.provider or 'openai'}/{info.model}",
             api_base=info.base_url,
             api_key=info.api_key,
             extra_headers=info.headers,
-            stream_options={"include_usage": True},
         )

{agentrun_inner_test-0.0.40 → agentrun_inner_test-0.0.64}/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_inner_test-0.0.40 → agentrun_inner_test-0.0.64}/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_inner_test-0.0.40 → agentrun_inner_test-0.0.64}/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_inner_test-0.0.40 → agentrun_inner_test-0.0.64}/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,
+    )