Update app/llm.py

app/llm.py

@@ -17,15 +17,23 @@ from langchain_chroma import Chroma
 from langchain_community.embeddings import GPT4AllEmbeddings
 
 
-def agent(): 
-    loader = WebBaseLoader("https://lilianweng.github.io/posts/2023-06-23-agent/")
-    data = loader.load()
-    text_splitter = RecursiveCharacterTextSplitter(chunk_size=500, chunk_overlap=0)
-    all_splits = text_splitter.split_documents(data)
-
-
-def download_embedding():
-    vectorstore =Chroma.from_documents(documents=all_splits, embedding=GPT4AllEmbeddings())
+class RagChat:
+    def agent(self): 
+        loader = WebBaseLoader("https://lilianweng.github.io/posts/2023-06-23-agent/")
+        data = loader.load()
+        text_splitter = RecursiveCharacterTextSplitter(chunk_size=500, chunk_overlap=0)
+        all_splits = text_splitter.split_documents(data)
+        return all_splits
+        
+    def download_embedding(self):
+        vectorstore = Chroma.from_documents(documents=self.agent, embedding=GPT4AllEmbeddings())
+        return vectorstore
+        
+    def search(self, question):
+        docs = vectorstore.similarity_search(question)
+        return len(docs)
+        
+