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

# 导入必要的库
from hello_agents import SimpleAgent, HelloAgentsLLM
from hello_agents.tools import BaseTool
import os
from dotenv import load_dotenv

# 加载环境变量
load_dotenv()

print("✅ 环境配置完成")

class CustomTool(BaseTool):
    """自定义工具类"""
    
    name = "tool_name"
    description = "工具描述"
    
    def run(self, query: str) -> str:
        """工具执行逻辑"""
        # 实现你的工具逻辑
        return f"处理结果：{query}"

print("✅ 工具定义完成")

# 创建LLM
llm = HelloAgentsLLM()

# 定义系统提示词
system_prompt = """你是一个智能助手。

你的任务是：
1. 理解用户的需求
2. 使用合适的工具
3. 提供有帮助的回答
"""

# 创建智能体
agent = SimpleAgent(
    name="示例智能体",
    llm=llm,
    system_prompt=system_prompt
)

# 添加工具
agent.add_tool(CustomTool())

print("✅ 智能体构建完成")

# 示例1：基础功能
print("=== 示例1：基础功能 ===")
result = agent.run("你的测试输入")
print(result)

# 示例2：复杂场景
print("\n=== 示例2：复杂场景 ===")
result = agent.run("更复杂的测试输入")
print(result)

# 评估代码
# 例如：测试准确率、响应时间等

print("✅ 评估完成")