Update agent.py

agent.py

@@ -173,11 +173,12 @@ def build_graph(provider: str = "groq"):
 
     def retriever(state: MessagesState):
         similar = vector_store.similarity_search(state["messages"][0].content)
-        print("🔍 Retrieved vector documents:", similar)
-        example_msg = HumanMessage(
-            content=f"Here’s a similar QA pair for grounding:\n\n{similar[0].page_content}"
-        )
-        return {"messages": [sys_msg] + state["messages"] + [example_msg]}
+        examples = [
+            HumanMessage(content=f"Similar QA:\n{doc.page_content}")
+            for doc in similar[:2]
+        ]
+        return {"messages": [sys_msg] + state["messages"] + examples}
+
 
     def assistant(state: MessagesState):
         try: