add llmmath

app.py

@@ -15,6 +15,7 @@ from langchain.tools import BaseTool
 from langchain.tools import DuckDuckGoSearchRun
 from langchain.utilities import WikipediaAPIWrapper
 from langchain.python import PythonREPL
+from langchain.chains import LLMMathChain
 
 import pinecone      
 from pinecone.core.client.configuration import Configuration as OpenApiConfiguration
@@ -57,7 +58,7 @@ python_tool = Tool(
 )
 
 # tools = [DB_Search(), wikipedia_tool, duckduckgo_tool, python_tool]
-tools = [DB_Search(), duckduckgo_tool, python_tool]
+
 
 os.environ["OPENAI_API_TYPE"] = "azure"
 os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
@@ -71,6 +72,17 @@ chat = AzureChatOpenAI(
     deployment_name="Chattester",
     temperature=0,
 )
+llm = chat
+
+llm_math = LLMMathChain(llm = llm)
+
+math_tool = Tool(
+    name ='Calculator',
+    func = llm_math.run,
+    description ='Useful for when you need to answer questions about math.'
+)
+
+tools = [DB_Search(), duckduckgo_tool, python_tool, math_tool]
 
 embeddings = OpenAIEmbeddings(deployment="model_embedding", chunk_size=15)
 
@@ -86,7 +98,7 @@ index = pinecone.Index(index_name)
 # print(pinecone.whoami())
 # print(index.describe_index_stats())
 
-llm = chat
+
 agent = initialize_agent(tools, llm, 
                          agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
                          verbose = True,