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

agentkit/toolkit/cli/cli_invoke.py

@@ -0,0 +1,168 @@
+# 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 - Invoke command implementation."""
+
+from pathlib import Path
+from typing import Optional, Any
+import json
+import typer
+from rich.console import Console
+from agentkit.toolkit.config import get_config
+
+# Note: 不要在文件开头导很重的包，不然会导致命令很卡(import包很慢)
+
+console = Console()
+
+
+def invoke_command(
+    config_file: Path = typer.Option("agentkit.yaml", help="Configuration file"),
+    message: str = typer.Argument(None, help="Simple message to send to agent"),
+    payload: str = typer.Option(
+        None, "--payload", "-p", help="JSON payload to send (advanced option)"
+    ),
+    headers: str = typer.Option(
+        None, "--headers", "-h", help="JSON headers for request (advanced option)"
+    ),
+    apikey: str = typer.Option(
+        None, "--apikey", "-ak", help="API key for authentication"
+    ),
+) -> Any:
+    """Send a test request to deployed Agent.
+    
+    Examples:
+        # Simple message
+        agentkit invoke "今天天气如何？"
+        
+        # Custom payload
+        agentkit invoke --payload '{"prompt": "杭州天气?"}'
+        
+        # With custom headers
+        agentkit invoke --payload '{"prompt": "杭州天气?"}' --headers '{"user_id": "test123"}'
+    """
+    from agentkit.toolkit.workflows import WORKFLOW_REGISTRY, Workflow
+    console.print("[cyan]Invoking agent...[/cyan]")
+    
+    # 验证参数：message和payload不能同时提供
+    if message and payload:
+        console.print("[red]Error: Cannot specify both message and payload. Use either message or --payload.[/red]")
+        raise typer.Exit(1)
+    
+    # 验证参数：必须提供message或payload
+    if not message and not payload:
+        console.print("[red]Error: Must provide either a message or --payload option.[/red]")
+        raise typer.Exit(1)
+    
+    # 处理payload
+    if message:
+        # 简单消息模式：自动组织为payload
+        final_payload = {"prompt": message}
+        console.print(f"[blue]Using simple message mode: {message}[/blue]")
+    else:
+        # 高级模式：使用用户提供的payload
+        try:
+            final_payload = json.loads(payload) if isinstance(payload, str) else payload
+            console.print(f"[blue]Using custom payload: {final_payload}[/blue]")
+        except json.JSONDecodeError as e:
+            console.print(f"[red]Error: Invalid JSON payload: {e}[/red]")
+            raise typer.Exit(1)
+    
+    # 处理headers
+    final_headers = {"user_id": "agentkit_user", "session_id": "agentkit_sample_session"}
+    if headers:
+        try:
+            final_headers = json.loads(headers) if isinstance(headers, str) else headers
+            console.print(f"[blue]Using custom headers: {final_headers}[/blue]")
+        except json.JSONDecodeError as e:
+            console.print(f"[red]Error: Invalid JSON headers: {e}[/red]")
+            raise typer.Exit(1)
+    else:
+        console.print(f"[blue]Using default headers: {final_headers}[/blue]")
+    
+    config = get_config(config_path=config_file)
+    common_config = config.get_common_config()
+    
+    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, response = workflow.invoke(workflow_config, {
+        "payload": final_payload, 
+        "headers": final_headers, 
+        "apikey": apikey
+    })
+    if not success:
+        raise typer.Exit(1)
+    
+    console.print("[green]✅ Invocation successful[/green]")
+    
+    # 处理流式响应（生成器）
+    if hasattr(response, '__iter__') and not isinstance(response, (dict, str, list)):
+        console.print("[cyan]📡 Streaming response detected...[/cyan]\n")
+        result_list = []
+        complete_text = []
+        
+        for event in response:
+            result_list.append(event)
+            
+            # 如果是字符串且以 "data: " 开头，尝试解析（fallback处理）
+            if isinstance(event, str):
+                if event.strip().startswith("data: "):
+                    try:
+                        json_str = event.strip()[6:].strip()  # 移除 "data: " 前缀
+                        event = json.loads(json_str)
+                    except json.JSONDecodeError:
+                        # 解析失败，跳过此事件
+                        continue
+                else:
+                    # 不是 SSE 格式的字符串，跳过
+                    continue
+            
+            # 解析事件内容
+            if isinstance(event, dict):
+                # 提取文本内容
+                content = event.get("content", {})
+                part = event.get("partial", False)
+                if part and isinstance(content, dict):
+                    parts = content.get("parts", [])
+                    for part in parts:
+                        if isinstance(part, dict) and "text" in part:
+                            text = part["text"]
+                            complete_text.append(text)
+                            console.print(text, end="", style="green")
+                            
+                    
+                # 显示错误信息
+                if "error" in event:
+                    console.print(f"\n[red]Error: {event['error']}[/red]")
+        
+        # 显示完整响应
+        if complete_text:
+            console.print("\n\n[cyan]━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━[/cyan]")
+            console.print(f"[cyan]📝 Complete response:[/cyan] {''.join(complete_text)}")
+        
+        return str(result_list)
+    
+    # 处理非流式响应
+    console.print("[cyan]📝 Response:[/cyan]")
+    if isinstance(response, dict):
+        console.print(json.dumps(response, indent=2, ensure_ascii=False))
+    else:
+        console.print(response)
+    
+    return str(response)

