CHECKING FOR CONTEXT

Index.py

@@ -220,6 +220,7 @@ def _search(query, extractor, question=None):
 
     # return {"question": question, "answer": llm_chain.run(question)}
     print(extractor([("answer", query, _prompt(question), False)]))
+       
     return extractor([("answer", query, _prompt(question), False)])[0][1]
 
 
@@ -248,4 +249,8 @@ def rag(domain: str, question: str):
     # else:
     print("before doing Q&A")
     answer = _search(question, extractor)
+    text = _prompt(question)
+    text += "\n" + "\n".join(x["text"]for x in embeddings.search(question))
+    print(text)
+    
     return {"question": question, "answer": answer}