@kb-labs/adapters-voyage-ai 2.96.0

package/README.md

@@ -0,0 +1,5 @@
+# @kb-labs/adapters-voyage-ai
+
+> Part of [KB Labs](https://github.com/KirillBaranov/kb-labs) ecosystem.
+
+Voyage AI adapter implementing the `IEmbeddings` interface.

package/dist/embeddings-manifest.d.ts

@@ -0,0 +1,5 @@
+import { AdapterManifest } from '@kb-labs/sdk/adapters';
+
+declare const manifest: AdapterManifest;
+
+export { manifest };

package/dist/embeddings-manifest.js

@@ -0,0 +1,44 @@
+// src/embeddings-manifest.ts
+var manifest = {
+  manifestVersion: "1.0.0",
+  id: "voyage-ai-embeddings",
+  name: "Voyage AI Embeddings",
+  version: "0.1.0",
+  description: "Voyage AI text embeddings adapter (voyage-3, voyage-3.5, voyage-code-3, etc.)",
+  author: "KB Labs",
+  license: "MIT",
+  type: "core",
+  implements: "IEmbeddings",
+  capabilities: {
+    batch: true
+  },
+  configSchema: {
+    apiKey: {
+      type: "string",
+      description: "Voyage AI API key (defaults to VOYAGE_API_KEY env var)"
+    },
+    model: {
+      type: "string",
+      default: "voyage-3",
+      description: "Embedding model to use",
+      enum: [
+        "voyage-3",
+        "voyage-3-lite",
+        "voyage-3.5",
+        "voyage-code-3",
+        "voyage-finance-2",
+        "voyage-law-2",
+        "voyage-multilingual-2"
+      ]
+    },
+    inputType: {
+      type: "string",
+      description: "Input type hint for retrieval tasks",
+      enum: ["query", "document"]
+    }
+  }
+};
+
+export { manifest };
+//# sourceMappingURL=embeddings-manifest.js.map
+//# sourceMappingURL=embeddings-manifest.js.map

package/dist/embeddings-manifest.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/embeddings-manifest.ts"],"names":[],"mappings":";AAEO,IAAM,QAAA,GAA4B;AAAA,EACvC,eAAA,EAAiB,OAAA;AAAA,EACjB,EAAA,EAAI,sBAAA;AAAA,EACJ,IAAA,EAAM,sBAAA;AAAA,EACN,OAAA,EAAS,OAAA;AAAA,EACT,WAAA,EACE,+EAAA;AAAA,EACF,MAAA,EAAQ,SAAA;AAAA,EACR,OAAA,EAAS,KAAA;AAAA,EACT,IAAA,EAAM,MAAA;AAAA,EACN,UAAA,EAAY,aAAA;AAAA,EACZ,YAAA,EAAc;AAAA,IACZ,KAAA,EAAO;AAAA,GACT;AAAA,EACA,YAAA,EAAc;AAAA,IACZ,MAAA,EAAQ;AAAA,MACN,IAAA,EAAM,QAAA;AAAA,MACN,WAAA,EAAa;AAAA,KACf;AAAA,IACA,KAAA,EAAO;AAAA,MACL,IAAA,EAAM,QAAA;AAAA,MACN,OAAA,EAAS,UAAA;AAAA,MACT,WAAA,EAAa,wBAAA;AAAA,MACb,IAAA,EAAM;AAAA,QACJ,UAAA;AAAA,QACA,eAAA;AAAA,QACA,YAAA;AAAA,QACA,eAAA;AAAA,QACA,kBAAA;AAAA,QACA,cAAA;AAAA,QACA;AAAA;AACF,KACF;AAAA,IACA,SAAA,EAAW;AAAA,MACT,IAAA,EAAM,QAAA;AAAA,MACN,WAAA,EAAa,qCAAA;AAAA,MACb,IAAA,EAAM,CAAC,OAAA,EAAS,UAAU;AAAA;AAC5B;AAEJ","file":"embeddings-manifest.js","sourcesContent":["import type { AdapterManifest } from \"@kb-labs/sdk/adapters\";\n\nexport const manifest: AdapterManifest = {\n  manifestVersion: \"1.0.0\",\n  id: \"voyage-ai-embeddings\",\n  name: \"Voyage AI Embeddings\",\n  version: \"0.1.0\",\n  description:\n    \"Voyage AI text embeddings adapter (voyage-3, voyage-3.5, voyage-code-3, etc.)\",\n  author: \"KB Labs\",\n  license: \"MIT\",\n  type: \"core\",\n  implements: \"IEmbeddings\",\n  capabilities: {\n    batch: true,\n  },\n  configSchema: {\n    apiKey: {\n      type: \"string\",\n      description: \"Voyage AI API key (defaults to VOYAGE_API_KEY env var)\",\n    },\n    model: {\n      type: \"string\",\n      default: \"voyage-3\",\n      description: \"Embedding model to use\",\n      enum: [\n        \"voyage-3\",\n        \"voyage-3-lite\",\n        \"voyage-3.5\",\n        \"voyage-code-3\",\n        \"voyage-finance-2\",\n        \"voyage-law-2\",\n        \"voyage-multilingual-2\",\n      ],\n    },\n    inputType: {\n      type: \"string\",\n      description: \"Input type hint for retrieval tasks\",\n      enum: [\"query\", \"document\"],\n    },\n  },\n};\n"]}

