# 安装依赖（如果需要）
# !pip install -q hello-agents[all] requests beautifulsoup4 python-dotenv

# 导入必要的库
import os
import sys
from dotenv import load_dotenv

# 加载环境变量
load_dotenv()

# 添加项目根目录到Python路径
sys.path.append('.')

print("✅ 环境配置完成")
print(f"📁 当前工作目录: {os.getcwd()}")
print(f"🐍 Python版本: {sys.version}")

# 导入工具定义
from src.tools.browser_tool import BrowserTool
from src.tools.terminal_tool import TerminalTool

# 创建工具实例
browser_tool = BrowserTool()
terminal_tool = TerminalTool()

print("📋 可用工具列表:")
print(f"1. {browser_tool.name}: {browser_tool.description}")
print(f"2. {terminal_tool.name}: {terminal_tool.description}")

# 导入智能体类
from src.agents.agent_universal import UniversalAgent

# 创建智能体实例
agent = UniversalAgent()

print("🚀 智能体创建成功!")
print(f"📛 智能体名称: {agent.name}")
print(f"🔧 已注册工具数量: {len(agent.tool_registry.tools)}")

# 显示已注册的工具
print("\n📋 已注册的工具:")
for tool_name, tool in agent.tool_registry.tools.items():
    print(f"  - {tool_name}: {tool.description}")

# 演示搜索功能
query = "Python人工智能最新发展"
print(f"🔍 搜索查询: {query}")
print("=" * 50)

try:
    result = agent.run(query)
    print("\n📋 搜索结果:")
    print(result)
except Exception as e:
    print(f"❌ 搜索失败: {e}")

# 演示终端功能（安全命令）
command = "pwd"
print(f"💻 执行命令: {command}")
print("=" * 50)

try:
    result = agent.run(f"执行 {command}")
    print("\n📋 执行结果:")
    print(result)
except Exception as e:
    print(f"❌ 命令执行失败: {e}")

# 演示复杂任务
complex_task = "搜索LangChain框架的最新版本信息，然后查看当前目录的文件列表"
print(f"🎯 复杂任务: {complex_task}")
print("=" * 50)

try:
    result = agent.run(complex_task)
    print("\n📋 处理结果:")
    print(result)
except Exception as e:
    print(f"❌ 任务处理失败: {e}")

import time

# 简单的性能测试
test_queries = [
    "Python基础语法",
    "当前目录文件列表",
    "人工智能发展趋势"
]

print("📊 性能测试开始...")
print("=" * 50)

for i, query in enumerate(test_queries, 1):
    print(f"\n🧪 测试 {i}: {query}")
    
    start_time = time.time()
    try:
        result = agent.run(query)
        end_time = time.time()
        
        response_time = end_time - start_time
        result_length = len(result) if result else 0
        
        print(f"✅ 成功 - 响应时间: {response_time:.2f}秒, 结果长度: {result_length}字符")
        print(f"📝 结果预览: {result[:100]}..." if len(result) > 100 else f"📝 结果: {result}")
    except Exception as e:
        end_time = time.time()
        response_time = end_time - start_time
        print(f"❌ 失败 - 响应时间: {response_time:.2f}秒, 错误: {e}")

print("\n📊 性能测试完成")