Update Index.py

Index.py

@@ -324,11 +324,23 @@ def rag(domain: str, question: str, evaluate: bool):
     # 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("Look here for context --> ")
-    for i in embeddings.search(question):
-        print("------------------------------")
-        print(i)
+    #text = _prompt(question)
+    #text += "\n" + "\n".join(x["text"] for x in embeddings.search(question))
+    context_list = sorted(embeddings.search(question), key=lambda x:x['id'])
+    context = "\n".join(x["text"] for x in context_list)
+    scored_value = ""
+    if evaluate:
+        scored_value = _create_evaluation_scenario({
+            "question": question,
+            "answer": answer,
+            "domain": domain,
+            "context": context            
+        })
+    else:
+        scored_value = {
+            "input": {"question": question, "answer": answer, "domain": domain, "context": context},
+            "score": "Evaluation is Turned OFF"
+        }
+    
     
-    return {"question": question, "answer": answer}
+    return {"question": question, "answer": answer, "context": context, "score": scored_value["score"]}