Merge pull request HamaWhiteGG#58 from kael-aiur/dev

fix bug: exception at BaseCombineDocumentsChain when use BaseRetrievalQA, as missing key [question]
mgrygles-lab · Jul 25, 2023 · 4fb013f · 4fb013f
2 parents c17b614 + 0b32cfe
commit 4fb013f
Showing 1 changed file with 3 additions and 0 deletions.
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
@@ -79,6 +79,9 @@ public Map<String, String> innerCall(Map<String, Object> inputs) {
 
         List<Document> docs = getDocs(question);
         inputs.put("input_documents", docs);
+        if (!inputs.containsKey("question")) {
+            inputs.put("question", question);
+        }
         String answer = combineDocumentsChain.run(inputs);
 
         Map<String, String> result = Maps.newHashMap();