package/dist/embeddings.d.ts

@@ -0,0 +1,41 @@
+import { IEmbeddings } from '@kb-labs/sdk/adapters';
+export { manifest } from './embeddings-manifest.js';
+
+/**
+ * @module @kb-labs/adapters-voyage-ai/embeddings
+ * Voyage AI implementation of IEmbeddings interface.
+ */
+
+declare const EMBEDDING_MODELS: {
+    readonly "voyage-3": 1024;
+    readonly "voyage-3-lite": 512;
+    readonly "voyage-3.5": 2048;
+    readonly "voyage-code-3": 1024;
+    readonly "voyage-finance-2": 1024;
+    readonly "voyage-law-2": 1024;
+    readonly "voyage-multilingual-2": 1024;
+};
+type EmbeddingModel = keyof typeof EMBEDDING_MODELS;
+type InputType = "query" | "document";
+interface VoyageAIEmbeddingsConfig {
+    /** Voyage AI API key (defaults to VOYAGE_API_KEY env var) */
+    apiKey?: string;
+    /** Embedding model to use */
+    model?: EmbeddingModel;
+    /** Input type hint for retrieval tasks */
+    inputType?: InputType;
+}
+declare class VoyageAIEmbeddings implements IEmbeddings {
+    private apiKey;
+    private model;
+    private inputType;
+    readonly dimensions: number;
+    constructor(config?: VoyageAIEmbeddingsConfig);
+    private request;
+    embed(text: string): Promise<number[]>;
+    embedBatch(texts: string[]): Promise<number[][]>;
+    getDimensions(): Promise<number>;
+}
+declare function createAdapter(config?: VoyageAIEmbeddingsConfig): VoyageAIEmbeddings;
+
+export { EMBEDDING_MODELS, type EmbeddingModel, type InputType, VoyageAIEmbeddings, type VoyageAIEmbeddingsConfig, createAdapter, createAdapter as default };

package/dist/embeddings.js

