import type { LanguageModelV3 } from '../../language-model/v3/language-model-v3'; import type { LanguageModelV3CallOptions } from '../../language-model/v3/language-model-v3-call-options'; import type { LanguageModelV3GenerateResult } from '../../language-model/v3/language-model-v3-generate-result'; import type { LanguageModelV3StreamResult } from '../../language-model/v3/language-model-v3-stream-result'; /** * Experimental middleware for LanguageModelV3. * This type defines the structure for middleware that can be used to modify * the behavior of LanguageModelV3 operations. */ export type LanguageModelV3Middleware = { /** * Middleware specification version. Use `v3` for the current version. */ readonly specificationVersion: 'v3'; /** * Override the provider name if desired. * @param options.model - The language model instance. */ overrideProvider?: (options: { model: LanguageModelV3 }) => string; /** * Override the model ID if desired. * @param options.model - The language model instance. */ overrideModelId?: (options: { model: LanguageModelV3 }) => string; /** * Override the supported URLs if desired. * @param options.model - The language model instance. */ overrideSupportedUrls?: (options: { model: LanguageModelV3; }) => PromiseLike> | Record; /** * Transforms the parameters before they are passed to the language model. * @param options - Object containing the type of operation and the parameters. * @param options.type - The type of operation ('generate' or 'stream'). * @param options.params - The original parameters for the language model call. * @returns A promise that resolves to the transformed parameters. */ transformParams?: (options: { type: 'generate' | 'stream'; params: LanguageModelV3CallOptions; model: LanguageModelV3; }) => PromiseLike; /** * Wraps the generate operation of the language model. * @param options - Object containing the generate function, parameters, and model. * @param options.doGenerate - The original generate function. * @param options.doStream - The original stream function. * @param options.params - The parameters for the generate call. If the * `transformParams` middleware is used, this will be the transformed parameters. * @param options.model - The language model instance. * @returns A promise that resolves to the result of the generate operation. */ wrapGenerate?: (options: { doGenerate: () => PromiseLike; doStream: () => PromiseLike; params: LanguageModelV3CallOptions; model: LanguageModelV3; }) => PromiseLike; /** * Wraps the stream operation of the language model. * * @param options - Object containing the stream function, parameters, and model. * @param options.doGenerate - The original generate function. * @param options.doStream - The original stream function. * @param options.params - The parameters for the stream call. If the * `transformParams` middleware is used, this will be the transformed parameters. * @param options.model - The language model instance. * @returns A promise that resolves to the result of the stream operation. */ wrapStream?: (options: { doGenerate: () => PromiseLike; doStream: () => PromiseLike; params: LanguageModelV3CallOptions; model: LanguageModelV3; }) => PromiseLike; };