@providerprotocol/ai 0.0.16 → 0.0.18

package/dist/openrouter/index.d.ts

@@ -1,4 +1,29 @@
-import { b as Provider, d as ModelReference, a as LLMHandler } from '../provider-vTZ74u-w.js';
+import { d as Provider, f as ModelReference, b as LLMHandler, c as EmbeddingHandler } from '../provider-D5MO3-pS.js';
+
+/**
+ * @fileoverview OpenRouter Embeddings API Handler
+ *
+ * This module implements the embedding handler for OpenRouter's embeddings API.
+ * OpenRouter provides access to multiple embedding providers through an OpenAI-compatible endpoint.
+ *
+ * @see {@link https://openrouter.ai/docs/api/reference/embeddings OpenRouter Embeddings API Reference}
+ * @module providers/openrouter/embed
+ */
+
+/**
+ * OpenRouter embedding parameters.
+ * Passed through unchanged to the API.
+ */
+interface OpenRouterEmbedParams {
+    /** Output dimensions (model-dependent) */
+    dimensions?: number;
+    /** Encoding format: 'float' or 'base64' */
+    encoding_format?: 'float' | 'base64';
+    /** A unique identifier representing your end-user */
+    user?: string;
+    /** Input type hint for some models */
+    input_type?: string;
+}
 
 /**
  * OpenRouter-specific types for the Unified Provider Protocol.
@@ -289,10 +314,11 @@ interface OpenRouterProvider extends Provider<OpenRouterProviderOptions> {
     readonly version: string;
     /**
      * Supported modalities for this provider.
-     * OpenRouter currently only supports LLM (text generation).
+     * OpenRouter supports LLM (text generation) and Embedding.
      */
     readonly modalities: {
         llm: LLMHandler<OpenRouterLLMParamsUnion>;
+        embedding: EmbeddingHandler<OpenRouterEmbedParams>;
     };
 }
 /**
@@ -346,4 +372,4 @@ interface OpenRouterProvider extends Provider<OpenRouterProviderOptions> {
  */
 declare const openrouter: OpenRouterProvider;
 
-export { type OpenRouterAPIMode, type OpenRouterCompletionsParams, type OpenRouterConfig, type OpenRouterModelOptions, type OpenRouterModelReference, type OpenRouterProviderPreferences, type OpenRouterResponsesParams, openrouter };
+export { type OpenRouterAPIMode, type OpenRouterCompletionsParams, type OpenRouterConfig, type OpenRouterEmbedParams, type OpenRouterModelOptions, type OpenRouterModelReference, type OpenRouterProviderPreferences, type OpenRouterResponsesParams, openrouter };

package/dist/openrouter/index.js

@@ -3,17 +3,19 @@ import {
   isAssistantMessage,
   isToolResultMessage,
   isUserMessage
-} from "../chunk-SVYROCLD.js";
+} from "../chunk-UMKWXGO3.js";
 import {
   parseSSEStream
 } from "../chunk-Z7RBRCRN.js";
 import {
-  UPPError,
   doFetch,
   doStreamFetch,
   normalizeHttpError,
   resolveApiKey
-} from "../chunk-MOU4U3PO.js";
+} from "../chunk-5FEAOEXV.js";
+import {
+  UPPError
+} from "../chunk-DZQHVGNV.js";
 
 // src/providers/openrouter/transform.completions.ts
 function transformRequest(request, modelId) {
@@ -1336,11 +1338,121 @@ function createResponsesLLMHandler() {
   };
 }
 
