agentkit-sdk-python 0.1.5__py3-none-any.whl

agentkit/runtime/runtime_v1.py

@@ -0,0 +1,178 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import json
+from typing import Dict
+
+from agentkit.client import BaseAgentkitClient
+from agentkit.utils import get_logger
+
+from agentkit.runtime.types import (
+    CreateAgentkitRuntimeRequest,
+    CreateAgentkitRuntimeResponse,
+)
+
+# 导入自动生成的类型（强类型版本）
+from agentkit.runtime.runtime_all_types import (
+    GetAgentKitRuntimeRequest,
+    GetAgentKitRuntimeResponse,
+    ListAgentKitRuntimesRequest,
+    ListAgentKitRuntimesResponse,
+    UpdateAgentKitRuntimeRequest,
+    UpdateAgentKitRuntimeResponse,
+    DeleteAgentKitRuntimeRequest,
+    DeleteAgentKitRuntimeResponse,
+    ReleaseAgentKitRuntimeRequest,
+    ReleaseAgentKitRuntimeResponse,
+    GetAgentKitRuntimeVersionRequest,
+    GetAgentKitRuntimeVersionResponse,
+    ListAgentKitRuntimeVersionsRequest,
+    ListAgentKitRuntimeVersionsResponse,
+    GetAgentKitRuntimeCozeTokenRequest,
+    GetAgentKitRuntimeCozeTokenResponse,
+)
+
+logger = get_logger(__name__)
+
+
+class AgentkitRuntime(BaseAgentkitClient):
+    """AgentKit Runtime Management Service"""
+    
+    # Define all API actions for this service
+    API_ACTIONS: Dict[str, str] = {
+        "CreateAgentKitRuntime": "CreateRuntime",
+        "GetAgentKitRuntime": "GetRuntime",
+        "UpdateAgentKitRuntime": "UpdateRuntime",
+        "DeleteAgentKitRuntime": "DeleteRuntime",
+        "ListAgentKitRuntimes": "ListRuntimes",
+        "ReleaseAgentKitRuntime": "ReleaseRuntime",
+        "GetAgentKitRuntimeVersion": "GetRuntimeVersion",
+        "ListAgentKitRuntimeVersions": "ListRuntimeVersions",
+        "GetAgentKitRuntimeCozeToken": "GetRuntimeCozeToken",
+    }
+    
+    def __init__(
+        self,
+        access_key: str = "",
+        secret_key: str = "",
+        region: str = "",
+        session_token: str = "",
+    ) -> None:
+        super().__init__(
+            access_key=access_key,
+            secret_key=secret_key,
+            region=region,
+            session_token=session_token,
+            service_name="runtime",
+        )
+
+    def create(
+        self, request: CreateAgentkitRuntimeRequest, params: dict
+    ) -> CreateAgentkitRuntimeResponse:
+        """Create a new AgentKit Runtime on Volcengine."""
+        res = self.json(
+            api="CreateAgentKitRuntime", params=params, body=request.model_dump()
+        )
+        if not res:
+            raise Exception("Empty response from create agentkit runtime request.")
+
+        response_data = json.loads(res)
+        return CreateAgentkitRuntimeResponse(**response_data.get('Result', {}))
+
+    def get(self, request: GetAgentKitRuntimeRequest) -> GetAgentKitRuntimeResponse:
+        """Get details of a specific AgentKit Runtime from Volcengine."""
+        logger.debug("Get runtime request: %s", json.dumps(request.model_dump(by_alias=True)))
+        try:
+            resp = self._invoke_api(
+                api_action="GetAgentKitRuntime",
+                request=request,
+                response_type=GetAgentKitRuntimeResponse,
+            )
+        except Exception as e:
+            if "InvalidAgentKitRuntime.NotFound" in str(e):
+                return None
+            raise Exception(f"Failed to get agentkit runtime: {str(e)}")
+        return resp
+
+    def update(
+        self, request: UpdateAgentKitRuntimeRequest
+    ) -> UpdateAgentKitRuntimeResponse:
+        """Update an existing AgentKit Runtime on Volcengine."""
+        logger.debug("Update runtime request: %s", json.dumps(request.model_dump(by_alias=True, exclude_none=True)))
+        return self._invoke_api(
+            api_action="UpdateAgentKitRuntime",
+            request=request,
+            response_type=UpdateAgentKitRuntimeResponse,
+        )
+
+    def delete(
+        self, request: DeleteAgentKitRuntimeRequest
+    ) -> DeleteAgentKitRuntimeResponse:
+        """Delete an AgentKit Runtime on Volcengine."""
+        return self._invoke_api(
+            api_action="DeleteAgentKitRuntime",
+            request=request,
+            response_type=DeleteAgentKitRuntimeResponse,
+        )
+
+    def list(
+        self, request: ListAgentKitRuntimesRequest
+    ) -> ListAgentKitRuntimesResponse:
+        """List all AgentKit Runtimes from Volcengine."""
+        return self._invoke_api(
+            api_action="ListAgentKitRuntimes",
+            request=request,
+            response_type=ListAgentKitRuntimesResponse,
+        )
+
+    def release(
+        self, request: ReleaseAgentKitRuntimeRequest
+    ) -> ReleaseAgentKitRuntimeResponse:
+        """Release an AgentKit Runtime on Volcengine."""
+        return self._invoke_api(
+            api_action="ReleaseAgentKitRuntime",
+            request=request,
+            response_type=ReleaseAgentKitRuntimeResponse,
+        )
+
+    def get_version(
+        self, request: GetAgentKitRuntimeVersionRequest
+    ) -> GetAgentKitRuntimeVersionResponse:
+        """Get the version of a specific AgentKit Runtime."""
+        return self._invoke_api(
+            api_action="GetAgentKitRuntimeVersion",
+            request=request,
+            response_type=GetAgentKitRuntimeVersionResponse,
+        )
+
+    def list_versions(
+        self, request: ListAgentKitRuntimeVersionsRequest
+    ) -> ListAgentKitRuntimeVersionsResponse:
+        """List all versions of a specific AgentKit Runtime."""
+        return self._invoke_api(
+            api_action="ListAgentKitRuntimeVersions",
+            request=request,
+            response_type=ListAgentKitRuntimeVersionsResponse,
+        )
+
+    def get_coze_token(
+        self, request: GetAgentKitRuntimeCozeTokenRequest
+    ) -> GetAgentKitRuntimeCozeTokenResponse:
+        """Get Coze JWT token for a specific AgentKit Runtime."""
+        return self._invoke_api(
+            api_action="GetAgentKitRuntimeCozeToken",
+            request=request,
+            response_type=GetAgentKitRuntimeCozeTokenResponse,
+        )
+

