Make model name configurable

box · Feb 17, 2025 · e47c7e7 · e47c7e7
1 parent 9498b78
commit e47c7e7
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/...ain/java/com/box/l10n/mojito/service/oaitranslate/AiTranslateConfigurationProperties.java b/...ain/java/com/box/l10n/mojito/service/oaitranslate/AiTranslateConfigurationProperties.java
@@ -9,6 +9,7 @@
 public class AiTranslateConfigurationProperties {
   String openaiClientToken;
   String schedulerName = QuartzSchedulerManager.DEFAULT_SCHEDULER_NAME;
+  String modelName = "gpt-4o-2024-08-06";
 
   public String getOpenaiClientToken() {
     return openaiClientToken;
@@ -25,4 +26,12 @@ public String getSchedulerName() {
   public void setSchedulerName(String schedulerName) {
     this.schedulerName = schedulerName;
   }
+
+  public String getModelName() {
+    return modelName;
+  }
+
+  public void setModelName(String modelName) {
+    this.modelName = modelName;
+  }
 }
diff --git a/webapp/src/main/java/com/box/l10n/mojito/service/oaitranslate/AiTranslateService.java b/webapp/src/main/java/com/box/l10n/mojito/service/oaitranslate/AiTranslateService.java
@@ -299,7 +299,7 @@ private Mono<MyRecord> getChatCompletionForTextUnitDTO(
 
     ChatCompletionsRequest chatCompletionsRequest =
         chatCompletionsRequest()
-            .model("gpt-4o-2024-08-06")
+            .model(aiTranslateConfigurationProperties.getModelName())
             .maxTokens(16384)
             .messages(
                 List.of(
@@ -517,7 +517,7 @@ String generateBatchFileContent(List<TextUnitDTO> textUnitDTOS) {
 
               ChatCompletionsRequest chatCompletionsRequest =
                   chatCompletionsRequest()
-                      .model("gpt-4o-2024-08-06")
+                      .model(aiTranslateConfigurationProperties.getModelName())
                       .maxTokens(16384)
                       .messages(
                           List.of(