agentrun-sdk 0.0.4__py3-none-any.whl

agentrun/model/model_service.py

@@ -0,0 +1,438 @@
+"""
+This file is auto generated by the code generation script.
+Do not modify this file manually.
+Use the `make codegen` command to regenerate.
+
+当前文件为自动生成的控制 API 客户端代码。请勿手动修改此文件。
+使用 `make codegen` 命令重新生成。
+
+source: agentrun/model/__model_service_async_template.py
+
+Model Service 高层 API / Model Service High-Level API
+
+此模块定义模型服务资源的高级API。
+This module defines the high-level API for model service resources.
+"""
+
+from typing import List, Optional
+
+from agentrun.model.api.data import BaseInfo, ModelCompletionAPI
+from agentrun.utils.config import Config
+from agentrun.utils.model import PageableInput
+from agentrun.utils.resource import ResourceBase
+
+from .model import (
+    BackendType,
+    ModelServiceCreateInput,
+    ModelServiceImmutableProps,
+    ModelServiceListInput,
+    ModelServiceMutableProps,
+    ModelServicesSystemProps,
+    ModelServiceUpdateInput,
+    ModelType,
+)
+
+
+class ModelService(
+    ModelServiceImmutableProps,
+    ModelServiceMutableProps,
+    ModelServicesSystemProps,
+    ResourceBase,
+):
+    """模型服务"""
+
+    @classmethod
+    def __get_client(cls):
+        from .client import ModelClient
+
+        return ModelClient()
+
+    @classmethod
+    async def create_async(
+        cls, input: ModelServiceCreateInput, config: Optional[Config] = None
+    ):
+        """创建模型服务（异步）
+
+        Args:
+            input: 模型服务输入参数
+            config: 配置
+
+        Returns:
+            ModelService: 创建的模型服务对象
+        """
+        return await cls.__get_client().create_async(input, config=config)
+
+    @classmethod
+    def create(
+        cls, input: ModelServiceCreateInput, config: Optional[Config] = None
+    ):
+        """创建模型服务（同步）
+
+        Args:
+            input: 模型服务输入参数
+            config: 配置
+
+        Returns:
+            ModelService: 创建的模型服务对象
+        """
+        return cls.__get_client().create(input, config=config)
+
+    @classmethod
+    async def delete_by_name_async(
+        cls, model_service_name: str, config: Optional[Config] = None
+    ):
+        """根据名称删除模型服务（异步）
+
+        Args:
+            model_service_name: 模型服务名称
+            config: 配置
+        """
+        return await cls.__get_client().delete_async(
+            model_service_name, backend_type=BackendType.SERVICE, config=config
+        )
+
+    @classmethod
+    def delete_by_name(
+        cls, model_service_name: str, config: Optional[Config] = None
+    ):
+        """根据名称删除模型服务（同步）
+
+        Args:
+            model_service_name: 模型服务名称
+            config: 配置
+        """
+        return cls.__get_client().delete(
+            model_service_name, backend_type=BackendType.SERVICE, config=config
+        )
+
+    @classmethod
+    async def update_by_name_async(
+        cls,
+        model_service_name: str,
+        input: ModelServiceUpdateInput,
+        config: Optional[Config] = None,
+    ):
+        """根据名称更新模型服务（异步）
+
+        Args:
+            model_service_name: 模型服务名称
+            input: 模型服务更新输入参数
+            config: 配置
+
+        Returns:
+            ModelService: 更新后的模型服务对象
+        """
+        return await cls.__get_client().update_async(
+            model_service_name, input, config=config
+        )
+
+    @classmethod
+    def update_by_name(
+        cls,
+        model_service_name: str,
+        input: ModelServiceUpdateInput,
+        config: Optional[Config] = None,
+    ):
+        """根据名称更新模型服务（同步）
+
+        Args:
+            model_service_name: 模型服务名称
+            input: 模型服务更新输入参数
+            config: 配置
+
+        Returns:
+            ModelService: 更新后的模型服务对象
+        """
+        return cls.__get_client().update(
+            model_service_name, input, config=config
+        )
+
+    @classmethod
+    async def get_by_name_async(
+        cls, model_service_name: str, config: Optional[Config] = None
+    ):
+        """根据名称获取模型服务（异步）
+
+        Args:
+            model_service_name: 模型服务名称
+            config: 配置
+
+        Returns:
+            ModelService: 模型服务对象
+        """
+        return await cls.__get_client().get_async(
+            model_service_name, backend_type=BackendType.SERVICE, config=config
+        )
+
+    @classmethod
+    def get_by_name(
+        cls, model_service_name: str, config: Optional[Config] = None
+    ):
+        """根据名称获取模型服务（同步）
+
+        Args:
+            model_service_name: 模型服务名称
+            config: 配置
+
+        Returns:
+            ModelService: 模型服务对象
+        """
+        return cls.__get_client().get(
+            model_service_name, backend_type=BackendType.SERVICE, 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=ModelServiceListInput(
+                **kwargs,
+                **page_input.model_dump(),
+            ),
+            config=config,
+        )
+
+    @classmethod
+    def _list_page(
+        cls, page_input: PageableInput, config: Config | None = None, **kwargs
+    ):
+        return cls.__get_client().list(
+            input=ModelServiceListInput(
+                **kwargs,
+                **page_input.model_dump(),
+            ),
+            config=config,
+        )
+
+    @classmethod
+    async def list_all_async(
+        cls,
+        *,
+        model_type: Optional[ModelType] = None,
+        provider: Optional[str] = None,
+        config: Optional[Config] = None,
+    ) -> List["ModelService"]:
+        return await cls._list_all_async(
+            lambda m: m.model_service_id or "",
+            config=config,
+            model_type=model_type,
+            provider=provider,
+        )
+
+    @classmethod
+    def list_all(
+        cls,
+        *,
+        model_type: Optional[ModelType] = None,
+        provider: Optional[str] = None,
+        config: Optional[Config] = None,
+    ) -> List["ModelService"]:
+        return cls._list_all(
+            lambda m: m.model_service_id or "",
+            config=config,
+            model_type=model_type,
+            provider=provider,
+        )
+
+    async def update_async(
+        self, input: ModelServiceUpdateInput, config: Optional[Config] = None
+    ):
+        """更新模型服务（异步）
+
+        Args:
+            input: 模型服务更新输入参数
+            config: 配置
+
+        Returns:
+            ModelService: 更新后的模型服务对象
+        """
+        if self.model_service_name is None:
+            raise ValueError(
+                "model_service_name is required to update a ModelService"
+            )
+
+        result = await self.update_by_name_async(
+            self.model_service_name, input, config=config
+        )
+        self.update_self(result)
+
+        return self
+
+    def update(
+        self, input: ModelServiceUpdateInput, config: Optional[Config] = None
+    ):
+        """更新模型服务（同步）
+
+        Args:
+            input: 模型服务更新输入参数
+            config: 配置
+
+        Returns:
+            ModelService: 更新后的模型服务对象
+        """
+        if self.model_service_name is None:
+            raise ValueError(
+                "model_service_name is required to update a ModelService"
+            )
+
+        result = self.update_by_name(
+            self.model_service_name, input, config=config
+        )
+        self.update_self(result)
+
+        return self
+
+    async def delete_async(self, config: Optional[Config] = None):
+        """删除模型服务（异步）
+
+        Args:
+            config: 配置
+        """
+        if self.model_service_name is None:
+            raise ValueError(
+                "model_service_name is required to delete a ModelService"
+            )
+
+        return await self.delete_by_name_async(
+            self.model_service_name, config=config
+        )
+
+    def delete(self, config: Optional[Config] = None):
+        """删除模型服务（同步）
+
+        Args:
+            config: 配置
+        """
+        if self.model_service_name is None:
+            raise ValueError(
+                "model_service_name is required to delete a ModelService"
+            )
+
+        return self.delete_by_name(self.model_service_name, config=config)
+
+    async def get_async(self, config: Optional[Config] = None):
+        """刷新模型服务信息（异步）
+
+        Args:
+            config: 配置
+
+        Returns:
+            ModelService: 刷新后的模型服务对象
+        """
+        if self.model_service_name is None:
+            raise ValueError(
+                "model_service_name is required to refresh a ModelService"
+            )
+
+        result = await self.get_by_name_async(
+            self.model_service_name, config=config
+        )
+        self.update_self(result)
+
+        return self
+
+    def get(self, config: Optional[Config] = None):
+        """刷新模型服务信息（同步）
+
+        Args:
+            config: 配置
+
+        Returns:
+            ModelService: 刷新后的模型服务对象
+        """
+        if self.model_service_name is None:
+            raise ValueError(
+                "model_service_name is required to refresh a ModelService"
+            )
+
+        result = self.get_by_name(self.model_service_name, config=config)
+        self.update_self(result)
+
+        return self
+
+    async def refresh_async(self, config: Optional[Config] = None):
+        """刷新模型服务信息（异步）
+
+        Args:
+            config: 配置
+
+        Returns:
+            ModelService: 刷新后的模型服务对象
+        """
+        return await self.get_async(config=config)
+
+    def refresh(self, config: Optional[Config] = None):
+        """刷新模型服务信息（同步）
+
+        Args:
+            config: 配置
+
+        Returns:
+            ModelService: 刷新后的模型服务对象
+        """
+        return self.get(config=config)
+
+    def model_info(self, config: Optional[Config] = None) -> BaseInfo:
+        cfg = Config.with_configs(self._config, config)
+
+        assert self.provider_settings is not None
+        assert self.provider_settings.base_url is not None
+
+        api_key = self.provider_settings.api_key or ""
+        if not api_key and self.credential_name:
+            from agentrun.credential import Credential
+
+            credential = Credential.get_by_name(
+                self.credential_name, config=cfg
+            )
+            api_key = credential.credential_secret or ""
+
+        default_model = (
+            self.provider_settings.model_names[0]
+            if self.provider_settings.model_names is not None
+            and len(self.provider_settings.model_names) > 0
+            else None
+        )
+
+        return BaseInfo(
+            api_key=api_key,
+            base_url=self.provider_settings.base_url,
+            model=default_model,
+            headers=cfg.get_headers(),
+        )
+
+    def completions(
+        self,
+        messages: list,
+        model: Optional[str] = None,
+        stream: bool = False,
+        **kwargs,
+    ):
+        info = self.model_info(config=kwargs.get("config"))
+
+        m = ModelCompletionAPI(
+            api_key=info.api_key or "",
+            base_url=info.base_url or "",
+            model=model or info.model or self.model_service_name or "",
+        )
+
+        return m.completions(**kwargs, messages=messages, stream=stream)
+
+    def responses(
+        self,
+        messages: list,
+        model: Optional[str] = None,
+        stream: bool = False,
+        **kwargs,
+    ):
+        info = self.model_info(config=kwargs.get("config"))
+
+        m = ModelCompletionAPI(
+            api_key=info.api_key or "",
+            base_url=info.base_url or "",
+            model=model or info.model or self.model_service_name or "",
+            provider=(self.provider or "openai").lower(),
+        )
+
+        return m.responses(**kwargs, messages=messages, stream=stream)