agentkit/runtime/types.py

@@ -0,0 +1,188 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from typing import Optional
+
+from pydantic import BaseModel, Field
+
+# ==================
+# Atomic configs
+# ==================
+
+class KeyAuth_(BaseModel):
+    ApiKey: str = Field(alias="ApiKey")
+    ApiKeyName: str = Field(alias="ApiKeyName")
+    ApiKeyLocation: str = Field(alias="ApiKeyLocation")
+
+class AuthorizerConfiguration(BaseModel):
+    KeyAuth: Optional[KeyAuth_] = Field(default=None, alias="KeyAuth")
+    
+    model_config = {
+        "arbitrary_types_allowed": True
+    }
+
+
+# ==================
+# API configs
+# ==================
+
+
+class CreateAgentkitRuntimeRequest(BaseModel):
+    name: str = Field(alias="Name")
+    description: Optional[str] = Field(alias="Description")
+    artifact_type: str = Field(alias="ArtifactType")
+    artifact_url: str = Field(alias="ArtifactUrl")
+    # command: str = Field(alias="Command")
+    project_name: str = Field(alias="ProjectName")
+    client_token: str = Field(alias="ClientToken")
+    role_name: str = Field(alias="RoleName")
+    authorizer_configuration: AuthorizerConfiguration = Field(alias="AuthorizerConfiguration")
+    envs: Optional[list[dict]] = Field(alias="Envs")
+    tags: Optional[list[dict]] = Field(alias="Tags")
+    enable_apmplus: Optional[bool] = Field(False, alias="ApmplusEnable")
+    
+    model_config = {
+        "populate_by_name": True,
+        "arbitrary_types_allowed": True
+    }
+
+
+class CreateAgentkitRuntimeResponse(BaseModel):
+    id: str = Field(alias="RuntimeId")
+    model_config = {
+        "populate_by_name": True,
+        "arbitrary_types_allowed": True
+    }
+
+
+class GetAgentkitRuntimeRequest(BaseModel):
+    runtime_id: str = Field(alias="RuntimeId")
+    
+    model_config = {
+        "populate_by_name": True,
+        "arbitrary_types_allowed": True
+    }
+
+
+# todo
+class GetAgentkitRuntimeResponse(BaseModel):
+    runtime_id: str = Field(alias="RuntimeId")
+    name: str = Field(alias="Name")
+    description: Optional[str] = Field(alias="Description")
+    artifact_type: str = Field(alias="ArtifactType")
+    artifact_url: str = Field(alias="ArtifactUrl")
+    command: str = Field(alias="Command")
+    project_name: str = Field(alias="ProjectName")
+    status: str = Field(alias="Status")
+    created_at: str = Field(alias="CreatedAt")
+    updated_at: str = Field(alias="UpdatedAt")
+    role_name: str = Field(alias="RoleName")
+    endpoint: str = Field(alias="Endpoint")
+    authorizer_configuration: Optional[AuthorizerConfiguration] = Field(default=None, alias="AuthorizerConfiguration")
+    envs: Optional[list[dict]] = Field(alias="Envs")
+    tags: Optional[list[dict]] = Field(alias="Tags")
+    enable_apmplus: Optional[bool] = Field(alias="ApmplusEnable")
+    
+    model_config = {
+        "populate_by_name": True,
+        "arbitrary_types_allowed": True
+    }
+
+
+class UpdateAgentkitRuntimeRequest(BaseModel):
+    description: Optional[str]
+    artifact_url: Optional[str]
+    role_name: Optional[str]
+    to_release: bool = False
+    client_token: str
+    model_config = {
+        "populate_by_name": True,
+        "arbitrary_types_allowed": True
+    }
+
+
+class UpdateAgentkitRuntimeResponse(BaseModel):
+    id: str
+    model_config = {
+        "populate_by_name": True,
+        "arbitrary_types_allowed": True
+    }
+
+
+class DeleteAgentkitRuntimeRequest(BaseModel):
+    id: str = Field(alias="RuntimeId")
+    client_token: Optional[str] = Field(default="",alias="ClientToken")
+
+
+class DeleteAgentkitRuntimeResponse(BaseModel):
+    id: str = Field(alias="RuntimeId")
+    model_config = {
+        "populate_by_name": True,
+        "arbitrary_types_allowed": True
+    }
+
+
+class ListAgentkitRuntimesRequest(BaseModel):
+    name_contains: Optional[str]
+    status: Optional[str]
+    creation_time_before: Optional[str]
+    creation_time_after: Optional[str]
+    last_update_time_before: Optional[str]
+    last_update_time_after: Optional[str]
+    next_token: Optional[str]
+    max_results: Optional[int]
+
+
+class ListAgentkitRuntimesResponse(BaseModel):
+    next_token: str
+
+
+class ReleaseAgentkitRuntimeRequest(BaseModel):
+    id: str
+    version_number: int
+
+
+class ReleaseAgentkitRuntimeResponse(BaseModel):
+    id: str
+
+
+class ListAgentkitRuntimeErrorDetailsRequest(BaseModel):
+    id: str
+    version_number: int
+    next_token: str
+    max_results: int
+
+
+class ListAgentkitRuntimeErrorDetailsResponse(BaseModel):
+    next_token: str
+
+
+class GetAgentkitRuntimeVersionRequest(BaseModel):
+    id: str
+    version_number: int
+
+
+class GetAgentkitRuntimeVersionResponse(BaseModel):
+    id: str
+    name: str
+    description: str
+    version_number: int
+
+
+class ListAgentkitRuntimeVersionsRequest(BaseModel):
+    version: str
+
+
+class ListAgentkitRuntimeVersionsResponse(BaseModel):
+    version: str

