"""
API接口测试脚本
"""

import requests
import json


def test_agent_api():
    """测试Agent API"""
    
    print("=" * 80)
    print("API接口测试")
    print("=" * 80)
    print()
    
    # API地址
    url = "http://localhost:8001/api/agent"
    
    # 测试用例
    test_cases = [
        {
            "name": "美食类内容",
            "query": "我想做一些美食相关的短视频，主要是家常菜的制作教程"
        },
        {
            "name": "校园类内容",
            "query": "我想拍一些关于大学生活的有趣视频，记录校园日常"
        },
        {
            "name": "健身类内容",
            "query": "我想做一些关于健身的短视频，分享简单的居家锻炼方法"
        }
    ]
    
    # 选择测试用例
    print("可用的测试用例：")
    for i, case in enumerate(test_cases, 1):
        print(f"{i}. {case['name']}: {case['query']}")
    print()
    
    choice = input("选择测试用例（1-3，直接回车选择1）: ").strip()
    
    if not choice:
        choice = "1"
    
    try:
        test_index = int(choice) - 1
        if test_index < 0 or test_index >= len(test_cases):
            test_index = 0
    except ValueError:
        test_index = 0
    
    selected_test = test_cases[test_index]
    
    print(f"\n测试: {selected_test['name']}")
    print(f"查询: {selected_test['query']}")
    print()
    
    # 构建请求
    payload = {
        "query": selected_test['query'],
        "max_iterations": 15
    }
    
    print("正在调用API...")
    print(f"URL: {url}")
    print(f"Payload: {json.dumps(payload, ensure_ascii=False, indent=2)}")
    print()
    
    try:
        # 发送请求
        response = requests.post(url, json=payload, timeout=300)
        
        print("=" * 80)
        print("响应")
        print("=" * 80)
        print()
        
        if response.status_code == 200:
            result = response.json()
            
            if result.get("success"):
                print("✓ 成功！")
                print()
                print(result.get("final_answer", ""))
                print()
                print(f"迭代次数: {result.get('iteration')}")
                print(f"工具调用: {len(result.get('tool_calls', []))}")
            else:
                print("✗ 失败")
                print(f"错误: {result.get('error')}")
        else:
            print(f"✗ HTTP错误: {response.status_code}")
            print(response.text)
            
    except requests.exceptions.ConnectionError:
        print("✗ 连接失败")
        print()
        print("请确保API服务已启动：")
        print("  python api.py")
        print()
    except Exception as e:
        print(f"✗ 出错: {e}")


if __name__ == "__main__":
    test_agent_api()