#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
测试AI智能体API
"""

import json
import requests
import sys

def test_api():
    """测试API功能"""
    base_url = "http://localhost:5001"
    
    print("🧪 测试AI智能体API")
    print("=" * 50)
    
    # 1. 测试健康检查
    try:
        response = requests.get(f"{base_url}/api/health", timeout=5)
        if response.status_code == 200:
            print("✅ 健康检查通过")
            print(f"   响应: {response.json()}")
        else:
            print(f"❌ 健康检查失败: {response.status_code}")
            return False
    except Exception as e:
        print(f"❌ 无法连接到API服务: {e}")
        return False
    
    # 2. 测试聊天功能
    test_messages = [
        "你好",
        "你们公司是做什么的？",
        "营销功能包括什么？",
        "How much does it cost?",
        "สวัสดี"
    ]
    
    for message in test_messages:
        print(f"\n🤖 测试消息: {message}")
        try:
            response = requests.post(
                f"{base_url}/api/chat",
                json={"message": message},
                timeout=30
            )
            
            if response.status_code == 200:
                data = response.json()
                if data.get("success"):
                    result = data["data"]
                    print(f"   语言: {result['language']}")
                    print(f"   意图: {result['intent']}")
                    print(f"   回复: {result['response'][:100]}...")
                    if result.get('related_questions'):
                        print(f"   相关问题: {len(result['related_questions'])}个")
                else:
                    print(f"   ❌ API返回错误: {data.get('error')}")
            else:
                print(f"   ❌ HTTP错误: {response.status_code}")
                
        except Exception as e:
            print(f"   ❌ 请求失败: {e}")
    
    print("\n✅ API测试完成")
    return True

if __name__ == "__main__":
    test_api()