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

agentrun/utils/__init__.py

@@ -0,0 +1,5 @@
+"""Utils 模块 / Utils Module
+
+此模块提供通用工具函数和基础设施。
+This module provides common utility functions and infrastructure.
+"""

agentrun/utils/__resource_async_template.py

@@ -0,0 +1,158 @@
+"""资源基类模板 / Resource Base Template
+
+此模板用于生成资源对象的基类代码。
+This template is used to generate base class code for resource objects.
+"""
+
+from abc import abstractmethod
+import asyncio
+import time
+from typing import Awaitable, Callable, List, Optional
+
+from typing_extensions import Self
+
+from agentrun.utils.config import Config
+from agentrun.utils.exception import DeleteResourceError, ResourceNotExistError
+from agentrun.utils.log import logger
+
+from .model import BaseModel, PageableInput, Status
+
+
+class ResourceBase(BaseModel):
+    status: Optional[Status] = None
+    _config: Optional[Config] = None
+
+    @classmethod
+    @abstractmethod
+    async def _list_page_async(
+        cls,
+        page_input: PageableInput,
+        config: Optional[Config] = None,
+        **kwargs,
+    ) -> list:
+        ...
+
+    @classmethod
+    async def _list_all_async(
+        cls,
+        uniq_id_callback: Callable[[Self], str],
+        config: Optional[Config] = None,
+        **kwargs,
+    ) -> list:
+        all_results: List[Self] = []
+        page = 1
+        page_size = 50
+        while True:
+            page_results = await cls._list_page_async(
+                PageableInput(
+                    page_number=page,
+                    page_size=page_size,
+                ),
+                config=config,
+                **kwargs,
+            )
+            page += 1
+            all_results.extend(page_results)  # type: ignore
+            if len(page_results) < page_size:
+                break
+
+        result_set = set()
+        results: list = []
+        for item in all_results:
+            uniq_id = uniq_id_callback(item)
+            if uniq_id not in result_set:
+                result_set.add(uniq_id)
+                results.append(item)
+
+        return results
+
+    @abstractmethod
+    async def refresh_async(self, config: Optional[Config] = None) -> Self:
+        ...
+
+    @abstractmethod
+    async def delete_async(self, config: Optional[Config] = None) -> Self:
+        ...
+
+    async def __wait_until_async(
+        self,
+        check_finished_callback_async: Callable[[Self], Awaitable[bool]],
+        interval_seconds: int = 5,
+        timeout_seconds: int = 300,
+    ) -> Self:
+        """等待智能体运行时进入就绪状态"""
+
+        start_time = time.time()
+        while True:
+            if await check_finished_callback_async(self):
+                return self
+
+            if time.time() - start_time > timeout_seconds:
+                raise TimeoutError("等待就绪超时")
+
+            await asyncio.sleep(interval_seconds)
+
+    async def wait_until_ready_or_failed_async(
+        self,
+        callback: Optional[Callable[[Self], None]] = None,
+        interval_seconds: int = 5,
+        timeout_seconds: int = 300,
+    ):
+        """等待智能体运行时进入就绪状态"""
+
+        async def check_ready_callback(resource: Self) -> bool:
+            await resource.refresh_async()
+            if callback:
+                callback(resource)
+            logger.debug("当前状态：%s", resource.status)
+
+            return Status.is_final_status(resource.status)
+
+        await self.__wait_until_async(
+            check_ready_callback,
+            interval_seconds=interval_seconds,
+            timeout_seconds=timeout_seconds,
+        )
+
+    async def delete_and_wait_until_finished_async(
+        self,
+        callback: Optional[Callable[[Self], None]] = None,
+        interval_seconds: int = 5,
+        timeout_seconds: int = 300,
+    ):
+        """等待智能体运行时被删除"""
+        try:
+            await self.delete_async()
+        except ResourceNotExistError:
+            return
+
+        async def check_deleted_callback(resource: Self) -> bool:
+            try:
+                await resource.refresh_async()
+                if callback:
+                    callback(resource)
+            except ResourceNotExistError:
+                return True
+
+            if resource.status == Status.DELETING:
+                return False
+
+            raise DeleteResourceError(f"Resource status is {resource.status}")
+
+        await self.__wait_until_async(
+            check_deleted_callback,
+            interval_seconds=interval_seconds,
+            timeout_seconds=timeout_seconds,
+        )
+
+    def set_config(self, config: Config) -> Self:
+        """设置配置
+
+        Args:
+            config: 配置
+
+        Returns:
+            Self: 当前对象
+        """
+        self._config = config
+        return self

