Add several new tools

app.py

@@ -1,4 +1,4 @@
-from typing import Any, Coroutine
+# from typing import Any, Coroutine
 import openai
 import os
 from langchain.vectorstores import Chroma
@@ -10,8 +10,11 @@ from langchain.chains import RetrievalQA
 from langchain.vectorstores import Pinecone
 from langchain.agents import initialize_agent
 from langchain.agents import AgentType
-from langchain.agents import tool
+from langchain.agents import Tool
 from langchain.tools import BaseTool
+from langchain.tools import DuckDuckGoSearchRun
+from langchain.utilities import WikipediaAPIWrapper
+from langchain.python import PythonREPL
 
 import pinecone      
 from pinecone.core.client.configuration import Configuration as OpenApiConfiguration
@@ -20,7 +23,7 @@ import time
 
 class DB_Search(BaseTool):
     name = "Vector Database Search"
-    description = "This is the vector database to search local information"
+    description = "This is the vector database to search information firstly"
     def _run(self, query: str) -> str:
         response, source = QAQuery_p(query)
         # response = "test db_search feedback"
@@ -29,9 +32,31 @@ class DB_Search(BaseTool):
     def _arun(self, query: str):
         raise NotImplementedError("N/A")
 
-tools = [DB_Search()]
 
 
+Wikipedia = WikipediaAPIWrapper()
+Netsearch = DuckDuckGoSearchRun()
+Python_REPL = PythonREPL()
+
+wikipedia_tool = Tool(
+    name = "Wikipedia Search",
+    func = Wikipedia.run(),
+    description = "Useful to search a topic, country or person when there is no availble information in vector database"
+)
+
+duckduckgo_tool = Tool(
+    name = "Duckduckgo Internet Search",
+    func = Python_REPL.run(),
+    description = "Useful to search information in internet when it is not available in other tools"    
+)
+
+python_tool = Tool(
+    name = "Python REPL",
+    func = Netsearch.run(),
+    description = "Useful when you need python to answer questions. You should input python code."    
+)
+
+tools = [DB_Search(), wikipedia_tool, duckduckgo_tool, python_tool]
 
 os.environ["OPENAI_API_TYPE"] = "azure"
 os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")