@apertis/ai-sdk-provider 1.0.0 → 1.1.0

package/dist/index.d.cts

@@ -1,4 +1,4 @@
-import { LanguageModelV3 } from '@ai-sdk/provider';
+import { ProviderV3, LanguageModelV3, EmbeddingModelV3 } from '@ai-sdk/provider';
 
 type ApertisModelId = "gpt-5.2" | "gpt-5.2-codex" | "gpt-5.1" | "claude-opus-4-5-20251101" | "claude-sonnet-4.5" | "claude-haiku-4.5" | "gemini-3-pro-preview" | "gemini-3-flash-preview" | "gemini-2.5-flash-preview" | (string & {});
 interface ApertisProviderSettings {
@@ -36,9 +36,51 @@ interface ApertisChatSettings {
     topLogprobs?: number;
 }
 
-interface ApertisProvider {
+interface ApertisCompletionSettings {
+    /**
+     * Echo back the prompt in addition to the completion.
+     */
+    echo?: boolean;
+    /**
+     * Include the log probabilities on the logprobs most likely tokens.
+     */
+    logprobs?: number;
+    /**
+     * The suffix that comes after a completion of inserted text.
+     */
+    suffix?: string;
+    /**
+     * A unique identifier representing your end-user.
+     */
+    user?: string;
+}
+type ApertisCompletionModelId = "gpt-3.5-turbo-instruct" | "davinci-002" | "babbage-002" | (string & {});
+
+interface ApertisEmbeddingSettings {
+    /**
+     * Override the maximum number of embeddings per call.
+     */
+    maxEmbeddingsPerCall?: number;
+    /**
+     * Override the parallelism of embedding calls.
+     */
+    supportsParallelCalls?: boolean;
+    /**
+     * The number of dimensions the resulting output embeddings should have.
+     * Only supported in text-embedding-3 and later models.
+     */
+    dimensions?: number;
+    /**
+     * A unique identifier representing your end-user.
+     */
+    user?: string;
+}
+type ApertisEmbeddingModelId = "text-embedding-3-small" | "text-embedding-3-large" | "text-embedding-ada-002" | (string & {});
+
+interface ApertisProvider extends ProviderV3 {
     /**
      * Creates a chat model for text generation.
+     * Default call creates a chat model.
      */
     (modelId: ApertisModelId, settings?: ApertisChatSettings): LanguageModelV3;
     /**
@@ -46,9 +88,28 @@ interface ApertisProvider {
      */
     chat(modelId: ApertisModelId, settings?: ApertisChatSettings): LanguageModelV3;
     /**
-     * Creates a chat model for text generation (alias for languageModel).
+     * Creates a language model (alias for chat).
+     * Required by ProviderV3 interface.
+     */
+    languageModel(modelId: string): LanguageModelV3;
+    /**
+     * Creates a completion model for text completions.
+     */
+    completion(modelId: ApertisCompletionModelId, settings?: ApertisCompletionSettings): LanguageModelV3;
+    /**
+     * Creates an embedding model.
+     * Required by ProviderV3 interface.
+     */
+    embeddingModel(modelId: string): EmbeddingModelV3;
+    /**
+     * Creates a text embedding model.
+     */
+    textEmbeddingModel(modelId: ApertisEmbeddingModelId, settings?: ApertisEmbeddingSettings): EmbeddingModelV3;
+    /**
+     * Image models are not supported by Apertis.
+     * Required by ProviderV3 interface.
      */
-    languageModel(modelId: ApertisModelId, settings?: ApertisChatSettings): LanguageModelV3;
+    imageModel(modelId: string): never;
 }
 declare function createApertis(options?: ApertisProviderSettings): ApertisProvider;
 /**
@@ -56,4 +117,4 @@ declare function createApertis(options?: ApertisProviderSettings): ApertisProvid
  */
 declare const apertis: ApertisProvider;
 
-export { type ApertisChatSettings, type ApertisModelId, type ApertisProvider, type ApertisProviderSettings, apertis, createApertis };
+export { type ApertisChatSettings, type ApertisCompletionModelId, type ApertisCompletionSettings, type ApertisEmbeddingModelId, type ApertisEmbeddingSettings, type ApertisModelId, type ApertisProvider, type ApertisProviderSettings, apertis, createApertis };

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { LanguageModelV3 } from '@ai-sdk/provider';
+import { ProviderV3, LanguageModelV3, EmbeddingModelV3 } from '@ai-sdk/provider';
 
 type ApertisModelId = "gpt-5.2" | "gpt-5.2-codex" | "gpt-5.1" | "claude-opus-4-5-20251101" | "claude-sonnet-4.5" | "claude-haiku-4.5" | "gemini-3-pro-preview" | "gemini-3-flash-preview" | "gemini-2.5-flash-preview" | (string & {});
 interface ApertisProviderSettings {
@@ -36,9 +36,51 @@ interface ApertisChatSettings {
     topLogprobs?: number;
 }
 
-interface ApertisProvider {
+interface ApertisCompletionSettings {
+    /**
+     * Echo back the prompt in addition to the completion.
+     */
+    echo?: boolean;
+    /**
+     * Include the log probabilities on the logprobs most likely tokens.
+     */
+    logprobs?: number;
+    /**
+     * The suffix that comes after a completion of inserted text.
+     */
+    suffix?: string;
+    /**
+     * A unique identifier representing your end-user.
+     */
+    user?: string;
+}
+type ApertisCompletionModelId = "gpt-3.5-turbo-instruct" | "davinci-002" | "babbage-002" | (string & {});
+
+interface ApertisEmbeddingSettings {
+    /**
+     * Override the maximum number of embeddings per call.
+     */
+    maxEmbeddingsPerCall?: number;
+    /**
+     * Override the parallelism of embedding calls.
+     */
+    supportsParallelCalls?: boolean;
+    /**
+     * The number of dimensions the resulting output embeddings should have.
+     * Only supported in text-embedding-3 and later models.
+     */
+    dimensions?: number;
+    /**
+     * A unique identifier representing your end-user.
+     */
+    user?: string;
+}
+type ApertisEmbeddingModelId = "text-embedding-3-small" | "text-embedding-3-large" | "text-embedding-ada-002" | (string & {});
+
+interface ApertisProvider extends ProviderV3 {
     /**
      * Creates a chat model for text generation.
+     * Default call creates a chat model.
      */
     (modelId: ApertisModelId, settings?: ApertisChatSettings): LanguageModelV3;
     /**
@@ -46,9 +88,28 @@ interface ApertisProvider {
      */
     chat(modelId: ApertisModelId, settings?: ApertisChatSettings): LanguageModelV3;
     /**
-     * Creates a chat model for text generation (alias for languageModel).
+     * Creates a language model (alias for chat).
+     * Required by ProviderV3 interface.
+     */
+    languageModel(modelId: string): LanguageModelV3;
+    /**
+     * Creates a completion model for text completions.
+     */
+    completion(modelId: ApertisCompletionModelId, settings?: ApertisCompletionSettings): LanguageModelV3;
+    /**
+     * Creates an embedding model.
+     * Required by ProviderV3 interface.
+     */
+    embeddingModel(modelId: string): EmbeddingModelV3;
+    /**
+     * Creates a text embedding model.
+     */
+    textEmbeddingModel(modelId: ApertisEmbeddingModelId, settings?: ApertisEmbeddingSettings): EmbeddingModelV3;
+    /**
+     * Image models are not supported by Apertis.
+     * Required by ProviderV3 interface.
      */
-    languageModel(modelId: ApertisModelId, settings?: ApertisChatSettings): LanguageModelV3;
+    imageModel(modelId: string): never;
 }
 declare function createApertis(options?: ApertisProviderSettings): ApertisProvider;
 /**
@@ -56,4 +117,4 @@ declare function createApertis(options?: ApertisProviderSettings): ApertisProvid
  */
 declare const apertis: ApertisProvider;
 
-export { type ApertisChatSettings, type ApertisModelId, type ApertisProvider, type ApertisProviderSettings, apertis, createApertis };
+export { type ApertisChatSettings, type ApertisCompletionModelId, type ApertisCompletionSettings, type ApertisEmbeddingModelId, type ApertisEmbeddingSettings, type ApertisModelId, type ApertisProvider, type ApertisProviderSettings, apertis, createApertis };

package/dist/index.js

@@ -458,6 +458,327 @@ var ApertisChatLanguageModel = class {
   }
 };
 
+// src/apertis-completion-language-model.ts
+import {
+  createEventSourceResponseHandler as createEventSourceResponseHandler2,
+  createJsonResponseHandler as createJsonResponseHandler2,
+  generateId as generateId2,
+  postJsonToApi as postJsonToApi2
+} from "@ai-sdk/provider-utils";
+
+// src/schemas/completion-response.ts
+import { z as z3 } from "zod";
+var openAICompletionResponseSchema = z3.object({
+  id: z3.string(),
+  object: z3.literal("text_completion"),
+  created: z3.number(),
+  model: z3.string(),
+  choices: z3.array(
+    z3.object({
+      text: z3.string(),
+      index: z3.number(),
+      logprobs: z3.object({
+        tokens: z3.array(z3.string()).optional(),
+        token_logprobs: z3.array(z3.number()).optional(),
+        top_logprobs: z3.array(z3.record(z3.number())).optional(),
+        text_offset: z3.array(z3.number()).optional()
+      }).nullable().optional(),
+      finish_reason: z3.string().nullable().optional()
+    })
+  ),
+  usage: z3.object({
+    prompt_tokens: z3.number(),
+    completion_tokens: z3.number(),
+    total_tokens: z3.number()
+  }).optional()
+});
+var openAICompletionChunkSchema = z3.object({
+  id: z3.string(),
+  object: z3.literal("text_completion"),
+  created: z3.number(),
+  model: z3.string(),
+  choices: z3.array(
+    z3.object({
+      text: z3.string(),
+      index: z3.number(),
+      logprobs: z3.object({
+        tokens: z3.array(z3.string()).optional(),
+        token_logprobs: z3.array(z3.number()).optional(),
+        top_logprobs: z3.array(z3.record(z3.number())).optional(),
+        text_offset: z3.array(z3.number()).optional()
+      }).nullable().optional(),
+      finish_reason: z3.string().nullable().optional()
+    })
+  ),
+  usage: z3.object({
+    prompt_tokens: z3.number(),
+    completion_tokens: z3.number(),
+    total_tokens: z3.number()
+  }).optional().nullable()
+});
+
+// src/apertis-completion-language-model.ts
+var ApertisCompletionLanguageModel = class {
+  constructor(modelId, settings, config) {
+    this.modelId = modelId;
+    this.settings = settings;
+    this.config = config;
+  }
+  specificationVersion = "v3";
+  supportedUrls = {};
+  get provider() {
+    return this.config.provider;
+  }
+  async doGenerate(options) {
+    const body = this.buildRequestBody(options, false);
+    const { value: response } = await postJsonToApi2({
+      url: `${this.config.baseURL}/completions`,
+      headers: this.config.headers(),
+      body,
+      failedResponseHandler: apertisFailedResponseHandler,
+      successfulResponseHandler: createJsonResponseHandler2(
+        openAICompletionResponseSchema
+      ),
+      fetch: this.config.fetch,
+      abortSignal: options.abortSignal
+    });
+    const choice = response.choices[0];
+    const content = [];
+    if (choice.text) {
+      content.push({
+        type: "text",
+        text: choice.text
+      });
+    }
+    return {
+      content,
+      finishReason: this.mapFinishReason(choice.finish_reason),
+      usage: {
+        inputTokens: {
+          total: response.usage?.prompt_tokens ?? 0,
+          noCache: void 0,
+          cacheRead: void 0,
+          cacheWrite: void 0
+        },
+        outputTokens: {
+          total: response.usage?.completion_tokens ?? 0,
+          text: void 0,
+          reasoning: void 0
+        }
+      },
+      warnings: [],
+      request: { body }
+    };
+  }
+  async doStream(options) {
+    const body = this.buildRequestBody(options, true);
+    const { value: response } = await postJsonToApi2({
+      url: `${this.config.baseURL}/completions`,
+      headers: this.config.headers(),
+      body,
+      failedResponseHandler: apertisFailedResponseHandler,
+      successfulResponseHandler: createEventSourceResponseHandler2(
+        openAICompletionChunkSchema
+      ),
+      fetch: this.config.fetch,
+      abortSignal: options.abortSignal
+    });
+    let textId = null;
+    const transformStream = new TransformStream({
+      transform(parseResult, controller) {
+        if (!parseResult.success) {
+          return;
+        }
+        const chunk = parseResult.value;
+        const choice = chunk.choices[0];
+        if (!choice) return;
+        if (choice.text) {
+          if (!textId) {
+            textId = generateId2();
+            controller.enqueue({
+              type: "text-start",
+              id: textId
+            });
+          }
+          controller.enqueue({
+            type: "text-delta",
+            id: textId,
+            delta: choice.text
+          });
+        }
+        if (choice.finish_reason) {
+          if (textId) {
+            controller.enqueue({
+              type: "text-end",
+              id: textId
+            });
+          }
+          controller.enqueue({
+            type: "finish",
+            finishReason: {
+              unified: choice.finish_reason === "stop" ? "stop" : choice.finish_reason === "length" ? "length" : "other",
+              raw: choice.finish_reason ?? void 0
+            },
+            usage: {
+              inputTokens: {
+                total: chunk.usage?.prompt_tokens ?? 0,
+                noCache: void 0,
+                cacheRead: void 0,
+                cacheWrite: void 0
+              },
+              outputTokens: {
+                total: chunk.usage?.completion_tokens ?? 0,
+                text: void 0,
+                reasoning: void 0
+              }
+            }
+          });
+        }
+      },
+      flush(controller) {
+        if (textId) {
+          controller.enqueue({
+            type: "text-end",
+            id: textId
+          });
+        }
+      }
+    });
+    return {
+      stream: response.pipeThrough(transformStream),
+      request: { body }
+    };
+  }
+  buildRequestBody(options, stream) {
+    const prompt = this.convertPromptToText(options.prompt);
+    const body = {
+      model: this.modelId,
+      prompt,
+      stream
+    };
+    if (stream) body.stream_options = { include_usage: true };
+    if (options.maxOutputTokens !== void 0)
+      body.max_tokens = options.maxOutputTokens;
+    if (options.temperature !== void 0)
+      body.temperature = options.temperature;
+    if (options.topP !== void 0) body.top_p = options.topP;
+    if (options.frequencyPenalty !== void 0)
+      body.frequency_penalty = options.frequencyPenalty;
+    if (options.presencePenalty !== void 0)
+      body.presence_penalty = options.presencePenalty;
+    if (options.stopSequences !== void 0) body.stop = options.stopSequences;
+    if (options.seed !== void 0) body.seed = options.seed;
+    if (this.settings.echo !== void 0) body.echo = this.settings.echo;
+    if (this.settings.logprobs !== void 0)
+      body.logprobs = this.settings.logprobs;
+    if (this.settings.suffix !== void 0) body.suffix = this.settings.suffix;
+    if (this.settings.user !== void 0) body.user = this.settings.user;
+    return body;
+  }
+  convertPromptToText(prompt) {
+    const parts = [];
+    for (const message of prompt) {
+      if (message.role === "system") {
+        parts.push(message.content);
+      } else if (message.role === "user") {
+        for (const part of message.content) {
+          if (part.type === "text") {
+            parts.push(part.text);
+          }
+        }
+      } else if (message.role === "assistant") {
+        for (const part of message.content) {
+          if (part.type === "text") {
+            parts.push(part.text);
+          }
+        }
+      }
+    }
+    return parts.join("\n\n");
+  }
+  mapFinishReason(finishReason) {
+    const raw = finishReason ?? void 0;
+    switch (finishReason) {
+      case "stop":
+        return { unified: "stop", raw };
+      case "length":
+        return { unified: "length", raw };
+      default:
+        return { unified: "other", raw };
+    }
+  }
+};
+
+// src/apertis-embedding-model.ts
+import {
+  createJsonResponseHandler as createJsonResponseHandler3,
+  postJsonToApi as postJsonToApi3
+} from "@ai-sdk/provider-utils";
+
+// src/schemas/embedding-response.ts
+import { z as z4 } from "zod";
+var openAIEmbeddingResponseSchema = z4.object({
+  object: z4.literal("list"),
+  data: z4.array(
+    z4.object({
+      object: z4.literal("embedding"),
+      embedding: z4.array(z4.number()),
+      index: z4.number()
+    })
+  ),
+  model: z4.string(),
+  usage: z4.object({
+    prompt_tokens: z4.number(),
+    total_tokens: z4.number()
+  }).optional()
+});
+
+// src/apertis-embedding-model.ts
+var ApertisEmbeddingModel = class {
+  constructor(modelId, settings, config) {
+    this.modelId = modelId;
+    this.settings = settings;
+    this.config = config;
+    this.maxEmbeddingsPerCall = settings.maxEmbeddingsPerCall ?? 2048;
+    this.supportsParallelCalls = settings.supportsParallelCalls ?? true;
+  }
+  specificationVersion = "v3";
+  maxEmbeddingsPerCall;
+  supportsParallelCalls;
+  get provider() {
+    return this.config.provider;
+  }
+  async doEmbed(options) {
+    const body = {
+      model: this.modelId,
+      input: options.values,
+      encoding_format: "float"
+    };
+    if (this.settings.dimensions !== void 0) {
+      body.dimensions = this.settings.dimensions;
+    }
+    if (this.settings.user !== void 0) {
+      body.user = this.settings.user;
+    }
+    const { value: response } = await postJsonToApi3({
+      url: `${this.config.baseURL}/embeddings`,
+      headers: this.config.headers(),
+      body,
+      failedResponseHandler: apertisFailedResponseHandler,
+      successfulResponseHandler: createJsonResponseHandler3(
+        openAIEmbeddingResponseSchema
+      ),
+      fetch: this.config.fetch,
+      abortSignal: options.abortSignal
+    });
+    return {
+      embeddings: response.data.map((item) => item.embedding),
+      usage: response.usage ? { tokens: response.usage.prompt_tokens } : void 0,
+      warnings: []
+    };
+  }
+};
+
 // src/apertis-provider.ts
 function createApertis(options = {}) {
   const baseURL = withoutTrailingSlash(options.baseURL) ?? "https://api.apertis.ai/v1";
@@ -476,11 +797,30 @@ function createApertis(options = {}) {
     headers: getHeaders,
     fetch: options.fetch
   });
+  const createCompletionModel = (modelId, settings = {}) => new ApertisCompletionLanguageModel(modelId, settings, {
+    provider: "apertis.completion",
+    baseURL,
+    headers: getHeaders,
+    fetch: options.fetch
+  });
+  const createEmbeddingModel = (modelId, settings = {}) => new ApertisEmbeddingModel(modelId, settings, {
+    provider: "apertis.embedding",
+    baseURL,
+    headers: getHeaders,
+    fetch: options.fetch
+  });
   const provider = Object.assign(
     (modelId, settings) => createChatModel(modelId, settings),
     {
+      specificationVersion: "v3",
       chat: createChatModel,
-      languageModel: createChatModel
+      languageModel: (modelId) => createChatModel(modelId),
+      completion: createCompletionModel,
+      embeddingModel: (modelId) => createEmbeddingModel(modelId),
+      textEmbeddingModel: createEmbeddingModel,
+      imageModel: () => {
+        throw new Error("Image models are not supported by Apertis");
+      }
     }
   );
   return provider;