@core-ai/core-ai 0.7.1 → 0.10.0

package/dist/index.d.ts

@@ -102,12 +102,49 @@ interface EmbedProviderOptions {
 interface ImageProviderOptions {
     [key: string]: Record<string, unknown> | undefined;
 }
+type ChatModelMiddleware = {
+    generate?: (args: {
+        execute: (options?: GenerateOptions) => Promise<GenerateResult>;
+        options: GenerateOptions;
+        model: ChatModel;
+    }) => Promise<GenerateResult>;
+    stream?: (args: {
+        execute: (options?: GenerateOptions) => Promise<ChatStream>;
+        options: GenerateOptions;
+        model: ChatModel;
+    }) => Promise<ChatStream>;
+    generateObject?: <TSchema extends z.ZodType>(args: {
+        execute: (options?: GenerateObjectOptions<TSchema>) => Promise<GenerateObjectResult<TSchema>>;
+        options: GenerateObjectOptions<TSchema>;
+        model: ChatModel;
+    }) => Promise<GenerateObjectResult<TSchema>>;
+    streamObject?: <TSchema extends z.ZodType>(args: {
+        execute: (options?: StreamObjectOptions<TSchema>) => Promise<ObjectStream<TSchema>>;
+        options: StreamObjectOptions<TSchema>;
+        model: ChatModel;
+    }) => Promise<ObjectStream<TSchema>>;
+};
+type EmbeddingModelMiddleware = {
+    embed?: (args: {
+        execute: (options?: EmbedOptions) => Promise<EmbedResult>;
+        options: EmbedOptions;
+        model: EmbeddingModel;
+    }) => Promise<EmbedResult>;
+};
+type ImageModelMiddleware = {
+    generate?: (args: {
+        execute: (options?: ImageGenerateOptions) => Promise<ImageGenerateResult>;
+        options: ImageGenerateOptions;
+        model: ImageModel;
+    }) => Promise<ImageGenerateResult>;
+};
 type BaseGenerateOptions = {
     messages: Message[];
     temperature?: number;
     maxTokens?: number;
     topP?: number;
     reasoning?: ReasoningConfig;
+    metadata?: Record<string, unknown>;
     providerOptions?: GenerateProviderOptions;
     signal?: AbortSignal;
 };
@@ -247,6 +284,7 @@ type EmbeddingModel = {
 type EmbedOptions = {
     input: string | string[];
     dimensions?: number;
+    metadata?: Record<string, unknown>;
     providerOptions?: EmbedProviderOptions;
 };
 type EmbedResult = {
@@ -270,6 +308,7 @@ type ImageGenerateOptions = {
     prompt: string;
     n?: number;
     size?: string;
+    metadata?: Record<string, unknown>;
     providerOptions?: ImageProviderOptions;
 };
 type ImageGenerateResult = {
@@ -281,15 +320,21 @@ type GeneratedImage = {
     revisedPrompt?: string;
 };
 
-declare class LLMError extends Error {
+declare class CoreAIError extends Error {
     readonly cause?: unknown;
-    constructor(message: string, cause?: unknown);
+    readonly provider?: string;
+    constructor(message: string, cause?: unknown, provider?: string);
 }
-declare class StreamAbortedError extends LLMError {
-    constructor(message?: string, cause?: unknown);
+declare class ValidationError extends CoreAIError {
+    constructor(message: string, cause?: unknown, provider?: string);
 }
-declare class ProviderError extends LLMError {
-    readonly provider: string;
+declare class AbortedError extends CoreAIError {
+    constructor(cause?: unknown, provider?: string);
+}
+declare class StreamAbortedError extends AbortedError {
+    constructor(cause?: unknown, provider?: string);
+}
+declare class ProviderError extends CoreAIError {
     readonly statusCode?: number;
     constructor(message: string, provider: string, statusCode?: number, cause?: unknown);
 }
@@ -298,7 +343,8 @@ type StructuredOutputErrorOptions = {
     cause?: unknown;
     rawOutput?: string;
 };
-declare class StructuredOutputError extends ProviderError {
+declare class StructuredOutputError extends CoreAIError {
+    readonly statusCode?: number;
     readonly rawOutput?: string;
     constructor(message: string, provider: string, options?: StructuredOutputErrorOptions);
 }
@@ -359,6 +405,21 @@ declare function createChatStream(source: AsyncIterable<StreamEvent> | (() => Pr
     signal?: AbortSignal;
 }): ChatStream;
 
+declare function wrapChatModel(config: {
+    model: ChatModel;
+    middleware: ChatModelMiddleware | ChatModelMiddleware[];
+}): ChatModel;
+
+declare function wrapEmbeddingModel(config: {
+    model: EmbeddingModel;
+    middleware: EmbeddingModelMiddleware | EmbeddingModelMiddleware[];
+}): EmbeddingModel;
+
+declare function wrapImageModel(config: {
+    model: ImageModel;
+    middleware: ImageModelMiddleware | ImageModelMiddleware[];
+}): ImageModel;
+
 declare function getProviderMetadata<T extends Record<string, unknown>>(providerMetadata: Record<string, Record<string, unknown>> | undefined, provider: string): T | undefined;
 
 type EmbedParams = EmbedOptions & {
@@ -371,4 +432,4 @@ type GenerateImageParams = ImageGenerateOptions & {
 };
 declare function generateImage(params: GenerateImageParams): Promise<ImageGenerateResult>;
 
-export { type AssistantContentPart, type AssistantMessage, type AssistantTextPart, type BaseGenerateOptions, type ChatInputTokenDetails, type ChatModel, type ChatOutputTokenDetails, type ChatStream, type ChatUsage, type EmbedOptions, type EmbedProviderOptions, type EmbedResult, type EmbeddingModel, type EmbeddingUsage, type FilePart, type FinishReason, type GenerateObjectOptions, type GenerateObjectResult, type GenerateOptions, type GenerateProviderOptions, type GenerateResult, type GeneratedImage, type ImageGenerateOptions, type ImageGenerateResult, type ImageModel, type ImagePart, type ImageProviderOptions, LLMError, type Message, type ObjectStream, type ObjectStreamEvent, ProviderError, type ReasoningConfig, type ReasoningEffort, type ReasoningPart, StreamAbortedError, type StreamEvent, type StreamObjectOptions, StructuredOutputError, StructuredOutputNoObjectGeneratedError, StructuredOutputParseError, StructuredOutputValidationError, type SystemMessage, type TextPart, type ToolCall, type ToolCallPart, type ToolChoice, type ToolDefinition, type ToolResultMessage, type ToolSet, type UserContentPart, type UserMessage, asObject, assistantMessage, createChatStream, createObjectStream, defineTool, embed, generate, generateImage, generateObject, getProviderMetadata, resultToMessage, safeParseJsonObject, stream, streamObject, stripModelDateSuffix, zodSchemaToJsonSchema };
+export { AbortedError, type AssistantContentPart, type AssistantMessage, type AssistantTextPart, type BaseGenerateOptions, type ChatInputTokenDetails, type ChatModel, type ChatModelMiddleware, type ChatOutputTokenDetails, type ChatStream, type ChatUsage, CoreAIError, type EmbedOptions, type EmbedProviderOptions, type EmbedResult, type EmbeddingModel, type EmbeddingModelMiddleware, type EmbeddingUsage, type FilePart, type FinishReason, type GenerateObjectOptions, type GenerateObjectResult, type GenerateOptions, type GenerateProviderOptions, type GenerateResult, type GeneratedImage, type ImageGenerateOptions, type ImageGenerateResult, type ImageModel, type ImageModelMiddleware, type ImagePart, type ImageProviderOptions, type Message, type ObjectStream, type ObjectStreamEvent, ProviderError, type ReasoningConfig, type ReasoningEffort, type ReasoningPart, StreamAbortedError, type StreamEvent, type StreamObjectOptions, StructuredOutputError, StructuredOutputNoObjectGeneratedError, StructuredOutputParseError, StructuredOutputValidationError, type SystemMessage, type TextPart, type ToolCall, type ToolCallPart, type ToolChoice, type ToolDefinition, type ToolResultMessage, type ToolSet, type UserContentPart, type UserMessage, ValidationError, asObject, assistantMessage, createChatStream, createObjectStream, defineTool, embed, generate, generateImage, generateObject, getProviderMetadata, resultToMessage, safeParseJsonObject, stream, streamObject, stripModelDateSuffix, wrapChatModel, wrapEmbeddingModel, wrapImageModel, zodSchemaToJsonSchema };

package/dist/index.js

@@ -1,33 +1,48 @@
 // src/errors.ts
-var LLMError = class extends Error {
+var CoreAIError = class extends Error {
   cause;
-  constructor(message, cause) {
+  provider;
+  constructor(message, cause, provider) {
     super(message);
-    this.name = "LLMError";
+    this.name = "CoreAIError";
     this.cause = cause;
+    this.provider = provider;
+  }
+};
+var ValidationError = class extends CoreAIError {
+  constructor(message, cause, provider) {
+    super(message, cause, provider);
+    this.name = "ValidationError";
+  }
+};
+var AbortedError = class extends CoreAIError {
+  constructor(cause, provider) {
+    super("operation aborted", cause, provider);
+    this.name = "AbortedError";
   }
 };
-var StreamAbortedError = class extends LLMError {
-  constructor(message = "stream aborted", cause) {
-    super(message, cause);
+var StreamAbortedError = class extends AbortedError {
+  constructor(cause, provider) {
+    super(cause, provider);
     this.name = "StreamAbortedError";
+    this.message = "stream aborted";
   }
 };
-var ProviderError = class extends LLMError {
-  provider;
+var ProviderError = class extends CoreAIError {
   statusCode;
   constructor(message, provider, statusCode, cause) {
-    super(message, cause);
+    super(message, cause, provider);
     this.name = "ProviderError";
-    this.provider = provider;
     this.statusCode = statusCode;
   }
 };
-var StructuredOutputError = class extends ProviderError {
+var StructuredOutputError = class extends CoreAIError {
+  statusCode;
   rawOutput;
   constructor(message, provider, options = {}) {
-    super(message, provider, options.statusCode, options.cause);
+    super(message, options.cause, provider);
     this.name = "StructuredOutputError";
+    this.statusCode = options.statusCode;
     this.rawOutput = options.rawOutput;
   }
 };
@@ -108,19 +123,19 @@ function isEmptyText(value) {
 }
 function assertNonEmptyMessages(messages) {
   if (messages.length === 0) {
-    throw new LLMError("messages must not be empty");
+    throw new ValidationError("messages must not be empty");
   }
 }
 function assertNonEmptyEmbedInput(input) {
   const isEmptyString = typeof input === "string" && isEmptyText(input);
   const isEmptyArray = Array.isArray(input) && input.length === 0;
   if (isEmptyString || isEmptyArray) {
-    throw new LLMError("input must not be empty");
+    throw new ValidationError("input must not be empty");
   }
 }
 function assertNonEmptyPrompt(prompt) {
   if (isEmptyText(prompt)) {
-    throw new LLMError("prompt must not be empty");
+    throw new ValidationError("prompt must not be empty");
   }
 }
 
@@ -140,10 +155,7 @@ function callModelWithOptions(params, call) {
 // src/generate.ts
 async function generate(params) {
   assertNonEmptyMessages(params.messages);
-  return callModelWithOptions(
-    params,
-    (model, options) => model.generate(options)
-  );
+  return callModelWithOptions(params, (model, options) => model.generate(options));
 }
 
 // src/generate-object.ts
@@ -240,7 +252,7 @@ function createStream(options) {
     if (terminalState.status !== "running") {
       return;
     }
-    settleRejected(new StreamAbortedError("stream aborted"));
+    settleRejected(new StreamAbortedError());
     void closeSourceIterator();
   }
   if (signal) {
@@ -361,7 +373,7 @@ function createObjectStream(source, options = {}) {
     },
     finalizeResult() {
       if (objectState.status !== "ready") {
-        throw new LLMError(
+        throw new CoreAIError(
           "object stream completed without emitting a final object"
         );
       }
@@ -522,6 +534,114 @@ function createChatStream(source, options = {}) {
   });
 }
 
+// src/wrap-model-utils.ts
+function buildMiddlewareChain(config) {
+  const { model, operations, finalExecute } = config;
+  return operations.reduceRight(
+    (nextExecute, operation) => {
+      return (options) => operation({
+        execute: (overrideOptions) => nextExecute(overrideOptions ?? options),
+        options,
+        model
+      });
+    },
+    finalExecute
+  );
+}
+function normalizeMiddleware(middleware) {
+  return Array.isArray(middleware) ? middleware : [middleware];
+}
+
+// src/wrap-chat-model.ts
+function wrapChatModel(config) {
+  const middlewares = normalizeMiddleware(config.middleware);
+  const { model } = config;
+  const generateOps = middlewares.flatMap(
+    (mw) => mw.generate ? [mw.generate] : []
+  );
+  const streamOps = middlewares.flatMap(
+    (mw) => mw.stream ? [mw.stream] : []
+  );
+  return {
+    provider: model.provider,
+    modelId: model.modelId,
+    generate(options) {
+      return buildMiddlewareChain({
+        model,
+        operations: generateOps,
+        finalExecute: (opts) => model.generate(opts)
+      })(options);
+    },
+    stream(options) {
+      return buildMiddlewareChain({
+        model,
+        operations: streamOps,
+        finalExecute: (opts) => model.stream(opts)
+      })(options);
+    },
+    generateObject(options) {
+      const operations = middlewares.flatMap(
+        (mw) => mw.generateObject ? [mw.generateObject] : []
+      );
+      return buildMiddlewareChain({
+        model,
+        operations,
+        finalExecute: (opts) => model.generateObject(opts)
+      })(options);
+    },
+    streamObject(options) {
+      const operations = middlewares.flatMap(
+        (mw) => mw.streamObject ? [mw.streamObject] : []
+      );
+      return buildMiddlewareChain({
+        model,
+        operations,
+        finalExecute: (opts) => model.streamObject(opts)
+      })(options);
+    }
+  };
+}
+
+// src/wrap-embedding-model.ts
+function wrapEmbeddingModel(config) {
+  const middlewares = normalizeMiddleware(config.middleware);
+  const { model } = config;
+  const embedOps = middlewares.flatMap(
+    (mw) => mw.embed ? [mw.embed] : []
+  );
+  return {
+    provider: model.provider,
+    modelId: model.modelId,
+    embed(options) {
+      return buildMiddlewareChain({
+        model,
+        operations: embedOps,
+        finalExecute: (opts) => model.embed(opts)
+      })(options);
+    }
+  };
+}
+
+// src/wrap-image-model.ts
+function wrapImageModel(config) {
+  const middlewares = normalizeMiddleware(config.middleware);
+  const { model } = config;
+  const generateOps = middlewares.flatMap(
+    (mw) => mw.generate ? [mw.generate] : []
+  );
+  return {
+    provider: model.provider,
+    modelId: model.modelId,
+    generate(options) {
+      return buildMiddlewareChain({
+        model,
+        operations: generateOps,
+        finalExecute: (opts) => model.generate(opts)
+      })(options);
+    }
+  };
+}
+
 // src/provider-metadata.ts
 function getProviderMetadata(providerMetadata, provider) {
   return providerMetadata?.[provider];
@@ -542,13 +662,15 @@ async function generateImage(params) {
   );
 }
 export {
-  LLMError,
+  AbortedError,
+  CoreAIError,
   ProviderError,
   StreamAbortedError,
   StructuredOutputError,
   StructuredOutputNoObjectGeneratedError,
   StructuredOutputParseError,
   StructuredOutputValidationError,
+  ValidationError,
   asObject,
   assistantMessage,
   createChatStream,
@@ -564,5 +686,8 @@ export {
   stream,
   streamObject,
   stripModelDateSuffix,
+  wrapChatModel,
+  wrapEmbeddingModel,
+  wrapImageModel,
   zodSchemaToJsonSchema
 };

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@core-ai/core-ai",
-    "version": "0.7.1",
+    "version": "0.10.0",
     "description": "Type-safe LLM abstraction layer over native provider SDKs",
     "license": "MIT",
     "author": "Omnifact (https://omnifact.ai)",