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("那是一个真实的单词吗？")

# 加载环境变量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()