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

agentrun/credential/__client_async_template.py

@@ -0,0 +1,177 @@
+"""Credential 客户端 / Credential Client
+
+此模块提供凭证管理的客户端API。
+This module provides the client API for credential management.
+"""
+
+from typing import Optional
+
+from alibabacloud_agentrun20250910.models import (
+    CreateCredentialInput,
+    ListCredentialsRequest,
+    UpdateCredentialInput,
+)
+
+from agentrun.utils.config import Config
+from agentrun.utils.exception import HTTPError
+
+from .api.control import CredentialControlAPI
+from .credential import Credential
+from .model import (
+    CredentialCreateInput,
+    CredentialListInput,
+    CredentialListOutput,
+    CredentialUpdateInput,
+)
+
+
+class CredentialClient:
+    """Credential 客户端 / Credential Client
+
+    提供凭证的创建、删除、更新和查询功能。
+    Provides create, delete, update and query functions for credentials.
+    """
+
+    def __init__(self, config: Optional[Config] = None):
+        """初始化客户端 / Initialize client
+
+        Args:
+            config: 配置对象,可选 / Configuration object, optional
+        """
+        self.__control_api = CredentialControlAPI(config)
+
+    async def create_async(
+        self, input: CredentialCreateInput, config: Optional[Config] = None
+    ):
+        """创建凭证(异步) / Create credential asynchronously
+
+        Args:
+            input: 凭证输入参数 / Credential input parameters
+            config: 配置对象,可选 / Configuration object, optional
+
+        Returns:
+            Credential: 创建的凭证对象 / Created credential object
+
+        Raises:
+            ResourceAlreadyExistError: 资源已存在 / Resource already exists
+            HTTPError: HTTP 请求错误 / HTTP request error
+        """
+        try:
+            result = await self.__control_api.create_credential_async(
+                CreateCredentialInput().from_map({
+                    **input.model_dump(),
+                    **input.credential_config.model_dump(),
+                }),
+                config=config,
+            )
+
+            return Credential.from_inner_object(result)
+        except HTTPError as e:
+            raise e.to_resource_error(
+                "Credential", input.credential_name
+            ) from e
+
+    async def delete_async(
+        self, credential_name: str, config: Optional[Config] = None
+    ):
+        """删除凭证（异步）
+
+        Args:
+            credential_name: 凭证名称
+            config: 配置
+
+        Raises:
+            ResourceNotExistError: 凭证不存在
+        """
+        try:
+            result = await self.__control_api.delete_credential_async(
+                credential_name, config=config
+            )
+
+            return Credential.from_inner_object(result)
+
+        except HTTPError as e:
+            raise e.to_resource_error("Credential", credential_name) from e
+
+    async def update_async(
+        self,
+        credential_name: str,
+        input: CredentialUpdateInput,
+        config: Optional[Config] = None,
+    ):
+        """更新凭证（异步）
+
+        Args:
+            credential_name: 凭证名称
+            input: 凭证更新输入参数
+            config: 配置
+
+        Returns:
+            Credential: 更新后的凭证对象
+
+        Raises:
+            ResourceNotExistError: 凭证不存在
+        """
+        try:
+            result = await self.__control_api.update_credential_async(
+                credential_name,
+                UpdateCredentialInput().from_map({
+                    **input.model_dump(),
+                    **(
+                        input.credential_config.model_dump()
+                        if input.credential_config
+                        else {}
+                    ),
+                }),
+                config=config,
+            )
+
+            return Credential.from_inner_object(result)
+        except HTTPError as e:
+            raise e.to_resource_error("Credential", credential_name) from e
+
+    async def get_async(
+        self, credential_name: str, config: Optional[Config] = None
+    ):
+        """获取凭证（异步）
+
+        Args:
+            credential_name: 凭证名称
+            config: 配置
+
+        Returns:
+            Credential: 凭证对象
+
+        Raises:
+            ResourceNotExistError: 凭证不存在
+        """
+        try:
+            result = await self.__control_api.get_credential_async(
+                credential_name, config=config
+            )
+            return Credential.from_inner_object(result)
+        except HTTPError as e:
+            raise e.to_resource_error("Credential", credential_name) from e
+
+    async def list_async(
+        self,
+        input: Optional[CredentialListInput] = None,
+        config: Optional[Config] = None,
+    ):
+        """列出凭证（异步）
+
+        Args:
+            input: 分页查询参数
+            config: 配置
+
+        Returns:
+            List[Credential]: 凭证列表
+        """
+        if input is None:
+            input = CredentialListInput()
+
+        results = await self.__control_api.list_credentials_async(
+            ListCredentialsRequest().from_map(input.model_dump()),
+            config=config,
+        )
+        return [CredentialListOutput.from_inner_object(item) for item in results.items]  # type: ignore

agentrun/credential/__credential_async_template.py