agentrun/utils/config.py

@@ -0,0 +1,258 @@
+"""配置管理模块 / Configuration Management Module
+
+此模块提供 AgentRun SDK 的全局配置管理功能。
+This module provides global configuration management for AgentRun SDK.
+"""
+
+import os
+from typing import Dict, Optional
+
+from dotenv import load_dotenv
+
+load_dotenv()
+
+
+def get_env_with_default(default: str, *key: str) -> str:
+    """从环境变量获取值,支持多个候选键 / Get value from environment variables with multiple fallback keys
+
+    Args:
+        default: 默认值 / Default value
+        *key: 候选环境变量名 / Candidate environment variable names
+
+    Returns:
+        str: 环境变量值或默认值 / Environment variable value or default value
+    """
+    for k in key:
+        v = os.getenv(k)
+        if v is not None:
+            return v
+    return default
+
+
+class Config:
+    """AgentRun SDK 全局配置类 / AgentRun SDK Global Configuration Class
+
+    用于管理账号凭证和客户端配置。
+    Used for managing account credentials and client configuration.
+
+    支持从参数或环境变量读取配置。
+    Supports reading configuration from parameters or environment variables.
+
+    Examples:
+        >>> # 从参数创建配置 / Create config from parameters
+        >>> config = Config(
+        ...     account_id="your-account-id",
+        ...     access_key_id="your-key-id",
+        ...     access_key_secret="your-secret"
+        ... )
+        >>> # 或从环境变量读取 / Or read from environment variables
+        >>> config = Config()
+    """
+
+    __slots__ = (
+        "_access_key_id",
+        "_access_key_secret",
+        "_security_token",
+        "_account_id",
+        "_token",
+        "_region_id",
+        "_timeout",
+        "_read_timeout",
+        "_control_endpoint",
+        "_data_endpoint",
+        "_devs_endpoint",
+        "_headers",
+        "__weakref__",
+    )
+
+    def __init__(
+        self,
+        access_key_id: Optional[str] = None,
+        access_key_secret: Optional[str] = None,
+        security_token: Optional[str] = None,
+        account_id: Optional[str] = None,
+        token: Optional[str] = None,
+        region_id: Optional[str] = None,
+        timeout: Optional[int] = 600,
+        read_timeout: Optional[int] = 100000,
+        control_endpoint: Optional[str] = None,
+        data_endpoint: Optional[str] = None,
+        devs_endpoint: Optional[str] = None,
+        headers: Optional[Dict[str, str]] = None,
+    ) -> None:
+        """初始化配置 / Initialize configuration
+
+        Args:
+            access_key_id: Access Key ID / Access Key ID
+                未提供时从环境变量读取: AGENTRUN_ACCESS_KEY_ID 或 ALIBABA_CLOUD_ACCESS_KEY_ID
+                Read from env vars if not provided: AGENTRUN_ACCESS_KEY_ID or ALIBABA_CLOUD_ACCESS_KEY_ID
+            access_key_secret: Access Key Secret / Access Key Secret
+                未提供时从环境变量读取: AGENTRUN_ACCESS_KEY_SECRET 或 ALIBABA_CLOUD_ACCESS_KEY_SECRET
+                Read from env vars if not provided: AGENTRUN_ACCESS_KEY_SECRET or ALIBABA_CLOUD_ACCESS_KEY_SECRET
+            security_token: 安全令牌 / Security token
+                未提供时从环境变量读取: AGENTRUN_SECURITY_TOKEN 或 ALIBABA_CLOUD_SECURITY_TOKEN
+                Read from env vars if not provided: AGENTRUN_SECURITY_TOKEN or ALIBABA_CLOUD_SECURITY_TOKEN
+            account_id: 账号 ID / Account ID
+                未提供时从环境变量读取: AGENTRUN_ACCOUNT_ID 或 FC_ACCOUNT_ID
+                Read from env vars if not provided: AGENTRUN_ACCOUNT_ID or FC_ACCOUNT_ID
+            token: 自定义令牌,用于数据链路调用 / Custom token for data API calls
+            region_id: 区域 ID,默认 cn-hangzhou / Region ID, defaults to cn-hangzhou
+            timeout: 请求超时时间(秒),默认 600 / Request timeout in seconds, defaults to 600
+            read_timeout: 读取超时时间(秒),默认 100000 / Read timeout in seconds, defaults to 100000
+            control_endpoint: 自定义控制链路端点,可选 / Custom control endpoint, optional
+            data_endpoint: 自定义数据链路端点,可选 / Custom data endpoint, optional
+            devs_endpoint: 自定义 DevS 端点,可选 / Custom DevS endpoint, optional
+            headers: 自定义请求头,可选 / Custom request headers, optional
+        """
+
+        if access_key_id is None:
+            access_key_id = get_env_with_default(
+                "", "AGENTRUN_ACCESS_KEY_ID", "ALIBABA_CLOUD_ACCESS_KEY_ID"
+            )
+        if access_key_secret is None:
+            access_key_secret = get_env_with_default(
+                "",
+                "AGENTRUN_ACCESS_KEY_SECRET",
+                "ALIBABA_CLOUD_ACCESS_KEY_SECRET",
+            )
+        if security_token is None:
+            security_token = get_env_with_default(
+                "", "AGENTRUN_SECURITY_TOKEN", "ALIBABA_CLOUD_SECURITY_TOKEN"
+            )
+        if account_id is None:
+            account_id = get_env_with_default(
+                "", "AGENTRUN_ACCOUNT_ID", "FC_ACCOUNT_ID"
+            )
+        if region_id is None:
+            region_id = get_env_with_default(
+                "cn-hangzhou", "AGENTRUN_REGION", "FC_REGION"
+            )
+        if control_endpoint is None:
+            control_endpoint = get_env_with_default(
+                "", "AGENTRUN_CONTROL_ENDPOINT"
+            )
+        if data_endpoint is None:
+            data_endpoint = get_env_with_default("", "AGENTRUN_DATA_ENDPOINT")
+        if devs_endpoint is None:
+            devs_endpoint = get_env_with_default("", "DEVS_ENDPOINT")
+
+        self._access_key_id = access_key_id
+        self._access_key_secret = access_key_secret
+        self._security_token = security_token
+        self._account_id = account_id
+        self._token = token
+        self._region_id = region_id
+        self._timeout = timeout
+        self._read_timeout = read_timeout
+        self._control_endpoint = control_endpoint
+        self._data_endpoint = data_endpoint
+        self._devs_endpoint = devs_endpoint
+        self._headers = headers or {}
+
+    @classmethod
+    def with_configs(cls, *configs: Optional["Config"]) -> "Config":
+        return cls().update(*configs)
+
+    def update(self, *configs: Optional["Config"]) -> "Config":
+        """
+        使用给定的配置对象,返回新的实例,优先使用靠后的值
+
+        Args:
+            configs: 要合并的配置对象
+
+        Returns:
+            合并后的新配置对象
+        """
+
+        for config in configs:
+            if config is None:
+                continue
+
+            for attr in filter(
+                lambda x: x != "__weakref__",
+                self.__slots__,
+            ):
+                value = getattr(config, attr)
+                if value is not None:
+                    if type(value) is dict:
+                        getattr(self, attr).update(getattr(config, attr) or {})
+                    else:
+                        setattr(self, attr, value)
+
+        return self
+
+    def __repr__(self) -> str:
+
+        return "Config{%s}" % (
+            ", ".join([
+                f'"{key}": "{getattr(self, key)}"'
+                for key in self.__slots__
+                if key != "__weakref__"
+            ])
+        )
+
+    def get_access_key_id(self) -> str:
+        """获取 Access Key ID"""
+        return self._access_key_id
+
+    def get_access_key_secret(self) -> str:
+        """获取 Access Key Secret"""
+        return self._access_key_secret
+
+    def get_security_token(self) -> str:
+        """获取安全令牌"""
+        return self._security_token
+
+    def get_account_id(self) -> str:
+        """获取账号 ID"""
+        if not self._account_id:
+            raise ValueError(
+                "account id is not set, please add AGENTRUN_ACCOUNT_ID env"
+                " variable or set it in code."
+            )
+
+        return self._account_id
+
+    def get_token(self) -> Optional[str]:
+        """获取自定义令牌"""
+        return self._token
+
+    def get_region_id(self) -> str:
+        """获取区域 ID"""
+        if self._region_id:
+            return self._region_id
+
+        return "cn-hangzhou"
+
+    def get_timeout(self) -> Optional[int]:
+        """获取请求超时时间"""
+        return self._timeout or 600
+
+    def get_read_timeout(self) -> Optional[int]:
+        """获取请求超时时间"""
+        return self._read_timeout or 100000
+
+    def get_control_endpoint(self) -> str:
+        """获取控制链路端点"""
+        if self._control_endpoint:
+            return self._control_endpoint
+
+        return f"https://agentrun.{self.get_region_id()}.aliyuncs.com"
+
+    def get_data_endpoint(self) -> str:
+        """获取数据链路端点"""
+        if self._data_endpoint:
+            return self._data_endpoint
+
+        return f"https://{self.get_account_id()}.agentrun-data.{self.get_region_id()}.aliyuncs.com"
+
+    def get_devs_endpoint(self) -> str:
+        """获取 Devs 端点"""
+        if self._devs_endpoint:
+            return self._devs_endpoint
+
+        return f"https://devs.{self.get_region_id()}.aliyuncs.com"
+
+    def get_headers(self) -> Dict[str, str]:
+        """获取自定义请求头"""
+        return self._headers or {}

