Enable AI to have internet access - Google search integrated langchain

Basic dependency

• python 3.1.12 +
• langchain 0.1.12 +
• SERPAPI_API_KEY
• OPEN_AI_API_KEY

SERPAPI_API_KEY get from https://serpapi.com/

Install the following dependencies:

pip install langchain
pip install pymupdf
pip install openai
pip install --upgrade --quiet  langchain langchain-community langchainhub langchain-openai chromadb bs4
pip install --upgrade langchain-openai tiktoken chromadb langchain
pip install fake_useragent

Get start

# set the api key
os.environ["SERPAPI_API_KEY"] = '003dc0d2d9e0dc818aa2b497342346dfgdf799af1849c5d1249d34dd7'
openai.api_key="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"

from langchain_openai import ChatOpenAI
from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
from langchain.memory import ChatMessageHistory
from langchain_text_splitters import RecursiveCharacterTextSplitter
from langchain_community.vectorstores import Chroma
from langchain_openai import OpenAIEmbeddings

# build tool agent

from langchain_community.utilities import SerpAPIWrapper
from langchain.agents import create_openai_tools_agent
from langchain.agents import AgentExecutor,Tool
from langchain.memory import ConversationBufferWindowMemory

chat = ChatOpenAI(model="gpt-3.5-turbo-1106",streaming=True,max_tokens=4090,max_retries=2)
# 加载 serpapi 工具
search = SerpAPIWrapper()

tools = [Tool(
    name="google_search",
    description="Search Google for recent results.",
    func=search.run,
    max_results=1,
    max_iterations=2,
    max_retries=2,
    max_consecutive_errors=2,
    max_tokens=2000, #must
    return_direct=False
)]

prompt = ChatPromptTemplate.from_messages(
    [
        (
            "system",
            "You are a helpful assistant. You may not need to use tools for every query - the user may just want to chat!",
        ),
         ("user", "{input}"),
         MessagesPlaceholder(variable_name="agent_scratchpad"),
    ]
)

memory = ConversationBufferWindowMemory(k=2, return_messages=True)
agent = create_openai_tools_agent(tools = tools,llm = chat,prompt=prompt)
agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True,max_iterations=2,memory=memory)
response = agent_executor.invoke({"input": 'Who is the current President of the United States?'})
print(response)

Now AI can first use Google search and then answer your questions.

• 本文作者： reiner
• 本文链接： https://reiner.host/posts/undefined.html
• 版权声明： 转载请注明出处，并附上原文链接