agentkit/toolkit/cli/cli_launch.py

@@ -0,0 +1,34 @@
+# 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 - Launch command implementation."""
+
+from pathlib import Path
+import typer
+from rich.console import Console
+from agentkit.toolkit.cli.cli_build import build_command
+from agentkit.toolkit.cli.cli_deploy import deploy_command
+
+# Note: 不要在文件开头导很重的包，不然会导致命令很卡(import包很慢)
+
+console = Console()
+
+
+def launch_command(
+    config_file: Path = typer.Option("agentkit.yaml", help="Configuration file"),
+):
+    """Build and deploy in one command."""
+    console.print("[green]Launching agent...[/green]")
+    build_command(config_file=config_file)
+    deploy_command(config_file=config_file)

agentkit/toolkit/cli/cli_status.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 - Status 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 status_command(
+    config_file: Path = typer.Option("agentkit.yaml", help="Configuration file"),
+):
+    """Show current status of the agent runtime."""
+    from agentkit.toolkit.workflows import WORKFLOW_REGISTRY
+    try:
+        config = get_config(config_path=config_file)
+        common_config = config.get_common_config()
+        workflow_type = common_config.launch_type
+
+        if workflow_type not in WORKFLOW_REGISTRY:
+            console.print(f"[red]Error: Unknown workflow type '{workflow_type}'[/red]")
+            raise typer.Exit(1)
+
+        workflow = WORKFLOW_REGISTRY[workflow_type]()
+        workflow_config = config.get_workflow_config(workflow_type)
+        status_result = workflow.status(workflow_config)
+
+        if isinstance(status_result, dict) and status_result.get('error'):
+            console.print(f"[red]Status query failed: {status_result['error']}[/red]")
+            raise typer.Exit(1)
+
+        console.print(status_result)
+
+    except Exception as e:
+        console.print(f"[red]Status query failed: {e}[/red]")
+        raise typer.Exit(1)

agentkit/toolkit/cli/cli_version.py

