AI SDK Kotlin/ai.torad.aisdk/LanguageModelMiddleware

LanguageModelMiddleware

interface LanguageModelMiddleware(source)

Since

0.3.0-beta01

Functions

overrideModelId

open fun overrideModelId(model: LanguageModel): String?

Override the model id the wrapper presents (null = keep the wrapped model's).

overrideProvider

open fun overrideProvider(model: LanguageModel): String?

Override the provider name the wrapper presents (null = keep the wrapped model's).

overrideSupportedUrls

open fun overrideSupportedUrls(model: LanguageModel): Map<String, List<String>>?

Override the supported-URL patterns the wrapper presents (null = keep the wrapped model's).

transformParams

open suspend fun transformParams(operation: MiddlewareOperation, params: LanguageModelCallParams, model: LanguageModel): LanguageModelCallParams

Transform the call params before wrapGenerate/wrapStream run, for the given operation. The transformed params flow into both the wrap hooks and the downstream call. This is the params-only seam — a middleware can implement just this (e.g. ai.torad.aisdk.middleware.defaultSettingsMiddleware). Default: pass through. Mirrors v6's transformParams({ type, params, model }).

wrapGenerate

open suspend fun wrapGenerate(context: MiddlewareCallContext): LanguageModelResult

Wrap the one-shot generate call. Default: pass through.

wrapStream

open fun wrapStream(context: MiddlewareCallContext): Flow<StreamEvent>

Wrap the streaming call. Default: pass through.