agentrun-inner-test 0.0.62__py3-none-any.whl

agentrun/memory_collection/model.py

@@ -0,0 +1,162 @@
+"""MemoryCollection 模型定义 / MemoryCollection Model Definitions
+
+定义记忆集合相关的数据模型和枚举。
+Defines data models and enumerations related to memory collections.
+"""
+
+from typing import List, Optional
+
+from agentrun.utils.config import Config
+from agentrun.utils.model import BaseModel, PageableInput
+
+
+class EmbedderConfigConfig(BaseModel):
+    """嵌入模型内部配置 / Embedder Inner Configuration"""
+
+    model: Optional[str] = None
+    """模型名称"""
+
+
+class EmbedderConfig(BaseModel):
+    """嵌入模型配置 / Embedder Configuration"""
+
+    config: Optional[EmbedderConfigConfig] = None
+    """配置"""
+    model_service_name: Optional[str] = None
+    """模型服务名称"""
+
+
+class LLMConfigConfig(BaseModel):
+    """LLM 内部配置 / LLM Inner Configuration"""
+
+    model: Optional[str] = None
+    """模型名称"""
+
+
+class LLMConfig(BaseModel):
+    """LLM 配置 / LLM Configuration"""
+
+    config: Optional[LLMConfigConfig] = None
+    """配置"""
+    model_service_name: Optional[str] = None
+    """模型服务名称"""
+
+
+class NetworkConfiguration(BaseModel):
+    """网络配置 / Network Configuration"""
+
+    vpc_id: Optional[str] = None
+    """VPC ID"""
+    vswitch_ids: Optional[List[str]] = None
+    """交换机 ID 列表"""
+    security_group_id: Optional[str] = None
+    """安全组 ID"""
+    network_mode: Optional[str] = None
+    """网络模式"""
+
+
+class VectorStoreConfigConfig(BaseModel):
+    """向量存储内部配置 / Vector Store Inner Configuration"""
+
+    endpoint: Optional[str] = None
+    """端点"""
+    instance_name: Optional[str] = None
+    """实例名称"""
+    collection_name: Optional[str] = None
+    """集合名称"""
+    vector_dimension: Optional[int] = None
+    """向量维度"""
+
+
+class VectorStoreConfig(BaseModel):
+    """向量存储配置 / Vector Store Configuration"""
+
+    provider: Optional[str] = None
+    """提供商"""
+    config: Optional[VectorStoreConfigConfig] = None
+    """配置"""
+
+
+class MemoryCollectionMutableProps(BaseModel):
+    """MemoryCollection 可变属性"""
+
+    description: Optional[str] = None
+    """描述"""
+    embedder_config: Optional[EmbedderConfig] = None
+    """嵌入模型配置"""
+    execution_role_arn: Optional[str] = None
+    """执行角色 ARN"""
+    llm_config: Optional[LLMConfig] = None
+    """LLM 配置"""
+    network_configuration: Optional[NetworkConfiguration] = None
+    """网络配置"""
+    vector_store_config: Optional[VectorStoreConfig] = None
+    """向量存储配置"""
+
+
+class MemoryCollectionImmutableProps(BaseModel):
+    """MemoryCollection 不可变属性"""
+
+    memory_collection_name: Optional[str] = None
+    """Memory Collection 名称"""
+    type: Optional[str] = None
+    """类型"""
+
+
+class MemoryCollectionSystemProps(BaseModel):
+    """MemoryCollection 系统属性"""
+
+    memory_collection_id: Optional[str] = None
+    """Memory Collection ID"""
+    created_at: Optional[str] = None
+    """创建时间"""
+    last_updated_at: Optional[str] = None
+    """最后更新时间"""
+
+
+class MemoryCollectionCreateInput(
+    MemoryCollectionImmutableProps, MemoryCollectionMutableProps
+):
+    """MemoryCollection 创建输入参数"""
+
+    pass
+
+
+class MemoryCollectionUpdateInput(MemoryCollectionMutableProps):
+    """MemoryCollection 更新输入参数"""
+
+    pass
+
+
+class MemoryCollectionListInput(PageableInput):
+    """MemoryCollection 列表查询输入参数"""
+
+    memory_collection_name: Optional[str] = None
+    """Memory Collection 名称"""
+
+
+class MemoryCollectionListOutput(BaseModel):
+    """MemoryCollection 列表输出"""
+
+    memory_collection_id: Optional[str] = None
+    memory_collection_name: Optional[str] = None
+    description: Optional[str] = None
+    type: Optional[str] = None
+    created_at: Optional[str] = None
+    last_updated_at: Optional[str] = None
+
+    async def to_memory_collection_async(self, config: Optional[Config] = None):
+        """转换为完整的 MemoryCollection 对象（异步）"""
+        from .client import MemoryCollectionClient
+
+        return await MemoryCollectionClient(config).get_async(
+            self.memory_collection_name or "", config=config
+        )
+
+    def to_memory_collection(self, config: Optional[Config] = None):
+        """转换为完整的 MemoryCollection 对象"""
+        from .client import MemoryCollectionClient
+
+        return MemoryCollectionClient(config).get(
+            self.memory_collection_name or "", config=config
+        )

