Update rag.py

rag.py

@@ -42,8 +42,10 @@ class RAG:
             api_key=os.getenv("HF_TOKEN")
         )
 
-        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|>"
+        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|>"
+        #query = f"Context:\n{context}\n\nQuestion:\n{instruction}"
+        query = f"{sys_prompt}\n\nContext:\n{context}\n\nQuestion:\n{instruction}"
+        #query = f"{sys_prompt}\n\nQuestion:\n{instruction}\n\nContext:\n{context}"
         chat_completion = client.chat.completions.create(
             model="tgi",
             messages=[