search_bot.py

'''
From: https://langchain.readthedocs.io/en/latest/modules/agents/getting_started.html

Need SerpAPI API key for Google Searches: https://serpapi.com/pricing
'''

from langchain.agents import load_tools
from langchain.agents import initialize_agent
from langchain.llms import OpenAI

from utils import intialize_api_keys

if __name__ == "__main__":
    # Initialize API Key
    intialize_api_keys()

    # First, let's load the language model we're going to use to control the agent.
    llm = OpenAI(temperature=0)

    # Next, let's load some tools to use. Note that the `llm-math` tool uses an LLM, so we need to pass that in.
    tools = load_tools(["serpapi", "llm-math"], llm=llm)

    # Finally, let's initialize an agent with the tools, the language model, and the type of agent we want to use.
    # internally calls AgentExecutor.from_agent_and_tools
    agent = initialize_agent(tools, llm, agent="zero-shot-react-description", verbose=True)

    # Now let's test it out!
    agent.run("Who is Olivia Wilde's boyfriend? What is his current age raised to the 0.23 power?")