feat(chains): Support custom doc prompt in StuffDocumentsQAChain (#157)

davidmigloz · Sep 9, 2023 · faa9d2d · faa9d2d
1 parent 34bb440
commit faa9d2d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/packages/langchain/lib/src/chains/question_answering/stuff.dart b/packages/langchain/lib/src/chains/question_answering/stuff.dart
@@ -53,8 +53,12 @@ final _promptSelector = ConditionalPromptSelector(
 class StuffDocumentsQAChain extends StuffDocumentsChain {
   /// {@macro stuff_documents_qa_chain}
   StuffDocumentsQAChain({
-    required final BaseLanguageModel llm,
     final BasePromptTemplate? prompt,
+    required final BaseLanguageModel llm,
+    super.inputKey = StuffDocumentsChain.defaultInputKey,
+    super.outputKey = StuffDocumentsChain.defaultOutputKey,
+    super.documentPrompt = StuffDocumentsChain.defaultDocumentPrompt,
+    super.documentSeparator = StuffDocumentsChain.defaultDocumentSeparator,
   }) : super(
           llmChain: LLMChain(
             prompt: prompt ?? _promptSelector.getPrompt(llm),