introduce agent

app.py

@@ -1,3 +1,4 @@
+from typing import Any, Coroutine
 import openai
 import os
 from langchain.vectorstores import Chroma
@@ -7,11 +8,30 @@ from langchain.chat_models import AzureChatOpenAI
 from langchain.document_loaders import DirectoryLoader
 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.tools import BaseTool
+
 import pinecone      
 from pinecone.core.client.configuration import Configuration as OpenApiConfiguration
 import gradio as gr
 import time
 
+class DB_Search(BaseTool):
+    name = "Vector Databse Search"
+    description = "This is the vector database to search local information"
+    def _run(self, message: str) -> Any:
+        response, source = QAQuery_p(message)
+        return response
+
+    def _arun(self, query: str):
+        raise NotImplementedError("N/A")
+
+tools = [DB_Search()]
+
+
+
 os.environ["OPENAI_API_TYPE"] = "azure"
 os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
 os.environ["OPENAI_API_BASE"] = os.getenv("OPENAI_API_BASE")
@@ -37,7 +57,9 @@ 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)
 
 global vectordb
 vectordb = Chroma(persist_directory='db', embedding_function=embeddings)
@@ -68,7 +90,7 @@ def chathmi(message, history):
     # yield history
 
 def chathmi2(message, history):
-    response = QAQuery(message)
+    response = agent(message)
     time.sleep(0.3)
     print(history)
     yield response
@@ -77,7 +99,7 @@ def chathmi2(message, history):
 # chatbot = gr.Chatbot(color_map =("blue", "pink"))
 
 demo = gr.ChatInterface(
-    chathmi,
+    chathmi2,
     title="STLA BABY - YOUR FRIENDLY GUIDE ",
     description= "v0.2: Powered by MECH Core Team",
     )