Update RetrievalQAComponent parameters

src/backend/langflow/components/chains/RetrievalQA.py

@@ -1,13 +1,11 @@
 
 from langflow import CustomComponent
-from langchain.chains import BaseRetrievalQA
 from typing import Optional, Union, Callable
 from langflow.field_typing import (
-    BaseCombineDocumentsChain,
     BaseMemory,
-    BaseRetriever,
-)
-
+    BaseRetriever)
+from langchain.chains.retrieval_qa.base import BaseRetrievalQA
+from langchain.chains.combine_documents.base import BaseCombineDocumentsChain
 class RetrievalQAComponent(CustomComponent):
     display_name = "RetrievalQA"
     description = "Chain for question-answering against an index."
@@ -17,8 +15,8 @@ class RetrievalQAComponent(CustomComponent):
             "combine_documents_chain": {"display_name": "Combine Documents Chain"},
             "retriever": {"display_name": "Retriever"},
             "memory": {"display_name": "Memory", "required": False},
-            "input_key": {"display_name": "Input Key"},
-            "output_key": {"display_name": "Output Key"},
+            "input_key": {"display_name": "Input Key","advanced":True},
+            "output_key": {"display_name": "Output Key","advanced":True},
             "return_source_documents": {"display_name": "Return Source Documents"},
         }
 
@@ -27,9 +25,9 @@ class RetrievalQAComponent(CustomComponent):
         combine_documents_chain: BaseCombineDocumentsChain,
         retriever: BaseRetriever,
         memory: Optional[BaseMemory] = None,
-        input_key: str = "query",
-        output_key: str = "result",
-        return_source_documents: bool = True,
+        input_key: Optional[str] = "query",
+        output_key: Optional[str] = "result",
+        return_source_documents: Optional[bool] = True,
     ) -> Union[BaseRetrievalQA, Callable]:
         return BaseRetrievalQA(
             combine_documents_chain=combine_documents_chain,