correct chathmi2 issue

app.py

@@ -22,8 +22,8 @@ class DB_Search(BaseTool):
     name = "Vector Database Search"
     description = "This is the vector database to search local information"
     def _run(self, query: str) -> str:
-        # response, source = QAQuery_p(query)
-        response = "test db_search feedback"
+        response, source = QAQuery_p(query)
+        # response = "test db_search feedback"
         return response
 
     def _arun(self, query: str):
@@ -92,7 +92,7 @@ def chathmi(message, history):
     # yield history
 
 def chathmi2(message, history):
-    response = agent(message)
+    response = agent.run(input = message)
     time.sleep(0.3)
     print(history)
     yield response