前面课程中,已经把处理单独业务场景的工具函数都定义好了,接下来,就来定义一个Agent,通过描述的方式,引导大模型找相应的工具,来回答用户问题。

代码示例

1、定义Tool,并添加描述

注意,graph_func比较特殊,因为Agent会对输入文本进行加工,但图谱查询中,需要对原问题进行实体抽取,所以用匿名函数的方式,把原问题带过去。

内容不可见,请联系管理员开通权限。

2、定义Agent

内容不可见,请联系管理员开通权限。

3、调用测试

最后一个问题比较特殊,看上去跟疾病相关,但在库里查不到,以此来测试Agent思考的流程。

内容不可见,请联系管理员开通权限。

整体来看,效果还是不错的,Agent可以根据描述,把问题精准的定位到对应的处理逻辑上,并且,在没有查询结果的时候,会用Google搜索做兜底。至此,这个项目的核心功能就开发完了。

本文链接:http://www.ichenhua.cn/edu/note/719

版权声明:本文为「陈华编程」原创课程讲义,请给与知识创作者起码的尊重,未经许可不得传播或转售!