Add translation endpoint to Igbo API

src/config.ts

@@ -26,6 +26,9 @@ const SENDGRID_NEW_DEVELOPER_ACCOUNT_TEMPLATE_SOURCE = defineString(
 // Igbo API
 const ENV_MAIN_KEY = defineString('ENV_MAIN_KEY').value();
 
+// Nkọwa okwu AI Models
+const ENV_IGBO_TO_ENGLISH_URL = defineString('ENV_IGBO_TO_ENGLISH_URL').value();
+
 // Google Analytics
 const ANALYTICS_GA_TRACKING_ID = defineString('ANALYTICS_GA_TRACKING_ID').value();
 const ANALYTICS_GA_API_SECRET = defineString('ANALYTICS_GA_API_SECRET').value();
@@ -101,11 +104,12 @@ export const CORS_CONFIG = {
 export const API_ROUTE = isProduction ? '' : `http://localhost:${PORT}`;
 export const API_DOCS = 'https://docs.igboapi.com';
 
-// IgboSpeech
+// Nkọwa okwu AI Models
 export const SPEECH_TO_TEXT_API = isProduction
   ? 'https://speech.igboapi.com'
   : 'http://localhost:3333';
 
+export const IGBO_TO_ENGLISH_API = ENV_IGBO_TO_ENGLISH_URL;
 // SendGrid API
 export const SENDGRID_API_KEY = SENDGRID_API_KEY_SOURCE || '';
 export const SENDGRID_NEW_DEVELOPER_ACCOUNT_TEMPLATE =

src/controllers/__tests__/examples.test.ts

@@ -1,6 +1,5 @@
 import { exampleFixture } from '../../__tests__/shared/fixtures';
 import LanguageEnum from '../../shared/constants/LanguageEnum';
-import { SuggestionSourceEnum } from '../../shared/constants/SuggestionSourceEnum';
 import { convertToV1Example } from '../examples';
 
 describe('examples', () => {

src/controllers/__tests__/translation.test.ts

@@ -0,0 +1,62 @@
+import axios from 'axios';
+import {
+  requestFixture,
+  responseFixture,
+  nextFunctionFixture,
+} from '../../../__tests__/shared/fixtures';
+import { getTranslation } from '../translation';
+
+describe('translation', () => {
+  beforeEach(() => {
+    jest.resetAllMocks();
+  });
+  it('calls Nkọwa okwu ML translation server', async () => {
+    const req = requestFixture({
+      body: { igbo: 'aka' },
+      headers: {
+        'Content-Type': 'application/json',
+        'X-API-Key': 'main_key',
+      },
+    });
+    const res = responseFixture();
+    const next = nextFunctionFixture();
+    jest.spyOn(axios, 'request').mockResolvedValue({
+      data: { igbo: 'aka' },
+    });
+    await getTranslation(req, res, next);
+    expect(res.send).toHaveBeenCalled();
+  });
+
+  it('throws validation error when input is too long', async () => {
+    const req = requestFixture({
+      body: { igbo: 'aka'.repeat(100) },
+      headers: {
+        'Content-Type': 'application/json',
+        'X-API-Key': 'main_key',
+      },
+    });
+    const res = responseFixture();
+    const next = nextFunctionFixture();
+    jest.spyOn(axios, 'request').mockResolvedValue({
+      data: { igbo: 'aka'.repeat(100) },
+    });
+    await getTranslation(req, res, next);
+    expect(next).toHaveBeenCalled();
+  });
+  it('throws validation error when input string is empty', async () => {
+    const req = requestFixture({
+      body: { igbo: '' },
+      headers: {
+        'Content-Type': 'application/json',
+        'X-API-Key': 'main_key',
+      },
+    });
+    const res = responseFixture();
+    const next = nextFunctionFixture();
+    jest.spyOn(axios, 'request').mockResolvedValue({
+      data: { igbo: '' },
+    });
+    await getTranslation(req, res, next);
+    expect(next).toHaveBeenCalled();
+  });
+});

src/controllers/translation.ts

@@ -0,0 +1,52 @@
+import axios from 'axios';
+import { MiddleWare } from '../types';
+import { IGBO_TO_ENGLISH_API, MAIN_KEY } from '../config';
+
+interface TranslationMetadata {
+  igbo: string;
+}
+
+interface Translation {
+  translation: string;
+}
+
+// Due to limit on inputs used to train the model, the maximum
+// Igbo translation input is 120 characters
+const TRANSLATION_INPUT_MAX_LENGTH = 120;
+
+/**
+ * Talks to Igbo-to-English translation model to translate the provided text.
+ * @param req
+ * @param res
+ * @param next
+ * @returns English text translation of the provided Igbo text
+ */
+export const getTranslation: MiddleWare = async (req, res, next) => {
+  try {
+    const igboText: string = req.body['igbo'];
+    if (!igboText) {
+      throw new Error('Cannot translate empty string');
+    }
+
+    if (igboText.length > TRANSLATION_INPUT_MAX_LENGTH) {
+      throw new Error('Cannot translate text greater than 120 characters');
+    }
+
+    const payload: TranslationMetadata = { igbo: igboText };
+
+    // Talks to prediction endpoint
+    const { data: response } = await axios.request<Translation>({
+      method: 'POST',
+      url: IGBO_TO_ENGLISH_API,
+      headers: {
+        'Content-Type': 'application/json',
+        'X-API-Key': MAIN_KEY,
+      },
+      data: payload,
+    });
+
+    return res.send({ translation: response.translation });
+  } catch (err) {
+    return next(err);
+  }
+};

src/middleware/helpers/authorizeDeveloperUsage.ts

@@ -3,6 +3,7 @@ import { findDeveloperUsage } from './findDeveloperUsage';
 import { createDeveloperUsage } from './createDeveloperUsage';
 import { DeveloperUsageDocument } from '../../types/developerUsage';
 import ApiType from '../../shared/constants/ApiType';
+import ApiTypeToRoute from '../../shared/constants/ApiTypeToRoute';
 import ApiUsageLimit from '../../shared/constants/ApiUsageLimit';
 import { DeveloperDocument } from '../../types';
 
@@ -58,8 +59,27 @@ export const authorizeDeveloperUsage = async ({
 }: {
   path: string,
   developer: DeveloperDocument,
-}) =>
-  handleDeveloperUsage({
+}) => {
+  const extractedPath = getPath(path);
+  return handleDeveloperUsage({
     developer,
-    apiType: path.startsWith('speech-to-text') ? ApiType.SPEECH_TO_TEXT : ApiType.DICTIONARY,
+    apiType: getApiTypeFromRoute(extractedPath),
   });
+};
+
+const getApiTypeFromRoute = (route: string): ApiType => {
+  switch (route) {
+    case ApiTypeToRoute.SPEECH_TO_TEXT:
+      return ApiType.SPEECH_TO_TEXT;
+    case ApiTypeToRoute.TRANSLATE:
+      return ApiType.TRANSLATE;
+    default:
+      return ApiType.DICTIONARY;
+  }
+};
+
+const getPath = (path: string) => {
+  // Extract router path from full path
+  // ex. speech-to-text/params=param -> speech-to-text
+  return path.split(/[\/\?]/)[0];
+};

src/routers/routerV2.ts

@@ -1,12 +1,24 @@
 import { Router } from 'express';
+import rateLimit from 'express-rate-limit';
 import { getWords, getWord } from '../controllers/words';
 import { getExample, getExamples } from '../controllers/examples';
 import { getNsibidiCharacter, getNsibidiCharacters } from '../controllers/nsibidi';
 import { getTranscription } from '../controllers/speechToText';
+import { getTranslation } from '../controllers/translation';
 import validId from '../middleware/validId';
 import validateApiKey from '../middleware/validateApiKey';
 import analytics from '../middleware/analytics';
 import attachRedisClient from '../middleware/attachRedisClient';
+import { MiddleWare } from 'src/types';
+
+// TODO: add rate limiting with upstash
+// const ONE_DAY = 24 * 60 * 60 * 1000;
+// const REQUESTS_PER_MS_TRANSLATION = 5;
+
+// const translationRateLimiter: MiddleWare = rateLimit({
+//   windowMs: ONE_DAY,
+//   max: REQUESTS_PER_MS_TRANSLATION,
+// });
 
 const routerV2 = Router();
 
@@ -26,6 +38,7 @@
 
 // Speech-to-Text
 routerV2.post('/speech-to-text', analytics, validateApiKey, getTranscription);
+routerV2.post('/translate', analytics, validateApiKey, getTranslation);
 
 // Redirects to V1
 routerV2.post('/developers', (_, res) => res.redirect('/api/v1/developers'));

src/shared/constants/ApiType.ts

@@ -1,6 +1,7 @@
 enum ApiType {
   DICTIONARY = 'DICTIONARY',
   SPEECH_TO_TEXT = 'SPEECH_TO_TEXT',
+  TRANSLATE = 'TRANSLATE',
 }
 
 export default ApiType;

src/shared/constants/ApiTypeToRoute.ts

@@ -0,0 +1,8 @@
+import ApiType from './ApiType';
+
+const ApiTypeToRoute = {
+  [ApiType.SPEECH_TO_TEXT]: 'speech-to-text',
+  [ApiType.TRANSLATE]: 'translate',
+};
+
+export default ApiTypeToRoute;

src/shared/constants/ApiUsageLimit.ts

@@ -3,6 +3,7 @@ import ApiType from './ApiType';
 const ApiUsageLimit = {
   [ApiType.DICTIONARY]: 2500,
   [ApiType.SPEECH_TO_TEXT]: 20,
+  [ApiType.TRANSLATE]: 5,
 };
 
 export default ApiUsageLimit;