@@ -0,0 +1,117 @@
+// src/embeddings-manifest.ts
+var manifest = {
+  manifestVersion: "1.0.0",
+  id: "voyage-ai-embeddings",
+  name: "Voyage AI Embeddings",
+  version: "0.1.0",
+  description: "Voyage AI text embeddings adapter (voyage-3, voyage-3.5, voyage-code-3, etc.)",
+  author: "KB Labs",
+  license: "MIT",
+  type: "core",
+  implements: "IEmbeddings",
+  capabilities: {
+    batch: true
+  },
+  configSchema: {
+    apiKey: {
+      type: "string",
+      description: "Voyage AI API key (defaults to VOYAGE_API_KEY env var)"
+    },
+    model: {
+      type: "string",
+      default: "voyage-3",
+      description: "Embedding model to use",
+      enum: [
+        "voyage-3",
+        "voyage-3-lite",
+        "voyage-3.5",
+        "voyage-code-3",
+        "voyage-finance-2",
+        "voyage-law-2",
+        "voyage-multilingual-2"
+      ]
+    },
+    inputType: {
+      type: "string",
+      description: "Input type hint for retrieval tasks",
+      enum: ["query", "document"]
+    }
+  }
+};
+
+// src/embeddings.ts
+var EMBEDDING_MODELS = {
+  "voyage-3": 1024,
+  "voyage-3-lite": 512,
+  "voyage-3.5": 2048,
+  "voyage-code-3": 1024,
+  "voyage-finance-2": 1024,
+  "voyage-law-2": 1024,
+  "voyage-multilingual-2": 1024
+};
+var API_URL = "https://api.voyageai.com/v1/embeddings";
+var VoyageAIEmbeddings = class {
+  apiKey;
+  model;
+  inputType;
+  dimensions;
+  constructor(config = {}) {
+    const key = config.apiKey ?? process.env["VOYAGE_API_KEY"];
+    if (!key) {
+      throw new Error(
+        "Voyage AI API key is required. Set VOYAGE_API_KEY env var or pass apiKey in config."
+      );
+    }
+    this.apiKey = key;
+    this.model = config.model ?? "voyage-3";
+    this.inputType = config.inputType;
+    this.dimensions = EMBEDDING_MODELS[this.model];
+  }
+  async request(input) {
+    const body = {
+      model: this.model,
+      input
+    };
+    if (this.inputType) {
+      body["input_type"] = this.inputType;
+    }
+    const res = await fetch(API_URL, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        Authorization: `Bearer ${this.apiKey}`
+      },
+      body: JSON.stringify(body)
+    });
+    if (!res.ok) {
+      const text = await res.text().catch(() => res.statusText);
+      throw new Error(`Voyage AI API error ${res.status}: ${text}`);
+    }
+    const json = await res.json();
+    return json.data.sort((a, b) => a.index - b.index).map((d) => d.embedding);
+  }
+  async embed(text) {
+    const [embedding] = await this.request(text);
+    if (!embedding) {
+      throw new Error("Voyage AI embeddings API returned empty response");
+    }
+    return embedding;
+  }
+  async embedBatch(texts) {
+    if (texts.length === 0) {
+      return [];
+    }
+    return this.request(texts);
+  }
+  async getDimensions() {
+    return this.dimensions;
+  }
+};
+function createAdapter(config) {
+  return new VoyageAIEmbeddings(config);
+}
+var embeddings_default = createAdapter;
+
+export { EMBEDDING_MODELS, VoyageAIEmbeddings, createAdapter, embeddings_default as default, manifest };
+//# sourceMappingURL=embeddings.js.map
+//# sourceMappingURL=embeddings.js.map