+// src/providers/openrouter/embed.ts
+var OPENROUTER_EMBEDDINGS_API_URL = "https://openrouter.ai/api/v1/embeddings";
+function getDefaultDimensions(modelId) {
+  if (modelId.includes("text-embedding-3-large")) {
+    return 3072;
+  }
+  if (modelId.includes("text-embedding-3-small") || modelId.includes("ada-002")) {
+    return 1536;
+  }
+  if (modelId.includes("gemini-embedding")) {
+    return 3072;
+  }
+  return 1536;
+}
+function createEmbeddingHandler() {
+  let providerRef = null;
+  return {
+    supportedInputs: ["text"],
+    _setProvider(provider) {
+      providerRef = provider;
+    },
+    bind(modelId) {
+      if (!providerRef) {
+        throw new UPPError(
+          "Provider reference not set. Handler must be used with createProvider().",
+          "INVALID_REQUEST",
+          "openrouter",
+          "embedding"
+        );
+      }
+      const model = {
+        modelId,
+        maxBatchSize: 2048,
+        maxInputLength: 8191,
+        dimensions: getDefaultDimensions(modelId),
+        get provider() {
+          return providerRef;
+        },
+        async embed(request) {
+          const apiKey = await resolveApiKey(
+            request.config,
+            "OPENROUTER_API_KEY",
+            "openrouter",
+            "embedding"
+          );
+          const baseUrl = request.config.baseUrl ?? OPENROUTER_EMBEDDINGS_API_URL;
+          const inputTexts = request.inputs.map((input) => {
+            if (typeof input === "string") {
+              return input;
+            }
+            if ("text" in input) {
+              return input.text;
+            }
+            throw new UPPError(
+              "OpenRouter embeddings only support text input",
+              "INVALID_REQUEST",
+              "openrouter",
+              "embedding"
+            );
+          });
+          const body = {
+            model: modelId,
+            input: inputTexts
+          };
+          if (request.params?.dimensions !== void 0) {
+            body.dimensions = request.params.dimensions;
+          }
+          if (request.params?.encoding_format !== void 0) {
+            body.encoding_format = request.params.encoding_format;
+          }
+          if (request.params?.user !== void 0) {
+            body.user = request.params.user;
+          }
+          if (request.params?.input_type !== void 0) {
+            body.input_type = request.params.input_type;
+          }
+          const headers = {
+            "Content-Type": "application/json",
+            Authorization: `Bearer ${apiKey}`
+          };
+          if (request.config.headers) {
+            for (const [key, value] of Object.entries(request.config.headers)) {
+              if (value !== void 0) {
+                headers[key] = value;
+              }
+            }
+          }
+          const response = await doFetch(baseUrl, {
+            method: "POST",
+            headers,
+            body: JSON.stringify(body),
+            signal: request.signal
+          }, request.config, "openrouter", "embedding");
+          const data = await response.json();
+          return {
+            embeddings: data.data.map((d) => ({
+              vector: d.embedding,
+              index: d.index
+            })),
+            usage: { totalTokens: data.usage.total_tokens },
+            metadata: data.usage.cost !== void 0 ? { cost: data.usage.cost } : void 0
+          };
+        }
+      };
+      return model;
+    }
+  };
+}
+
 // src/providers/openrouter/index.ts
 function createOpenRouterProvider() {
   let currentApiMode = "completions";
   const completionsHandler = createCompletionsLLMHandler();
   const responsesHandler = createResponsesLLMHandler();
+  const embeddingHandler = createEmbeddingHandler();
   const fn = function(modelId, options) {
     const apiMode = options?.api ?? "completions";
     currentApiMode = apiMode;
@@ -1349,7 +1461,8 @@ function createOpenRouterProvider() {
   const modalities = {
     get llm() {
       return currentApiMode === "responses" ? responsesHandler : completionsHandler;
-    }
+    },
+    embedding: embeddingHandler
   };
   Object.defineProperties(fn, {
     name: {
@@ -1371,6 +1484,7 @@ function createOpenRouterProvider() {
   const provider = fn;
   completionsHandler._setProvider?.(provider);
   responsesHandler._setProvider?.(provider);
+  embeddingHandler._setProvider?.(provider);
   return provider;
 }
 var openrouter = createOpenRouterProvider();