Voyage Integration

src/globals.ts

@@ -54,6 +54,7 @@ export const ZHIPU: string = 'zhipu';
 export const NOVITA_AI: string = 'novita-ai';
 export const MONSTERAPI: string = 'monsterapi';
 export const PREDIBASE: string = 'predibase';
+export const VOYAGE: string = 'voyage';
 
 export const VALID_PROVIDERS = [
   ANTHROPIC,
@@ -86,6 +87,7 @@ export const VALID_PROVIDERS = [
   NOVITA_AI,
   MONSTERAPI,
   PREDIBASE,
+  VOYAGE,
 ];
 
 export const CONTENT_TYPES = {

src/providers/index.ts

@@ -29,6 +29,7 @@ import ZhipuConfig from './zhipu';
 import NovitaAIConfig from './novita-ai';
 import MonsterAPIConfig from './monsterapi';
 import PredibaseConfig from './predibase';
+import VoyageConfig from './voyage';
 
 const Providers: { [key: string]: ProviderConfigs } = {
   openai: OpenAIConfig,
@@ -61,6 +62,7 @@ const Providers: { [key: string]: ProviderConfigs } = {
   'novita-ai': NovitaAIConfig,
   monsterapi: MonsterAPIConfig,
   predibase: PredibaseConfig,
+  voyage: VoyageConfig,
 };
 
 export default Providers;

src/providers/voyage/api.ts

@@ -0,0 +1,18 @@
+import { ProviderAPIConfig } from '../types';
+
+const VoyageAPIConfig: ProviderAPIConfig = {
+    getBaseURL: () => 'https://api.voyageai.com/v1',
+    headers: ({ providerOptions }) => {
+      return { Authorization: `Bearer ${providerOptions.apiKey}` };
+    },
+    getEndpoint: ({ fn }) => {
+      switch (fn) {
+        case 'embed':
+          return '/embeddings';
+        default:
+          return '';
+      }
+    },
+  };
+
+  export default VoyageAPIConfig;

src/providers/voyage/embed.ts

@@ -0,0 +1,96 @@
+import { VOYAGE } from '../../globals';
+import { EmbedParams, EmbedResponse } from '../../types/embedRequestBody';
+import { ErrorResponse, ProviderConfig } from '../types';
+import {
+  generateErrorResponse,
+  generateInvalidProviderResponseError,
+} from '../utils';
+
+export const VoyageEmbedConfig: ProviderConfig = {
+  model: {
+    param: 'model',
+    required: true,
+  },
+  input: {
+    param: 'input',
+    required: true,
+    transform: (params: EmbedParams) => {
+      if (Array.isArray(params.input)) {
+        return params.input;
+      }
+
+      return [params.input];
+    },
+  },
+  input_type: {
+    param: 'input_type',
+    required: false,
+    default: null,
+  },
+  truncation: {
+    param: 'truncation',
+    required: false,
+    default: true,
+  },
+  encoding_format: {
+    param: 'encoding_format',
+    required: false,
+    default: null,
+  },
+};
+
+interface VoyageEmbedResponse {
+  object: 'list';
+  data: Array<{ object: 'embedding'; embedding: number[]; index: number }>;
+  model: string;
+  usage: {
+    total_tokens: number;
+  };
+}
+
+export interface VoyageValidationErrorResponse {
+  detail: string;
+}
+
+export const VoyageErrorResponseTransform: (
+  response: VoyageValidationErrorResponse | any
+) => ErrorResponse = (response) => {
+  let errorField: string | null = null;
+
+  let errorMessage = response.detail;
+  let errorType = 'Invalid Request';
+
+  return generateErrorResponse(
+    {
+      message: `${errorField ? `${errorField}: ` : ''}${errorMessage}`,
+      type: errorType,
+      param: null,
+      code: null,
+    },
+    VOYAGE
+  );
+};
+
+export const VoyageEmbedResponseTransform: (
+  response: VoyageEmbedResponse | VoyageValidationErrorResponse | any,
+  responseStatus: number
+) => EmbedResponse | ErrorResponse = (response, responseStatus) => {
+  if ('detail' in response && responseStatus !== 200) {
+    return VoyageErrorResponseTransform(response);
+  }
+
+  if ('data' in response) {
+    return {
+      object: 'list',
+      data: response.data,
+      model: response.model,
+      usage: {
+        prompt_tokens: response.usage.total_tokens,
+        total_tokens: response.usage.total_tokens,
+      },
+      provider: VOYAGE,
+    };
+  }
+
+  return generateInvalidProviderResponseError(response, VOYAGE);
+};

src/providers/voyage/index.ts

@@ -0,0 +1,13 @@
+import { ProviderConfigs } from '../types';
+import VoyageAPIConfig from './api';
+import { VoyageEmbedConfig, VoyageEmbedResponseTransform } from './embed';
+
+const VoyageConfig: ProviderConfigs = {
+  embed: VoyageEmbedConfig,
+  api: VoyageAPIConfig,
+  responseTransforms: {
+    embed: VoyageEmbedResponseTransform,
+  },
+};
+
+export default VoyageConfig;