Update rag.py

rag.py

@@ -42,7 +42,7 @@ class RAG:
             api_key=os.getenv("HF_TOKEN")
         )
 
-        query = f"{context}\n\n{instruction}"
+        query = f"Context:\n{context}\n\nQuestion:\n{instruction}"
         #sys_prompt = "You are a helpful assistant. Answer the question using only the context you are provided with. If it is not possible to do it with the context, just say 'I can't answer'. <|endoftext|>"
 
         chat_completion = client.chat.completions.create(