Update app/graph/nodes/synthesizer.py

app/graph/nodes/synthesizer.py

@@ -1,7 +1,11 @@
+
+
+
 # app/graph/nodes/synthesizer.py
 
 from app.core.llm_engine import llm
 from app.core.prompts.rag_prompt import rag_prompt
+from app.core.prompts.hybrid_prompt import hybrid_prompt
 from langchain_core.output_parsers import StrOutputParser
 
 
@@ -9,18 +13,25 @@ def synthesizer_node(state):
     route = state.get("route")
     general_answer = state.get("general_answer")
 
-    if route in ("general", "hybrid"):
+    # ✅ FIX: Use 'in' operator
+    if route == "general":
         return {
             **state,
-            "final_answer": general_answer or "I couldn't find a relevant answer."
+            "final_answer": general_answer or "I couldn't find relevant information."
         }
-
+    
+    # Prepare common inputs
     query = state["query"]
     context = state.get("context", "")
     history = state.get("history", "")
-
-
-    chain = rag_prompt | llm | StrOutputParser()
+    
+    # ✅ Route-specific prompt selection
+    if route == "hybrid":
+        chain = hybrid_prompt | llm | StrOutputParser()
+        print("🔀 Using HYBRID prompt (doc snippet + general knowledge)")
+    else:  # route == "rag"
+        chain = rag_prompt | llm | StrOutputParser()
+        print("📚 Using RAG prompt (document-only)")
     
     answer = chain.invoke({
         "context": context,
@@ -32,5 +43,4 @@ def synthesizer_node(state):
         **state,
         "final_answer": answer.strip()
     }
-
-
+