六、ReAct 核心思想是:推理+操作
1、实现 ReAct 算法的测试函数:使用 LangChain 库和 OpenAI API 来执行推理和操作。
# 以下`SERPAPI_API_KEY`仅为示例,请访问 https://serpapi.com 注册账号并替换为自己的 `API_KEY`(每月100次免费调用)def react_test():"""实现 ReAct 算法的测试函数。使用 LangChain 库和 OpenAI API 来执行推理和操作。"""os.environ["SERPAPI_API_KEY"] = os.getenv('SERPAPI_API_KEY')# 加载 LangChain 内置的 Toolstools = load_tools(["serpapi", "llm-math"], llm=llm)# 实例化 ZERO_SHOT_REACT Agentagent = initialize_agent(tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True)print(agent.run("谁是莱昂纳多·迪卡普里奥的女朋友?她现在年龄的0.43次方是多少"))
2、测试自我提问与搜索功能的函数:使用 SerpAPI 来执行搜索操作,并结合 LangChain 的工具和LLM进行问答。
defself_ask_with_search_test():"""测试自我提问与搜索功能的函数。使用SerpAPI来执行搜索操作,并结合LangChain的工具和LLM进行问答。"""os.environ["SERPAPI_API_KEY"]=os.getenv('SERPAPI_API_KEY')#实例化查询工具search=SerpAPIWrapper()tools=[Tool(name="IntermediateAnswer",func=search.run,description="usefulforwhenyouneedtoaskwithsearch",)]#实例化SELF_ASK_WITH_SEARCHAgentself_ask_with_search=initialize_agent(tools,llm,agent=AgentType.SELF_ASK_WITH_SEARCH,verbose=True,handle_parsing_errors=True)#实际运行Agent,查询问题(正确)self_ask_with_search.run("成都举办的大运会是第几届大运会?")3、测试函数工具的函数:使用 LangChain 和 OpenAI 来执行特定的函数操作。
@tooldef get_word_length(word: str) -> int:"""计算单词长度的工具函数。参数:word: 要计算长度的单词。返回:单词的长度。""""""Returns the length of a word."""return len(word)tools = [get_word_length]def function_test():"""测试函数工具的函数。使用 LangChain 和 OpenAI 来执行特定的函数操作。"""system_message = SystemMessage(content="你是非常强大的AI助手,但在计算单词长度方面不擅长。")prompt = OpenAIFunctionsAgent.create_prompt(system_message=system_message)agent = OpenAIFunctionsAgent(llm=llm, tools=tools, prompt=prompt)# 实例化 OpenAIFunctionsAgentagent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True)agent_executor.run("单词“educa”中有多少个字母?")
4、测试函数记忆功能的函数:使用 LangChain 的记忆机制来保持对话上下文。
deffunction_memory_test():"""测试函数记忆功能的函数。使用LangChain的记忆机制来保持对话上下文。"""system_message=SystemMessage(content="你是非常强大的AI助手,但在计算单词长度方面不擅长。")MEMORY_KEY="chat_history"prompt=OpenAIFunctionsAgent.create_prompt(system_message=system_message,extra_prompt_messages=[MessagesPlaceholder(variable_name=MEMORY_KEY)])memory=ConversationBufferMemory(memory_key=MEMORY_KEY,return_messages=True)agent=OpenAIFunctionsAgent(llm=llm,tools=tools,prompt=prompt)#实例化OpenAIFunctionsAgentagent_executor=AgentExecutor(agent=agent,tools=tools,verbose=True)agent_executor.run("单词“educa”中有多少个字母?")#agent_executor.run("那是一个真实的单词吗?")5、完整代码
# 加载环境变量import openaiimport osimport tiktoken# 加载 .env 文件from dotenv import load_dotenv, find_dotenvfrom langchain.prompts import PromptTemplatefrom langchain.llms import OpenAIfrom langchain.chains import LLMChain#from langchain.chat_models import AzureChatOpenAIfrom langchain.chat_models import ChatOpenAI #直接访问OpenAI的GPT服务from langchain.agents import load_toolsfrom langchain.agents import initialize_agentfrom langchain.agents import AgentType,Toolfrom langchain.utilities import SerpAPIWrapperfrom langchain.schema import SystemMessagefrom langchain.agents import OpenAIFunctionsAgentfrom langchain.agents import AgentExecutorfrom langchain.agents import toolfrom langchain.prompts import MessagesPlaceholderfrom langchain.memory import ConversationBufferMemory# 加载环境变量_ = load_dotenv(find_dotenv())# 从环境变量中获得你的 OpenAI Key和配置URLopenai.api_key = os.getenv('OPENAI_API_KEY')openai.api_base = os.getenv('OPENAI_API_URL')model = os.getenv('OPENAI_API_MODEL')# 初始化LLM链llm = ChatOpenAI(model_name=model, temperature=0) #直接访问OpenAI的GPT服务# ReAct 核心思想是 推理+操作,本示例以` Google Search` 和 `LLM Math` 作为可选操作集合(toolkits),实现 ReAct 功能。# 以下`SERPAPI_API_KEY`仅为示例,请访问 https://serpapi.com 注册账号并替换为自己的 `API_KEY`(每月100次免费调用)def react_test():"""实现 ReAct 算法的测试函数。使用 LangChain 库和 OpenAI API 来执行推理和操作。"""os.environ["SERPAPI_API_KEY"] = os.getenv('SERPAPI_API_KEY')# 加载 LangChain 内置的 Toolstools = load_tools(["serpapi", "llm-math"], llm=llm)# 实例化 ZERO_SHOT_REACT Agentagent = initialize_agent(tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True)print(agent.run("谁是莱昂纳多·迪卡普里奥的女朋友?她现在年龄的0.43次方是多少"))def self_ask_with_search_test():"""测试自我提问与搜索功能的函数。使用 SerpAPI 来执行搜索操作,并结合 LangChain 的工具和LLM进行问答。"""os.environ["SERPAPI_API_KEY"] = os.getenv('SERPAPI_API_KEY')# 实例化查询工具search = SerpAPIWrapper()tools = [Tool(name="Intermediate Answer",func=search.run,description="useful for when you need to ask with search",)]# 实例化 SELF_ASK_WITH_SEARCH Agentself_ask_with_search = initialize_agent(tools, llm, agent=AgentType.SELF_ASK_WITH_SEARCH, verbose=True,handle_parsing_errors=True)# 实际运行 Agent,查询问题(正确)self_ask_with_search.run("成都举办的大运会是第几届大运会?")@tooldef get_word_length(word: str) -> int:"""计算单词长度的工具函数。参数:word: 要计算长度的单词。返回:单词的长度。""""""Returns the length of a word."""return len(word)tools = [get_word_length]def function_test():"""测试函数工具的函数。使用 LangChain 和 OpenAI 来执行特定的函数操作。"""system_message = SystemMessage(content="你是非常强大的AI助手,但在计算单词长度方面不擅长。")prompt = OpenAIFunctionsAgent.create_prompt(system_message=system_message)agent = OpenAIFunctionsAgent(llm=llm, tools=tools, prompt=prompt)# 实例化 OpenAIFunctionsAgentagent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True)agent_executor.run("单词“educa”中有多少个字母?")def function_memory_test():"""测试函数记忆功能的函数。使用 LangChain 的记忆机制来保持对话上下文。"""system_message = SystemMessage(content="你是非常强大的AI助手,但在计算单词长度方面不擅长。")MEMORY_KEY = "chat_history"prompt = OpenAIFunctionsAgent.create_prompt(system_message=system_message,extra_prompt_messages=[MessagesPlaceholder(variable_name=MEMORY_KEY)])memory = ConversationBufferMemory(memory_key=MEMORY_KEY, return_messages=True)agent = OpenAIFunctionsAgent(llm=llm, tools=tools, prompt=prompt)# 实例化 OpenAIFunctionsAgentagent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True)agent_executor.run("单词“educa”中有多少个字母?")#agent_executor.run("那是一个真实的单词吗?")# python 入口函数if __name__ == '__main__':#react_test()#self_ask_with_search_test()#function_test()function_memory_test()
| 欢迎光临 链载Ai (https://www.lianzai.com/) | Powered by Discuz! X3.5 |