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

test(ai): isolated Bedrock bearer credentials

Browse Source
This commit is contained in:
can1357
2026-08-11 14:55:47 +02:00
parent cd7a86423f
commit 42357889b0
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/ai/test/bedrock-caller-headers.test.ts
+7 -3
View File
@@ -18,13 +18,17 @@ import { buildModel } from "@oh-my-pi/pi-catalog/build";
* until `afterAll` runs, which is the full-suite hazard `AGENTS.md` rules out.
*/
async function withSkippedAuth<T>(body: () => Promise<T>): Promise<T> {
const original = process.env.AWS_BEDROCK_SKIP_AUTH;
const originalSkipAuth = process.env.AWS_BEDROCK_SKIP_AUTH;
const originalBearerToken = process.env.AWS_BEARER_TOKEN_BEDROCK;
process.env.AWS_BEDROCK_SKIP_AUTH = "1";
delete process.env.AWS_BEARER_TOKEN_BEDROCK;
try {
return await body();
} finally {
if (original === undefined) delete process.env.AWS_BEDROCK_SKIP_AUTH;
else process.env.AWS_BEDROCK_SKIP_AUTH = original;
if (originalSkipAuth === undefined) delete process.env.AWS_BEDROCK_SKIP_AUTH;
else process.env.AWS_BEDROCK_SKIP_AUTH = originalSkipAuth;
if (originalBearerToken === undefined) delete process.env.AWS_BEARER_TOKEN_BEDROCK;
else process.env.AWS_BEARER_TOKEN_BEDROCK = originalBearerToken;
}
}