Skip to content

Commit 9a572eb

Browse files
committed
skip memory for OpenAI/Gemini E2E harness tests to reduce deploy time
1 parent 7335079 commit 9a572eb

3 files changed

Lines changed: 7 additions & 2 deletions

File tree

e2e-tests/harness-e2e-helper.ts

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ const baseCanRun = prereqs.npm && prereqs.git && hasAws;
1818
interface HarnessE2EConfig {
1919
modelProvider: 'bedrock' | 'open_ai' | 'gemini';
2020
requiredEnvVar?: string;
21+
skipMemory?: boolean;
2122
}
2223

2324
export function createHarnessE2ESuite(cfg: HarnessE2EConfig) {
@@ -57,6 +58,10 @@ export function createHarnessE2ESuite(cfg: HarnessE2EConfig) {
5758
createArgs.push('--api-key-arn', process.env[cfg.requiredEnvVar]!);
5859
}
5960

61+
if (cfg.skipMemory) {
62+
createArgs.push('--no-harness-memory');
63+
}
64+
6065
const result = await runAgentCoreCLI(createArgs, testDir);
6166

6267
expect(result.exitCode, `Create failed: ${result.stderr}`).toBe(0);

e2e-tests/harness-gemini.test.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
11
import { createHarnessE2ESuite } from './harness-e2e-helper.js';
22

3-
createHarnessE2ESuite({ modelProvider: 'gemini', requiredEnvVar: 'GEMINI_API_KEY_ARN' });
3+
createHarnessE2ESuite({ modelProvider: 'gemini', requiredEnvVar: 'GEMINI_API_KEY_ARN', skipMemory: true });

e2e-tests/harness-openai.test.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
11
import { createHarnessE2ESuite } from './harness-e2e-helper.js';
22

3-
createHarnessE2ESuite({ modelProvider: 'open_ai', requiredEnvVar: 'OPENAI_API_KEY_ARN' });
3+
createHarnessE2ESuite({ modelProvider: 'open_ai', requiredEnvVar: 'OPENAI_API_KEY_ARN', skipMemory: true });

0 commit comments

Comments
 (0)