promptfoo 0.78.2 → 0.78.3
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/package.json +2 -2
- package/dist/src/assertions.js +3 -3
- package/dist/src/assertions.js.map +1 -1
- package/dist/src/commands/eval.d.ts.map +1 -1
- package/dist/src/commands/eval.js +8 -3
- package/dist/src/commands/eval.js.map +1 -1
- package/dist/src/commands/generate/redteam.d.ts.map +1 -1
- package/dist/src/commands/generate/redteam.js +2 -2
- package/dist/src/commands/generate/redteam.js.map +1 -1
- package/dist/src/envars.d.ts +1 -0
- package/dist/src/envars.d.ts.map +1 -1
- package/dist/src/envars.js.map +1 -1
- package/dist/src/providers/scriptCompletion.d.ts +2 -0
- package/dist/src/providers/scriptCompletion.d.ts.map +1 -1
- package/dist/src/providers/scriptCompletion.js +81 -57
- package/dist/src/providers/scriptCompletion.js.map +1 -1
- package/dist/src/redteam/index.d.ts.map +1 -1
- package/dist/src/redteam/index.js +5 -1
- package/dist/src/redteam/index.js.map +1 -1
- package/dist/src/web/nextui/404/index.html +1 -1
- package/dist/src/web/nextui/404.html +1 -1
- package/dist/src/web/nextui/_next/static/chunks/{258-bb8ca628533ae496.js → 258-4b7fded41a7208ac.js} +1 -1
- package/dist/src/web/nextui/_next/static/chunks/app/eval/[id]/{page-6c4e63c9602f522f.js → page-9bc768bf1e13119e.js} +1 -1
- package/dist/src/web/nextui/_next/static/chunks/app/{layout-bd07f93483b60d27.js → layout-bece9e53b5d250aa.js} +1 -1
- package/dist/src/web/nextui/_next/static/chunks/app/page-b6714efe6c381b92.js +1 -0
- package/dist/src/web/nextui/auth/login/index.html +1 -1
- package/dist/src/web/nextui/auth/login/index.txt +2 -2
- package/dist/src/web/nextui/auth/signup/index.html +1 -1
- package/dist/src/web/nextui/auth/signup/index.txt +2 -2
- package/dist/src/web/nextui/datasets/index.html +1 -1
- package/dist/src/web/nextui/datasets/index.txt +2 -2
- package/dist/src/web/nextui/eval/index.html +1 -1
- package/dist/src/web/nextui/eval/index.txt +3 -3
- package/dist/src/web/nextui/index.html +1 -1
- package/dist/src/web/nextui/index.txt +2 -2
- package/dist/src/web/nextui/progress/index.html +1 -1
- package/dist/src/web/nextui/progress/index.txt +2 -2
- package/dist/src/web/nextui/prompts/index.html +1 -1
- package/dist/src/web/nextui/prompts/index.txt +2 -2
- package/dist/src/web/nextui/report/index.html +1 -1
- package/dist/src/web/nextui/report/index.txt +2 -2
- package/dist/src/web/nextui/setup/index.html +1 -1
- package/dist/src/web/nextui/setup/index.txt +2 -2
- package/dist/test/commands/generate/redteam.test.d.ts +2 -0
- package/dist/test/commands/generate/redteam.test.d.ts.map +1 -0
- package/dist/test/commands/generate/redteam.test.js +228 -0
- package/dist/test/commands/generate/redteam.test.js.map +1 -0
- package/dist/test/providers.scriptCompletion.test.d.ts +2 -0
- package/dist/test/providers.scriptCompletion.test.d.ts.map +1 -0
- package/dist/test/providers.scriptCompletion.test.js +77 -0
- package/dist/test/providers.scriptCompletion.test.js.map +1 -0
- package/package.json +2 -2
- package/dist/src/web/nextui/_next/static/chunks/app/page-81e3b8fee37eaf67.js +0 -1
- /package/dist/src/web/nextui/_next/static/{7WCweG-frl9PmaQvBJk5m → glbZwKe9JIxYb5HK0YQnG}/_buildManifest.js +0 -0
- /package/dist/src/web/nextui/_next/static/{7WCweG-frl9PmaQvBJk5m → glbZwKe9JIxYb5HK0YQnG}/_ssgManifest.js +0 -0
|
@@ -0,0 +1,228 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
3
|
+
if (k2 === undefined) k2 = k;
|
|
4
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
5
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
6
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
7
|
+
}
|
|
8
|
+
Object.defineProperty(o, k2, desc);
|
|
9
|
+
}) : (function(o, m, k, k2) {
|
|
10
|
+
if (k2 === undefined) k2 = k;
|
|
11
|
+
o[k2] = m[k];
|
|
12
|
+
}));
|
|
13
|
+
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
14
|
+
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
15
|
+
}) : function(o, v) {
|
|
16
|
+
o["default"] = v;
|
|
17
|
+
});
|
|
18
|
+
var __importStar = (this && this.__importStar) || function (mod) {
|
|
19
|
+
if (mod && mod.__esModule) return mod;
|
|
20
|
+
var result = {};
|
|
21
|
+
if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
|
|
22
|
+
__setModuleDefault(result, mod);
|
|
23
|
+
return result;
|
|
24
|
+
};
|
|
25
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
26
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
27
|
+
};
|
|
28
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
29
|
+
const fs_1 = __importDefault(require("fs"));
|
|
30
|
+
const cliState_1 = __importDefault(require("../../../src/cliState"));
|
|
31
|
+
const redteam_1 = require("../../../src/commands/generate/redteam");
|
|
32
|
+
const configModule = __importStar(require("../../../src/config"));
|
|
33
|
+
const logger_1 = __importDefault(require("../../../src/logger"));
|
|
34
|
+
const redteam_2 = require("../../../src/redteam");
|
|
35
|
+
const config_1 = require("../../../src/util/config");
|
|
36
|
+
jest.mock('fs');
|
|
37
|
+
jest.mock('yaml');
|
|
38
|
+
jest.mock('../../../src/redteam');
|
|
39
|
+
jest.mock('../../../src/util/config');
|
|
40
|
+
jest.mock('../../../src/logger');
|
|
41
|
+
jest.mock('../../../src/telemetry');
|
|
42
|
+
jest.mock('../../../src/config', () => ({
|
|
43
|
+
readConfigs: jest.fn(),
|
|
44
|
+
resolveConfigs: jest.fn(),
|
|
45
|
+
}));
|
|
46
|
+
describe('doGenerateRedteam', () => {
|
|
47
|
+
beforeEach(() => {
|
|
48
|
+
jest.clearAllMocks();
|
|
49
|
+
jest.mocked(configModule.resolveConfigs).mockResolvedValue({
|
|
50
|
+
basePath: '/mock/path',
|
|
51
|
+
testSuite: {
|
|
52
|
+
prompts: [],
|
|
53
|
+
providers: [],
|
|
54
|
+
},
|
|
55
|
+
config: {
|
|
56
|
+
redteam: {
|
|
57
|
+
plugins: [],
|
|
58
|
+
strategies: [],
|
|
59
|
+
},
|
|
60
|
+
},
|
|
61
|
+
});
|
|
62
|
+
});
|
|
63
|
+
it('should generate redteam tests and write to output file', async () => {
|
|
64
|
+
jest.mocked(configModule.readConfigs).mockResolvedValue([
|
|
65
|
+
{
|
|
66
|
+
prompts: [{ raw: 'Test prompt', label: 'Test label' }],
|
|
67
|
+
providers: [],
|
|
68
|
+
tests: [],
|
|
69
|
+
},
|
|
70
|
+
]);
|
|
71
|
+
const options = {
|
|
72
|
+
output: 'output.yaml',
|
|
73
|
+
config: 'config.yaml',
|
|
74
|
+
cache: true,
|
|
75
|
+
defaultConfig: {},
|
|
76
|
+
write: true,
|
|
77
|
+
};
|
|
78
|
+
jest.mocked(fs_1.default.readFileSync).mockReturnValue(JSON.stringify({
|
|
79
|
+
prompts: [{ raw: 'Test prompt' }],
|
|
80
|
+
providers: [],
|
|
81
|
+
tests: [],
|
|
82
|
+
}));
|
|
83
|
+
jest.mocked(redteam_2.synthesize).mockResolvedValue({
|
|
84
|
+
testCases: [
|
|
85
|
+
{
|
|
86
|
+
vars: { input: 'Test input' },
|
|
87
|
+
assert: [{ type: 'equals', value: 'Test output' }],
|
|
88
|
+
metadata: { pluginId: 'redteam' },
|
|
89
|
+
},
|
|
90
|
+
],
|
|
91
|
+
purpose: 'Test purpose',
|
|
92
|
+
entities: ['Test entity'],
|
|
93
|
+
});
|
|
94
|
+
await (0, redteam_1.doGenerateRedteam)(options);
|
|
95
|
+
expect(redteam_2.synthesize).toHaveBeenCalledWith(expect.objectContaining({
|
|
96
|
+
language: undefined,
|
|
97
|
+
numTests: undefined,
|
|
98
|
+
plugins: expect.any(Array),
|
|
99
|
+
prompts: [],
|
|
100
|
+
strategies: expect.any(Array),
|
|
101
|
+
}));
|
|
102
|
+
expect(config_1.writePromptfooConfig).toHaveBeenCalledWith(expect.objectContaining({
|
|
103
|
+
prompts: [{ raw: 'Test prompt' }],
|
|
104
|
+
providers: [],
|
|
105
|
+
tests: [
|
|
106
|
+
{
|
|
107
|
+
vars: { input: 'Test input' },
|
|
108
|
+
assert: [{ type: 'equals', value: 'Test output' }],
|
|
109
|
+
metadata: { pluginId: 'redteam' },
|
|
110
|
+
},
|
|
111
|
+
],
|
|
112
|
+
redteam: expect.objectContaining({
|
|
113
|
+
purpose: 'Test purpose',
|
|
114
|
+
entities: ['Test entity'],
|
|
115
|
+
}),
|
|
116
|
+
defaultTest: {
|
|
117
|
+
metadata: {
|
|
118
|
+
purpose: 'Test purpose',
|
|
119
|
+
entities: ['Test entity'],
|
|
120
|
+
},
|
|
121
|
+
},
|
|
122
|
+
}), 'output.yaml');
|
|
123
|
+
expect(logger_1.default.info).toHaveBeenCalledWith(expect.stringContaining('Wrote 1 new test cases'));
|
|
124
|
+
});
|
|
125
|
+
it('should write to config file when write option is true', async () => {
|
|
126
|
+
const options = {
|
|
127
|
+
config: 'config.yaml',
|
|
128
|
+
cache: true,
|
|
129
|
+
defaultConfig: {},
|
|
130
|
+
write: true,
|
|
131
|
+
};
|
|
132
|
+
jest.mocked(fs_1.default.readFileSync).mockReturnValue(JSON.stringify({}));
|
|
133
|
+
jest.mocked(redteam_2.synthesize).mockResolvedValue({
|
|
134
|
+
testCases: [],
|
|
135
|
+
purpose: 'Test purpose',
|
|
136
|
+
entities: [],
|
|
137
|
+
});
|
|
138
|
+
await (0, redteam_1.doGenerateRedteam)(options);
|
|
139
|
+
expect(config_1.writePromptfooConfig).toHaveBeenCalledWith(expect.objectContaining({
|
|
140
|
+
tests: [],
|
|
141
|
+
redteam: expect.objectContaining({
|
|
142
|
+
purpose: 'Test purpose',
|
|
143
|
+
entities: [],
|
|
144
|
+
}),
|
|
145
|
+
}), 'config.yaml');
|
|
146
|
+
});
|
|
147
|
+
it('should handle missing configuration file', async () => {
|
|
148
|
+
const options = {
|
|
149
|
+
cache: true,
|
|
150
|
+
defaultConfig: {},
|
|
151
|
+
write: true,
|
|
152
|
+
output: 'redteam.yaml',
|
|
153
|
+
};
|
|
154
|
+
await (0, redteam_1.doGenerateRedteam)(options);
|
|
155
|
+
expect(logger_1.default.info).toHaveBeenCalledWith(expect.stringContaining("Can't generate without configuration"));
|
|
156
|
+
});
|
|
157
|
+
it('should use purpose when no config is provided', async () => {
|
|
158
|
+
const options = {
|
|
159
|
+
purpose: 'Test purpose',
|
|
160
|
+
cache: true,
|
|
161
|
+
defaultConfig: {},
|
|
162
|
+
write: true,
|
|
163
|
+
output: 'redteam.yaml',
|
|
164
|
+
};
|
|
165
|
+
jest.mocked(redteam_2.synthesize).mockResolvedValue({
|
|
166
|
+
testCases: [
|
|
167
|
+
{
|
|
168
|
+
vars: { input: 'Test input' },
|
|
169
|
+
assert: [{ type: 'equals', value: 'Test output' }],
|
|
170
|
+
metadata: { pluginId: 'redteam' },
|
|
171
|
+
},
|
|
172
|
+
],
|
|
173
|
+
purpose: 'Test purpose',
|
|
174
|
+
entities: ['Test entity'],
|
|
175
|
+
});
|
|
176
|
+
await (0, redteam_1.doGenerateRedteam)(options);
|
|
177
|
+
expect(redteam_2.synthesize).toHaveBeenCalledWith({
|
|
178
|
+
language: undefined,
|
|
179
|
+
numTests: undefined,
|
|
180
|
+
purpose: 'Test purpose',
|
|
181
|
+
plugins: expect.any(Array),
|
|
182
|
+
prompts: expect.any(Array),
|
|
183
|
+
strategies: expect.any(Array),
|
|
184
|
+
});
|
|
185
|
+
expect(config_1.writePromptfooConfig).toHaveBeenCalledWith(expect.objectContaining({
|
|
186
|
+
tests: expect.arrayContaining([
|
|
187
|
+
expect.objectContaining({
|
|
188
|
+
vars: { input: 'Test input' },
|
|
189
|
+
assert: [{ type: 'equals', value: 'Test output' }],
|
|
190
|
+
metadata: { pluginId: 'redteam' },
|
|
191
|
+
}),
|
|
192
|
+
]),
|
|
193
|
+
redteam: expect.objectContaining({
|
|
194
|
+
purpose: 'Test purpose',
|
|
195
|
+
entities: ['Test entity'],
|
|
196
|
+
}),
|
|
197
|
+
}), 'redteam.yaml');
|
|
198
|
+
});
|
|
199
|
+
it('should update cliState with basePath', async () => {
|
|
200
|
+
const options = {
|
|
201
|
+
config: 'config.yaml',
|
|
202
|
+
cache: true,
|
|
203
|
+
defaultConfig: {},
|
|
204
|
+
write: true,
|
|
205
|
+
};
|
|
206
|
+
jest.mocked(configModule.resolveConfigs).mockResolvedValue({
|
|
207
|
+
basePath: '/mock/base/path',
|
|
208
|
+
testSuite: {
|
|
209
|
+
prompts: [],
|
|
210
|
+
providers: [],
|
|
211
|
+
},
|
|
212
|
+
config: {
|
|
213
|
+
redteam: {
|
|
214
|
+
plugins: [],
|
|
215
|
+
strategies: [],
|
|
216
|
+
},
|
|
217
|
+
},
|
|
218
|
+
});
|
|
219
|
+
jest.mocked(redteam_2.synthesize).mockResolvedValue({
|
|
220
|
+
testCases: [],
|
|
221
|
+
purpose: 'Test purpose',
|
|
222
|
+
entities: [],
|
|
223
|
+
});
|
|
224
|
+
await (0, redteam_1.doGenerateRedteam)(options);
|
|
225
|
+
expect(cliState_1.default.basePath).toBe('/mock/base/path');
|
|
226
|
+
});
|
|
227
|
+
});
|
|
228
|
+
//# sourceMappingURL=redteam.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"redteam.test.js","sourceRoot":"","sources":["../../../../test/commands/generate/redteam.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,4CAAoB;AACpB,qEAA6C;AAC7C,oEAA2E;AAC3E,kEAAoD;AACpD,iEAAyC;AACzC,kDAAkD;AAElD,qDAAgE;AAEhE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAChB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAClB,IAAI,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;AAClC,IAAI,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;AACtC,IAAI,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AACjC,IAAI,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;AAEpC,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;IACtB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC,iBAAiB,CAAC;YACzD,QAAQ,EAAE,YAAY;YACtB,SAAS,EAAE;gBACT,OAAO,EAAE,EAAE;gBACX,SAAS,EAAE,EAAE;aACd;YACD,MAAM,EAAE;gBACN,OAAO,EAAE;oBACP,OAAO,EAAE,EAAE;oBACX,UAAU,EAAE,EAAE;iBACf;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,iBAAiB,CAAC;YACtD;gBACE,OAAO,EAAE,CAAC,EAAE,GAAG,EAAE,aAAa,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;gBACtD,SAAS,EAAE,EAAE;gBACb,KAAK,EAAE,EAAE;aACV;SACK,CAAC,CAAC;QAEV,MAAM,OAAO,GAA2B;YACtC,MAAM,EAAE,aAAa;YACrB,MAAM,EAAE,aAAa;YACrB,KAAK,EAAE,IAAI;YACX,aAAa,EAAE,EAAE;YACjB,KAAK,EAAE,IAAI;SACZ,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,YAAE,CAAC,YAAY,CAAC,CAAC,eAAe,CAC1C,IAAI,CAAC,SAAS,CAAC;YACb,OAAO,EAAE,CAAC,EAAE,GAAG,EAAE,aAAa,EAAE,CAAC;YACjC,SAAS,EAAE,EAAE;YACb,KAAK,EAAE,EAAE;SACV,CAAC,CACH,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,oBAAU,CAAC,CAAC,iBAAiB,CAAC;YACxC,SAAS,EAAE;gBACT;oBACE,IAAI,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE;oBAC7B,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC;oBAClD,QAAQ,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE;iBAClC;aACF;YACD,OAAO,EAAE,cAAc;YACvB,QAAQ,EAAE,CAAC,aAAa,CAAC;SAC1B,CAAC,CAAC;QAEH,MAAM,IAAA,2BAAiB,EAAC,OAAO,CAAC,CAAC;QAEjC,MAAM,CAAC,oBAAU,CAAC,CAAC,oBAAoB,CACrC,MAAM,CAAC,gBAAgB,CAAC;YACtB,QAAQ,EAAE,SAAS;YACnB,QAAQ,EAAE,SAAS;YACnB,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC;YAC1B,OAAO,EAAE,EAAE;YACX,UAAU,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC;SAC9B,CAAC,CACH,CAAC;QACF,MAAM,CAAC,6BAAoB,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,gBAAgB,CAAC;YACtB,OAAO,EAAE,CAAC,EAAE,GAAG,EAAE,aAAa,EAAE,CAAC;YACjC,SAAS,EAAE,EAAE;YACb,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE;oBAC7B,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC;oBAClD,QAAQ,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE;iBAClC;aACF;YACD,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;gBAC/B,OAAO,EAAE,cAAc;gBACvB,QAAQ,EAAE,CAAC,aAAa,CAAC;aAC1B,CAAC;YACF,WAAW,EAAE;gBACX,QAAQ,EAAE;oBACR,OAAO,EAAE,cAAc;oBACvB,QAAQ,EAAE,CAAC,aAAa,CAAC;iBAC1B;aACF;SACF,CAAC,EACF,aAAa,CACd,CAAC;QACF,MAAM,CAAC,gBAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC,wBAAwB,CAAC,CAAC,CAAC;IAC9F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,OAAO,GAA2B;YACtC,MAAM,EAAE,aAAa;YACrB,KAAK,EAAE,IAAI;YACX,aAAa,EAAE,EAAE;YACjB,KAAK,EAAE,IAAI;SACZ,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,YAAE,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC;QACjE,IAAI,CAAC,MAAM,CAAC,oBAAU,CAAC,CAAC,iBAAiB,CAAC;YACxC,SAAS,EAAE,EAAE;YACb,OAAO,EAAE,cAAc;YACvB,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QAEH,MAAM,IAAA,2BAAiB,EAAC,OAAO,CAAC,CAAC;QAEjC,MAAM,CAAC,6BAAoB,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,gBAAgB,CAAC;YACtB,KAAK,EAAE,EAAE;YACT,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;gBAC/B,OAAO,EAAE,cAAc;gBACvB,QAAQ,EAAE,EAAE;aACb,CAAC;SACH,CAAC,EACF,aAAa,CACd,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,OAAO,GAAG;YACd,KAAK,EAAE,IAAI;YACX,aAAa,EAAE,EAAE;YACjB,KAAK,EAAE,IAAI;YACX,MAAM,EAAE,cAAc;SACvB,CAAC;QAEF,MAAM,IAAA,2BAAiB,EAAC,OAAO,CAAC,CAAC;QAEjC,MAAM,CAAC,gBAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CACtC,MAAM,CAAC,gBAAgB,CAAC,sCAAsC,CAAC,CAChE,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,OAAO,GAA2B;YACtC,OAAO,EAAE,cAAc;YACvB,KAAK,EAAE,IAAI;YACX,aAAa,EAAE,EAAE;YACjB,KAAK,EAAE,IAAI;YACX,MAAM,EAAE,cAAc;SACvB,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,oBAAU,CAAC,CAAC,iBAAiB,CAAC;YACxC,SAAS,EAAE;gBACT;oBACE,IAAI,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE;oBAC7B,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC;oBAClD,QAAQ,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE;iBAClC;aACF;YACD,OAAO,EAAE,cAAc;YACvB,QAAQ,EAAE,CAAC,aAAa,CAAC;SAC1B,CAAC,CAAC;QAEH,MAAM,IAAA,2BAAiB,EAAC,OAAO,CAAC,CAAC;QAEjC,MAAM,CAAC,oBAAU,CAAC,CAAC,oBAAoB,CAAC;YACtC,QAAQ,EAAE,SAAS;YACnB,QAAQ,EAAE,SAAS;YACnB,OAAO,EAAE,cAAc;YACvB,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC;YAC1B,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC;YAC1B,UAAU,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC;SAC9B,CAAC,CAAC;QACH,MAAM,CAAC,6BAAoB,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,gBAAgB,CAAC;YACtB,KAAK,EAAE,MAAM,CAAC,eAAe,CAAC;gBAC5B,MAAM,CAAC,gBAAgB,CAAC;oBACtB,IAAI,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE;oBAC7B,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC;oBAClD,QAAQ,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE;iBAClC,CAAC;aACH,CAAC;YACF,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;gBAC/B,OAAO,EAAE,cAAc;gBACvB,QAAQ,EAAE,CAAC,aAAa,CAAC;aAC1B,CAAC;SACH,CAAC,EACF,cAAc,CACf,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,OAAO,GAA2B;YACtC,MAAM,EAAE,aAAa;YACrB,KAAK,EAAE,IAAI;YACX,aAAa,EAAE,EAAE;YACjB,KAAK,EAAE,IAAI;SACZ,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC,iBAAiB,CAAC;YACzD,QAAQ,EAAE,iBAAiB;YAC3B,SAAS,EAAE;gBACT,OAAO,EAAE,EAAE;gBACX,SAAS,EAAE,EAAE;aACd;YACD,MAAM,EAAE;gBACN,OAAO,EAAE;oBACP,OAAO,EAAE,EAAE;oBACX,UAAU,EAAE,EAAE;iBACf;aACF;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,oBAAU,CAAC,CAAC,iBAAiB,CAAC;YACxC,SAAS,EAAE,EAAE;YACb,OAAO,EAAE,cAAc;YACvB,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QAEH,MAAM,IAAA,2BAAiB,EAAC,OAAO,CAAC,CAAC;QAEjC,MAAM,CAAC,kBAAQ,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"providers.scriptCompletion.test.d.ts","sourceRoot":"","sources":["../../test/providers.scriptCompletion.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,77 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
const crypto_1 = __importDefault(require("crypto"));
|
|
7
|
+
const fs_1 = __importDefault(require("fs"));
|
|
8
|
+
const scriptCompletion_1 = require("../src/providers/scriptCompletion");
|
|
9
|
+
jest.mock('fs');
|
|
10
|
+
jest.mock('crypto');
|
|
11
|
+
describe('parseScriptParts', () => {
|
|
12
|
+
it('should parse script parts correctly', () => {
|
|
13
|
+
const scriptPath = `node script.js "arg with 'spaces'" 'another arg' simple_arg`;
|
|
14
|
+
const result = (0, scriptCompletion_1.parseScriptParts)(scriptPath);
|
|
15
|
+
expect(result).toEqual(['node', 'script.js', "arg with 'spaces'", 'another arg', 'simple_arg']);
|
|
16
|
+
});
|
|
17
|
+
it('should handle script path with no arguments', () => {
|
|
18
|
+
const scriptPath = '/bin/bash script.sh';
|
|
19
|
+
const result = (0, scriptCompletion_1.parseScriptParts)(scriptPath);
|
|
20
|
+
expect(result).toEqual(['/bin/bash', 'script.sh']);
|
|
21
|
+
});
|
|
22
|
+
});
|
|
23
|
+
describe('getFileHashes', () => {
|
|
24
|
+
beforeEach(() => {
|
|
25
|
+
jest.clearAllMocks();
|
|
26
|
+
});
|
|
27
|
+
it('should return file hashes for existing files', () => {
|
|
28
|
+
const scriptParts = ['file1.js', 'file2.js', 'nonexistent.js'];
|
|
29
|
+
const mockFileContent1 = 'content1';
|
|
30
|
+
const mockFileContent2 = 'content2';
|
|
31
|
+
const mockHash1 = 'hash1';
|
|
32
|
+
const mockHash2 = 'hash2';
|
|
33
|
+
jest.mocked(fs_1.default.existsSync).mockImplementation((path) => path !== 'nonexistent.js');
|
|
34
|
+
jest.mocked(fs_1.default.statSync).mockReturnValue({
|
|
35
|
+
isFile: () => true,
|
|
36
|
+
isDirectory: () => false,
|
|
37
|
+
isBlockDevice: () => false,
|
|
38
|
+
isCharacterDevice: () => false,
|
|
39
|
+
isSymbolicLink: () => false,
|
|
40
|
+
isFIFO: () => false,
|
|
41
|
+
isSocket: () => false,
|
|
42
|
+
});
|
|
43
|
+
jest.mocked(fs_1.default.readFileSync).mockImplementation((path) => {
|
|
44
|
+
if (path === 'file1.js') {
|
|
45
|
+
return mockFileContent1;
|
|
46
|
+
}
|
|
47
|
+
if (path === 'file2.js') {
|
|
48
|
+
return mockFileContent2;
|
|
49
|
+
}
|
|
50
|
+
throw new Error('File not found');
|
|
51
|
+
});
|
|
52
|
+
const mockHashUpdate = {
|
|
53
|
+
update: jest.fn().mockReturnThis(),
|
|
54
|
+
digest: jest.fn(),
|
|
55
|
+
};
|
|
56
|
+
jest
|
|
57
|
+
.mocked(mockHashUpdate.digest)
|
|
58
|
+
.mockReturnValueOnce(mockHash1)
|
|
59
|
+
.mockReturnValueOnce(mockHash2);
|
|
60
|
+
jest.mocked(crypto_1.default.createHash).mockReturnValue(mockHashUpdate);
|
|
61
|
+
const result = (0, scriptCompletion_1.getFileHashes)(scriptParts);
|
|
62
|
+
expect(result).toEqual([mockHash1, mockHash2]);
|
|
63
|
+
expect(fs_1.default.existsSync).toHaveBeenCalledTimes(3);
|
|
64
|
+
expect(fs_1.default.readFileSync).toHaveBeenCalledTimes(2);
|
|
65
|
+
expect(crypto_1.default.createHash).toHaveBeenCalledTimes(2);
|
|
66
|
+
});
|
|
67
|
+
it('should return an empty array for non-existent files', () => {
|
|
68
|
+
const scriptParts = ['nonexistent1.js', 'nonexistent2.js'];
|
|
69
|
+
jest.mocked(fs_1.default.existsSync).mockReturnValue(false);
|
|
70
|
+
const result = (0, scriptCompletion_1.getFileHashes)(scriptParts);
|
|
71
|
+
expect(result).toEqual([]);
|
|
72
|
+
expect(fs_1.default.existsSync).toHaveBeenCalledTimes(2);
|
|
73
|
+
expect(fs_1.default.readFileSync).not.toHaveBeenCalled();
|
|
74
|
+
expect(crypto_1.default.createHash).not.toHaveBeenCalled();
|
|
75
|
+
});
|
|
76
|
+
});
|
|
77
|
+
//# sourceMappingURL=providers.scriptCompletion.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"providers.scriptCompletion.test.js","sourceRoot":"","sources":["../../test/providers.scriptCompletion.test.ts"],"names":[],"mappings":";;;;;AAAA,oDAA4B;AAC5B,4CAAoB;AACpB,wEAAoF;AAEpF,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAChB,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AAEpB,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,UAAU,GAAG,6DAA6D,CAAC;QACjF,MAAM,MAAM,GAAG,IAAA,mCAAgB,EAAC,UAAU,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,WAAW,EAAE,mBAAmB,EAAE,aAAa,EAAE,YAAY,CAAC,CAAC,CAAC;IAClG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,UAAU,GAAG,qBAAqB,CAAC;QACzC,MAAM,MAAM,GAAG,IAAA,mCAAgB,EAAC,UAAU,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,WAAW,GAAG,CAAC,UAAU,EAAE,UAAU,EAAE,gBAAgB,CAAC,CAAC;QAC/D,MAAM,gBAAgB,GAAG,UAAU,CAAC;QACpC,MAAM,gBAAgB,GAAG,UAAU,CAAC;QACpC,MAAM,SAAS,GAAG,OAAO,CAAC;QAC1B,MAAM,SAAS,GAAG,OAAO,CAAC;QAE1B,IAAI,CAAC,MAAM,CAAC,YAAE,CAAC,UAAU,CAAC,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,gBAAgB,CAAC,CAAC;QACnF,IAAI,CAAC,MAAM,CAAC,YAAE,CAAC,QAAQ,CAAC,CAAC,eAAe,CAAC;YACvC,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI;YAClB,WAAW,EAAE,GAAG,EAAE,CAAC,KAAK;YACxB,aAAa,EAAE,GAAG,EAAE,CAAC,KAAK;YAC1B,iBAAiB,EAAE,GAAG,EAAE,CAAC,KAAK;YAC9B,cAAc,EAAE,GAAG,EAAE,CAAC,KAAK;YAC3B,MAAM,EAAE,GAAG,EAAE,CAAC,KAAK;YACnB,QAAQ,EAAE,GAAG,EAAE,CAAC,KAAK;SACV,CAAC,CAAC;QACf,IAAI,CAAC,MAAM,CAAC,YAAE,CAAC,YAAY,CAAC,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE;YACvD,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;gBACxB,OAAO,gBAAgB,CAAC;YAC1B,CAAC;YACD,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;gBACxB,OAAO,gBAAgB,CAAC;YAC1B,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,MAAM,cAAc,GAAG;YACrB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;YAClC,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;SACQ,CAAC;QAC5B,IAAI;aACD,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC;aAC7B,mBAAmB,CAAC,SAAS,CAAC;aAC9B,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAClC,IAAI,CAAC,MAAM,CAAC,gBAAM,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;QAE/D,MAAM,MAAM,GAAG,IAAA,gCAAa,EAAC,WAAW,CAAC,CAAC;QAE1C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,YAAE,CAAC,UAAU,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,YAAE,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjD,MAAM,CAAC,gBAAM,CAAC,UAAU,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,WAAW,GAAG,CAAC,iBAAiB,EAAE,iBAAiB,CAAC,CAAC;QAC3D,IAAI,CAAC,MAAM,CAAC,YAAE,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,IAAA,gCAAa,EAAC,WAAW,CAAC,CAAC;QAE1C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC3B,MAAM,CAAC,YAAE,CAAC,UAAU,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,YAAE,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/C,MAAM,CAAC,gBAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACnD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
package/package.json
CHANGED
|
@@ -2,7 +2,7 @@
|
|
|
2
2
|
"name": "promptfoo",
|
|
3
3
|
"description": "LLM eval & testing toolkit",
|
|
4
4
|
"author": "Ian Webster",
|
|
5
|
-
"version": "0.78.
|
|
5
|
+
"version": "0.78.3",
|
|
6
6
|
"license": "MIT",
|
|
7
7
|
"type": "commonjs",
|
|
8
8
|
"repository": {
|
|
@@ -157,7 +157,7 @@
|
|
|
157
157
|
"python-shell": "^5.0.0",
|
|
158
158
|
"replicate": "^0.32.0",
|
|
159
159
|
"rfdc": "^1.4.1",
|
|
160
|
-
"rouge": "
|
|
160
|
+
"js-rouge": "3.0.0",
|
|
161
161
|
"semver": "^7.6.3",
|
|
162
162
|
"socket.io": "^4.7.5",
|
|
163
163
|
"tiny-invariant": "^1.3.3",
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
(self.webpackChunk_N_E=self.webpackChunk_N_E||[]).push([[931],{5944:function(e,n,t){Promise.resolve().then(t.t.bind(t,16662,23)),Promise.resolve().then(t.t.bind(t,27924,23)),Promise.resolve().then(t.t.bind(t,30910,23)),Promise.resolve().then(t.t.bind(t,81885,23)),Promise.resolve().then(t.t.bind(t,48607,23))},81885:function(e,n,t){"use strict";Object.defineProperty(n,"__esModule",{value:!0}),function(e,n){for(var t in n)Object.defineProperty(e,t,{enumerable:!0,get:n[t]})}(n,{suspense:function(){return o},NoSSR:function(){return i}}),t(69278),t(14978);let r=t(66995);function o(){let e=Error(r.NEXT_DYNAMIC_NO_SSR_CODE);throw e.digest=r.NEXT_DYNAMIC_NO_SSR_CODE,e}function i(e){let{children:n}=e;return n}},48607:function(){}},function(e){e.O(0,[470,730,744],function(){return e(e.s=5944)}),_N_E=e.O()}]);
|
|
File without changes
|
|
File without changes
|