test: update bedrock extra params fixture

This commit is contained in:
Peter Steinberger
2026-05-10 12:31:42 +01:00
parent 17d4450f1c
commit e26d5e60cc

View File

@@ -274,11 +274,8 @@ function createAnthropicFastModeWrapper(baseStreamFn: StreamFn | undefined, fast
return createAnthropicServiceTierWrapper(baseStreamFn, fastMode ? "auto" : "standard_only");
}
import { isAnthropicBedrockModel } from "./pi-embedded-runner/anthropic-family-cache-semantics.js";
import { createAnthropicToolPayloadCompatibilityWrapper } from "./pi-embedded-runner/anthropic-family-tool-payload-compat.js";
import {
createBedrockNoCacheWrapper,
isAnthropicBedrockModel,
} from "./pi-embedded-runner/bedrock-stream-wrappers.js";
import {
applyExtraParamsToAgent,
resolveAgentTransportOverride,
@@ -338,7 +335,7 @@ function installFullProviderRuntimeDepsForTest() {
if (params.provider === "amazon-bedrock") {
return isAnthropicBedrockModel(params.context.modelId)
? params.context.streamFn
: createBedrockNoCacheWrapper(params.context.streamFn);
: createTestBedrockNoCacheWrapper(params.context.streamFn);
}
if (params.provider === "google") {
return createGoogleThinkingPayloadWrapper(
@@ -396,6 +393,15 @@ function installFullProviderRuntimeDepsForTest() {
});
}
function createTestBedrockNoCacheWrapper(baseStreamFn: StreamFn | undefined): StreamFn {
const underlying = baseStreamFn ?? (() => ({}) as ReturnType<StreamFn>);
return (model, context, options) =>
underlying(model, context, {
...options,
cacheRetention: "none",
});
}
function withMinimalProviderRuntimeDepsForTest<T>(run: () => T): T {
extraParamsTesting.setProviderRuntimeDepsForTest({
prepareProviderExtraParams: () => undefined,