minor

Author: thiswillbeyourgithub

WDoc/WDoc.py

@@ -1399,7 +1399,7 @@ async def do_eval(inputs):
                 @optional_typecheck
                 def retrieve_documents(inputs):
                     return {
-                        "unfiltered_docs": retriever.invoke({"question_for_embedding": inputs["question_for_embedding"]}),
+                        "unfiltered_docs": retriever.invoke(inputs["question_for_embedding"]),
                         "question_to_answer": inputs["question_to_answer"],
                     }
                     return inputs
@@ -1458,7 +1458,7 @@ def retrieve_documents(inputs):
 
                 docs = output["filtered_docs"]
             else:
-                docs = retriever.invoke({"question_for_embedding": query})
+                docs = retriever.invoke(query)
                 if len(docs) < self.interaction_settings["top_k"]:
                     red(f"Only found {len(docs)} relevant documents")
 
@@ -1528,7 +1528,7 @@ def retrieve_documents(inputs):
             @optional_typecheck
             def retrieve_documents(inputs):
                 return {
-                    "unfiltered_docs": retriever.get_relevant_documents(inputs["question_for_embedding"]),
+                    "unfiltered_docs": retriever.invoke(inputs["question_for_embedding"]),
                     "question_to_answer": inputs["question_to_answer"],
                 }
                 return inputs