// Covers extra-params stream wrapper composition across provider families.

import type { StreamFn } from "openclaw/plugin-sdk/agent-core";

import type { Context, Model, SimpleStreamOptions } from "openclaw/plugin-sdk/llm";

import { createAssistantMessageEventStream } from "openclaw/plugin-sdk/llm";

baseStreamFn: StreamFn | undefined,

fastMode: boolean,

): StreamFn {

// xAI fast mode swaps known model ids before the request reaches the base

// stream function, mirroring the provider wrapper without network calls.

return (model, context, options) => {

if (!fastMode || model.api !== "openai-completions" || model.provider !== "xai") {

return (

}

function createTestXaiPayloadCompatibilityWrapper(baseStreamFn: StreamFn | undefined): StreamFn {

// xAI rejects OpenAI-specific reasoning and strict tool-schema fields, so the

// wrapper removes those from the outgoing payload.

return (model, context, options) => {

const underlying =

baseStreamFn ??

}

function createAnthropicBetaHeadersWrapper(baseStreamFn: StreamFn | undefined, betas: string[]) {

// Anthropic beta headers combine OAuth-required betas, default tool/thinking

// betas, and user-configured betas while stripping the managed 1M marker.

const underlying = baseStreamFn ?? (() => ({}) as ReturnType<StreamFn>);

return ((model, context, options) => {

const configuredBetas = betas.filter((beta) => beta !== ANTHROPIC_CONTEXT_1M_BETA);

>[0];

function installFullProviderRuntimeDepsForTest() {

// Install a test-only provider runtime that composes the same wrapper families

// the production provider hook layer owns.

extraParamsTesting.setProviderRuntimeDepsForTest({

prepareProviderExtraParams: (params) => {

if (params.provider !== "openai") {

payload?: Record<string, unknown>;

thinkingLevel?: Parameters<typeof applyExtraParamsToAgent>[5];

}) {

// Mutates a caller-owned payload through onPayload, matching how the runtime

// finalizes provider request bodies.

const payload = params.payload ?? { store: false };

const baseStreamFn: StreamFn = (model, _context, options) => {

options?.onPayload?.(payload, model);

extraParamsOverride?: Record<string, unknown>;

payload?: Record<string, unknown>;

}) {

// This bypasses provider wrappers so the test observes only core

// parallel_tool_calls alias handling.

return withMinimalProviderRuntimeDepsForTest(() => {

const payload = params.payload ?? {};

const baseStreamFn: StreamFn = (model, _context, options) => {