@@ -0,0 +1,87 @@
+# 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 - Version command implementation."""
+
+from pathlib import Path
+from importlib.metadata import version as get_version, PackageNotFoundError
+import sys
+from rich.panel import Panel
+from rich.console import Console
+
+# Note: 不要在文件开头导很重的包，不然会导致命令很卡(import包很慢)
+
+# Python 3.11+ has tomllib, older versions need tomli
+if sys.version_info >= (3, 11):
+    import tomllib
+else:
+    try:
+        import tomli as tomllib
+    except ImportError:
+        tomllib = None
+
+console = Console()
+
+
+def get_package_version() -> str:
+    """Get package version from installed package or pyproject.toml."""
+    # Try to get version from installed package
+    try:
+        return get_version("agentkit-sdk-python")
+    except PackageNotFoundError:
+        pass
+    
+    # Fallback: read from pyproject.toml using toml parser
+    if tomllib is not None:
+        try:
+            # Find pyproject.toml in the package directory
+            current_file = Path(__file__)
+            # Go up from agentkit/toolkit/cli/cli.py to workspace root
+            pyproject_path = current_file.parent.parent.parent.parent / "pyproject.toml"
+            
+            if pyproject_path.exists():
+                with open(pyproject_path, "rb") as f:
+                    data = tomllib.load(f)
+                    return data.get("project", {}).get("version", "unknown")
+        except Exception:
+            pass
+    
+    # Last resort: parse pyproject.toml manually for version line
+    try:
+        current_file = Path(__file__)
+        pyproject_path = current_file.parent.parent.parent.parent / "pyproject.toml"
+        
+        if pyproject_path.exists():
+            with open(pyproject_path, "r") as f:
+                for line in f:
+                    if line.strip().startswith("version"):
+                        # Parse line like: version = "0.0.7.34"
+                        parts = line.split("=")
+                        if len(parts) == 2:
+                            version_str = parts[1].strip().strip('"').strip("'")
+                            return version_str
+    except Exception:
+        pass
+    
+    return "unknown"
+
+
+def version_command():
+    """Show AgentKit version information."""
+    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"
+    ))

agentkit/toolkit/cli/utils.py

@@ -0,0 +1,47 @@
+from typing import Literal, get_args, get_origin
+
+from InquirerPy import resolver
+from pydantic import BaseModel
+
+
+def prompt_base_model(model: type[BaseModel]) -> dict:
+    prompts = []
+
+    for field_name, model_field in model.model_fields.items():
+        if get_origin(model_field.annotation) == Literal:
+            prompts.append(
+                {
+                    "type": "list",
+                    "name": field_name,
+                    "default": model_field.default if model_field.default else "",
+                    "message": model_field.description
+                    if model_field.description
+                    else field_name,
+                    "choices": list(get_args(model_field.annotation)),
+                }
+            )
+        elif model_field.annotation is bool:
+            prompts.append(
+                {
+                    "type": "confirm",
+                    "name": field_name,
+                    "default": model_field.default if model_field.default else False,
+                    "message": model_field.description
+                    if model_field.description
+                    else field_name,
+                }
+            )
+        else:
+            prompts.append(
+                {
+                    "type": "input",
+                    "name": field_name,
+                    "default": str(model_field.default) if model_field.default else "",
+                    "message": model_field.description
+                    if model_field.description
+                    else field_name,
+                }
+            )
+
+    responses = resolver.prompt(prompts)
+    return responses

agentkit/toolkit/config/__init__.py

@@ -0,0 +1,52 @@
+# 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 .config import (
+    AgentkitConfigManager,
+    CommonConfig,
+    ConfigUpdateResult,
+    get_config,
+    create_config_update_result,
+    set_global_config_file_path,
+    get_global_config_file_path,
+)
+
+from .workflow_configs import (
+    LocalDockerConfig_v1,
+    HybridVeAgentkitConfig_v1,
+    VeAgentkitConfig,
+)
+
+from .utils import is_valid_config, is_invalid_config, merge_runtime_envs
+from .constants import *
+
+__all__ = [
+    "AgentkitConfigManager",
+    "CommonConfig",
+    "ConfigUpdateResult",
+    "get_config",
+    "create_config_update_result",
+    "AUTO_CREATE_VE",
+    "DEFAULT_WORKSPACE_NAME",
+    "DEFAULT_CR_NAMESPACE",
+    "DEFAULT_IMAGE_TAG",
+    "is_valid_config",
+    "is_invalid_config",
+    "merge_runtime_envs",
+    # Workflow 配置类
+    "LocalDockerConfig_v1",
+    "HybridVeAgentkitConfig_v1",
+    "VeAgentkitConfig",
+]