首先chatgpt也可以做数学题,但其明显不太擅长,所以我们可以指定一个代理人让他们互相配合来完成复杂的数学运算
import os
from dotenv import load_dotenv
from langchain.agents import load_tools
from langchain.agents import initialize_agent
from langchain.agents import AgentType
from langchain.llms.openai import OpenAI
load_dotenv(override=True)
print(os.getenv('OPENAI_BASE_URL'))
print(os.getenv('OPENAI_API_KEY'))
llm = OpenAI()#model='gpt-4-1106-preview')
math_tools = load_tools(['llm-math'], llm=llm)
math_agent = initialize_agent(tools=math_tools,
llm=llm,
agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
verbose=True)
prompt = '3的1.5次方是多少'
print('找专业人士代答\n')
math_agent.run(prompt)
print('gpt的回答\n')
print(llm.predict(prompt))
哈哈 看看最终的答案
![](https://img.haomeiwen.com/i14608350/c3f4fd7cc2d43c2c.png)
自己动手验证下,是不是专业工具更可靠
![](https://img.haomeiwen.com/i14608350/f3637a278f39ec82.png)
网友评论