深入理解Agent:从0实现function call

Function的调用时Agent实现很重要的一步,只有了解了这个原理才可以更好的创建Agent。

我将不使用任何langchain等框架或者coze等平台,从0开始构建一个可以调用function的Agent。

源代码已经上传github:github.com/astordu/age...

一、场景

Agent的目标:可以回答关于天气的问题。 用到的function:调用某地方的天气情况,并且反馈

流程: 1.思考: 用户输入问题,模型先对问题进行分析 2.行动: 如果问到了天气问题,则分析出需要调用的function以及function要传入的参数 3.响应:function返回后,将答案整理好回复给用户。

函数自定:

  1. 我们先定义一个获取天气的函数(属于Tools中的一个,这里用于演示,不做真实调用):

    python 复制代码
         def get_weather(location):
                 return "天气晴朗"
  2. 再定义一个大模型的发送信息的方法:

    ini 复制代码
     def send_messages(messages):
             response = client.chat.completions.create(
             model="deepseek-chat",
             messages=messages
             )
             return response.choices[0].message
    
     client = OpenAI(
             api_key="<你的deepseek的key>",
             base_url="https://api.deepseek.com",
     )

二、设计思路

从"用户提出问题"到"思考"到"响应"其实是调用了多次LLM模型。

所以我们要求模型按照顺序去调用LLM:

css 复制代码
你在运行一个"思考","工具调用","响应"循环。每次只运行一个阶段

1."思考"阶段:你要仔细思考用户的问题
2."工具调用阶段":选择可以调用的工具,并且输出对应工具需要的参数
3."响应"阶段:根据工具调用返回的影响,回复用户问题。

已有的工具如下:
get_weather:
e.g. get_weather:天津
返回天津的天气情况

Example:
question:天津的天气怎么样?
thought:我应该调用工具查询天津的天气情况
Action:
{
	"function_name":"get_response_time"
	"function_params":{
		"location":"天津"
	}
}
调用Action的结果:"天气晴朗"
Answer:天津的天气晴朗

上边的逻辑正好可以当作system的提示语:

ini 复制代码
system="""
	你在运行一个"思考","工具调用","响应"循环。每次只运行一个阶段
	
	1."思考"阶段:你要仔细思考用户的问题
	2."工具调用阶段":选择可以调用的工具,并且输出对应工具需要的参数
	3."响应"阶段:根据工具调用返回的影响,回复用户问题。
	
	已有的工具如下:
	get_weather:
	e.g. get_weather:天津
	返回天津的天气情况
	
	Example:
	question:天津的天气怎么样?
	thought:我应该调用工具查询天津的天气情况
	Action:
	{
		"function_name":"get_response_time"
		"function_params":{
			"location":"天津"
		}
	}
	调用Action的结果:"天气晴朗"
	Answer:天津的天气晴朗
"""

第一步,向模型提问一个问题

ini 复制代码
question="北京天气怎么样"

messages = [{"role": "system", "content": system_prompt},
{"role": "user", "content": question}]

message = send_messages(messages)
print(f"Model-1th>\n {message.content}")

返回值:

json 复制代码
Model-1th>
 thought:我应该调用工具查询北京的天气情况
Action:
{
        "function_name":"get_weather",
        "function_params":{
                "location":"北京"
        }
}

可以看出模型已经进行了思考,并且返回了可以调用的工具了

第二步,如果从"第一步"的返回值中可以提取调用工具的json

第三步,调用真实工具,获取真实结果(这里是伪代码)

scss 复制代码
    invoke_function(**function_name,**function_params)

第四步,将工具调用的结果追加到message中,一起给到模型,让它总结回答:

python 复制代码
messages.append({"role": "assistant", "content": f"调用Action的结果:{tianqi}"})
message = send_messages(messages)

print(f"Model-2th>\n {message.content}")

返回值:

shell 复制代码
Model-2th> 
北京今天的天气晴朗。

三、tools功能的演进

随着LLM调用工具的普及,这种调用方法集成在大模型api接口中就变得越重要。

大部分模型厂商已经支持了function call,下面是deepseek工具调用的一个例子1

ini 复制代码
response = client.chat.completions.create(
	model="deepseek-chat",
	messages=messages,
	tools=tools
)

其中 tools是可以供模型选择的工具。

写在最后

从0开发写function的逻辑,需要让模型思考、观察、行动。其实这个流程的循环其实就是ReAct框架的原理。2

参考文章: 1 deepseek function文档: api-docs.deepseek.com/zh-cn/guide... 2# 讓 LLM 更好用的方法:ReAct prompting: edge.aif.tw/application...

相关推荐
wangruofeng8 小时前
一个 Markdown 文件攒下 37k 星,i-have-adhd 给 AI 输出立了 10 条规矩
github·aigc·ai编程
墨风如雪8 小时前
Grok Bot 从 0 到 1|用云电脑和插件,搭一个能干活的 X 助手
aigc
Csvn9 小时前
第 22 章 安全、合规与治理
人工智能·aigc·agent
Warson_L10 小时前
Python的TypedDict
python·langchain·llm
全栈弄潮儿11 小时前
需求不清时,如何让 AI 帮你补全问题,而不是瞎写代码
aigc·openai·ai编程
AI创界者12 小时前
前沿多模态大模型 MiniMax-H3无审查 本地部署与深度实测指南
人工智能·aigc·音视频
XLYcmy13 小时前
面向安全领域决策的大型语言模型漏洞价值评估对齐机制与智能化评级系统研发 研究方向与核心内容
网络安全·llm·sft·cve·cot·对齐·漏洞检测
程序猿编码13 小时前
LLM+Agent 全栈推理底座:Triton 算子优化・语义级缓存・分布式调度
大模型·llm·agent·推理
小白跃升坊14 小时前
大模型推理入门:从原理到实践
ai·大模型·llm·推理优化
武子康15 小时前
把 SGLang 接进 Agent,查一次订单要走几步
人工智能·llm·agent