@eidentic/model 0.1.1 → 0.2.0

package/dist/index.cjs

@@ -241,19 +241,22 @@ var AIModel = class {
 // src/embedder.ts
 var import_ai3 = require("ai");
 var AIEmbedder = class _AIEmbedder {
-  constructor(model, dim) {
+  constructor(model, dim, maxRetries) {
     this.model = model;
+    this.maxRetries = maxRetries;
     this.dim = dim;
   }
   model;
+  maxRetries;
   dim;
   /** Construct an embedder, probing the model once to discover its output dimension. */
-  static async create(model) {
-    const { embedding } = await (0, import_ai3.embed)({ model, value: "x" });
-    return new _AIEmbedder(model, embedding.length);
+  static async create(model, opts) {
+    const maxRetries = opts?.maxRetries ?? 2;
+    const { embedding } = await (0, import_ai3.embed)({ model, value: "x", maxRetries });
+    return new _AIEmbedder(model, embedding.length, maxRetries);
   }
   async embed(text) {
-    const { embedding } = await (0, import_ai3.embed)({ model: this.model, value: text });
+    const { embedding } = await (0, import_ai3.embed)({ model: this.model, value: text, maxRetries: this.maxRetries });
     return embedding;
   }
   /**
@@ -263,7 +266,7 @@ var AIEmbedder = class _AIEmbedder {
    */
   async embedBatch(texts) {
     if (texts.length === 0) return [];
-    const { embeddings } = await (0, import_ai3.embedMany)({ model: this.model, values: texts });
+    const { embeddings } = await (0, import_ai3.embedMany)({ model: this.model, values: texts, maxRetries: this.maxRetries });
     for (let i = 0; i < embeddings.length; i++) {
       if (embeddings[i].length !== this.dim) {
         throw new Error(`embedBatch: embedding[${i}] has length ${embeddings[i].length}, expected ${this.dim}`);

package/dist/index.d.cts

@@ -37,6 +37,22 @@ declare class AIModel implements ModelPort {
 
 /** The embedding-model type accepted by AI SDK v6 `embed`, minus the bare-string branch. */
 type AIEmbeddingModel = Exclude<Parameters<typeof embed>[0]["model"], string>;
+/**
+ * Provider-agnostic hosted embedder over AI SDK v6. Bring your own provider + key + model:
+ *   const embedder = await AIEmbedder.create(openai.embedding("text-embedding-3-small"));
+ * Works with any `@ai-sdk/*` embedding model (OpenAI, Cohere, Google, Mistral, ...).
+ * A first-class peer to the local `@eidentic/transformers` embedder; pick whichever fits.
+ */
+/** Options for {@link AIEmbedder.create}. */
+interface AIEmbedderOptions {
+    /**
+     * Max retry attempts per embedding call. The AI SDK retries transient
+     * failures (including provider rate limits / 429s) with exponential backoff
+     * and honours `retry-after`. Raise this for high-volume ingest against a
+     * rate-limited provider. Default: 2 (the AI SDK default).
+     */
+    maxRetries?: number;
+}
 /**
  * Provider-agnostic hosted embedder over AI SDK v6. Bring your own provider + key + model:
  *   const embedder = await AIEmbedder.create(openai.embedding("text-embedding-3-small"));
@@ -45,10 +61,11 @@ type AIEmbeddingModel = Exclude<Parameters<typeof embed>[0]["model"], string>;
  */
 declare class AIEmbedder implements EmbeddingPort {
     private readonly model;
+    private readonly maxRetries;
     readonly dim: number;
     private constructor();
     /** Construct an embedder, probing the model once to discover its output dimension. */
-    static create(model: AIEmbeddingModel): Promise<AIEmbedder>;
+    static create(model: AIEmbeddingModel, opts?: AIEmbedderOptions): Promise<AIEmbedder>;
     embed(text: string): Promise<number[]>;
     /**
      * Batch embedding via AI SDK v6 `embedMany({ model, values })` → `{ embeddings: number[][] }`.

package/dist/index.d.ts

@@ -37,6 +37,22 @@ declare class AIModel implements ModelPort {
 
 /** The embedding-model type accepted by AI SDK v6 `embed`, minus the bare-string branch. */
 type AIEmbeddingModel = Exclude<Parameters<typeof embed>[0]["model"], string>;
+/**
+ * Provider-agnostic hosted embedder over AI SDK v6. Bring your own provider + key + model:
+ *   const embedder = await AIEmbedder.create(openai.embedding("text-embedding-3-small"));
+ * Works with any `@ai-sdk/*` embedding model (OpenAI, Cohere, Google, Mistral, ...).
+ * A first-class peer to the local `@eidentic/transformers` embedder; pick whichever fits.
+ */
+/** Options for {@link AIEmbedder.create}. */
+interface AIEmbedderOptions {
+    /**
+     * Max retry attempts per embedding call. The AI SDK retries transient
+     * failures (including provider rate limits / 429s) with exponential backoff
+     * and honours `retry-after`. Raise this for high-volume ingest against a
+     * rate-limited provider. Default: 2 (the AI SDK default).
+     */
+    maxRetries?: number;
+}
 /**
  * Provider-agnostic hosted embedder over AI SDK v6. Bring your own provider + key + model:
  *   const embedder = await AIEmbedder.create(openai.embedding("text-embedding-3-small"));
@@ -45,10 +61,11 @@ type AIEmbeddingModel = Exclude<Parameters<typeof embed>[0]["model"], string>;
  */
 declare class AIEmbedder implements EmbeddingPort {
     private readonly model;
+    private readonly maxRetries;
     readonly dim: number;
     private constructor();
     /** Construct an embedder, probing the model once to discover its output dimension. */
-    static create(model: AIEmbeddingModel): Promise<AIEmbedder>;
+    static create(model: AIEmbeddingModel, opts?: AIEmbedderOptions): Promise<AIEmbedder>;
     embed(text: string): Promise<number[]>;
     /**
      * Batch embedding via AI SDK v6 `embedMany({ model, values })` → `{ embeddings: number[][] }`.

package/dist/index.js

@@ -211,19 +211,22 @@ var AIModel = class {
 // src/embedder.ts
 import { embed, embedMany } from "ai";
 var AIEmbedder = class _AIEmbedder {
-  constructor(model, dim) {
+  constructor(model, dim, maxRetries) {
     this.model = model;
+    this.maxRetries = maxRetries;
     this.dim = dim;
   }
   model;
+  maxRetries;
   dim;
   /** Construct an embedder, probing the model once to discover its output dimension. */
-  static async create(model) {
-    const { embedding } = await embed({ model, value: "x" });
-    return new _AIEmbedder(model, embedding.length);
+  static async create(model, opts) {
+    const maxRetries = opts?.maxRetries ?? 2;
+    const { embedding } = await embed({ model, value: "x", maxRetries });
+    return new _AIEmbedder(model, embedding.length, maxRetries);
   }
   async embed(text) {
-    const { embedding } = await embed({ model: this.model, value: text });
+    const { embedding } = await embed({ model: this.model, value: text, maxRetries: this.maxRetries });
     return embedding;
   }
   /**
@@ -233,7 +236,7 @@ var AIEmbedder = class _AIEmbedder {
    */
   async embedBatch(texts) {
     if (texts.length === 0) return [];
-    const { embeddings } = await embedMany({ model: this.model, values: texts });
+    const { embeddings } = await embedMany({ model: this.model, values: texts, maxRetries: this.maxRetries });
     for (let i = 0; i < embeddings.length; i++) {
       if (embeddings[i].length !== this.dim) {
         throw new Error(`embedBatch: embedding[${i}] has length ${embeddings[i].length}, expected ${this.dim}`);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@eidentic/model",
-  "version": "0.1.1",
+  "version": "0.2.0",
   "type": "module",
   "license": "Apache-2.0",
   "publishConfig": {
@@ -29,7 +29,7 @@
   ],
   "dependencies": {
     "ai": "^6.0.0",
-    "@eidentic/types": "0.1.1"
+    "@eidentic/types": "0.2.0"
   },
   "peerDependencies": {
     "ollama-ai-provider": ">=1.0.0"