enhance code

dinsy008 · Oct 4, 2023 · 4aec20c · 4aec20c
1 parent bc5e5cd
commit 4aec20c
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/langchain-core/src/main/java/com/hw/langchain/chains/query/constructor/base/BaseUtils.java b/langchain-core/src/main/java/com/hw/langchain/chains/query/constructor/base/BaseUtils.java
@@ -77,7 +77,8 @@ private static BasePromptTemplate getPrompt(String documentContents, List<Attrib
                 Map.of("i", examples.size() + 1, "content", documentContents, "attributes", attributeStr));
 
         var outputParser = StructuredQueryOutputParser.fromComponents(allowedComparators, allowedOperators);
-        return new FewShotPromptTemplate(examples, EXAMPLE_PROMPT, prefix, suffix, List.of("query"), outputParser);
+        return new FewShotPromptTemplate(examples, EXAMPLE_PROMPT, prefix, suffix, List.of("query"), "\n\n",
+                outputParser);
     }
 
     private static Map<String, Object> createTemplateArguments(List<Comparator> allowedComparators,

diff --git a/langchain-core/src/main/java/com/hw/langchain/prompts/few/shot/FewShotPromptTemplate.java b/langchain-core/src/main/java/com/hw/langchain/prompts/few/shot/FewShotPromptTemplate.java
@@ -63,15 +63,26 @@ public class FewShotPromptTemplate extends StringPromptTemplate {
     /**
      * String separator used to join the prefix, the examples, and suffix.
      */
-    private String exampleSeparator = "\n\n";
+    private String exampleSeparator;
 
     public FewShotPromptTemplate(List<Map<String, Object>> examples, PromptTemplate examplePrompt, String prefix,
-            String suffix, List<String> inputVariables, BaseOutputParser<?> outputParser) {
+            String suffix, List<String> inputVariables) {
+        this(examples, examplePrompt, prefix, suffix, inputVariables, "\n\n");
+    }
+
+    public FewShotPromptTemplate(List<Map<String, Object>> examples, PromptTemplate examplePrompt, String prefix,
+            String suffix, List<String> inputVariables, String exampleSeparator) {
+        this(examples, examplePrompt, prefix, suffix, inputVariables, exampleSeparator, null);
+    }
+
+    public FewShotPromptTemplate(List<Map<String, Object>> examples, PromptTemplate examplePrompt, String prefix,
+            String suffix, List<String> inputVariables, String exampleSeparator, BaseOutputParser<?> outputParser) {
         super(inputVariables, outputParser);
         this.examples = examples;
         this.examplePrompt = examplePrompt;
         this.prefix = prefix;
         this.suffix = suffix;
+        this.exampleSeparator = exampleSeparator;
     }
 
     public List<Map<String, Object>> getExamples() {