Upload 2 files

app/graph/graph_builder.py

@@ -7,31 +7,43 @@ from app.graph.nodes.synthesizer import synthesizer_node
 from app.graph.nodes.evaluator import evaluator_node
 from app.graph.nodes.general_agent import general_agent_node
 
+from app.graph.nodes.hybrid_agent import hybrid_agent_node
+
+def route_decision(state):
+    return state["route"]
 
 def build_graph():
     builder = StateGraph(GraphState)
 
     # nodes
     builder.add_node("router", router_node)
-    builder.add_node("rag_agent", rag_agent_node)
     builder.add_node("synthesizer", synthesizer_node)
     builder.add_node("evaluator", evaluator_node)
+
+    builder.add_node("rag_agent", rag_agent_node)
     builder.add_node("general_agent", general_agent_node)
+    builder.add_node("hybrid_agent", hybrid_agent_node)
+
 
     # edges
     builder.set_entry_point("router")
 
     builder.add_conditional_edges(
         "router",
-        lambda state: state["route"],
+        route_decision,
         {
             "rag": "rag_agent",
+            "hybrid": "hybrid_agent",
             "general": "general_agent"
         }
     )
 
+      # ✅ both paths merge at synthesizer
     builder.add_edge("rag_agent", "synthesizer")
+    builder.add_edge("hybrid_agent", "synthesizer")
     builder.add_edge("general_agent", "synthesizer")
+
+    # ✅ synthesizer → evaluator → end
     builder.add_edge("synthesizer", "evaluator")
     builder.add_edge("evaluator", END)
 

app/graph/state.py

@@ -1,4 +1,4 @@
-from typing import TypedDict, Optional
+from typing import TypedDict, Optional, List
 
 class GraphState(TypedDict):
     query: str
@@ -7,7 +7,8 @@ class GraphState(TypedDict):
     route: Optional[str]
 
     context: Optional[str]
-    sources: Optional[str]
+    sources: Optional[List]
+    score: Optional[float]
 
     history: Optional[str]