follow-up question

utils/generator.py

@@ -203,9 +203,14 @@ CRITICAL: Use ONLY [number] format. Never include page numbers, years, document
 
 DO NOT add a "References" section, bibliography, or sources list at the end.
 
+FOLLOW-UP QUESTIONS:
+- If the context contains related information beyond what you included in your answer, suggest 1 relevant follow-up question the user might want to explore.
+- Base the question on related information you found in the context or natural extensions of the user's query.
+- Format the follow-up question clearly at the end of your response under "You might also want to know:"
+- Keep the follow-up question concise and directly related to the audit reports.
+
 - If the context is insufficient, say "I don't have sufficient information to answer the question. Please try rephrasing your query."
 """
-    
     user_content = f"### CONTEXT\n{context}\n\n### USER QUESTION\n{question}"
     return [SystemMessage(content=system_content), HumanMessage(content=user_content)]