awoooi/scripts/test_agent_sdk.py

#!/usr/bin/env python3
"""
Phase 9 Agent SDK POC Verification Script
==========================================

驗證 claude-agent-sdk 是否能在 AWOOOI 專案中正常運作。

Usage:
    python scripts/test_agent_sdk.py

Environment Variables:
    ANTHROPIC_API_KEY - 如果設置，將測試基本 API 調用
"""

import os
import sys
from typing import Any


def test_import() -> bool:
    """Test basic import of claude-agent-sdk."""
    print("=" * 60)
    print("Phase 9 Agent SDK POC Verification")
    print("=" * 60)
    print()

    try:
        import claude_agent_sdk

        version = getattr(claude_agent_sdk, "__version__", "N/A")
        print(f"[PASS] claude_agent_sdk imported successfully")
        print(f"       Version: {version}")
        return True
    except ImportError as e:
        print(f"[FAIL] Failed to import claude_agent_sdk: {e}")
        return False


def test_core_exports() -> bool:
    """Test that core exports are available."""
    print()
    print("-" * 60)
    print("Testing core exports...")
    print("-" * 60)

    required_exports = [
        "query",
        "ClaudeAgentOptions",
        "ClaudeSDKClient",
        "Message",
        "UserMessage",
        "AssistantMessage",
        "SystemMessage",
    ]

    all_passed = True
    for export in required_exports:
        try:
            from claude_agent_sdk import query  # noqa: F401

            module = __import__("claude_agent_sdk", fromlist=[export])
            getattr(module, export)
            print(f"  [PASS] {export}")
        except (ImportError, AttributeError) as e:
            print(f"  [FAIL] {export}: {e}")
            all_passed = False

    return all_passed


def test_types() -> bool:
    """Test type definitions are available."""
    print()
    print("-" * 60)
    print("Testing type definitions...")
    print("-" * 60)

    try:
        from claude_agent_sdk import (
            AssistantMessage,
            ClaudeAgentOptions,
            Message,
            PermissionMode,
            StreamEvent,
            TextBlock,
            ThinkingConfig,
            ToolUseBlock,
            UserMessage,
        )

        print(f"  [PASS] ClaudeAgentOptions: {ClaudeAgentOptions}")
        print(f"  [PASS] Message: {Message}")
        print(f"  [PASS] UserMessage: {UserMessage}")
        print(f"  [PASS] AssistantMessage: {AssistantMessage}")
        print(f"  [PASS] TextBlock: {TextBlock}")
        print(f"  [PASS] ToolUseBlock: {ToolUseBlock}")
        print(f"  [PASS] StreamEvent: {StreamEvent}")
        print(f"  [PASS] ThinkingConfig: {ThinkingConfig}")
        print(f"  [PASS] PermissionMode: {PermissionMode}")
        return True
    except ImportError as e:
        print(f"  [FAIL] Type import failed: {e}")
        return False


def test_mcp_integration() -> bool:
    """Test MCP server configuration types."""
    print()
    print("-" * 60)
    print("Testing MCP integration types...")
    print("-" * 60)

    try:
        from claude_agent_sdk import (
            McpServerConfig,
            McpServerInfo,
            McpServerStatus,
            McpStatusResponse,
            McpToolInfo,
        )

        print(f"  [PASS] McpServerConfig: {McpServerConfig}")
        print(f"  [PASS] McpServerInfo: {McpServerInfo}")
        print(f"  [PASS] McpServerStatus: {McpServerStatus}")
        print(f"  [PASS] McpStatusResponse: {McpStatusResponse}")
        print(f"  [PASS] McpToolInfo: {McpToolInfo}")
        return True
    except ImportError as e:
        print(f"  [FAIL] MCP type import failed: {e}")
        return False