agentrun/model/__client_async_template.py

@@ -0,0 +1,357 @@
+"""Model Service 客户端 / Model Service Client
+
+此模块提供模型服务和模型代理的客户端API。
+This module provides the client API for model services and model proxies.
+"""
+
+from typing import List, Literal, Optional, overload, Union
+
+from alibabacloud_agentrun20250910.models import (
+    CreateModelProxyInput,
+    CreateModelServiceInput,
+    ListModelProxiesRequest,
+    ListModelServicesRequest,
+    UpdateModelProxyInput,
+    UpdateModelServiceInput,
+)
+import pydash
+
+from agentrun.utils.config import Config
+from agentrun.utils.exception import HTTPError
+
+from .api.control import ModelControlAPI
+from .model import (
+    BackendType,
+    ModelProxyCreateInput,
+    ModelProxyListInput,
+    ModelProxyUpdateInput,
+    ModelServiceCreateInput,
+    ModelServiceListInput,
+    ModelServiceUpdateInput,
+    ProxyMode,
+)
+from .model_proxy import ModelProxy
+from .model_service import ModelService
+
+
+class ModelClient:
+    """Model Service 客户端 / Model Service Client
+
+    提供模型服务和模型代理的创建、删除、更新和查询功能。
+    Provides create, delete, update and query functions for model services and model proxies.
+    """
+
+    def __init__(self, config: Optional[Config] = None):
+        """初始化客户端 / Initialize client
+
+        Args:
+            config: 配置对象,可选 / Configuration object, optional
+        """
+        self.__control_api = ModelControlAPI(config)
+
+    @overload
+    async def create_async(
+        self, input: ModelServiceCreateInput, config: Optional[Config] = None
+    ) -> ModelService:
+        ...
+
+    @overload
+    async def create_async(
+        self, input: ModelProxyCreateInput, config: Optional[Config] = None
+    ) -> ModelProxy:
+        ...
+
+    async def create_async(
+        self,
+        input: Union[ModelServiceCreateInput, ModelProxyCreateInput],
+        config: Optional[Config] = None,
+    ):
+        """创建模型服务（异步）
+
+        Args:
+            input: 模型服务输入参数
+            config: 配置
+
+        Returns:
+            model: 创建的对象
+        """
+        try:
+            if isinstance(input, ModelProxyCreateInput):
+                if not input.proxy_mode:
+                    input.proxy_mode = (
+                        ProxyMode.MULTI
+                        if len(pydash.get(input, "proxy_config.endpoints", []))
+                        > 1
+                        else ProxyMode.SINGLE
+                    )
+
+                result = await self.__control_api.create_model_proxy_async(
+                    CreateModelProxyInput().from_map(input.model_dump()),
+                    config=config,
+                )
+                return ModelProxy.from_inner_object(result)
+            else:
+                result = await self.__control_api.create_model_service_async(
+                    CreateModelServiceInput().from_map(input.model_dump()),
+                    config=config,
+                )
+                return ModelService.from_inner_object(result)
+        except HTTPError as e:
+            raise e.to_resource_error(
+                "Model",
+                (
+                    input.model_proxy_name
+                    if isinstance(input, ModelProxyCreateInput)
+                    else input.model_service_name
+                ),
+            ) from e
+
+    @overload
+    async def delete_async(
+        self,
+        name: str,
+        backend_type: Literal[BackendType.PROXY],
+        config: Optional[Config] = None,
+    ) -> ModelProxy:
+        ...
+
+    @overload
+    async def delete_async(
+        self,
+        name: str,
+        backend_type: Literal[BackendType.SERVICE],
+        config: Optional[Config] = None,
+    ) -> ModelService:
+        ...
+
+    @overload
+    async def delete_async(
+        self,
+        name: str,
+        backend_type: None = None,
+        config: Optional[Config] = None,
+    ) -> Union[ModelProxy, ModelService]:
+        ...
+
+    async def delete_async(
+        self,
+        name: str,
+        backend_type: Optional[BackendType] = None,
+        config: Optional[Config] = None,
+    ):
+        """删除模型服务（异步）
+
+        Args:
+            model_service_name: 模型服务名称
+            config: 配置
+
+        Raises:
+            ResourceNotExistError: 模型服务不存在
+        """
+
+        error: Optional[HTTPError] = None
+        if backend_type == BackendType.PROXY or backend_type is None:
+            try:
+                result = await self.__control_api.delete_model_proxy_async(
+                    model_proxy_name=name, config=config
+                )
+                return ModelProxy.from_inner_object(result)
+            except HTTPError as e:
+                error = e
+
+        if backend_type == BackendType.PROXY and error is not None:
+            raise error.to_resource_error("Model", name) from error
+
+        try:
+            result = await self.__control_api.delete_model_service_async(
+                model_service_name=name, config=config
+            )
+            return ModelService.from_inner_object(result)
+        except HTTPError as e:
+            raise e.to_resource_error("Model", name) from error
+
+    @overload
+    async def update_async(
+        self,
+        name: str,
+        input: ModelServiceUpdateInput,
+        config: Optional[Config] = None,
+    ) -> ModelService:
+        ...
+
+    @overload
+    async def update_async(
+        self,
+        name: str,
+        input: ModelProxyUpdateInput,
+        config: Optional[Config] = None,
+    ) -> ModelProxy:
+        ...
+
+    async def update_async(
+        self,
+        name: str,
+        input: Union[ModelServiceUpdateInput, ModelProxyUpdateInput],
+        config: Optional[Config] = None,
+    ):
+        """更新模型服务（异步）
+
+        Args:
+            model_service_name: 模型服务名称
+            input: 模型服务更新输入参数
+            config: 配置
+
+        Returns:
+            ModelService: 更新后的模型服务对象
+
+        Raises:
+            ResourceNotExistError: 模型服务不存在
+        """
+
+        if isinstance(input, ModelProxyUpdateInput):
+            try:
+                if not input.proxy_mode:
+                    input.proxy_mode = (
+                        ProxyMode.MULTI
+                        if len(pydash.get(input, "proxy_config.endpoints", []))
+                        > 1
+                        else ProxyMode.SINGLE
+                    )
+                result = await self.__control_api.update_model_proxy_async(
+                    model_proxy_name=name,
+                    input=UpdateModelProxyInput().from_map(input.model_dump()),
+                    config=config,
+                )
+                return ModelProxy.from_inner_object(result)
+            except HTTPError as e:
+                raise e.to_resource_error("Model", name) from e
+
+        else:
+            try:
+                result = await self.__control_api.update_model_service_async(
+                    model_service_name=name,
+                    input=UpdateModelServiceInput().from_map(
+                        input.model_dump()
+                    ),
+                    config=config,
+                )
+                return ModelService.from_inner_object(result)
+            except HTTPError as e:
+                raise e.to_resource_error("Model", name) from e
+
+    @overload
+    async def get_async(
+        self,
+        name: str,
+        backend_type: Literal[BackendType.SERVICE],
+        config: Optional[Config] = None,
+    ) -> ModelService:
+        ...
+
+    @overload
+    async def get_async(
+        self,
+        name: str,
+        backend_type: Literal[BackendType.PROXY],
+        config: Optional[Config] = None,
+    ) -> ModelProxy:
+        ...
+
+    @overload
+    async def get_async(
+        self,
+        name: str,
+        backend_type: None = None,
+        config: Optional[Config] = None,
+    ) -> Union[ModelService, ModelProxy]:
+        ...
+
+    async def get_async(
+        self,
+        name: str,
+        backend_type: Optional[BackendType] = None,
+        config: Optional[Config] = None,
+    ):
+        """获取模型服务（异步）
+
+        Args:
+            model_service_name: 模型服务名称
+            config: 配置
+
+        Returns:
+            ModelService: 模型服务对象
+
+        Raises:
+            ResourceNotExistError: 模型服务不存在
+        """
+
+        error: Optional[HTTPError] = None
+        if backend_type == BackendType.PROXY or backend_type is None:
+            try:
+                result = await self.__control_api.get_model_proxy_async(
+                    model_proxy_name=name, config=config
+                )
+                return ModelProxy.from_inner_object(result)
+            except HTTPError as e:
+                error = e
+
+        if backend_type == BackendType.PROXY and error is not None:
+            raise error.to_resource_error("Model", name) from error
+
+        try:
+            result = await self.__control_api.get_model_service_async(
+                model_service_name=name, config=config
+            )
+            return ModelService.from_inner_object(result)
+        except HTTPError as e:
+            raise e.to_resource_error("Model", name) from e
+
+    @overload
+    async def list_async(
+        self,
+        input: ModelProxyListInput,
+        config: Optional[Config] = None,
+    ) -> List[ModelProxy]:
+        ...
+
+    @overload
+    async def list_async(
+        self,
+        input: ModelServiceListInput,
+        config: Optional[Config] = None,
+    ) -> List[ModelService]:
+        ...
+
+    async def list_async(
+        self,
+        input: Union[ModelServiceListInput, ModelProxyListInput],
+        config: Optional[Config] = None,
+    ):
+        """列出模型服务（异步）
+
+        Args:
+            input: 分页查询参数
+            config: 配置
+
+        Returns:
+            List[ModelService]: 模型服务列表
+        """
+
+        if isinstance(input, ModelProxyListInput):
+            result = await self.__control_api.list_model_proxies_async(
+                ListModelProxiesRequest().from_map(input.model_dump()),
+                config=config,
+            )
+            return [
+                ModelProxy.from_inner_object(item)
+                for item in result.items or []
+            ]
+        else:
+            result = await self.__control_api.list_model_services_async(
+                ListModelServicesRequest().from_map(input.model_dump()),
+                config=config,
+            )
+            return [
+                ModelService.from_inner_object(item)
+                for item in result.items or []
+            ]

