bug fix

Alfred_Agent.py

@@ -84,9 +84,9 @@ def retriever(state: AgentState):
     """Retriever node"""
     similar_question = vector_search(state["messages"][0].content)
 
-    if similar_question:  # Check if the list is not empty
+    if similar_question:
         example_msg = HumanMessage(
-            content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0]}",
+            content=f"Here I provide a similar question and answer for reference: \n\n{similar_question}",
         )
         return {"messages": [ai_message] + state["messages"] + [example_msg]}
     else:
@@ -118,4 +118,4 @@ alfred = builder.compile()
 # response = alfred.invoke({"messages": messages})
 
 # print("🎩 Alfred's Response:")
-# print(response['messages'][-1].content)
+# print(response['messages'][-1].content)

other_tools.py

@@ -68,10 +68,10 @@ def web_search(query: str) -> str:
 
 from upload_metadata_n_setup_retrivers import retriever
 def vector_search(query:str) -> str:
-    search_docs = retriever.invoke(query=query)
+    search_docs = retriever.invoke(input=query)
     formatted_search_docs = "\n\n---\n\n".join(
-        [
+    [
             f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>{doc.page_content}</Document>'
             for doc in search_docs
         ])
-    return {"vector_results": formatted_search_docs}
+    return formatted_search_docs