大模型学习笔记八:手撕AutoGPT

一、功能需求

  • 目的
    设计一个 Agent,自动选择使用以下工具回答用户的问题
  • 问题
1、查看目录下的文件
2、基于给定的文档回答用户问题
3、查看与分析 Excel 文件
4、撰写文档
5、调用 Email 客户端发邮件

二、演示用例

  • 实验资料(实验中使用三个文档演示 Agent 的能力)
./data
 |__2023年8月-9月销售记录.xlsx
 |__供应商名录.xlsx
 |__供应商资格要求.pdf
  • 资料展示
    在这里插入图片描述

  • 测试用例输入举例

1、9 月份的销售额是多少
2、销售总额最大的产品是什么
3、帮我找最近一个月出销售额不达标的供应商
4、给对方发一封邮通知此事
5、对比 8 月和 9 月销售情况,写一份报告

三、核心模块流程图

在这里插入图片描述

四、代码分析

1)Agent类目录创建智能体对象

初始化准备工作:

1、语言模型:llm
2、工具集 tools
3、工作数据地址:work_dir
4、最多能思考多少步:max_thiought_steps
5、常识记忆的向量数据库连接池:memery_retriever
6、output_parse专门解析action传出来的json格式的数据
7、robust_parser用来矫正输出json格式的错误
8、__iniy_prompt_templates用来初始化模板(需要用接口适配langchain输出中文有时候打印Unicode码,看的不舒服,函数名字是__chinese_friendly)
9、初始化chain(用主要prompt和结束的prompt串起来)

在这里插入图片描述

2)开始主流程

①初始化常识和短识的记忆(短识记忆初始化为空,常识是连接向量数据库)
②记录思考步数
③开始主循环推理,若输出结果为FINISH就break,否则就执行选择出来的action,并更新短识记忆
④若大于最大思考步数,则输出未能完成任务
⑤更新长识记忆
在这里插入图片描述

3)在prompt的main目录输入主prompt和最后prompt

在这里插入图片描述

  • 注意点
    ①有时候目录会在查找过程中丢失,所以访问文件确保文件路径调整
    ②对于一些准则(总结方法论)告诉他约束
    ③输出形式要按步骤输出思考过程,然后加自我反思
    ④观察执行记录和自我反思,进一步思考
    ⑤根据反思和思考,一步步推理

4)增加实际的工具集tools(也就是函数)

①实际工具类文件是Tools.py,比如如下的FileTool.py定义了返回搜索数据文件的名字的函数
在这里插入图片描述

在这里插入图片描述
②定义读取PDF的函数,类似RAG的部分,这里拿langchain原生接口代替(不推荐,FileQATool.py)
在这里插入图片描述
③按照客户需求生成文档的函数
在这里插入图片描述

④分析Excel的函数工具(根据表名字、行列名字信息、具体格信息分析文件)ExcelTool.py和生成代码去分析数据的工具PythonTool.py,并且在prompts/tools目录下填上excel分析的prompt
在这里插入图片描述

在这里插入图片描述
⑤发邮件的工具EmailTool.py
在这里插入图片描述

5)新增文件夹Utils工具来打印中间结果

新文件callbackfunc.py

在这里插入图片描述

和带颜色的打印

在这里插入图片描述

本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
THE END
分享
二维码
< <上一篇
下一篇>>