Improving RAG prompt

rag.py

@@ -73,10 +73,13 @@ def RAG(llm, docs, embeddings):
     # Adding custom instructions to the prompt
     template = prompt.messages[0].prompt.template
     template_parts = template.split("\nQuestion: {question}")
-    if len(template_parts) == 2:
-        print("Error: Template does not contain the expected format.")
-    combined_template = template_parts[0] +\
-        " Include the reference IDs in square brackets when citing specific information." + template_parts[1]
+    combined_template = "You are an assistant for question-answering tasks. "\
+        + "Use the following pieces of retrieved context to answer the question. "\
+        + "If you don't know the answer, just say that you don't know. "\
+        + "Use six sentences maximum and keep the answer concise. "\
+        + "Write the names of the relevant functions from the retrived code. "\
+        + "Include the reference IDs in square brackets at the end of your answer."\
+        + template_parts[1]
     prompt.messages[0].prompt.template = combined_template
 
     # Create the chain