Upload Index.py

Index.py

@@ -248,4 +248,9 @@ def rag(domain: str, question: str):
     # else:
     print("before doing Q&A")
     answer = _search(question, extractor)
-    return {"question": question, "answer": answer}
+
+    text = _prompt(question)
+    text += "\n" + " ".join(x["text"] for x in embeddings.search(question))
+    print("context \n")
+    print(text)
+    return {"question": question, "answer": answer, "context": text}