update prompt

core/graph.py

@@ -19,12 +19,12 @@ class RAGAgent:
         self.llm = get_llm()
         self.app = self.build_graph()
 
-    def retrieve(self, state: GraphState):
+    def retriever_node(self, state: GraphState):
         query = state["current_query"]
         docs = self.retriever.invoke(query)
         return {"documents": docs}
 
-    def generate(self, state: GraphState):
+    def generator_node(self, state: GraphState):
         question = state["question"]
         docs = state["documents"]
         
@@ -34,7 +34,7 @@ class RAGAgent:
         response = chain.invoke({"context": context, "question": question})
         return {"generation": response}
 
-    def reflect(self, state: GraphState):
+    def reflector_node(self, state: GraphState):
         question = state["question"]
         generation = state["generation"]
         docs = state["documents"]
@@ -51,7 +51,7 @@ class RAGAgent:
         normalized_score = "yes" if "yes" in score.lower() else "no"
         return {"reflection_score": normalized_score}
 
-    def rewrite_query(self, state: GraphState):
+    def rewriter_node(self, state: GraphState):
         question = state["question"]
         previous_query = state["current_query"]
         failed_gen = state["generation"]
@@ -78,24 +78,24 @@ class RAGAgent:
         workflow = StateGraph(GraphState)
 
 
-        workflow.add_node("retrieve", self.retrieve)
-        workflow.add_node("generate", self.generate)
-        workflow.add_node("reflect", self.reflect)
-        workflow.add_node("rewrite_query", self.rewrite_query)
+        workflow.add_node("retriever", self.retriever_node)
+        workflow.add_node("generator", self.generator_node)
+        workflow.add_node("reflector", self.reflector_node)
+        workflow.add_node("rewriter", self.rewriter_node)
 
-        workflow.set_entry_point("retrieve")
-        workflow.add_edge("retrieve", "generate")
-        workflow.add_edge("generate", "reflect")
+        workflow.set_entry_point("retriever")
+        workflow.add_edge("retriever", "generator")
+        workflow.add_edge("generator", "reflector")
         
         workflow.add_conditional_edges(
-            "reflect",
+            "reflector",
             self.decide_to_rewrite,
             {
-                "rewrite": "rewrite_query",
+                "rewrite": "rewriter",
                 "end": END
             }
         )
-        workflow.add_edge("rewrite_query", "retrieve")
+        workflow.add_edge("rewriter", "retriever")
 
         return workflow.compile()
 

prompts.py

@@ -5,14 +5,15 @@ RAG_SYSTEM = """You are a professional research and learning assistant. Answer t
 If the context does not contain the answer, say "I cannot answer this based on the provided material."
 
 Requirements:
-1. Use a professional and objective tone. Use **structured bullet points** or numbered lists to organize your answer for better readability.
+1. Use a professional and objective tone.
 2. **In-text Citations:** Use Unicode Superscript Numbers (¹, ², ³, ⁴, ⁵, ⁶, ⁷, ⁸, ⁹, ¹⁰) strictly. Place them immediately after the punctuation or relevant phrase.
    - Do NOT use `[^1]` (Markdown footnotes) or `[1]` (Brackets). 
    - Example: ...at compile time¹.
 3. **References Section:** At the very end, include a section titled "References".
-4. **Reference Format:** List the citations **sequentially in the order they appear in your answer** using normal numbers.
+4. **Reference Format:** List the citations sequentially using normal numbers.
    - Format: `1. Document: <filename>, Page: <page>`
-5. Be concise, comprehensive, and well-structured. """
+5. **Emphasis:** Use **bold** to highlight key points, core concepts, or critical data.
+6. Be concise, comprehensive, and well-structured"""
 
 RAG_HUMAN = """Context:
 {context}