Update ragagent.py

ragagent.py

@@ -82,13 +82,13 @@ class RAGAgent():
         Here is the question: {question} <|eot_id|><|start_header_id|>assistant<|end_header_id|>""",
         input_variables=["generation", "question"],
     )
-    '''
+    
     def reset_chains():
         RAGAgent.retrieval_grader = RAGAgent.retrieval_grader_prompt | CustomLlama3(bearer_token = RAGAgent.HF_TOKEN) | JsonOutputParser()
         RAGAgent.rag_chain = RAGAgent.answer_prompt | CustomLlama3(bearer_token = RAGAgent.HF_TOKEN) | StrOutputParser()
         RAGAgent.hallucination_grader = RAGAgent.hallucination_prompt | CustomLlama3(bearer_token = RAGAgent.HF_TOKEN) | JsonOutputParser()
         RAGAgent.answer_grader = RAGAgent.answer_grader_prompt | CustomLlama3(bearer_token = RAGAgent.HF_TOKEN) | JsonOutputParser()
-    '''
+    
     
     def __init__(self, docs):
         docs_list = [item for sublist in docs for item in sublist]
@@ -100,15 +100,13 @@ class RAGAgent():
         
         embedding_function = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
         collection_name = re.sub(r'[^a-zA-Z0-9]', '', doc_splits[0].metadata.get('source'))
-
-        # persistent_client = chromadb.PersistentClient(settings=Settings(allow_reset=True))
         persistent_client = chromadb.PersistentClient()
-        # persistent_client.reset()
+
         if collection_name in [c.name for c in persistent_client.list_collections()]:
             print("\ndeleted: ",collection_name)
             persistent_client.delete_collection(collection_name)
             
-        collection = persistent_client.create_collection(collection_name)
+        persistent_client.create_collection(collection_name)
         print("\ncreated: ",collection_name)
 
         # Add to vectorDB
@@ -121,13 +119,8 @@ class RAGAgent():
         vectorstore.add_documents(doc_splits)
         
         RAGAgent.retriever = vectorstore.as_retriever()
-        # RAGAgent.reset_chains()
+        RAGAgent.reset_chains()
         RAGAgent.logs=""
-
-    RAGAgent.retrieval_grader = RAGAgent.retrieval_grader_prompt | CustomLlama3(bearer_token = RAGAgent.HF_TOKEN) | JsonOutputParser()
-    RAGAgent.rag_chain = RAGAgent.answer_prompt | CustomLlama3(bearer_token = RAGAgent.HF_TOKEN) | StrOutputParser()
-    RAGAgent.hallucination_grader = RAGAgent.hallucination_prompt | CustomLlama3(bearer_token = RAGAgent.HF_TOKEN) | JsonOutputParser()
-    RAGAgent.answer_grader = RAGAgent.answer_grader_prompt | CustomLlama3(bearer_token = RAGAgent.HF_TOKEN) | JsonOutputParser()
     
     def add_log(log):
         RAGAgent.logs += log + "\n"