Update rag/rag_production.py

rag/rag_production.py

@@ -18,8 +18,9 @@ logger = logging.getLogger(__name__)
 rag_prompt_template = '''
 You are a legal assistant trained to answer questions using legal documents.
 
-If the answer cannot be determined from the available legal text, you must answer without include the
-According to Article <article>, Chapter <chapter>, <title> phrase, for example:
+If the answer cannot be determined from the available legal text, you must answer without include this phrase:
+  > According to Article <article>, Chapter <chapter>, <title> 
+For example:
    > The answer cannot be determined from the available legal text.
 
 Otherwise, you must STRICTLY follow this 3-step structure:
@@ -37,7 +38,7 @@ Otherwise, you must STRICTLY follow this 3-step structure:
 - Do NOT repeat or paraphrase the law reference later.
 - Do NOT invent any legal information — use ONLY the provided context and metadata.
 - Do NOT add phrase like 'The most relevant point is that' when mentioning the context.
-- If the context is insufficient, respond with:
+- If the context is insufficient, nrever include the legal citation and respond with:
   > "The answer cannot be determined from the available legal text."
 ---
 
@@ -62,7 +63,7 @@ prompt = ChatPromptTemplate.from_messages([
     ('user', "Context:\n{context}\n\nQuestion:\n{input}\n\nAnswer:")
 ])
 
-def get_rag_chain(model_name='llama-3.3-70b-versatile', k=1):
+def get_rag_chain(model_name='llama-3.3-70b-versatile', k=3):
     db = get_vectorstore()
 
     groq_api_key = os.getenv('GROQ_API_KEY')