1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
/**
* @license
* Copyright 2025 Google LLC
* SPDX-License-Identifier: Apache-2.0
*/
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
import { getCoreSystemPrompt } from './prompts.js'; // Adjust import path
import * as process from 'node:process';
// Mock tool names if they are dynamically generated or complex
vi.mock('../tools/ls', () => ({ LSTool: { Name: 'list_directory' } }));
vi.mock('../tools/edit', () => ({ EditTool: { Name: 'replace' } }));
vi.mock('../tools/glob', () => ({ GlobTool: { Name: 'glob' } }));
vi.mock('../tools/grep', () => ({ GrepTool: { Name: 'search_file_content' } }));
vi.mock('../tools/read-file', () => ({ ReadFileTool: { Name: 'read_file' } }));
vi.mock('../tools/read-many-files', () => ({
ReadManyFilesTool: { Name: 'read_many_files' },
}));
vi.mock('../tools/shell', () => ({
ShellTool: { Name: 'execute_bash_command' },
}));
vi.mock('../tools/write-file', () => ({
WriteFileTool: { Name: 'write_file' },
}));
describe('Core System Prompt (prompts.ts)', () => {
// Store original env vars that we modify
let originalSandboxEnv: string | undefined;
beforeEach(() => {
// Store original value before each test
originalSandboxEnv = process.env.SANDBOX;
});
afterEach(() => {
// Restore original value after each test
if (originalSandboxEnv === undefined) {
delete process.env.SANDBOX;
} else {
process.env.SANDBOX = originalSandboxEnv;
}
});
it('should return the base prompt when no userMemory is provided', () => {
delete process.env.SANDBOX; // Ensure default state for snapshot
const prompt = getCoreSystemPrompt();
expect(prompt).not.toContain('---\n\n'); // Separator should not be present
expect(prompt).toContain('You are an interactive CLI agent'); // Check for core content
expect(prompt).toMatchSnapshot(); // Use snapshot for base prompt structure
});
it('should return the base prompt when userMemory is empty string', () => {
delete process.env.SANDBOX;
const prompt = getCoreSystemPrompt('');
expect(prompt).not.toContain('---\n\n');
expect(prompt).toContain('You are an interactive CLI agent');
expect(prompt).toMatchSnapshot();
});
it('should return the base prompt when userMemory is whitespace only', () => {
delete process.env.SANDBOX;
const prompt = getCoreSystemPrompt(' \n \t ');
expect(prompt).not.toContain('---\n\n');
expect(prompt).toContain('You are an interactive CLI agent');
expect(prompt).toMatchSnapshot();
});
it('should append userMemory with separator when provided', () => {
delete process.env.SANDBOX;
const memory = 'This is custom user memory.\nBe extra polite.';
const expectedSuffix = `\n\n---\n\n${memory}`;
const prompt = getCoreSystemPrompt(memory);
expect(prompt.endsWith(expectedSuffix)).toBe(true);
expect(prompt).toContain('You are an interactive CLI agent'); // Ensure base prompt follows
expect(prompt).toMatchSnapshot(); // Snapshot the combined prompt
});
it('should include sandbox-specific instructions when SANDBOX env var is set', () => {
process.env.SANDBOX = 'true'; // Generic sandbox value
const prompt = getCoreSystemPrompt();
expect(prompt).toContain('# Sandbox');
expect(prompt).not.toContain('# MacOS Seatbelt');
expect(prompt).not.toContain('# Outside of Sandbox');
expect(prompt).toMatchSnapshot();
});
it('should include seatbelt-specific instructions when SANDBOX env var is "sandbox-exec"', () => {
process.env.SANDBOX = 'sandbox-exec';
const prompt = getCoreSystemPrompt();
expect(prompt).toContain('# MacOS Seatbelt');
expect(prompt).not.toContain('# Sandbox');
expect(prompt).not.toContain('# Outside of Sandbox');
expect(prompt).toMatchSnapshot();
});
it('should include non-sandbox instructions when SANDBOX env var is not set', () => {
delete process.env.SANDBOX; // Ensure it's not set
const prompt = getCoreSystemPrompt();
expect(prompt).toContain('# Outside of Sandbox');
expect(prompt).not.toContain('# Sandbox');
expect(prompt).not.toContain('# MacOS Seatbelt');
expect(prompt).toMatchSnapshot();
});
});
|