agentrun/utils/control_api.py

@@ -0,0 +1,78 @@
+"""控制 API 基类模块 / Control API Base Module
+
+此模块定义控制链路 API 的基类。
+This module defines the base class for control API.
+"""
+
+from typing import Optional
+
+from alibabacloud_agentrun20250910.client import Client as AgentRunClient
+from alibabacloud_devs20230714.client import Client as DevsClient
+from alibabacloud_tea_openapi import utils_models as open_api_util_models
+
+from agentrun.utils.config import Config
+
+
+class ControlAPI:
+    """控制链路客户端基类 / Control API Client Base Class
+
+    提供控制链路和 DevS API 客户端的获取功能。
+    Provides functionality to get control API and DevS API clients.
+    """
+
+    def __init__(self, config: Optional[Config] = None):
+        """初始化 Control API 客户端 / Initialize Control API client
+
+        Args:
+            config: 全局配置对象,可选 / Global configuration object, optional
+        """
+        self.config = config
+
+    def _get_client(self, config: Optional[Config] = None) -> "AgentRunClient":
+        """获取 Control API 客户端实例 / Get Control API client instance
+
+        Args:
+            config: 配置对象,可选,用于覆盖默认配置 / Configuration object, optional, to override default config
+
+        Returns:
+            AgentRunClient: Control API 客户端实例 / Control API client instance
+        """
+
+        cfg = Config.with_configs(self.config, config)
+        endpoint = cfg.get_control_endpoint()
+        if endpoint.startswith("http://") or endpoint.startswith("https://"):
+            endpoint = endpoint.split("://", 1)[1]
+        return AgentRunClient(
+            open_api_util_models.Config(
+                access_key_id=cfg.get_access_key_id(),
+                access_key_secret=cfg.get_access_key_secret(),
+                security_token=cfg.get_security_token(),
+                region_id=cfg.get_region_id(),
+                endpoint=endpoint,
+                connect_timeout=cfg.get_timeout(),  # type: ignore
+            )
+        )
+
+    def _get_devs_client(self, config: Optional[Config] = None) -> "DevsClient":
+        """
+        获取 Devs API 客户端实例
+
+        Returns:
+            DevsClient: Devs API 客户端实例
+        """
+
+        cfg = Config.with_configs(self.config, config)
+        endpoint = cfg.get_devs_endpoint()
+        if endpoint.startswith("http://") or endpoint.startswith("https://"):
+            endpoint = endpoint.split("://", 1)[1]
+        return DevsClient(
+            open_api_util_models.Config(
+                access_key_id=cfg.get_access_key_id(),
+                access_key_secret=cfg.get_access_key_secret(),
+                security_token=cfg.get_security_token(),
+                region_id=cfg.get_region_id(),
+                endpoint=endpoint,
+                connect_timeout=cfg.get_timeout(),  # type: ignore
+                read_timeout=cfg.get_read_timeout(),  # type: ignore
+            )
+        )