WIP AI Translate

Author: ja-openai

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

@@ -0,0 +1,50 @@
+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.rest.repository.RepositoryWithIdNotFoundException;
+import com.box.l10n.mojito.service.oaitranslate.AiTranslateConfigurationProperties;
+import com.box.l10n.mojito.service.oaitranslate.AiTranslateService;
+import com.box.l10n.mojito.service.pollableTask.PollableFuture;
+import com.box.l10n.mojito.service.repository.RepositoryRepository;
+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 PollableTask aiTranslate(@RequestBody ProtoAiTranslateRequest protoAiTranslateRequest)
+      throws RepositoryWithIdNotFoundException {
+
+    repositoryRepository
+        .findById(protoAiTranslateRequest.repositoryId())
+        .orElseThrow(
+            () -> new RepositoryWithIdNotFoundException(protoAiTranslateRequest.repositoryId()));
+
+    PollableFuture<Void> pollableFuture =
+        aiTranslateService.aiTranslateAsync(protoAiTranslateRequest.repositoryId());
+    return pollableFuture.getPollableTask();
+  }
+
+  public record ProtoAiTranslateRequest(long repositoryId) {}
+}

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,42 @@
+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() {
+    return new OpenAIClient.Builder()
+        .apiKey(aiTranslateConfigurationProperties.getOpenaiClientToken())
+        .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,32 @@
+package com.box.l10n.mojito.service.oaitranslate;
+
+import static com.box.l10n.mojito.service.oaitranslate.AiTranslateJob.*;
+
+import com.box.l10n.mojito.quartz.QuartzPollableJob;
+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<AiTranslateJobInput, Void> {
+
+  static Logger logger = LoggerFactory.getLogger(AiTranslateJob.class);
+
+  @Autowired AiTranslateService aiTranslateService;
+
+  @Override
+  public Void call(AiTranslateJobInput aiTranslateJobInput) throws Exception {
+    logger.debug(
+        "Start AiTranslateJob with repository id:  {}", aiTranslateJobInput.repositoryId());
+    aiTranslateService.aiTranslate(aiTranslateJobInput.repositoryId());
+    return null;
+  }
+
+  public record AiTranslateJobInput(long repositoryId) {}
+}