remove low rated or neutral responses

planning_ai/nodes/reduce_node.py

@@ -4,7 +4,12 @@ from planning_ai.states import OverallState
 
 def generate_final_summary(state: OverallState):
     if len(state["documents"]) == len(state["summaries_fixed"]):
-        response = reduce_chain.invoke({"context": state["summaries_fixed"]})
+        summaries = [
+            summary["summary"].summary
+            for summary in state["summaries_fixed"]
+            if summary["summary"].stance != "NEUTRAL" and summary["summary"].rating >= 5
+        ]
+        response = reduce_chain.invoke({"context": summaries})
         return {
             "final_summary": response,
             "summaries_fixed": state["summaries_fixed"],
@@ -12,3 +17,12 @@ def generate_final_summary(state: OverallState):
             "hallucinations": state["hallucinations"],
             "documents": state["documents"],
         }
+
+
+def add_snippets(state: OverallState):
+    final_summary = state["final_summary"]
+    summaries = state["summaries_fixed"]
+
+    response = snippet_chain.invoke(
+        {"final_summary": final_summary, "summaries": summaries}
+    )