"""
快速测试创作灵感生成功能
"""

import asyncio
from ai_agent import create_agent


async def quick_test():
    """快速测试"""
    
    print("=" * 80)
    print("创作灵感生成 - 快速测试")
    print("=" * 80)
    print()
    
    # 获取用户输入
    print("请输入你的创作需求（或直接回车使用示例）：")
    print("示例：我想做一些美食相关的短视频，主要是家常菜的制作教程")
    print()
    
    user_input = input("你的需求: ").strip()
    
    if not user_input:
        user_input = "我想做一些美食相关的短视频，主要是家常菜的制作教程"
        print(f"使用示例需求: {user_input}")
    
    print("\n" + "=" * 80)
    print("正在生成创作灵感...")
    print("=" * 80)
    print()
    
    try:
        # 创建Agent
        agent = create_agent()
        
        # 运行Agent
        result = await agent.run(
            user_input=user_input,
            system_prompt_file="prompts/agent_prompt.md",
            max_iterations=15
        )
        
        print("\n" + "=" * 80)
        print("结果")
        print("=" * 80)
        print()
        
        if result["success"]:
            print(result["final_answer"])
            print("\n" + "=" * 80)
            print(f"✓ 成功！共调用了 {len(result['tool_calls'])} 个工具")
        else:
            print(f"✗ 失败: {result['error']}")
            
    except Exception as e:
        print(f"\n✗ 出错: {e}")
        import traceback
        traceback.print_exc()


if __name__ == "__main__":
    asyncio.run(quick_test())