baseline_agent.py

from llama_index.agent import FnRetrieverOpenAIAgent
from llama_index.llms import OpenAI

# Initialize the LLM
llm = OpenAI(model="gpt-3.5-turbo-0613")

# Initialize the FnRetrieverOpenAIAgent
top_agent = FnRetrieverOpenAIAgent.from_retriever(
   obj_index.as_retriever(similarity_top_k=4),
   system_prompt=""" \
You are an agent designed to answer queries about a set of given patents.
Please always use the tools provided to answer a question. Do not rely on prior knowledge.\
""",
   llm=llm,
   verbose=True,
)
# baseline
response = base_query_engine.query(
    "Tell me about the Death Ray"
)
print(str(response))