add Retrieval QA

zhangxiaojiawow · Jul 4, 2023 · 7e0ee7c · 7e0ee7c
1 parent 1b41a05
commit 7e0ee7c
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 4 deletions.
diff --git a/...re/src/main/java/com/hw/langchain/chains/combine/documents/stuff/StuffDocumentsChain.java b/...re/src/main/java/com/hw/langchain/chains/combine/documents/stuff/StuffDocumentsChain.java
@@ -64,7 +64,8 @@ public StuffDocumentsChain(LLMChain llmChain, String documentVariableName) {
         this(llmChain, getDefaultDocumentPrompt(), documentVariableName, "\n\n");
     }
 
-    public StuffDocumentsChain(LLMChain llmChain, BasePromptTemplate documentPrompt, String documentVariableName, String documentSeparator) {
+    public StuffDocumentsChain(LLMChain llmChain, BasePromptTemplate documentPrompt, String documentVariableName,
+            String documentSeparator) {
         this.llmChain = llmChain;
         this.documentPrompt = documentPrompt;
         this.documentVariableName = documentVariableName;

diff --git a/langchain-core/src/main/java/com/hw/langchain/chains/question/answering/ChainType.java b/langchain-core/src/main/java/com/hw/langchain/chains/question/answering/ChainType.java
@@ -37,7 +37,7 @@ public enum ChainType {
      * Chain type for "refine".
      */
     REFINE("refine"),
-    
+
     /**
      * Chain type for "map_rerank".
      */

diff --git a/langchain-core/src/main/java/com/hw/langchain/chains/question/answering/StuffPrompt.java b/langchain-core/src/main/java/com/hw/langchain/chains/question/answering/StuffPrompt.java
@@ -36,7 +36,7 @@
  */
 public class StuffPrompt {
 
-    private StuffPrompt(){
+    private StuffPrompt() {
     }
 
     private static final String PROMPT_TEMPLATE =

diff --git a/langchain-core/src/test/java/com/hw/langchain/chains/retrieval/qa/base/RetrievalQATest.java b/langchain-core/src/test/java/com/hw/langchain/chains/retrieval/qa/base/RetrievalQATest.java
@@ -21,8 +21,8 @@
 import com.hw.langchain.embeddings.openai.OpenAIEmbeddings;
 import com.hw.langchain.llms.openai.OpenAI;
 import com.hw.langchain.vectorstores.pinecone.Pinecone;
-import com.hw.pinecone.PineconeClient;
 import com.hw.langchain.vectorstores.pinecone.PineconeTest;
+import com.hw.pinecone.PineconeClient;
 
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Disabled;
-Original file line number
+Diff line change
@@ Expand Up / @@ -37,7 +37,7 @@ public enum ChainType { @@
          * Chain type for "refine".
          */
         REFINE("refine"),
         /**
          * Chain type for "map_rerank".
          */
@@ Expand Down @@