agentrun/model/__init__.py

@@ -0,0 +1,57 @@
+"""Model Service 模块 / Model Service Module"""
+
+from .api import ModelCompletionAPI, ModelControlAPI, ModelDataAPI
+from .client import ModelClient
+from .model import (
+    BackendType,
+    ModelFeatures,
+    ModelInfoConfig,
+    ModelParameterRule,
+    ModelProperties,
+    ModelProxyCreateInput,
+    ModelProxyListInput,
+    ModelProxyUpdateInput,
+    ModelServiceCreateInput,
+    ModelServiceListInput,
+    ModelServiceUpdateInput,
+    ModelType,
+    Provider,
+    ProviderSettings,
+    ProxyConfig,
+    ProxyConfigEndpoint,
+    ProxyConfigFallback,
+    ProxyConfigPolicies,
+)
+from .model_proxy import ModelProxy
+from .model_service import ModelService
+
+__all__ = [
+    # base
+    "ModelClient",
+    "ModelService",
+    "ModelProxy",
+    "ModelControlAPI",
+    "ModelCompletionAPI",
+    "ModelDataAPI",
+    # enum
+    "BackendType",
+    "ModelType",
+    "Provider",
+    # inner model
+    "ProviderSettings",
+    "ModelFeatures",
+    "ModelProperties",
+    "ModelParameterRule",
+    "ModelInfoConfig",
+    "ProxyConfigEndpoint",
+    "ProxyConfigFallback",
+    "ProxyConfigPolicies",
+    "ProxyConfig",
+    # api model
+    "ModelServiceCreateInput",
+    "ModelServiceUpdateInput",
+    "ModelServiceListInput",
+    "ModelProxyCreateInput",
+    "ModelProxyUpdateInput",
+    "ModelProxyListInput",
+]