WIP AI Translate

Author: ja-openai

webapp/src/main/java/com/box/l10n/mojito/rest/textunit/AiTranslateWS.java

@@ -0,0 +1,54 @@
+package com.box.l10n.mojito.rest.textunit;
+
+import com.box.l10n.mojito.entity.PollableTask;
+import com.box.l10n.mojito.quartz.QuartzPollableTaskScheduler;
+import com.box.l10n.mojito.service.oaitranslate.AiTranslateConfigurationProperties;
+import com.box.l10n.mojito.service.oaitranslate.AiTranslateService;
+import com.box.l10n.mojito.service.oaitranslate.AiTranslateService.AiTranslateInput;
+import com.box.l10n.mojito.service.pollableTask.PollableFuture;
+import com.box.l10n.mojito.service.repository.RepositoryRepository;
+import java.util.List;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.HttpStatus;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestMethod;
+import org.springframework.web.bind.annotation.ResponseStatus;
+import org.springframework.web.bind.annotation.RestController;
+
+@RestController
+public class AiTranslateWS {
+
+  /** logger */
+  static Logger logger = LoggerFactory.getLogger(AiTranslateWS.class);
+
+  @Autowired AiTranslateService aiTranslateService;
+
+  @Autowired RepositoryRepository repositoryRepository;
+
+  @Autowired QuartzPollableTaskScheduler quartzPollableTaskScheduler;
+
+  @Autowired AiTranslateConfigurationProperties aiTranslateConfigurationProperties;
+
+  @RequestMapping(method = RequestMethod.POST, value = "/api/proto-ai-translate")
+  @ResponseStatus(HttpStatus.OK)
+  public ProtoAiTranslateResponse aiTranslate(@RequestBody ProtoAiTranslateRequest protoAiTranslateRequest) {
+
+    PollableFuture<Void> pollableFuture =
+        aiTranslateService.aiTranslateAsync(
+            new AiTranslateInput(
+                protoAiTranslateRequest.repositoryName(),
+                protoAiTranslateRequest.targetBcp47tags(),
+                protoAiTranslateRequest.sourceTextMaxCountPerLocale()));
+
+    return new ProtoAiTranslateResponse(pollableFuture.getPollableTask());
+  }
+
+  public record ProtoAiTranslateRequest(
+      String repositoryName, List<String> targetBcp47tags, int sourceTextMaxCountPerLocale) {}
+
+  public record ProtoAiTranslateResponse(PollableTask pollableTask) {
+  }
+}

webapp/src/main/java/com/box/l10n/mojito/service/blobstorage/StructuredBlobStorage.java

@@ -48,5 +48,6 @@ public enum Prefix {
     MULTI_BRANCH_STATE,
     TEXT_UNIT_DTOS_CACHE,
     CLOB_STORAGE_WS,
+    AI_TRANSLATE_WS,
   }
 }

webapp/src/main/java/com/box/l10n/mojito/service/oaitranslate/AiTranslateConfig.java

@@ -0,0 +1,44 @@
+package com.box.l10n.mojito.service.oaitranslate;
+
+import com.box.l10n.mojito.json.ObjectMapper;
+import com.box.l10n.mojito.openai.OpenAIClient;
+import java.time.Duration;
+import org.springframework.beans.factory.annotation.Qualifier;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import reactor.util.retry.Retry;
+import reactor.util.retry.RetryBackoffSpec;
+
+@Configuration
+public class AiTranslateConfig {
+
+  AiTranslateConfigurationProperties aiTranslateConfigurationProperties;
+
+  public AiTranslateConfig(AiTranslateConfigurationProperties aiTranslateConfigurationProperties) {
+    this.aiTranslateConfigurationProperties = aiTranslateConfigurationProperties;
+  }
+
+  @Bean
+  @Qualifier("AiTranslate")
+  OpenAIClient openAIClient() {
+    String openaiClientToken = aiTranslateConfigurationProperties.getOpenaiClientToken();
+    if (openaiClientToken == null) {
+      return null;
+    }
+    return new OpenAIClient.Builder().apiKey(openaiClientToken).build();
+  }
+
+  @Bean
+  @Qualifier("AiTranslate")
+  ObjectMapper objectMapper() {
+    ObjectMapper objectMapper = new ObjectMapper();
+    AiTranslateService.configureObjectMapper(objectMapper);
+    return objectMapper;
+  }
+
+  @Bean
+  @Qualifier("AiTranslate")
+  RetryBackoffSpec retryBackoffSpec() {
+    return Retry.backoff(5, Duration.ofMillis(500)).maxBackoff(Duration.ofSeconds(5));
+  }
+}

webapp/src/main/java/com/box/l10n/mojito/service/oaitranslate/AiTranslateConfigurationProperties.java

@@ -0,0 +1,28 @@
+package com.box.l10n.mojito.service.oaitranslate;
+
+import com.box.l10n.mojito.quartz.QuartzSchedulerManager;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.stereotype.Component;
+
+@Component
+@ConfigurationProperties("l10n.ai-translate")
+public class AiTranslateConfigurationProperties {
+  String openaiClientToken;
+  String schedulerName = QuartzSchedulerManager.DEFAULT_SCHEDULER_NAME;
+
+  public String getOpenaiClientToken() {
+    return openaiClientToken;
+  }
+
+  public void setOpenaiClientToken(String openaiClientToken) {
+    this.openaiClientToken = openaiClientToken;
+  }
+
+  public String getSchedulerName() {
+    return schedulerName;
+  }
+
+  public void setSchedulerName(String schedulerName) {
+    this.schedulerName = schedulerName;
+  }
+}

webapp/src/main/java/com/box/l10n/mojito/service/oaitranslate/AiTranslateJob.java

@@ -0,0 +1,27 @@
+package com.box.l10n.mojito.service.oaitranslate;
+
+import com.box.l10n.mojito.quartz.QuartzPollableJob;
+import com.box.l10n.mojito.service.oaitranslate.AiTranslateService.AiTranslateInput;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+/**
+ * Class to process a batch of strings for machine translation against a set of target languages.
+ *
+ * @author garion
+ */
+@Component
+public class AiTranslateJob extends QuartzPollableJob<AiTranslateInput, Void> {
+
+  static Logger logger = LoggerFactory.getLogger(AiTranslateJob.class);
+
+  @Autowired AiTranslateService aiTranslateService;
+
+  @Override
+  public Void call(AiTranslateInput aiTranslateJobInput) throws Exception {
+    aiTranslateService.aiTranslate(aiTranslateJobInput);
+    return null;
+  }
+}