test_api.py

#!/usr/bin/env python3
"""
Script để test các agent APIs với schema đơn giản
"""

import asyncio
import json
import aiohttp

API_BASE = "http://localhost:8000/api/v1"

async def test_agents():
    """Test tất cả các agents"""
    async with aiohttp.ClientSession() as session:
        
        # 1. Liệt kê tất cả agents
        print("=== DANH SÁCH AGENTS ===")
        async with session.get(f"{API_BASE}/agents/") as resp:
            agents = await resp.json()
            print(json.dumps(agents, indent=2, ensure_ascii=False))
        
        print("\n" + "="*50 + "\n")
        
        # 2. Test LLM Agent
        print("=== TEST LLM AGENT ===")
        llm_request = {
            "text": "Hãy viết một câu chuyện ngắn về trí tuệ nhân tạo"
        }
        
        async with session.post(f"{API_BASE}/agents/llm_agent/process", 
                               json=llm_request) as resp:
            result = await resp.json()
            print(json.dumps(result, indent=2, ensure_ascii=False))
        
        print("\n" + "="*50 + "\n")
        
        # 3. Test Vision Agent  
        print("=== TEST VISION AGENT ===")
        vision_request = {
            "image": "https://example.com/sample-image.jpg"
        }
        
        async with session.post(f"{API_BASE}/agents/vision_agent/process", 
                               json=vision_request) as resp:
            result = await resp.json()
            print(json.dumps(result, indent=2, ensure_ascii=False))
        
        print("\n" + "="*50 + "\n")
        
        # 4. Test File Processing Agent
        print("=== TEST FILE PROCESSING AGENT ===")
        file_request = {
            "file": "https://example.com/sample-document.pdf"
        }
        
        async with session.post(f"{API_BASE}/agents/file_processing_agent/process", 
                               json=file_request) as resp:
            result = await resp.json()
            print(json.dumps(result, indent=2, ensure_ascii=False))
        
        print("\n" + "="*50 + "\n")
        
        # 5. Test với nhiều input types (chỉ text sẽ được xử lý)
        print("=== TEST MIXED INPUT ===")
        mixed_request = {
            "text": "Phân tích hình ảnh này",
            "image": "https://example.com/image.jpg"
        }
        
        async with session.post(f"{API_BASE}/agents/llm_agent/process", 
                               json=mixed_request) as resp:
            result = await resp.json()
            print("LLM Agent chỉ xử lý text:")
            print(json.dumps(result, indent=2, ensure_ascii=False))

if __name__ == "__main__":
    print("Bắt đầu test API...")
    print("Đảm bảo server đang chạy: uvicorn app.main:app --reload")
    print("="*60)
    
    try:
        asyncio.run(test_agents())
    except Exception as e:
        print(f"Lỗi khi test: {e}")
        print("Hãy kiểm tra server có đang chạy không?")