agentrun-inner-test 0.0.56__py3-none-any.whl → 0.0.64__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.56"
+__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/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/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/memory_collection/__client_async_template.py

@@ -0,0 +1,178 @@
+"""MemoryCollection 客户端 / MemoryCollection Client
+
+此模块提供记忆集合管理的客户端API。
+This module provides the client API for memory collection management.
+"""
+
+from typing import Optional
+
+from alibabacloud_agentrun20250910.models import (
+    CreateMemoryCollectionInput,
+    ListMemoryCollectionsRequest,
+    UpdateMemoryCollectionInput,
+)
+
+from agentrun.utils.config import Config
+from agentrun.utils.exception import HTTPError
+
+from .api.control import MemoryCollectionControlAPI
+from .memory_collection import MemoryCollection
+from .model import (
+    MemoryCollectionCreateInput,
+    MemoryCollectionListInput,
+    MemoryCollectionListOutput,
+    MemoryCollectionUpdateInput,
+)
+
+
+class MemoryCollectionClient:
+    """MemoryCollection 客户端 / MemoryCollection Client
+
+    提供记忆集合的创建、删除、更新和查询功能。
+    Provides create, delete, update and query functions for memory collections.
+    """
+
+    def __init__(self, config: Optional[Config] = None):
+        """初始化客户端 / Initialize client
+
+        Args:
+            config: 配置对象,可选 / Configuration object, optional
+        """
+        self.__control_api = MemoryCollectionControlAPI(config)
+
+    async def create_async(
+        self,
+        input: MemoryCollectionCreateInput,
+        config: Optional[Config] = None,
+    ):
+        """创建记忆集合(异步) / Create memory collection asynchronously
+
+        Args:
+            input: 记忆集合输入参数 / Memory collection input parameters
+            config: 配置对象,可选 / Configuration object, optional
+
+        Returns:
+            MemoryCollection: 创建的记忆集合对象 / Created memory collection object
+
+        Raises:
+            ResourceAlreadyExistError: 资源已存在 / Resource already exists
+            HTTPError: HTTP 请求错误 / HTTP request error
+        """
+        try:
+            result = await self.__control_api.create_memory_collection_async(
+                CreateMemoryCollectionInput().from_map(input.model_dump()),
+                config=config,
+            )
+
+            return MemoryCollection.from_inner_object(result)
+        except HTTPError as e:
+            raise e.to_resource_error(
+                "MemoryCollection", input.memory_collection_name
+            ) from e
+
+    async def delete_async(
+        self, memory_collection_name: str, config: Optional[Config] = None
+    ):
+        """删除记忆集合（异步）
+
+        Args:
+            memory_collection_name: 记忆集合名称
+            config: 配置
+
+        Raises:
+            ResourceNotExistError: 记忆集合不存在
+        """
+        try:
+            result = await self.__control_api.delete_memory_collection_async(
+                memory_collection_name, config=config
+            )
+
+            return MemoryCollection.from_inner_object(result)
+
+        except HTTPError as e:
+            raise e.to_resource_error(
+                "MemoryCollection", memory_collection_name
+            ) from e
+
+    async def update_async(
+        self,
+        memory_collection_name: str,
+        input: MemoryCollectionUpdateInput,
+        config: Optional[Config] = None,
+    ):
+        """更新记忆集合（异步）
+
+        Args:
+            memory_collection_name: 记忆集合名称
+            input: 记忆集合更新输入参数
+            config: 配置
+
+        Returns:
+            MemoryCollection: 更新后的记忆集合对象
+
+        Raises:
+            ResourceNotExistError: 记忆集合不存在
+        """
+        try:
+            result = await self.__control_api.update_memory_collection_async(
+                memory_collection_name,
+                UpdateMemoryCollectionInput().from_map(input.model_dump()),
+                config=config,
+            )
+
+            return MemoryCollection.from_inner_object(result)
+        except HTTPError as e:
+            raise e.to_resource_error(
+                "MemoryCollection", memory_collection_name
+            ) from e
+
+    async def get_async(
+        self, memory_collection_name: str, config: Optional[Config] = None
+    ):
+        """获取记忆集合（异步）
+
+        Args:
+            memory_collection_name: 记忆集合名称
+            config: 配置
+
+        Returns:
+            MemoryCollection: 记忆集合对象
+
+        Raises:
+            ResourceNotExistError: 记忆集合不存在
+        """
+        try:
+            result = await self.__control_api.get_memory_collection_async(
+                memory_collection_name, config=config
+            )
+            return MemoryCollection.from_inner_object(result)
+        except HTTPError as e:
+            raise e.to_resource_error(
+                "MemoryCollection", memory_collection_name
+            ) from e
+
+    async def list_async(
+        self,
+        input: Optional[MemoryCollectionListInput] = None,
+        config: Optional[Config] = None,
+    ):
+        """列出记忆集合（异步）
+
+        Args:
+            input: 分页查询参数
+            config: 配置
+
+        Returns:
+            List[MemoryCollectionListOutput]: 记忆集合列表
+        """
+        if input is None:
+            input = MemoryCollectionListInput()
+
+        results = await self.__control_api.list_memory_collections_async(
+            ListMemoryCollectionsRequest().from_map(input.model_dump()),
+            config=config,
+        )
+        return [
+            MemoryCollectionListOutput.from_inner_object(item)
+            for item in results.items  # type: ignore
+        ]

agentrun/memory_collection/__init__.py

@@ -0,0 +1,37 @@
+"""MemoryCollection 模块 / MemoryCollection Module
+
+提供记忆集合管理功能。
+Provides memory collection management functionality.
+"""
+
+from .client import MemoryCollectionClient
+from .memory_collection import MemoryCollection
+from .model import (
+    EmbedderConfig,
+    EmbedderConfigConfig,
+    LLMConfig,
+    LLMConfigConfig,
+    MemoryCollectionCreateInput,
+    MemoryCollectionListInput,
+    MemoryCollectionListOutput,
+    MemoryCollectionUpdateInput,
+    NetworkConfiguration,
+    VectorStoreConfig,
+    VectorStoreConfigConfig,
+)
+
+__all__ = [
+    "MemoryCollection",
+    "MemoryCollectionClient",
+    "MemoryCollectionCreateInput",
+    "MemoryCollectionUpdateInput",
+    "MemoryCollectionListInput",
+    "MemoryCollectionListOutput",
+    "EmbedderConfig",
+    "EmbedderConfigConfig",
+    "LLMConfig",
+    "LLMConfigConfig",
+    "NetworkConfiguration",
+    "VectorStoreConfig",
+    "VectorStoreConfigConfig",
+]