test tools

app.py

@@ -19,10 +19,11 @@ import gradio as gr
 import time
 
 class DB_Search(BaseTool):
-    name = "Vector Databse Search"
+    name = "Vector Database Search"
     description = "This is the vector database to search local information"
-    def _run(self, message: str) -> Any:
-        response, source = QAQuery_p(message)
+    def _run(self, query: str) -> str:
+        # response, source = QAQuery_p(query)
+        response = "test db_search feedback"
         return response
 
     def _arun(self, query: str):
@@ -57,9 +58,10 @@ index = pinecone.Index(index_name)
 # print(pinecone.whoami())
 # print(index.describe_index_stats())
 
-agent = initialize_agent(tools, chat, 
-                         agent="zero-shot-react-description",
-                         verbose = True)
+llm = chat
+agent = initialize_agent(tools, llm, 
+                         agent=AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION,
+                         verbose = True,)
 
 global vectordb
 vectordb = Chroma(persist_directory='db', embedding_function=embeddings)