Determine whether the question field exists

mgrygles-lab · Jul 25, 2023 · 0b32cfe · 0b32cfe
1 parent 2bfb096
commit 0b32cfe
Showing 1 changed file with 3 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
@@ -79,7 +79,9 @@ public Map<String, String> innerCall(Map<String, Object> inputs) {
 
         List<Document> docs = getDocs(question);
         inputs.put("input_documents", docs);
-        inputs.put("question", question);
+        if (!inputs.containsKey("question")) {
+            inputs.put("question", question);
+        }
         String answer = combineDocumentsChain.run(inputs);
 
         Map<String, String> result = Maps.newHashMap();