feat(FgForrest#3): add chatgpt translator a configurations

Kamest · Mar 10, 2024 · f456970 · f456970
1 parent 58ea24c
commit f456970
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/src/main/java/one/edee/babylon/export/translator/DeeplTranslator.java b/src/main/java/one/edee/babylon/export/translator/DeeplTranslator.java
@@ -13,6 +13,8 @@
 import java.util.List;
 import java.util.stream.Collectors;
 
+import static java.util.Optional.ofNullable;
+
 /**
  * I apologize in advance for the lack of documentation in this code.
  * I had every intention of providing clear and concise explanations
@@ -43,13 +45,16 @@ public List<String> translate(@Nullable String defaultLang, @NotNull List<String
         if (lang.equals("en")) {
             lang = "en-GB";
         }
+
+        String contextMessage = System.getProperty("babylon.deepl.context");
         try {
             return service.translateText(
                             original,
                             defaultLang,
                             lang,
                             new TextTranslationOptions()
-                                    .setContext("You are eshop translator. Do not translate text from other languages then source lang, or technical texts."))
+                                    .setContext(ofNullable(contextMessage)
+                                            .orElse("You are eshop translator. Do not translate text from other languages then source lang, or technical texts.")))
                     .stream()
                     .map(TextResult::getText)
                     .collect(Collectors.toList());

diff --git a/src/main/java/one/edee/babylon/export/translator/OpenAiTranslator.java b/src/main/java/one/edee/babylon/export/translator/OpenAiTranslator.java
@@ -56,9 +56,10 @@ public List<String> translate(@Nullable String defaultLang, @NotNull List<String
                 lang);
 
 
+        String joined = String.join("~", original);
         List<ChatMessage> messages = Arrays.asList(
                 new ChatMessage("system", formattedSystemMessage),
-                new ChatMessage("user", String.join("~", original))
+                new ChatMessage("user", joined)
         );
 
         ChatCompletionResult chatCompletion;
@@ -81,7 +82,10 @@ public List<String> translate(@Nullable String defaultLang, @NotNull List<String
             }
             throw e;
         }
-        return Arrays.stream(chatCompletion.getChoices().get(0).getMessage().getContent().split("~")).collect(Collectors.toList());
+        String result = chatCompletion.getChoices().get(0).getMessage().getContent();
+        List<String> output = Arrays.stream(result.split("~")).collect(Collectors.toList());
+        Assert.isTrue(output.size() == original.size(), "Size not equal " + joined + " " + result);
+        return output;
     }
 
     @Override