From d7f1a5cc8ce4589f2ea5b8988fef3249354673e5 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Mon, 26 Feb 2024 16:17:05 -0300
Subject: [PATCH] Add "retriever" input to the build method

---
 .../langflow/components/chains/RetrievalQAWithSourcesChain.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/backend/langflow/components/chains/RetrievalQAWithSourcesChain.py b/src/backend/langflow/components/chains/RetrievalQAWithSourcesChain.py
index f447e831e..faf3ab7dd 100644
--- a/src/backend/langflow/components/chains/RetrievalQAWithSourcesChain.py
+++ b/src/backend/langflow/components/chains/RetrievalQAWithSourcesChain.py
@@ -21,10 +21,12 @@ class RetrievalQAWithSourcesChainComponent(CustomComponent):
             },
             "memory": {"display_name": "Memory"},
             "return_source_documents": {"display_name": "Return Source Documents"},
+            "retriever": {"display_name": "Retriever"},
         }
 
     def build(
         self,
+        inputs: str,
         retriever: BaseRetriever,
         llm: BaseLanguageModel,
         chain_type: str,
@@ -50,7 +52,7 @@ class RetrievalQAWithSourcesChainComponent(CustomComponent):
         references_str = ""
         if return_source_documents:
             references_str = self.create_references_from_records(records)
-        result_str = result.get("result")
+        result_str = result.get("answer")
         final_result = "\n".join([result_str, references_str])
         self.status = final_result
         return final_result