def test_hook_types() -> bool:
    """Test hook-related types for agent customization."""
    print()
    print("-" * 60)
    print("Testing hook types...")
    print("-" * 60)

    try:
        from claude_agent_sdk import (
            HookCallback,
            HookContext,
            HookInput,
            PermissionRequestHookInput,
            PostToolUseHookInput,
            PreToolUseHookInput,
            StopHookInput,
        )

        print(f"  [PASS] HookCallback: {HookCallback}")
        print(f"  [PASS] HookContext: {HookContext}")
        print(f"  [PASS] HookInput: {HookInput}")
        print(f"  [PASS] PreToolUseHookInput: {PreToolUseHookInput}")
        print(f"  [PASS] PostToolUseHookInput: {PostToolUseHookInput}")
        print(f"  [PASS] PermissionRequestHookInput: {PermissionRequestHookInput}")
        print(f"  [PASS] StopHookInput: {StopHookInput}")
        return True
    except ImportError as e:
        print(f"  [FAIL] Hook type import failed: {e}")
        return False


def test_session_management() -> bool:
    """Test session management functions."""
    print()
    print("-" * 60)
    print("Testing session management functions...")
    print("-" * 60)

    try:
        from claude_agent_sdk import (
            get_session_info,
            get_session_messages,
            list_sessions,
            rename_session,
            tag_session,
        )

        print(f"  [PASS] list_sessions: {list_sessions}")
        print(f"  [PASS] get_session_info: {get_session_info}")
        print(f"  [PASS] get_session_messages: {get_session_messages}")
        print(f"  [PASS] rename_session: {rename_session}")
        print(f"  [PASS] tag_session: {tag_session}")
        return True
    except ImportError as e:
        print(f"  [FAIL] Session management import failed: {e}")
        return False


def test_api_key_available() -> bool:
    """Check if ANTHROPIC_API_KEY is available."""
    print()
    print("-" * 60)
    print("Checking environment variables...")
    print("-" * 60)

    api_key = os.environ.get("ANTHROPIC_API_KEY")
    if api_key:
        # Don't print the actual key
        masked = api_key[:8] + "..." + api_key[-4:] if len(api_key) > 12 else "***"
        print(f"  [INFO] ANTHROPIC_API_KEY is set (masked: {masked})")
        return True
    else:
        print("  [INFO] ANTHROPIC_API_KEY not set - skipping API call test")
        return False


def list_required_env_vars() -> None:
    """List all required environment variables for full functionality."""
    print()
    print("=" * 60)
    print("Required Environment Variables for Full Functionality")
    print("=" * 60)
    print()

    env_vars = [
        ("ANTHROPIC_API_KEY", "Required for API calls", True),
        ("CLAUDE_API_KEY", "Alternative API key (optional)", False),
        ("CLAUDE_CONFIG_DIR", "Custom config directory (optional)", False),
    ]

    print("| Variable           | Description                      | Required |")
    print("|-------------------|----------------------------------|----------|")
    for var, desc, required in env_vars:
        status = "Yes" if required else "No"
        is_set = "[SET]" if os.environ.get(var) else "[NOT SET]"
        print(f"| {var:<17} | {desc:<32} | {status:<8} | {is_set}")
    print()


def main() -> int:
    """Run all tests and return exit code."""
    results: dict[str, bool] = {}

    # Run tests
    results["import"] = test_import()
    results["core_exports"] = test_core_exports()
    results["types"] = test_types()
    results["mcp_integration"] = test_mcp_integration()
    results["hooks"] = test_hook_types()
    results["session_management"] = test_session_management()

    # Check API key
    has_api_key = test_api_key_available()

    # List required env vars
    list_required_env_vars()

    # Summary
    print("=" * 60)
    print("Summary")
    print("=" * 60)
    print()

    passed = sum(1 for v in results.values() if v)
    total = len(results)

    for test_name, passed_test in results.items():
        status = "[PASS]" if passed_test else "[FAIL]"
        print(f"  {status} {test_name}")

    print()
    print(f"Results: {passed}/{total} tests passed")
    print()

    if passed == total:
        print("[SUCCESS] claude-agent-sdk is ready for Phase 9 integration!")
        print()
        print("Next steps:")
        print("  1. Add 'claude-agent-sdk>=0.1.50' to apps/api/pyproject.toml")
        print("  2. Set ANTHROPIC_API_KEY in environment")
        print("  3. Create agent definitions in apps/api/src/agents/")
        return 0
    else:
        print("[WARNING] Some tests failed. Please review the output above.")
        return 1


if __name__ == "__main__":
    sys.exit(main())