Update app.py

app.py

@@ -21,23 +21,42 @@ from llama_index import GPTSimpleVectorIndex, SimpleDirectoryReader
 
 index = GPTSimpleVectorIndex.load_from_disk('index.json')
 
-def querying_db(query: str):
+ef querying_db(query: str):
     response = index.query(query)
     return response
 
+
 tools = [
     Tool(
         name="QueryingDB",
         func=querying_db,
-        description="This function takes a query string as input and returns the most relevant answer from the documentation as output"
+        description="useful for when you need to answer questions from the database. The answer is given in bullet points.",
+        return_direct=True
     )
+
 ]
 
-llm = OpenAI(temperature=0)
+prefix = """Give a detailed answer to the question"""
+suffix = """Give answer in bullet points"
+
+Question: {input}
+{agent_scratchpad}"""
+
+prompt = ZeroShotAgent.create_prompt(
+    tools,
+    prefix=prefix,
+    suffix=suffix,
+    input_variables=["input", "agent_scratchpad"]
+)
+llm_chain = LLMChain(llm=OpenAI(temperature=0), prompt=prompt)
+tool_names = [tool.name for tool in tools]
+agent = ZeroShotAgent(llm_chain=llm_chain, allowed_tools=tool_names)
+
 
 def get_answer(query_string):
-    agent = initialize_agent(tools, llm, agent="zero-shot-react-description")
-    result = agent.run(query_string)
+    agent_executor = AgentExecutor.from_agent_and_tools(agent=agent, tools=tools, verbose=True)
+    result = agent_executor.run(query_string)
+
     return result
 
 def qa_app(query):