smashimo/oh-my-pi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test(ai): scope Bedrock auth bypass to stream calls

Browse Source
This commit is contained in:
can1357
2026-07-31 19:00:39 +02:00
parent 9eddf297fd
commit 7818cac342
3 changed files with 6 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/ai/test/bedrock-inference-profile.test.ts
+2 -16
View File
@@ -1,25 +1,10 @@
import { afterAll, beforeAll, describe, expect, test } from "bun:test";
import { describe, expect, test } from "bun:test";
import { streamBedrock } from "@oh-my-pi/pi-ai/providers/amazon-bedrock";
import type { Context, FetchImpl, Model } from "@oh-my-pi/pi-ai/types";
import { buildModel } from "@oh-my-pi/pi-catalog/build";
import { Effort } from "@oh-my-pi/pi-catalog/effort";
import { withEnv } from "./helpers";
// These suites capture the request payload from a fire-and-forget stream, so a
// credential lookup that rejects after the test ends surfaces as an unhandled
// error against whatever runs next. Skip Bedrock auth: the payload is built
// before signing and no request leaves the process.
const originalSkipAuth = process.env.AWS_BEDROCK_SKIP_AUTH;
beforeAll(() => {
process.env.AWS_BEDROCK_SKIP_AUTH = "1";
});
afterAll(() => {
if (originalSkipAuth === undefined) delete process.env.AWS_BEDROCK_SKIP_AUTH;
else process.env.AWS_BEDROCK_SKIP_AUTH = originalSkipAuth;
});
const profileArn = "arn:aws:bedrock:us-east-2:1234567890:application-inference-profile/company-opus-48";
const profileModel: Model<"bedrock-converse-stream"> = buildModel({
id: profileArn,
@@ -100,6 +85,7 @@ describe("Bedrock inference profile ARNs", () => {
const { promise, resolve } = Promise.withResolvers<unknown>();
void streamBedrock(profileModel, context, {
bearerToken: "test-token",
signal: controller.signal,
reasoning: Effort.High,
maxTokens: 16,
packages/ai/test/bedrock-prompt-cache.test.ts
+2 -16
View File
@@ -1,25 +1,10 @@
import { afterAll, beforeAll, describe, expect, test } from "bun:test";
import { describe, expect, test } from "bun:test";
import { streamBedrock } from "@oh-my-pi/pi-ai/providers/amazon-bedrock";
import type { Context, Model } from "@oh-my-pi/pi-ai/types";
import { buildModel } from "@oh-my-pi/pi-catalog/build";
import { getBundledModel } from "@oh-my-pi/pi-catalog/models";
import { withEnv } from "./helpers";
// These suites capture the request payload from a fire-and-forget stream, so a
// credential lookup that rejects after the test ends surfaces as an unhandled
// error against whatever runs next. Skip Bedrock auth: the payload is built
// before signing and no request leaves the process.
const originalSkipAuth = process.env.AWS_BEDROCK_SKIP_AUTH;
beforeAll(() => {
process.env.AWS_BEDROCK_SKIP_AUTH = "1";
});
afterAll(() => {
if (originalSkipAuth === undefined) delete process.env.AWS_BEDROCK_SKIP_AUTH;
else process.env.AWS_BEDROCK_SKIP_AUTH = originalSkipAuth;
});
interface CachePoint {
cachePoint: { type: "default"; ttl?: "1h" };
}
@@ -63,6 +48,7 @@ function capturePayload(
): Promise<Payload> {
const { promise, resolve } = Promise.withResolvers<Payload>();
void streamBedrock(bedrockModel, context, {
bearerToken: "test-token",
signal: abortedSignal(),
cacheRetention,
onPayload: payload => {
packages/ai/test/bedrock-system-prompt.test.ts
+2 -16
View File
@@ -1,23 +1,8 @@
import { afterAll, beforeAll, describe, expect, test } from "bun:test";
import { describe, expect, test } from "bun:test";
import { streamBedrock } from "@oh-my-pi/pi-ai/providers/amazon-bedrock";
import type { Context, Model } from "@oh-my-pi/pi-ai/types";
import { buildModel } from "@oh-my-pi/pi-catalog/build";
// These suites capture the request payload from a fire-and-forget stream, so a
// credential lookup that rejects after the test ends surfaces as an unhandled
// error against whatever runs next. Skip Bedrock auth: the payload is built
// before signing and no request leaves the process.
const originalSkipAuth = process.env.AWS_BEDROCK_SKIP_AUTH;
beforeAll(() => {
process.env.AWS_BEDROCK_SKIP_AUTH = "1";
});
afterAll(() => {
if (originalSkipAuth === undefined) delete process.env.AWS_BEDROCK_SKIP_AUTH;
else process.env.AWS_BEDROCK_SKIP_AUTH = originalSkipAuth;
});
interface Payload {
system?: Array<{ text: string } | { cachePoint: unknown }>;
}
@@ -52,6 +37,7 @@ async function capturePayload(systemPrompt: Context["systemPrompt"]): Promise<Pa
};
const { promise, resolve } = Promise.withResolvers<Payload | undefined>();
const stream = streamBedrock(model(), context, {
bearerToken: "test-token",
signal: abortedSignal(),
onPayload: payload => {
resolve(payload as Payload);