@@ -0,0 +1,216 @@
+"""Credential 高层 API / Credential High-Level API
+
+此模块定义凭证资源的高级API。
+This module defines the high-level API for credential resources.
+"""
+
+from typing import List, Optional
+
+from agentrun.utils.config import Config
+from agentrun.utils.model import PageableInput
+from agentrun.utils.resource import ResourceBase
+
+from .model import (
+    CredentialAuthType,
+    CredentialCreateInput,
+    CredentialImmutableProps,
+    CredentialListInput,
+    CredentialListOutput,
+    CredentialMutableProps,
+    CredentialSourceType,
+    CredentialSystemProps,
+    CredentialUpdateInput,
+)
+
+
+class Credential(
+    CredentialMutableProps,
+    CredentialImmutableProps,
+    CredentialSystemProps,
+    ResourceBase,
+):
+    """凭证资源 / Credential Resource
+
+    提供凭证的完整生命周期管理,包括创建、删除、更新、查询。
+    Provides complete lifecycle management for credentials, including create, delete, update, and query.
+    """
+
+    @classmethod
+    def __get_client(cls):
+        """获取客户端实例 / Get client instance
+
+        Returns:
+            CredentialClient: 客户端实例 / Client instance
+        """
+        from .client import CredentialClient
+
+        return CredentialClient()
+
+    @classmethod
+    async def create_async(
+        cls, input: CredentialCreateInput, config: Optional[Config] = None
+    ):
+        """创建凭证（异步）
+
+        Args:
+            input: 凭证输入参数
+            config: 配置
+
+        Returns:
+            Credential: 创建的凭证对象
+        """
+        return await cls.__get_client().create_async(input, config=config)
+
+    @classmethod
+    async def delete_by_name_async(
+        cls, credential_name: str, config: Optional[Config] = None
+    ):
+        """根据名称删除凭证（异步）
+
+        Args:
+            credential_name: 凭证名称
+            config: 配置
+        """
+        return await cls.__get_client().delete_async(
+            credential_name, config=config
+        )
+
+    @classmethod
+    async def update_by_name_async(
+        cls,
+        credential_name: str,
+        input: CredentialUpdateInput,
+        config: Optional[Config] = None,
+    ):
+        """根据名称更新凭证（异步）
+
+        Args:
+            credential_name: 凭证名称
+            input: 凭证更新输入参数
+            config: 配置
+
+        Returns:
+            Credential: 更新后的凭证对象
+        """
+        return await cls.__get_client().update_async(
+            credential_name, input, config=config
+        )
+
+    @classmethod
+    async def get_by_name_async(
+        cls, credential_name: str, config: Optional[Config] = None
+    ):
+        """根据名称获取凭证（异步）
+
+        Args:
+            credential_name: 凭证名称
+            config: 配置
+
+        Returns:
+            Credential: 凭证对象
+        """
+        return await cls.__get_client().get_async(
+            credential_name, config=config
+        )
+
+    @classmethod
+    async def _list_page_async(
+        cls, page_input: PageableInput, config: Config | None = None, **kwargs
+    ):
+        return await cls.__get_client().list_async(
+            input=CredentialListInput(
+                **kwargs,
+                **page_input.model_dump(),
+            ),
+            config=config,
+        )
+
+    @classmethod
+    async def list_all_async(
+        cls,
+        *,
+        credential_auth_type: Optional[CredentialAuthType] = None,
+        credential_name: Optional[str] = None,
+        credential_source_type: Optional[CredentialSourceType] = None,
+        provider: Optional[str] = None,
+        config: Optional[Config] = None,
+    ) -> List[CredentialListOutput]:
+        return await cls._list_all_async(
+            lambda cred: cred.credential_id or "",
+            config=config,
+            credential_auth_type=credential_auth_type,
+            credential_name=credential_name,
+            credential_source_type=credential_source_type,
+            provider=provider,
+        )
+
+    async def update_async(
+        self, input: CredentialUpdateInput, config: Optional[Config] = None
+    ):
+        """更新凭证（异步）
+
+        Args:
+            input: 凭证更新输入参数
+            config: 配置
+
+        Returns:
+            Credential: 更新后的凭证对象
+        """
+        if self.credential_name is None:
+            raise ValueError(
+                "credential_name is required to update a Credential"
+            )
+
+        result = await self.update_by_name_async(
+            self.credential_name, input, config=config
+        )
+        self.update_self(result)
+
+        return self
+
+    async def delete_async(self, config: Optional[Config] = None):
+        """删除凭证（异步）
+
+        Args:
+            config: 配置
+        """
+        if self.credential_name is None:
+            raise ValueError(
+                "credential_name is required to delete a Credential"
+            )
+
+        return await self.delete_by_name_async(
+            self.credential_name, config=config
+        )
+
+    async def get_async(self, config: Optional[Config] = None):
+        """刷新凭证信息（异步）
+
+        Args:
+            config: 配置
+
+        Returns:
+            Credential: 刷新后的凭证对象
+        """
+        if self.credential_name is None:
+            raise ValueError(
+                "credential_name is required to refresh a Credential"
+            )
+
+        result = await self.get_by_name_async(
+            self.credential_name, config=config
+        )
+        self.update_self(result)
+
+        return self
+
+    async def refresh_async(self, config: Optional[Config] = None):
+        """刷新凭证信息（异步）
+
+        Args:
+            config: 配置
+
+        Returns:
+            Credential: 刷新后的凭证对象
+        """
+        return await self.get_async(config=config)

agentrun/credential/__init__.py

@@ -0,0 +1,28 @@
+"""Credential 模块 / Credential Module"""
+
+from .api import CredentialControlAPI
+from .client import CredentialClient
+from .credential import Credential
+from .model import (
+    CredentialBasicAuth,
+    CredentialConfig,
+    CredentialCreateInput,
+    CredentialListInput,
+    CredentialUpdateInput,
+    RelatedResource,
+)
+
+__all__ = [
+    # base
+    "Credential",
+    "CredentialClient",
+    "CredentialControlAPI",
+    # inner model
+    "CredentialBasicAuth",
+    "RelatedResource",
+    "CredentialConfig",
+    # api model
+    "CredentialCreateInput",
+    "CredentialUpdateInput",
+    "CredentialListInput",
+]

agentrun/credential/api/__init__.py

@@ -0,0 +1,5 @@
+"""Credential API 模块 / Credential API Module"""
+
+from .control import CredentialControlAPI
+
+__all__ = ["CredentialControlAPI"]