agentkit/toolkit/__init__.py

@@ -0,0 +1,13 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.

agentkit/toolkit/cli/__init__.py

@@ -0,0 +1,13 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.

agentkit/toolkit/cli/__main__.py

@@ -0,0 +1,7 @@
+#!/usr/bin/env python3
+"""AgentKit CLI 入口点"""
+
+from .cli import app
+
+if __name__ == "__main__":
+    app()

agentkit/toolkit/cli/cli.py

@@ -0,0 +1,97 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""AgentKit CLI - Main entry point for AgentKit Starter Toolkit."""
+
+import logging
+import typer
+from rich.panel import Panel
+from rich.console import Console
+
+logging.basicConfig(level=logging.ERROR)
+
+# Import command modules
+from agentkit.toolkit.cli.cli_init import init_command, show_logo
+from agentkit.toolkit.cli.cli_invoke import invoke_command
+from agentkit.toolkit.cli.cli_config import config_command
+from agentkit.toolkit.cli.cli_version import version_command, get_package_version
+from agentkit.toolkit.cli.cli_build import build_command
+from agentkit.toolkit.cli.cli_deploy import deploy_command
+from agentkit.toolkit.cli.cli_launch import launch_command
+from agentkit.toolkit.cli.cli_status import status_command
+from agentkit.toolkit.cli.cli_destroy import destroy_command
+
+# Note: 不要在文件开头导很重的包，不然会导致命令很卡(import包很慢)
+
+app = typer.Typer(
+    name="agentkit",
+    help="AgentKit CLI - Deploy AI agents with ease",
+    add_completion=False,
+)
+
+console = Console()
+
+
+# get_package_version is now imported from cli_version
+
+
+def version_callback(value: bool):
+    """Callback for --version flag."""
+    if value:
+        pkg_version = get_package_version()
+        console.print(Panel(
+            f"[bold cyan]AgentKit SDK[/bold cyan]\n[green]Version: {pkg_version}[/green]",
+            title="📦 Version Info",
+            border_style="cyan"
+        ))
+        raise typer.Exit()
+
+
+@app.callback(invoke_without_command=True)
+def main(
+    ctx: typer.Context,
+    version_flag: bool = typer.Option(
+        False, 
+        "--version", 
+        "-v",
+        help="Show version information",
+        callback=version_callback,
+        is_eager=True
+    )
+):
+    """AgentKit CLI - Deploy AI agents with ease."""
+    # If no subcommand is provided, show logo
+    if ctx.invoked_subcommand is None:
+        show_logo()
+        pkg_version = get_package_version()
+        console.print(f"Version: {pkg_version}\n")
+        console.print("Run [bold]agentkit --help[/bold] to see available commands.\n")
+
+
+# Register commands
+app.command(name="init")(init_command)
+app.command(name="invoke")(invoke_command)
+app.command(name="config")(config_command)
+app.command(name="version")(version_command)
+app.command(name="build")(build_command)
+app.command(name="deploy")(deploy_command)
+app.command(name="launch")(launch_command)
+app.command(name="status")(status_command)
+app.command(name="destroy")(destroy_command)
+
+
+
+
+if __name__ == "__main__":
+    app()

agentkit/toolkit/cli/cli_build.py

@@ -0,0 +1,53 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""AgentKit CLI - Build command implementation."""
+
+from pathlib import Path
+import typer
+from rich.console import Console
+from agentkit.toolkit.config import get_config
+
+# Note: 不要在文件开头导很重的包，不然会导致命令很卡(import包很慢)
+
+console = Console()
+
+
+def build_command(
+    config_file: Path = typer.Option("agentkit.yaml", help="Configuration file"),
+    platform: str = typer.Option("auto", help="Build platform"),
+):
+    """Build Docker image for the Agent."""
+    from agentkit.toolkit.workflows import WORKFLOW_REGISTRY, Workflow
+    console.print(f"[cyan]Building image with {config_file}[/cyan]")
+    
+    config = get_config(config_path=config_file)
+    common_config = config.get_common_config()
+
+    if not common_config.entry_point:
+        console.print("[red]Entry point not configured, cannot build image[/red]")
+        raise typer.Exit(1)
+    
+    workflow_name = common_config.launch_type
+    if workflow_name not in WORKFLOW_REGISTRY:
+        console.print(f"[red]Error: Unknown workflow type '{workflow_name}'[/red]")
+        raise typer.Exit(1)
+    
+    workflow_config = config.get_workflow_config(workflow_name)
+    
+    workflow:Workflow = WORKFLOW_REGISTRY[workflow_name]()
+    success = workflow.build(workflow_config)
+    
+    if not success:
+        raise typer.Exit(1)