Merge pull request HamaWhiteGG#50 from zhangxiaojiawow/qa-retriver-bu…

…gfix pass the input key to inner chain in RetrivalQAChain
wangmiao-1981 · Jul 21, 2023 · 624713a · 624713a
2 parents 16d17ea + cdf51fc
commit 624713a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/langchain-core/src/main/java/com/hw/langchain/chains/retrieval/qa/base/BaseRetrievalQA.java b/langchain-core/src/main/java/com/hw/langchain/chains/retrieval/qa/base/BaseRetrievalQA.java
@@ -78,7 +78,8 @@ public Map<String, String> innerCall(Map<String, Object> inputs) {
         var question = inputs.get(inputKey).toString();
 
         List<Document> docs = getDocs(question);
-        String answer = combineDocumentsChain.run(Map.of("input_documents", docs, "question", question));
+        inputs.put("input_documents", docs);
+        String answer = combineDocumentsChain.run(inputs);
 
         Map<String, String> result = Maps.newHashMap();
         result.put(outputKey, answer);