@core-ai/core-ai 0.8.0 → 0.10.1

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 = {
@@ -366,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 & {
@@ -378,4 +432,4 @@ type GenerateImageParams = ImageGenerateOptions & {
 };
 declare function generateImage(params: GenerateImageParams): Promise<ImageGenerateResult>;
 
-export { AbortedError, type AssistantContentPart, type AssistantMessage, type AssistantTextPart, type BaseGenerateOptions, type ChatInputTokenDetails, type ChatModel, type ChatOutputTokenDetails, type ChatStream, type ChatUsage, CoreAIError, 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, 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, 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

@@ -155,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
@@ -537,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];
@@ -581,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.8.0",
+    "version": "0.10.1",
     "description": "Type-safe LLM abstraction layer over native provider SDKs",
     "license": "MIT",
     "author": "Omnifact (https://omnifact.ai)",