package/dist/embeddings.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/embeddings-manifest.ts","../src/embeddings.ts"],"names":[],"mappings":";AAEO,IAAM,QAAA,GAA4B;AAAA,EACvC,eAAA,EAAiB,OAAA;AAAA,EACjB,EAAA,EAAI,sBAAA;AAAA,EACJ,IAAA,EAAM,sBAAA;AAAA,EACN,OAAA,EAAS,OAAA;AAAA,EACT,WAAA,EACE,+EAAA;AAAA,EACF,MAAA,EAAQ,SAAA;AAAA,EACR,OAAA,EAAS,KAAA;AAAA,EACT,IAAA,EAAM,MAAA;AAAA,EACN,UAAA,EAAY,aAAA;AAAA,EACZ,YAAA,EAAc;AAAA,IACZ,KAAA,EAAO;AAAA,GACT;AAAA,EACA,YAAA,EAAc;AAAA,IACZ,MAAA,EAAQ;AAAA,MACN,IAAA,EAAM,QAAA;AAAA,MACN,WAAA,EAAa;AAAA,KACf;AAAA,IACA,KAAA,EAAO;AAAA,MACL,IAAA,EAAM,QAAA;AAAA,MACN,OAAA,EAAS,UAAA;AAAA,MACT,WAAA,EAAa,wBAAA;AAAA,MACb,IAAA,EAAM;AAAA,QACJ,UAAA;AAAA,QACA,eAAA;AAAA,QACA,YAAA;AAAA,QACA,eAAA;AAAA,QACA,kBAAA;AAAA,QACA,cAAA;AAAA,QACA;AAAA;AACF,KACF;AAAA,IACA,SAAA,EAAW;AAAA,MACT,IAAA,EAAM,QAAA;AAAA,MACN,WAAA,EAAa,qCAAA;AAAA,MACb,IAAA,EAAM,CAAC,OAAA,EAAS,UAAU;AAAA;AAC5B;AAEJ;;;AClCO,IAAM,gBAAA,GAAmB;AAAA,EAC9B,UAAA,EAAY,IAAA;AAAA,EACZ,eAAA,EAAiB,GAAA;AAAA,EACjB,YAAA,EAAc,IAAA;AAAA,EACd,eAAA,EAAiB,IAAA;AAAA,EACjB,kBAAA,EAAoB,IAAA;AAAA,EACpB,cAAA,EAAgB,IAAA;AAAA,EAChB,uBAAA,EAAyB;AAC3B;AAeA,IAAM,OAAA,GAAU,wCAAA;AAMT,IAAM,qBAAN,MAAgD;AAAA,EAC7C,MAAA;AAAA,EACA,KAAA;AAAA,EACA,SAAA;AAAA,EACC,UAAA;AAAA,EAET,WAAA,CAAY,MAAA,GAAmC,EAAC,EAAG;AACjD,IAAA,MAAM,GAAA,GAAM,MAAA,CAAO,MAAA,IAAU,OAAA,CAAQ,IAAI,gBAAgB,CAAA;AACzD,IAAA,IAAI,CAAC,GAAA,EAAK;AACR,MAAA,MAAM,IAAI,KAAA;AAAA,QACR;AAAA,OACF;AAAA,IACF;AACA,IAAA,IAAA,CAAK,MAAA,GAAS,GAAA;AACd,IAAA,IAAA,CAAK,KAAA,GAAQ,OAAO,KAAA,IAAS,UAAA;AAC7B,IAAA,IAAA,CAAK,YAAY,MAAA,CAAO,SAAA;AACxB,IAAA,IAAA,CAAK,UAAA,GAAa,gBAAA,CAAiB,IAAA,CAAK,KAAK,CAAA;AAAA,EAC/C;AAAA,EAEA,MAAc,QAAQ,KAAA,EAA+C;AACnE,IAAA,MAAM,IAAA,GAAgC;AAAA,MACpC,OAAO,IAAA,CAAK,KAAA;AAAA,MACZ;AAAA,KACF;AACA,IAAA,IAAI,KAAK,SAAA,EAAW;AAClB,MAAA,IAAA,CAAK,YAAY,IAAI,IAAA,CAAK,SAAA;AAAA,IAC5B;AAEA,IAAA,MAAM,GAAA,GAAM,MAAM,KAAA,CAAM,OAAA,EAAS;AAAA,MAC/B,MAAA,EAAQ,MAAA;AAAA,MACR,OAAA,EAAS;AAAA,QACP,cAAA,EAAgB,kBAAA;AAAA,QAChB,aAAA,EAAe,CAAA,OAAA,EAAU,IAAA,CAAK,MAAM,CAAA;AAAA,OACtC;AAAA,MACA,IAAA,EAAM,IAAA,CAAK,SAAA,CAAU,IAAI;AAAA,KAC1B,CAAA;AAED,IAAA,IAAI,CAAC,IAAI,EAAA,EAAI;AACX,MAAA,MAAM,IAAA,GAAO,MAAM,GAAA,CAAI,IAAA,GAAO,KAAA,CAAM,MAAM,IAAI,UAAU,CAAA;AACxD,MAAA,MAAM,IAAI,KAAA,CAAM,CAAA,oBAAA,EAAuB,IAAI,MAAM,CAAA,EAAA,EAAK,IAAI,CAAA,CAAE,CAAA;AAAA,IAC9D;AAEA,IAAA,MAAM,IAAA,GAAQ,MAAM,GAAA,CAAI,IAAA,EAAK;AAC7B,IAAA,OAAO,IAAA,CAAK,IAAA,CAAK,IAAA,CAAK,CAAC,GAAG,CAAA,KAAM,CAAA,CAAE,KAAA,GAAQ,CAAA,CAAE,KAAK,CAAA,CAAE,GAAA,CAAI,CAAC,CAAA,KAAM,EAAE,SAAS,CAAA;AAAA,EAC3E;AAAA,EAEA,MAAM,MAAM,IAAA,EAAiC;AAC3C,IAAA,MAAM,CAAC,SAAS,CAAA,GAAI,MAAM,IAAA,CAAK,QAAQ,IAAI,CAAA;AAC3C,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA,MAAM,IAAI,MAAM,kDAAkD,CAAA;AAAA,IACpE;AACA,IAAA,OAAO,SAAA;AAAA,EACT;AAAA,EAEA,MAAM,WAAW,KAAA,EAAsC;AACrD,IAAA,IAAI,KAAA,CAAM,WAAW,CAAA,EAAG;AAAC,MAAA,OAAO,EAAC;AAAA,IAAE;AACnC,IAAA,OAAO,IAAA,CAAK,QAAQ,KAAK,CAAA;AAAA,EAC3B;AAAA,EAEA,MAAM,aAAA,GAAiC;AACrC,IAAA,OAAO,IAAA,CAAK,UAAA;AAAA,EACd;AACF;AAEO,SAAS,cAAc,MAAA,EAAuD;AACnF,EAAA,OAAO,IAAI,mBAAmB,MAAM,CAAA;AACtC;AAIA,IAAO,kBAAA,GAAQ","file":"embeddings.js","sourcesContent":["import type { AdapterManifest } from \"@kb-labs/sdk/adapters\";\n\nexport const manifest: AdapterManifest = {\n  manifestVersion: \"1.0.0\",\n  id: \"voyage-ai-embeddings\",\n  name: \"Voyage AI Embeddings\",\n  version: \"0.1.0\",\n  description:\n    \"Voyage AI text embeddings adapter (voyage-3, voyage-3.5, voyage-code-3, etc.)\",\n  author: \"KB Labs\",\n  license: \"MIT\",\n  type: \"core\",\n  implements: \"IEmbeddings\",\n  capabilities: {\n    batch: true,\n  },\n  configSchema: {\n    apiKey: {\n      type: \"string\",\n      description: \"Voyage AI API key (defaults to VOYAGE_API_KEY env var)\",\n    },\n    model: {\n      type: \"string\",\n      default: \"voyage-3\",\n      description: \"Embedding model to use\",\n      enum: [\n        \"voyage-3\",\n        \"voyage-3-lite\",\n        \"voyage-3.5\",\n        \"voyage-code-3\",\n        \"voyage-finance-2\",\n        \"voyage-law-2\",\n        \"voyage-multilingual-2\",\n      ],\n    },\n    inputType: {\n      type: \"string\",\n      description: \"Input type hint for retrieval tasks\",\n      enum: [\"query\", \"document\"],\n    },\n  },\n};\n","/**\n * @module @kb-labs/adapters-voyage-ai/embeddings\n * Voyage AI implementation of IEmbeddings interface.\n */\n\nimport type { IEmbeddings } from \"@kb-labs/sdk/adapters\";\n\nexport const EMBEDDING_MODELS = {\n  \"voyage-3\": 1024,\n  \"voyage-3-lite\": 512,\n  \"voyage-3.5\": 2048,\n  \"voyage-code-3\": 1024,\n  \"voyage-finance-2\": 1024,\n  \"voyage-law-2\": 1024,\n  \"voyage-multilingual-2\": 1024,\n} as const;\n\nexport type EmbeddingModel = keyof typeof EMBEDDING_MODELS;\n\nexport type InputType = \"query\" | \"document\";\n\nexport interface VoyageAIEmbeddingsConfig {\n  /** Voyage AI API key (defaults to VOYAGE_API_KEY env var) */\n  apiKey?: string;\n  /** Embedding model to use */\n  model?: EmbeddingModel;\n  /** Input type hint for retrieval tasks */\n  inputType?: InputType;\n}\n\nconst API_URL = \"https://api.voyageai.com/v1/embeddings\";\n\ninterface VoyageResponse {\n  data: Array<{ embedding: number[]; index: number }>;\n}\n\nexport class VoyageAIEmbeddings implements IEmbeddings {\n  private apiKey: string;\n  private model: EmbeddingModel;\n  private inputType: InputType | undefined;\n  readonly dimensions: number;\n\n  constructor(config: VoyageAIEmbeddingsConfig = {}) {\n    const key = config.apiKey ?? process.env[\"VOYAGE_API_KEY\"];\n    if (!key) {\n      throw new Error(\n        \"Voyage AI API key is required. Set VOYAGE_API_KEY env var or pass apiKey in config.\",\n      );\n    }\n    this.apiKey = key;\n    this.model = config.model ?? \"voyage-3\";\n    this.inputType = config.inputType;\n    this.dimensions = EMBEDDING_MODELS[this.model];\n  }\n\n  private async request(input: string | string[]): Promise<number[][]> {\n    const body: Record<string, unknown> = {\n      model: this.model,\n      input,\n    };\n    if (this.inputType) {\n      body[\"input_type\"] = this.inputType;\n    }\n\n    const res = await fetch(API_URL, {\n      method: \"POST\",\n      headers: {\n        \"Content-Type\": \"application/json\",\n        Authorization: `Bearer ${this.apiKey}`,\n      },\n      body: JSON.stringify(body),\n    });\n\n    if (!res.ok) {\n      const text = await res.text().catch(() => res.statusText);\n      throw new Error(`Voyage AI API error ${res.status}: ${text}`);\n    }\n\n    const json = (await res.json()) as VoyageResponse;\n    return json.data.sort((a, b) => a.index - b.index).map((d) => d.embedding);\n  }\n\n  async embed(text: string): Promise<number[]> {\n    const [embedding] = await this.request(text);\n    if (!embedding) {\n      throw new Error(\"Voyage AI embeddings API returned empty response\");\n    }\n    return embedding;\n  }\n\n  async embedBatch(texts: string[]): Promise<number[][]> {\n    if (texts.length === 0) {return [];}\n    return this.request(texts);\n  }\n\n  async getDimensions(): Promise<number> {\n    return this.dimensions;\n  }\n}\n\nexport function createAdapter(config?: VoyageAIEmbeddingsConfig): VoyageAIEmbeddings {\n  return new VoyageAIEmbeddings(config);\n}\n\nexport { manifest } from \"./embeddings-manifest.js\";\n\nexport default createAdapter;\n"]}

