updating rag

rag.py

@@ -31,7 +31,8 @@ class RAGModel:
         # Create embeddings
         self.embeddings = OpenAIEmbeddings(model="text-embedding-ada-002", openai_api_key=openai_api_key)
         text_documents = [str(doc) for doc in chunked_documents]
-
+        print(text_documents)
+        
         # Create cache-backed embeddings
         self.store = LocalFileStore("./cache/")
         self.embedder = CacheBackedEmbeddings.from_bytes_store(
@@ -72,7 +73,7 @@ class RAGModel:
         # Format prompt
         prompt = self.prompt_template.format(context=context, question=question)
         
-#        print(context)
+        print(context)
         # Get response from chat model
 #        response = self.chat_model(prompt)
         # Parse response
@@ -80,7 +81,7 @@ class RAGModel:
         
     #    chain = prompt=prompt | self.chat_model | parser=self.parser
      #   result = chain.invoke()    
-        dict_context = {"question": question} 
+#        dict_context = {"question": question} 
         #chain = ({"context": context,"question":Runnab
         chain =({"context": lambda x: context,"question": RunnablePassthrough()}          
                   | self.prompt_template