agent
88 lines
4.6 KiB
Python
88 lines
4.6 KiB
Python
"""
|
||
Step 03 练习题：扩展 SimpleAgent
|
||
|
||
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
||
|
||
🎯 练习目标：
|
||
1. 巩固 Agent 循环的运行机制
|
||
2. 增强 Brain 的决策能力
|
||
3. 体验 Tool 的注册流程
|
||
|
||
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
||
"""
|
||
|
||
# ═══════════════════════════════════════════════════════════════════════════════
|
||
# 练习 1：实现 DateTimeTool 并注册到 Agent
|
||
# ═══════════════════════════════════════════════════════════════════════════════
|
||
|
||
"""
|
||
任务：
|
||
在 SimpleAgent 中添加一个 DateTimeTool，提供「现在几点 / 今天日期」能力。
|
||
|
||
要求：
|
||
1. 继承 BaseTool
|
||
2. name = "datetime"，description 描述清楚能做什么
|
||
3. execute(**kwargs) 接收 operation，支持：
|
||
- "now" -> 返回当前时间字符串（"%Y-%m-%d %H:%M:%S"）
|
||
- "today" -> 返回当前日期字符串（"%Y-%m-%d"）
|
||
- "weekday" -> 返回今天是星期几（中文，如 "星期一"）
|
||
4. 注册到 SimpleAgent.tools 字典中
|
||
5. 测试：用户输入「现在几点」时 Brain 能正确选择 datetime 工具
|
||
"""
|
||
|
||
|
||
# ═══════════════════════════════════════════════════════════════════════════════
|
||
# 练习 2：改进 Brain 的工具匹配
|
||
# ═══════════════════════════════════════════════════════════════════════════════
|
||
|
||
"""
|
||
任务：
|
||
当前 AgentBrain.decide() 用「关键词 + 表达式正则」匹配 calculator，
|
||
对「iOS / Android 兼容性」「产品 A+」这种文本会误判。
|
||
|
||
要求：
|
||
1. 在 AgentBrain.decide() 中加入你新加的 DateTimeTool 的路由
|
||
2. 修复 calculator 匹配的脆弱性（例如优先匹配明确的算式语法）
|
||
3. 让 Brain 在没有工具可调时返回 {"action": "respond", "response": "..."}
|
||
|
||
提示：
|
||
- 可用正则在 user_input 中提取首个形如「数字 运算符 数字」的子串
|
||
- 用 keyword in user_input 检测「现在」「今天」「星期」触发 datetime
|
||
"""
|
||
|
||
|
||
# ═══════════════════════════════════════════════════════════════════════════════
|
||
# 练习 3：让 Agent 暴露对话快照
|
||
# ═══════════════════════════════════════════════════════════════════════════════
|
||
|
||
"""
|
||
任务：
|
||
给 SimpleAgent 增加 snapshot()/restore(snap) 方法，用于保存和恢复会话。
|
||
|
||
要求：
|
||
1. snapshot() 返回 dict，包含 messages、tool_result、current_action
|
||
2. restore(snap) 用 snap 覆盖对应字段
|
||
3. 验证：snapshot -> 多轮对话 -> restore -> 状态回到 snapshot 时刻
|
||
|
||
提示：
|
||
- copy.deepcopy() 避免引用共享
|
||
- 只恢复可序列化的字段，不要把 self.brain / self.tools 一起覆盖
|
||
"""
|
||
|
||
|
||
# ═══════════════════════════════════════════════════════════════════════════════
|
||
# 测试
|
||
# ═══════════════════════════════════════════════════════════════════════════════
|
||
|
||
def test_exercises():
|
||
from step_03_simple_agent.concept import SimpleAgent
|
||
|
||
agent = SimpleAgent()
|
||
print("当前已注册工具:", list(agent.tools.keys()))
|
||
|
||
# TODO: 你的测试
|
||
|
||
|
||
if __name__ == "__main__":
|
||
test_exercises()
|