package/package.json

@@ -0,0 +1,53 @@
+{
+  "name": "@kb-labs/adapters-voyage-ai",
+  "description": "Voyage AI adapter implementing IEmbeddings interface",
+  "version": "2.96.0",
+  "type": "module",
+  "main": "./dist/embeddings.js",
+  "types": "./dist/embeddings.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/embeddings.d.ts",
+      "import": "./dist/embeddings.js"
+    },
+    "./embeddings": {
+      "types": "./dist/embeddings.d.ts",
+      "import": "./dist/embeddings.js"
+    }
+  },
+  "files": [
+    "dist",
+    "README.md"
+  ],
+  "devDependencies": {
+    "@types/node": "^24.3.3",
+    "eslint": "^9",
+    "rimraf": "^6.0.1",
+    "tsup": "^8.5.0",
+    "typescript": "^5.6.3",
+    "vitest": "^3.2.6",
+    "@kb-labs/devkit": "2.96.0",
+    "@kb-labs/sdk": "2.21.2"
+  },
+  "engines": {
+    "node": ">=20.0.0",
+    "pnpm": ">=9.0.0"
+  },
+  "kb": {
+    "manifest": "./dist/embeddings-manifest.js"
+  },
+  "peerDependencies": {
+    "@kb-labs/sdk": "^2.21.2"
+  },
+  "sideEffects": false,
+  "scripts": {
+    "build": "tsup",
+    "clean": "rimraf dist",
+    "dev": "tsup --watch",
+    "lint": "eslint src --ext .ts",
+    "lint:fix": "eslint . --fix",
+    "test": "vitest run --passWithNoTests",
+    "test:watch": "vitest",
+    "type-check": "tsc --noEmit"
+  }
+}