create new tool list

app.py

@@ -228,6 +228,18 @@ class DB_Search(BaseTool):
     def _arun(self, query: str):
         raise NotImplementedError("N/A")
 
+class DB_Search2(BaseTool):
+    name = "Vector Database Search"
+    description = "This is the internal database to search information firstly. If information is found, it is trustful."
+    def _run(self, query: str) -> str:
+        response, source = QAQuery_p(query)
+        # response = "test db_search feedback"
+        return response
+
+    def _arun(self, query: str):
+        raise NotImplementedError("N/A")
+
+
 def Text2Sound(text):
 
     speech_config = speechsdk.SpeechConfig(subscription=os.getenv('SPEECH_KEY'), region=os.getenv('SPEECH_REGION'))
@@ -312,6 +324,14 @@ Text2Sound_tool = Tool(
     description = "Useful when you need to convert text into sound file."
 )
 
+Text2Sound_tool2 = Tool(
+    name = "Text To Sound REST API",
+    # func = Text2Sound,
+    func = text_to_speech_2,
+    description = "Useful when you need to convert text into sound file."
+)
+
+
 Wikipedia = WikipediaAPIWrapper()
 Netsearch = DuckDuckGoSearchRun()
 Python_REPL = PythonREPL()
@@ -334,6 +354,26 @@ python_tool = Tool(
     description = "Useful when you need python to answer questions. You should input python code."    
 )
 
+wikipedia_tool2 = 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_tool2 = Tool(
+    name = "Duckduckgo Internet Search",
+    func = Netsearch.run,
+    description = "Useful to search information in internet when it is not available in other tools"    
+)
+
+python_tool2 = Tool(
+    name = "Python REPL",
+    func = Python_REPL.run,
+    description = "Useful when you need python to answer questions. You should input python code."    
+)
+
+
+
 # tools = [DB_Search(), wikipedia_tool, duckduckgo_tool, python_tool]
 
 
@@ -362,8 +402,12 @@ math_tool = Tool(
     description ='Useful for when you need to answer questions about math.'
 )
 
+
+
 tools = [DB_Search(), duckduckgo_tool, wikipedia_tool, python_tool, math_tool, Text2Sound_tool]
 
+tools2 = [DB_Search2(), duckduckgo_tool2, wikipedia_tool2, python_tool2, math_tool, Text2Sound_tool2]
+
 # tools = load_tools(["Vector Database Search","Wikipedia Search","Python REPL","llm-math"], llm=llm)
 
 embeddings = OpenAIEmbeddings(deployment="model_embedding", chunk_size=15)
@@ -442,7 +486,7 @@ prompt_openai = OpenAIMultiFunctionsAgent.create_prompt(
 
 input_variables=["input", "chat_history", "agent_scratchpad"]
 
-agent_ZEROSHOT_REACT = initialize_agent(tools, llm, 
+agent_ZEROSHOT_REACT = initialize_agent(tools2, llm, 
                         #  agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
                          agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
                          verbose = True,
@@ -466,13 +510,13 @@ llm_chain = LLMChain(llm=llm, prompt=prompt)
 
 # llm_chain_openai = LLMChain(llm=llm, prompt=prompt_openai, verbose=True)
 
-agent_core = ZeroShotAgent(llm_chain=llm_chain, tools=tools, verbose=True)
+agent_core = ZeroShotAgent(llm_chain=llm_chain, tools=tools2, verbose=True)
 
 agent_core_openai = OpenAIMultiFunctionsAgent(llm=llm, tools=tools, prompt=prompt_openai, verbose=True)
 
 agent_ZEROSHOT_AGENT = AgentExecutor.from_agent_and_tools(
     agent=agent_core, 
-    tools=tools, 
+    tools=tools2, 
     verbose=True, 
     # memory=memory,
     handle_parsing_errors = True,