agentrun/sandbox/__browser_sandbox_async_template.py

@@ -0,0 +1,113 @@
+"""浏览器沙箱高层API模板 / Browser Sandbox High-Level API Template
+
+此模板用于生成浏览器沙箱资源的高级API代码。
+This template is used to generate high-level API code for browser sandbox resources.
+"""
+
+import asyncio
+import time  # noqa: F401
+from typing import Optional
+
+from agentrun.sandbox.api import BrowserDataAPI
+from agentrun.sandbox.model import TemplateType
+from agentrun.utils.log import logger
+
+from .sandbox import Sandbox
+
+
+class BrowserSandbox(Sandbox):
+    _template_type = TemplateType.BROWSER
+
+    _data_api: Optional["BrowserDataAPI"] = None
+
+    async def __aenter__(self):
+        # Poll health check asynchronously
+        max_retries = 60  # Maximum 60 seconds
+        retry_count = 0
+
+        logger.debug("Waiting for browser to be ready...")
+
+        while retry_count < max_retries:
+            retry_count += 1
+
+            try:
+                health = await self.check_health_async()
+
+                if health["status"] == "ok":
+                    logger.debug(
+                        f"✓ Browser is ready! (took {retry_count} seconds)"
+                    )
+                    return self
+
+                logger.debug(
+                    "[%d/%d] Health status: %d %s",
+                    retry_count,
+                    max_retries,
+                    health.get("code"),
+                    health.get("message"),
+                )
+
+            except Exception as e:
+                logger.error(
+                    f"[{retry_count}/{max_retries}] Health check failed: {e}"
+                )
+
+            if retry_count < max_retries:
+                await asyncio.sleep(1)
+
+        raise RuntimeError(
+            f"Health check timeout after {max_retries} seconds. "
+            "Browser did not become ready in time."
+        )
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        if self.sandbox_id is None:
+            raise ValueError("Sandbox ID is not set")
+        logger.debug(f"Deleting browser sandbox {self.sandbox_id}...")
+        await self.delete_async()
+
+    @property
+    def data_api(self):
+        if self._data_api is None:
+            if self.sandbox_id is None:
+                raise ValueError("Sandbox ID is not set")
+
+            self._data_api = BrowserDataAPI(
+                sandbox_id=self.sandbox_id, config=self._config
+            )
+
+        return self._data_api
+
+    async def check_health_async(self):
+        return await self.data_api.check_health_async()
+
+    def get_cdp_url(self, record: Optional[bool] = False):
+        return self.data_api.get_cdp_url(record=record)
+
+    def get_vnc_url(self, record: Optional[bool] = False):
+        return self.data_api.get_vnc_url(record=record)
+
+    def sync_playwright(self, record: Optional[bool] = False):
+        return self.data_api.sync_playwright(record=record)
+
+    def async_playwright(self, record: Optional[bool] = False):
+        return self.data_api.async_playwright(record=record)
+
+    async def list_recordings_async(self):
+        return await self.data_api.list_recordings_async()
+
+    async def download_recording_async(self, filename: str, save_path: str):
+        """
+        Asynchronously download a recording video file and save it to local path.
+
+        Args:
+            filename: The name of the recording file to download
+            save_path: Local file path to save the downloaded video file (.mkv)
+
+        Returns:
+            Dictionary with 'saved_path' and 'size' keys
+        """
+        return await self.data_api.download_recording_async(filename, save_path)
+
+    async def delete_recording_async(self, filename: str):
+        return await self.data_api.delete_recording_async(filename)