model-test-bench 1.0.2
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/LICENSE +201 -0
- package/README.md +389 -0
- package/dist/bin/mtb.d.ts +3 -0
- package/dist/bin/mtb.d.ts.map +1 -0
- package/dist/bin/mtb.js +148 -0
- package/dist/bin/mtb.js.map +1 -0
- package/dist/server/index.d.ts +13 -0
- package/dist/server/index.d.ts.map +1 -0
- package/dist/server/index.js +72 -0
- package/dist/server/index.js.map +1 -0
- package/dist/server/interfaces/evaluator.d.ts +15 -0
- package/dist/server/interfaces/evaluator.d.ts.map +1 -0
- package/dist/server/interfaces/evaluator.js +2 -0
- package/dist/server/interfaces/evaluator.js.map +1 -0
- package/dist/server/interfaces/logger.d.ts +9 -0
- package/dist/server/interfaces/logger.d.ts.map +1 -0
- package/dist/server/interfaces/logger.js +2 -0
- package/dist/server/interfaces/logger.js.map +1 -0
- package/dist/server/interfaces/runner.d.ts +9 -0
- package/dist/server/interfaces/runner.d.ts.map +1 -0
- package/dist/server/interfaces/runner.js +2 -0
- package/dist/server/interfaces/runner.js.map +1 -0
- package/dist/server/interfaces/storage.d.ts +36 -0
- package/dist/server/interfaces/storage.d.ts.map +1 -0
- package/dist/server/interfaces/storage.js +2 -0
- package/dist/server/interfaces/storage.js.map +1 -0
- package/dist/server/routes/eval-queue.d.ts +23 -0
- package/dist/server/routes/eval-queue.d.ts.map +1 -0
- package/dist/server/routes/eval-queue.js +45 -0
- package/dist/server/routes/eval-queue.js.map +1 -0
- package/dist/server/routes/evaluations.d.ts +8 -0
- package/dist/server/routes/evaluations.d.ts.map +1 -0
- package/dist/server/routes/evaluations.js +221 -0
- package/dist/server/routes/evaluations.js.map +1 -0
- package/dist/server/routes/providers.d.ts +5 -0
- package/dist/server/routes/providers.d.ts.map +1 -0
- package/dist/server/routes/providers.js +179 -0
- package/dist/server/routes/providers.js.map +1 -0
- package/dist/server/routes/run-queue.d.ts +17 -0
- package/dist/server/routes/run-queue.d.ts.map +1 -0
- package/dist/server/routes/run-queue.js +34 -0
- package/dist/server/routes/run-queue.js.map +1 -0
- package/dist/server/routes/run-sse.d.ts +18 -0
- package/dist/server/routes/run-sse.d.ts.map +1 -0
- package/dist/server/routes/run-sse.js +57 -0
- package/dist/server/routes/run-sse.js.map +1 -0
- package/dist/server/routes/runs.d.ts +9 -0
- package/dist/server/routes/runs.d.ts.map +1 -0
- package/dist/server/routes/runs.js +380 -0
- package/dist/server/routes/runs.js.map +1 -0
- package/dist/server/routes/scenarios.d.ts +5 -0
- package/dist/server/routes/scenarios.d.ts.map +1 -0
- package/dist/server/routes/scenarios.js +181 -0
- package/dist/server/routes/scenarios.js.map +1 -0
- package/dist/server/services/eval-helpers.d.ts +22 -0
- package/dist/server/services/eval-helpers.d.ts.map +1 -0
- package/dist/server/services/eval-helpers.js +75 -0
- package/dist/server/services/eval-helpers.js.map +1 -0
- package/dist/server/services/eval-parsers-debate-impl.d.ts +11 -0
- package/dist/server/services/eval-parsers-debate-impl.d.ts.map +1 -0
- package/dist/server/services/eval-parsers-debate-impl.js +133 -0
- package/dist/server/services/eval-parsers-debate-impl.js.map +1 -0
- package/dist/server/services/eval-parsers.d.ts +24 -0
- package/dist/server/services/eval-parsers.d.ts.map +1 -0
- package/dist/server/services/eval-parsers.js +153 -0
- package/dist/server/services/eval-parsers.js.map +1 -0
- package/dist/server/services/eval-prompts.d.ts +9 -0
- package/dist/server/services/eval-prompts.d.ts.map +1 -0
- package/dist/server/services/eval-prompts.js +164 -0
- package/dist/server/services/eval-prompts.js.map +1 -0
- package/dist/server/services/evaluator.d.ts +10 -0
- package/dist/server/services/evaluator.d.ts.map +1 -0
- package/dist/server/services/evaluator.js +143 -0
- package/dist/server/services/evaluator.js.map +1 -0
- package/dist/server/services/fs-adapter.d.ts +20 -0
- package/dist/server/services/fs-adapter.d.ts.map +1 -0
- package/dist/server/services/fs-adapter.js +13 -0
- package/dist/server/services/fs-adapter.js.map +1 -0
- package/dist/server/services/instruction-parser.d.ts +26 -0
- package/dist/server/services/instruction-parser.d.ts.map +1 -0
- package/dist/server/services/instruction-parser.js +121 -0
- package/dist/server/services/instruction-parser.js.map +1 -0
- package/dist/server/services/log-rotator.d.ts +20 -0
- package/dist/server/services/log-rotator.d.ts.map +1 -0
- package/dist/server/services/log-rotator.js +60 -0
- package/dist/server/services/log-rotator.js.map +1 -0
- package/dist/server/services/logger.d.ts +15 -0
- package/dist/server/services/logger.d.ts.map +1 -0
- package/dist/server/services/logger.js +69 -0
- package/dist/server/services/logger.js.map +1 -0
- package/dist/server/services/model-factory.d.ts +10 -0
- package/dist/server/services/model-factory.d.ts.map +1 -0
- package/dist/server/services/model-factory.js +33 -0
- package/dist/server/services/model-factory.js.map +1 -0
- package/dist/server/services/runner.d.ts +9 -0
- package/dist/server/services/runner.d.ts.map +1 -0
- package/dist/server/services/runner.js +99 -0
- package/dist/server/services/runner.js.map +1 -0
- package/dist/server/services/seeder.d.ts +5 -0
- package/dist/server/services/seeder.d.ts.map +1 -0
- package/dist/server/services/seeder.js +79 -0
- package/dist/server/services/seeder.js.map +1 -0
- package/dist/server/services/storage-test-helpers.d.ts +15 -0
- package/dist/server/services/storage-test-helpers.d.ts.map +1 -0
- package/dist/server/services/storage-test-helpers.js +151 -0
- package/dist/server/services/storage-test-helpers.js.map +1 -0
- package/dist/server/services/storage.d.ts +35 -0
- package/dist/server/services/storage.d.ts.map +1 -0
- package/dist/server/services/storage.js +219 -0
- package/dist/server/services/storage.js.map +1 -0
- package/dist/server/services/tools.d.ts +6 -0
- package/dist/server/services/tools.d.ts.map +1 -0
- package/dist/server/services/tools.js +94 -0
- package/dist/server/services/tools.js.map +1 -0
- package/dist/server/services/transcript-formatter.d.ts +18 -0
- package/dist/server/services/transcript-formatter.d.ts.map +1 -0
- package/dist/server/services/transcript-formatter.js +227 -0
- package/dist/server/services/transcript-formatter.js.map +1 -0
- package/dist/server/services/update-checker.d.ts +3 -0
- package/dist/server/services/update-checker.d.ts.map +1 -0
- package/dist/server/services/update-checker.js +34 -0
- package/dist/server/services/update-checker.js.map +1 -0
- package/dist/server/types/evaluation.d.ts +94 -0
- package/dist/server/types/evaluation.d.ts.map +1 -0
- package/dist/server/types/evaluation.js +5 -0
- package/dist/server/types/evaluation.js.map +1 -0
- package/dist/server/types/index.d.ts +5 -0
- package/dist/server/types/index.d.ts.map +1 -0
- package/dist/server/types/index.js +5 -0
- package/dist/server/types/index.js.map +1 -0
- package/dist/server/types/provider.d.ts +23 -0
- package/dist/server/types/provider.d.ts.map +1 -0
- package/dist/server/types/provider.js +5 -0
- package/dist/server/types/provider.js.map +1 -0
- package/dist/server/types/run.d.ts +31 -0
- package/dist/server/types/run.d.ts.map +1 -0
- package/dist/server/types/run.js +5 -0
- package/dist/server/types/run.js.map +1 -0
- package/dist/server/types/scenario.d.ts +19 -0
- package/dist/server/types/scenario.d.ts.map +1 -0
- package/dist/server/types/scenario.js +5 -0
- package/dist/server/types/scenario.js.map +1 -0
- package/dist/src/server/index.d.ts +13 -0
- package/dist/src/server/index.d.ts.map +1 -0
- package/dist/src/server/index.js +72 -0
- package/dist/src/server/index.js.map +1 -0
- package/dist/src/server/interfaces/evaluator.d.ts +15 -0
- package/dist/src/server/interfaces/evaluator.d.ts.map +1 -0
- package/dist/src/server/interfaces/evaluator.js +2 -0
- package/dist/src/server/interfaces/evaluator.js.map +1 -0
- package/dist/src/server/interfaces/logger.d.ts +9 -0
- package/dist/src/server/interfaces/logger.d.ts.map +1 -0
- package/dist/src/server/interfaces/logger.js +2 -0
- package/dist/src/server/interfaces/logger.js.map +1 -0
- package/dist/src/server/interfaces/runner.d.ts +9 -0
- package/dist/src/server/interfaces/runner.d.ts.map +1 -0
- package/dist/src/server/interfaces/runner.js +2 -0
- package/dist/src/server/interfaces/runner.js.map +1 -0
- package/dist/src/server/interfaces/storage.d.ts +36 -0
- package/dist/src/server/interfaces/storage.d.ts.map +1 -0
- package/dist/src/server/interfaces/storage.js +2 -0
- package/dist/src/server/interfaces/storage.js.map +1 -0
- package/dist/src/server/routes/eval-queue.d.ts +23 -0
- package/dist/src/server/routes/eval-queue.d.ts.map +1 -0
- package/dist/src/server/routes/eval-queue.js +45 -0
- package/dist/src/server/routes/eval-queue.js.map +1 -0
- package/dist/src/server/routes/evaluations.d.ts +8 -0
- package/dist/src/server/routes/evaluations.d.ts.map +1 -0
- package/dist/src/server/routes/evaluations.js +221 -0
- package/dist/src/server/routes/evaluations.js.map +1 -0
- package/dist/src/server/routes/providers.d.ts +5 -0
- package/dist/src/server/routes/providers.d.ts.map +1 -0
- package/dist/src/server/routes/providers.js +179 -0
- package/dist/src/server/routes/providers.js.map +1 -0
- package/dist/src/server/routes/run-queue.d.ts +17 -0
- package/dist/src/server/routes/run-queue.d.ts.map +1 -0
- package/dist/src/server/routes/run-queue.js +34 -0
- package/dist/src/server/routes/run-queue.js.map +1 -0
- package/dist/src/server/routes/run-sse.d.ts +18 -0
- package/dist/src/server/routes/run-sse.d.ts.map +1 -0
- package/dist/src/server/routes/run-sse.js +57 -0
- package/dist/src/server/routes/run-sse.js.map +1 -0
- package/dist/src/server/routes/runs.d.ts +9 -0
- package/dist/src/server/routes/runs.d.ts.map +1 -0
- package/dist/src/server/routes/runs.js +380 -0
- package/dist/src/server/routes/runs.js.map +1 -0
- package/dist/src/server/routes/scenarios.d.ts +5 -0
- package/dist/src/server/routes/scenarios.d.ts.map +1 -0
- package/dist/src/server/routes/scenarios.js +181 -0
- package/dist/src/server/routes/scenarios.js.map +1 -0
- package/dist/src/server/services/eval-helpers.d.ts +22 -0
- package/dist/src/server/services/eval-helpers.d.ts.map +1 -0
- package/dist/src/server/services/eval-helpers.js +75 -0
- package/dist/src/server/services/eval-helpers.js.map +1 -0
- package/dist/src/server/services/eval-parsers-debate-impl.d.ts +11 -0
- package/dist/src/server/services/eval-parsers-debate-impl.d.ts.map +1 -0
- package/dist/src/server/services/eval-parsers-debate-impl.js +133 -0
- package/dist/src/server/services/eval-parsers-debate-impl.js.map +1 -0
- package/dist/src/server/services/eval-parsers.d.ts +24 -0
- package/dist/src/server/services/eval-parsers.d.ts.map +1 -0
- package/dist/src/server/services/eval-parsers.js +153 -0
- package/dist/src/server/services/eval-parsers.js.map +1 -0
- package/dist/src/server/services/eval-prompts.d.ts +9 -0
- package/dist/src/server/services/eval-prompts.d.ts.map +1 -0
- package/dist/src/server/services/eval-prompts.js +164 -0
- package/dist/src/server/services/eval-prompts.js.map +1 -0
- package/dist/src/server/services/evaluator.d.ts +10 -0
- package/dist/src/server/services/evaluator.d.ts.map +1 -0
- package/dist/src/server/services/evaluator.js +143 -0
- package/dist/src/server/services/evaluator.js.map +1 -0
- package/dist/src/server/services/fs-adapter.d.ts +20 -0
- package/dist/src/server/services/fs-adapter.d.ts.map +1 -0
- package/dist/src/server/services/fs-adapter.js +13 -0
- package/dist/src/server/services/fs-adapter.js.map +1 -0
- package/dist/src/server/services/instruction-parser.d.ts +26 -0
- package/dist/src/server/services/instruction-parser.d.ts.map +1 -0
- package/dist/src/server/services/instruction-parser.js +121 -0
- package/dist/src/server/services/instruction-parser.js.map +1 -0
- package/dist/src/server/services/log-rotator.d.ts +20 -0
- package/dist/src/server/services/log-rotator.d.ts.map +1 -0
- package/dist/src/server/services/log-rotator.js +60 -0
- package/dist/src/server/services/log-rotator.js.map +1 -0
- package/dist/src/server/services/logger.d.ts +15 -0
- package/dist/src/server/services/logger.d.ts.map +1 -0
- package/dist/src/server/services/logger.js +69 -0
- package/dist/src/server/services/logger.js.map +1 -0
- package/dist/src/server/services/model-factory.d.ts +10 -0
- package/dist/src/server/services/model-factory.d.ts.map +1 -0
- package/dist/src/server/services/model-factory.js +33 -0
- package/dist/src/server/services/model-factory.js.map +1 -0
- package/dist/src/server/services/runner.d.ts +9 -0
- package/dist/src/server/services/runner.d.ts.map +1 -0
- package/dist/src/server/services/runner.js +99 -0
- package/dist/src/server/services/runner.js.map +1 -0
- package/dist/src/server/services/seeder.d.ts +5 -0
- package/dist/src/server/services/seeder.d.ts.map +1 -0
- package/dist/src/server/services/seeder.js +79 -0
- package/dist/src/server/services/seeder.js.map +1 -0
- package/dist/src/server/services/storage.d.ts +35 -0
- package/dist/src/server/services/storage.d.ts.map +1 -0
- package/dist/src/server/services/storage.js +219 -0
- package/dist/src/server/services/storage.js.map +1 -0
- package/dist/src/server/services/tools.d.ts +6 -0
- package/dist/src/server/services/tools.d.ts.map +1 -0
- package/dist/src/server/services/tools.js +94 -0
- package/dist/src/server/services/tools.js.map +1 -0
- package/dist/src/server/services/transcript-formatter.d.ts +18 -0
- package/dist/src/server/services/transcript-formatter.d.ts.map +1 -0
- package/dist/src/server/services/transcript-formatter.js +227 -0
- package/dist/src/server/services/transcript-formatter.js.map +1 -0
- package/dist/src/server/services/update-checker.d.ts +3 -0
- package/dist/src/server/services/update-checker.d.ts.map +1 -0
- package/dist/src/server/services/update-checker.js +34 -0
- package/dist/src/server/services/update-checker.js.map +1 -0
- package/dist/src/server/types/evaluation.d.ts +94 -0
- package/dist/src/server/types/evaluation.d.ts.map +1 -0
- package/dist/src/server/types/evaluation.js +5 -0
- package/dist/src/server/types/evaluation.js.map +1 -0
- package/dist/src/server/types/index.d.ts +5 -0
- package/dist/src/server/types/index.d.ts.map +1 -0
- package/dist/src/server/types/index.js +5 -0
- package/dist/src/server/types/index.js.map +1 -0
- package/dist/src/server/types/provider.d.ts +23 -0
- package/dist/src/server/types/provider.d.ts.map +1 -0
- package/dist/src/server/types/provider.js +5 -0
- package/dist/src/server/types/provider.js.map +1 -0
- package/dist/src/server/types/run.d.ts +31 -0
- package/dist/src/server/types/run.d.ts.map +1 -0
- package/dist/src/server/types/run.js +5 -0
- package/dist/src/server/types/run.js.map +1 -0
- package/dist/src/server/types/scenario.d.ts +19 -0
- package/dist/src/server/types/scenario.d.ts.map +1 -0
- package/dist/src/server/types/scenario.js +5 -0
- package/dist/src/server/types/scenario.js.map +1 -0
- package/dist/web/assets/index-AJu1Yn5F.js +70 -0
- package/dist/web/assets/index-C_ioEISr.css +1 -0
- package/dist/web/index.html +15 -0
- package/docs/schemas/provider-api.example.json +12 -0
- package/docs/schemas/provider-openai.example.json +11 -0
- package/docs/schemas/scenario-baseline.example.json +24 -0
- package/docs/schemas/scenario-carwash-baseline.example.json +22 -0
- package/docs/schemas/scenario-carwash-with-system-prompt.example.json +24 -0
- package/docs/schemas/scenario-golden-rules-baseline.example.json +24 -0
- package/docs/schemas/scenario-golden-rules-with-system-prompt.example.json +28 -0
- package/docs/schemas/scenario-negative-analysis-baseline.example.json +23 -0
- package/docs/schemas/scenario-negative-analysis-with-system-prompt.example.json +25 -0
- package/docs/schemas/scenario-with-system-prompt.example.json +25 -0
- package/package.json +97 -0
|
@@ -0,0 +1,99 @@
|
|
|
1
|
+
// ---------------------------------------------------------------------------
|
|
2
|
+
// AiSdkRunner — executes a single run via Vercel AI SDK (streaming)
|
|
3
|
+
// ---------------------------------------------------------------------------
|
|
4
|
+
import { streamText, stepCountIs } from 'ai';
|
|
5
|
+
import { createModel } from './model-factory.js';
|
|
6
|
+
import { getEnabledTools } from './tools.js';
|
|
7
|
+
// ---------------------------------------------------------------------------
|
|
8
|
+
// Implementation
|
|
9
|
+
// ---------------------------------------------------------------------------
|
|
10
|
+
export class AiSdkRunner {
|
|
11
|
+
logger;
|
|
12
|
+
constructor(logger) {
|
|
13
|
+
this.logger = logger;
|
|
14
|
+
}
|
|
15
|
+
async executeRun(provider, scenario, run, callbacks, externalAbortController) {
|
|
16
|
+
const startTime = Date.now();
|
|
17
|
+
const messages = [];
|
|
18
|
+
const abortController = externalAbortController ?? new AbortController();
|
|
19
|
+
let timeoutId;
|
|
20
|
+
try {
|
|
21
|
+
// Set up timeout
|
|
22
|
+
const timeoutMs = provider.timeoutSeconds * 1000;
|
|
23
|
+
timeoutId = setTimeout(() => abortController.abort(), timeoutMs);
|
|
24
|
+
callbacks.onStatusChange('running');
|
|
25
|
+
// Build model and tools
|
|
26
|
+
const model = createModel(provider);
|
|
27
|
+
const tools = getEnabledTools(scenario.enabledTools);
|
|
28
|
+
const hasTools = Object.keys(tools).length > 0;
|
|
29
|
+
// Stream the query
|
|
30
|
+
const result = streamText({
|
|
31
|
+
model,
|
|
32
|
+
system: scenario.systemPrompt || undefined,
|
|
33
|
+
prompt: scenario.prompt,
|
|
34
|
+
temperature: provider.temperature,
|
|
35
|
+
maxOutputTokens: provider.maxTokens,
|
|
36
|
+
topP: provider.topP,
|
|
37
|
+
tools: hasTools ? tools : undefined,
|
|
38
|
+
stopWhen: stepCountIs(hasTools ? 10 : 1),
|
|
39
|
+
abortSignal: abortController.signal,
|
|
40
|
+
});
|
|
41
|
+
// Consume the stream, forwarding each part to the frontend via callbacks
|
|
42
|
+
for await (const part of result.fullStream) {
|
|
43
|
+
const record = {
|
|
44
|
+
timestamp: new Date().toISOString(),
|
|
45
|
+
message: part,
|
|
46
|
+
};
|
|
47
|
+
messages.push(record);
|
|
48
|
+
callbacks.onMessage(record);
|
|
49
|
+
}
|
|
50
|
+
// Await final aggregated values
|
|
51
|
+
const finalText = await result.text;
|
|
52
|
+
const usage = await result.usage;
|
|
53
|
+
const steps = await result.steps;
|
|
54
|
+
const durationMs = Date.now() - startTime;
|
|
55
|
+
const completedRun = {
|
|
56
|
+
...run,
|
|
57
|
+
status: 'completed',
|
|
58
|
+
messages,
|
|
59
|
+
resultText: finalText,
|
|
60
|
+
totalCostUsd: 0,
|
|
61
|
+
durationMs,
|
|
62
|
+
numTurns: steps.length,
|
|
63
|
+
updatedAt: new Date().toISOString(),
|
|
64
|
+
};
|
|
65
|
+
callbacks.onStatusChange('completed');
|
|
66
|
+
this.logger.info('Run completed', {
|
|
67
|
+
runId: run.id,
|
|
68
|
+
durationMs,
|
|
69
|
+
inputTokens: usage?.inputTokens ?? 0,
|
|
70
|
+
outputTokens: usage?.outputTokens ?? 0,
|
|
71
|
+
steps: steps.length,
|
|
72
|
+
});
|
|
73
|
+
return completedRun;
|
|
74
|
+
}
|
|
75
|
+
catch (err) {
|
|
76
|
+
const durationMs = Date.now() - startTime;
|
|
77
|
+
const errorMessage = err instanceof Error ? err.message : String(err);
|
|
78
|
+
const isAbort = abortController.signal.aborted;
|
|
79
|
+
const errorRun = {
|
|
80
|
+
...run,
|
|
81
|
+
status: isAbort ? 'cancelled' : 'failed',
|
|
82
|
+
messages,
|
|
83
|
+
resultText: '',
|
|
84
|
+
totalCostUsd: 0,
|
|
85
|
+
durationMs,
|
|
86
|
+
numTurns: 0,
|
|
87
|
+
error: isAbort ? `Timeout after ${provider.timeoutSeconds}s` : errorMessage,
|
|
88
|
+
updatedAt: new Date().toISOString(),
|
|
89
|
+
};
|
|
90
|
+
callbacks.onStatusChange(errorRun.status);
|
|
91
|
+
return errorRun;
|
|
92
|
+
}
|
|
93
|
+
finally {
|
|
94
|
+
if (timeoutId !== undefined)
|
|
95
|
+
clearTimeout(timeoutId);
|
|
96
|
+
}
|
|
97
|
+
}
|
|
98
|
+
}
|
|
99
|
+
//# sourceMappingURL=runner.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"runner.js","sourceRoot":"","sources":["../../../src/server/services/runner.ts"],"names":[],"mappings":"AAAA,8EAA8E;AAC9E,oEAAoE;AACpE,8EAA8E;AAE9E,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,IAAI,CAAC;AAI7C,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAE7C,8EAA8E;AAC9E,iBAAiB;AACjB,8EAA8E;AAE9E,MAAM,OAAO,WAAW;IACO;IAA7B,YAA6B,MAAe;QAAf,WAAM,GAAN,MAAM,CAAS;IAAG,CAAC;IAEhD,KAAK,CAAC,UAAU,CACd,QAAkB,EAClB,QAAkB,EAClB,GAAQ,EACR,SAAuB,EACvB,uBAAyC;QAEzC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,QAAQ,GAAuB,EAAE,CAAC;QACxC,MAAM,eAAe,GAAG,uBAAuB,IAAI,IAAI,eAAe,EAAE,CAAC;QAEzE,IAAI,SAAoD,CAAC;QAEzD,IAAI,CAAC;YACH,iBAAiB;YACjB,MAAM,SAAS,GAAG,QAAQ,CAAC,cAAc,GAAG,IAAI,CAAC;YACjD,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,eAAe,CAAC,KAAK,EAAE,EAAE,SAAS,CAAC,CAAC;YACjE,SAAS,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;YAEpC,wBAAwB;YACxB,MAAM,KAAK,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;YACpC,MAAM,KAAK,GAAG,eAAe,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;YACrD,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;YAE/C,mBAAmB;YACnB,MAAM,MAAM,GAAG,UAAU,CAAC;gBACxB,KAAK;gBACL,MAAM,EAAE,QAAQ,CAAC,YAAY,IAAI,SAAS;gBAC1C,MAAM,EAAE,QAAQ,CAAC,MAAM;gBACvB,WAAW,EAAE,QAAQ,CAAC,WAAW;gBACjC,eAAe,EAAE,QAAQ,CAAC,SAAS;gBACnC,IAAI,EAAE,QAAQ,CAAC,IAAI;gBACnB,KAAK,EAAE,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;gBACnC,QAAQ,EAAE,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;gBACxC,WAAW,EAAE,eAAe,CAAC,MAAM;aACpC,CAAC,CAAC;YAEH,yEAAyE;YACzE,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;gBAC3C,MAAM,MAAM,GAAqB;oBAC/B,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;oBACnC,OAAO,EAAE,IAA0C;iBACpD,CAAC;gBACF,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACtB,SAAS,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAC9B,CAAC;YAED,gCAAgC;YAChC,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC;YACpC,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;YACjC,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;YACjC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAE1C,MAAM,YAAY,GAAQ;gBACxB,GAAG,GAAG;gBACN,MAAM,EAAE,WAAW;gBACnB,QAAQ;gBACR,UAAU,EAAE,SAAS;gBACrB,YAAY,EAAE,CAAC;gBACf,UAAU;gBACV,QAAQ,EAAE,KAAK,CAAC,MAAM;gBACtB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC,CAAC;YACF,SAAS,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;YAEtC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,EAAE;gBAChC,KAAK,EAAE,GAAG,CAAC,EAAE;gBACb,UAAU;gBACV,WAAW,EAAE,KAAK,EAAE,WAAW,IAAI,CAAC;gBACpC,YAAY,EAAE,KAAK,EAAE,YAAY,IAAI,CAAC;gBACtC,KAAK,EAAE,KAAK,CAAC,MAAM;aACpB,CAAC,CAAC;YAEH,OAAO,YAAY,CAAC;QACtB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC1C,MAAM,YAAY,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACtE,MAAM,OAAO,GAAG,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC;YAE/C,MAAM,QAAQ,GAAQ;gBACpB,GAAG,GAAG;gBACN,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ;gBACxC,QAAQ;gBACR,UAAU,EAAE,EAAE;gBACd,YAAY,EAAE,CAAC;gBACf,UAAU;gBACV,QAAQ,EAAE,CAAC;gBACX,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,iBAAiB,QAAQ,CAAC,cAAc,GAAG,CAAC,CAAC,CAAC,YAAY;gBAC3E,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC,CAAC;YACF,SAAS,CAAC,cAAc,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YAC1C,OAAO,QAAQ,CAAC;QAClB,CAAC;gBAAS,CAAC;YACT,IAAI,SAAS,KAAK,SAAS;gBAAE,YAAY,CAAC,SAAS,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
import type { IStorage } from '../interfaces/storage.js';
|
|
2
|
+
import type { ILogger } from '../interfaces/logger.js';
|
|
3
|
+
/** Seed storage with example scenarios and providers when empty. */
|
|
4
|
+
export declare function seedIfEmpty(storage: IStorage, logger: ILogger): Promise<void>;
|
|
5
|
+
//# sourceMappingURL=seeder.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"seeder.d.ts","sourceRoot":"","sources":["../../../src/server/services/seeder.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AACzD,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AA8CvD,oEAAoE;AACpE,wBAAsB,WAAW,CAAC,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAoCnF"}
|
|
@@ -0,0 +1,79 @@
|
|
|
1
|
+
import fs from 'node:fs';
|
|
2
|
+
import path from 'node:path';
|
|
3
|
+
/**
|
|
4
|
+
* Seed files that ship with the project under docs/schemas/.
|
|
5
|
+
* Each entry maps a source file to the storage method used to persist it.
|
|
6
|
+
* Seeding only runs when the respective list is empty — existing data is never overwritten.
|
|
7
|
+
*/
|
|
8
|
+
const SEED_SCENARIOS = [
|
|
9
|
+
'scenario-baseline.example.json',
|
|
10
|
+
'scenario-with-system-prompt.example.json',
|
|
11
|
+
'scenario-carwash-baseline.example.json',
|
|
12
|
+
'scenario-carwash-with-system-prompt.example.json',
|
|
13
|
+
'scenario-negative-analysis-baseline.example.json',
|
|
14
|
+
'scenario-negative-analysis-with-system-prompt.example.json',
|
|
15
|
+
'scenario-golden-rules-baseline.example.json',
|
|
16
|
+
'scenario-golden-rules-with-system-prompt.example.json',
|
|
17
|
+
];
|
|
18
|
+
const SEED_PROVIDERS = [
|
|
19
|
+
'provider-api.example.json',
|
|
20
|
+
'provider-openai.example.json',
|
|
21
|
+
];
|
|
22
|
+
function resolveSchemasDir() {
|
|
23
|
+
// Try common locations: relative to cwd, relative to this file's compiled location
|
|
24
|
+
const candidates = [
|
|
25
|
+
path.join(process.cwd(), 'docs', 'schemas'),
|
|
26
|
+
path.join(process.cwd(), '..', 'docs', 'schemas'),
|
|
27
|
+
];
|
|
28
|
+
for (const dir of candidates) {
|
|
29
|
+
if (fs.existsSync(dir))
|
|
30
|
+
return dir;
|
|
31
|
+
}
|
|
32
|
+
return candidates[0]; // fallback
|
|
33
|
+
}
|
|
34
|
+
function loadJsonFile(filePath) {
|
|
35
|
+
try {
|
|
36
|
+
const raw = fs.readFileSync(filePath, 'utf-8');
|
|
37
|
+
return JSON.parse(raw);
|
|
38
|
+
}
|
|
39
|
+
catch {
|
|
40
|
+
return undefined;
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
/** Seed storage with example scenarios and providers when empty. */
|
|
44
|
+
export async function seedIfEmpty(storage, logger) {
|
|
45
|
+
const schemasDir = resolveSchemasDir();
|
|
46
|
+
// Seed scenarios
|
|
47
|
+
const existingScenarios = await storage.listScenarios();
|
|
48
|
+
if (existingScenarios.length === 0) {
|
|
49
|
+
let seeded = 0;
|
|
50
|
+
for (const file of SEED_SCENARIOS) {
|
|
51
|
+
const scenario = loadJsonFile(path.join(schemasDir, file));
|
|
52
|
+
if (scenario) {
|
|
53
|
+
await storage.saveScenario(scenario);
|
|
54
|
+
seeded++;
|
|
55
|
+
logger.info('Seeded scenario', { name: scenario.name, id: scenario.id });
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
if (seeded === 0) {
|
|
59
|
+
logger.info('No seed scenarios found in ' + schemasDir);
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
// Seed providers
|
|
63
|
+
const existingProviders = await storage.listProviders();
|
|
64
|
+
if (existingProviders.length === 0) {
|
|
65
|
+
let seeded = 0;
|
|
66
|
+
for (const file of SEED_PROVIDERS) {
|
|
67
|
+
const provider = loadJsonFile(path.join(schemasDir, file));
|
|
68
|
+
if (provider) {
|
|
69
|
+
await storage.saveProvider(provider);
|
|
70
|
+
seeded++;
|
|
71
|
+
logger.info('Seeded provider', { name: provider.name, id: provider.id });
|
|
72
|
+
}
|
|
73
|
+
}
|
|
74
|
+
if (seeded === 0) {
|
|
75
|
+
logger.info('No seed providers found in ' + schemasDir);
|
|
76
|
+
}
|
|
77
|
+
}
|
|
78
|
+
}
|
|
79
|
+
//# sourceMappingURL=seeder.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"seeder.js","sourceRoot":"","sources":["../../../src/server/services/seeder.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAM7B;;;;GAIG;AACH,MAAM,cAAc,GAAG;IACrB,gCAAgC;IAChC,0CAA0C;IAC1C,wCAAwC;IACxC,kDAAkD;IAClD,kDAAkD;IAClD,4DAA4D;IAC5D,6CAA6C;IAC7C,uDAAuD;CACxD,CAAC;AAEF,MAAM,cAAc,GAAG;IACrB,2BAA2B;IAC3B,8BAA8B;CAC/B,CAAC;AAEF,SAAS,iBAAiB;IACxB,mFAAmF;IACnF,MAAM,UAAU,GAAG;QACjB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,SAAS,CAAC;QAC3C,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,CAAC;KAClD,CAAC;IACF,KAAK,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;QAC7B,IAAI,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC;YAAE,OAAO,GAAG,CAAC;IACrC,CAAC;IACD,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW;AACnC,CAAC;AAED,SAAS,YAAY,CAAI,QAAgB;IACvC,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAC/C,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAM,CAAC;IAC9B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,oEAAoE;AACpE,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,OAAiB,EAAE,MAAe;IAClE,MAAM,UAAU,GAAG,iBAAiB,EAAE,CAAC;IAEvC,iBAAiB;IACjB,MAAM,iBAAiB,GAAG,MAAM,OAAO,CAAC,aAAa,EAAE,CAAC;IACxD,IAAI,iBAAiB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACnC,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,KAAK,MAAM,IAAI,IAAI,cAAc,EAAE,CAAC;YAClC,MAAM,QAAQ,GAAG,YAAY,CAAW,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC;YACrE,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;gBACrC,MAAM,EAAE,CAAC;gBACT,MAAM,CAAC,IAAI,CAAC,iBAAiB,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC;YAC3E,CAAC;QACH,CAAC;QACD,IAAI,MAAM,KAAK,CAAC,EAAE,CAAC;YACjB,MAAM,CAAC,IAAI,CAAC,6BAA6B,GAAG,UAAU,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED,iBAAiB;IACjB,MAAM,iBAAiB,GAAG,MAAM,OAAO,CAAC,aAAa,EAAE,CAAC;IACxD,IAAI,iBAAiB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACnC,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,KAAK,MAAM,IAAI,IAAI,cAAc,EAAE,CAAC;YAClC,MAAM,QAAQ,GAAG,YAAY,CAAW,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC;YACrE,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;gBACrC,MAAM,EAAE,CAAC;gBACT,MAAM,CAAC,IAAI,CAAC,iBAAiB,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC;YAC3E,CAAC;QACH,CAAC;QACD,IAAI,MAAM,KAAK,CAAC,EAAE,CAAC;YACjB,MAAM,CAAC,IAAI,CAAC,6BAA6B,GAAG,UAAU,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import type { FsAdapter } from './fs-adapter.js';
|
|
2
|
+
import type { Provider, Scenario, Run, Evaluation } from '../types/index.js';
|
|
3
|
+
export declare function makeProvider(overrides?: Partial<Provider>): Provider;
|
|
4
|
+
export declare function makeScenario(overrides?: Partial<Scenario>): Scenario;
|
|
5
|
+
export declare function makeRun(providerIdOrOverrides?: string | Partial<Run>, scenarioId?: string): Run;
|
|
6
|
+
export declare function makeEvaluation(overrides?: Partial<Evaluation>): Evaluation;
|
|
7
|
+
export interface MockFsState {
|
|
8
|
+
files: Map<string, string>;
|
|
9
|
+
dirs: Set<string>;
|
|
10
|
+
}
|
|
11
|
+
export declare function createMockFs(): {
|
|
12
|
+
fs: FsAdapter;
|
|
13
|
+
state: MockFsState;
|
|
14
|
+
};
|
|
15
|
+
//# sourceMappingURL=storage-test-helpers.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"storage-test-helpers.d.ts","sourceRoot":"","sources":["../../../src/server/services/storage-test-helpers.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,KAAK,EACV,QAAQ,EACR,QAAQ,EACR,GAAG,EACH,UAAU,EACX,MAAM,mBAAmB,CAAC;AAM3B,wBAAgB,YAAY,CAAC,SAAS,GAAE,OAAO,CAAC,QAAQ,CAAM,GAAG,QAAQ,CAaxE;AAED,wBAAgB,YAAY,CAAC,SAAS,GAAE,OAAO,CAAC,QAAQ,CAAM,GAAG,QAAQ,CAgBxE;AAED,wBAAgB,OAAO,CACrB,qBAAqB,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,EAC7C,UAAU,CAAC,EAAE,MAAM,GAClB,GAAG,CAqBL;AAED,wBAAgB,cAAc,CAAC,SAAS,GAAE,OAAO,CAAC,UAAU,CAAM,GAAG,UAAU,CA6B9E;AAID,MAAM,WAAW,WAAW;IAC1B,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC3B,IAAI,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;CACnB;AAED,wBAAgB,YAAY,IAAI;IAAE,EAAE,EAAE,SAAS,CAAC;IAAC,KAAK,EAAE,WAAW,CAAA;CAAE,CAyEpE"}
|
|
@@ -0,0 +1,151 @@
|
|
|
1
|
+
import { vi } from 'vitest';
|
|
2
|
+
// ─── Factory helpers ─────────────────────────────────────────────────
|
|
3
|
+
const MOCK_KEY = 'mock-value-for-testing';
|
|
4
|
+
export function makeProvider(overrides = {}) {
|
|
5
|
+
return {
|
|
6
|
+
id: 'provider-1',
|
|
7
|
+
name: 'Test Provider',
|
|
8
|
+
description: 'A test provider',
|
|
9
|
+
providerName: 'anthropic',
|
|
10
|
+
model: 'claude-sonnet-4-6',
|
|
11
|
+
apiKey: MOCK_KEY,
|
|
12
|
+
timeoutSeconds: 300,
|
|
13
|
+
createdAt: '2026-01-01T00:00:00.000Z',
|
|
14
|
+
updatedAt: '2026-01-01T00:00:00.000Z',
|
|
15
|
+
...overrides,
|
|
16
|
+
};
|
|
17
|
+
}
|
|
18
|
+
export function makeScenario(overrides = {}) {
|
|
19
|
+
return {
|
|
20
|
+
id: 'scenario-1',
|
|
21
|
+
name: 'Test Scenario',
|
|
22
|
+
category: 'planning',
|
|
23
|
+
prompt: 'Do something',
|
|
24
|
+
systemPrompt: '',
|
|
25
|
+
enabledTools: [],
|
|
26
|
+
expectedAnswer: 'done',
|
|
27
|
+
criticalRequirements: [],
|
|
28
|
+
gradingGuidelines: '',
|
|
29
|
+
scoringDimensions: [],
|
|
30
|
+
createdAt: '2026-01-01T00:00:00.000Z',
|
|
31
|
+
updatedAt: '2026-01-01T00:00:00.000Z',
|
|
32
|
+
...overrides,
|
|
33
|
+
};
|
|
34
|
+
}
|
|
35
|
+
export function makeRun(providerIdOrOverrides, scenarioId) {
|
|
36
|
+
const overrides = typeof providerIdOrOverrides === 'string'
|
|
37
|
+
? { providerId: providerIdOrOverrides, scenarioId: scenarioId ?? 'scenario-1' }
|
|
38
|
+
: providerIdOrOverrides ?? {};
|
|
39
|
+
return {
|
|
40
|
+
id: 'run-1',
|
|
41
|
+
providerId: 'provider-1',
|
|
42
|
+
scenarioId: 'scenario-1',
|
|
43
|
+
status: 'completed',
|
|
44
|
+
providerSnapshot: makeProvider(),
|
|
45
|
+
scenarioSnapshot: makeScenario(),
|
|
46
|
+
messages: [],
|
|
47
|
+
resultText: 'done',
|
|
48
|
+
totalCostUsd: 0.01,
|
|
49
|
+
durationMs: 1000,
|
|
50
|
+
numTurns: 3,
|
|
51
|
+
createdAt: '2026-01-01T00:00:00.000Z',
|
|
52
|
+
updatedAt: '2026-01-01T00:00:01.000Z',
|
|
53
|
+
...overrides,
|
|
54
|
+
};
|
|
55
|
+
}
|
|
56
|
+
export function makeEvaluation(overrides = {}) {
|
|
57
|
+
return {
|
|
58
|
+
id: 'eval-1',
|
|
59
|
+
runId: 'run-1',
|
|
60
|
+
status: 'completed',
|
|
61
|
+
evaluators: [],
|
|
62
|
+
rounds: [],
|
|
63
|
+
answerComparison: { matches: true, explanation: 'Good', similarity: 0.9 },
|
|
64
|
+
criticalResults: [],
|
|
65
|
+
setupCompliance: {
|
|
66
|
+
instructionCompliance: {
|
|
67
|
+
followed: [],
|
|
68
|
+
violated: [],
|
|
69
|
+
notApplicable: [],
|
|
70
|
+
overallCompliance: 1,
|
|
71
|
+
},
|
|
72
|
+
},
|
|
73
|
+
synthesis: {
|
|
74
|
+
dimensionScores: {},
|
|
75
|
+
weightedTotal: 8,
|
|
76
|
+
confidence: 0.9,
|
|
77
|
+
dissenting: [],
|
|
78
|
+
},
|
|
79
|
+
ledger: [],
|
|
80
|
+
totalCostUsd: 0.05,
|
|
81
|
+
createdAt: '2026-01-01T00:00:00.000Z',
|
|
82
|
+
updatedAt: '2026-01-01T00:00:02.000Z',
|
|
83
|
+
...overrides,
|
|
84
|
+
};
|
|
85
|
+
}
|
|
86
|
+
export function createMockFs() {
|
|
87
|
+
const state = {
|
|
88
|
+
files: new Map(),
|
|
89
|
+
dirs: new Set(),
|
|
90
|
+
};
|
|
91
|
+
const mockFs = {
|
|
92
|
+
mkdir: vi.fn((_dirPath) => {
|
|
93
|
+
state.dirs.add(_dirPath);
|
|
94
|
+
return Promise.resolve();
|
|
95
|
+
}),
|
|
96
|
+
writeFile: vi.fn((_filePath, data) => {
|
|
97
|
+
state.files.set(_filePath, data);
|
|
98
|
+
return Promise.resolve();
|
|
99
|
+
}),
|
|
100
|
+
readFile: vi.fn((filePath) => {
|
|
101
|
+
const data = state.files.get(filePath);
|
|
102
|
+
if (data === undefined) {
|
|
103
|
+
return Promise.reject(new Error(`ENOENT: no such file: ${filePath}`));
|
|
104
|
+
}
|
|
105
|
+
return Promise.resolve(data);
|
|
106
|
+
}),
|
|
107
|
+
readdir: vi.fn((dirPath) => {
|
|
108
|
+
const prefix = dirPath.endsWith('/') ? dirPath : dirPath + '/';
|
|
109
|
+
const entries = [];
|
|
110
|
+
for (const key of state.files.keys()) {
|
|
111
|
+
if (key.startsWith(prefix)) {
|
|
112
|
+
const rest = key.slice(prefix.length);
|
|
113
|
+
if (!rest.includes('/')) {
|
|
114
|
+
entries.push(rest);
|
|
115
|
+
}
|
|
116
|
+
}
|
|
117
|
+
}
|
|
118
|
+
return Promise.resolve(entries);
|
|
119
|
+
}),
|
|
120
|
+
unlink: vi.fn((filePath) => {
|
|
121
|
+
if (!state.files.has(filePath)) {
|
|
122
|
+
return Promise.reject(new Error(`ENOENT: no such file: ${filePath}`));
|
|
123
|
+
}
|
|
124
|
+
state.files.delete(filePath);
|
|
125
|
+
return Promise.resolve();
|
|
126
|
+
}),
|
|
127
|
+
rename: vi.fn((oldPath, newPath) => {
|
|
128
|
+
const data = state.files.get(oldPath);
|
|
129
|
+
if (data === undefined) {
|
|
130
|
+
return Promise.reject(new Error(`ENOENT: no such file: ${oldPath}`));
|
|
131
|
+
}
|
|
132
|
+
state.files.set(newPath, data);
|
|
133
|
+
state.files.delete(oldPath);
|
|
134
|
+
return Promise.resolve();
|
|
135
|
+
}),
|
|
136
|
+
stat: vi.fn((filePath) => {
|
|
137
|
+
if (!state.files.has(filePath)) {
|
|
138
|
+
return Promise.reject(new Error(`ENOENT: no such file: ${filePath}`));
|
|
139
|
+
}
|
|
140
|
+
return Promise.resolve({ isFile: () => true });
|
|
141
|
+
}),
|
|
142
|
+
access: vi.fn((filePath) => {
|
|
143
|
+
if (!state.files.has(filePath)) {
|
|
144
|
+
return Promise.reject(new Error(`ENOENT: no such file: ${filePath}`));
|
|
145
|
+
}
|
|
146
|
+
return Promise.resolve();
|
|
147
|
+
}),
|
|
148
|
+
};
|
|
149
|
+
return { fs: mockFs, state };
|
|
150
|
+
}
|
|
151
|
+
//# sourceMappingURL=storage-test-helpers.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"storage-test-helpers.js","sourceRoot":"","sources":["../../../src/server/services/storage-test-helpers.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAS5B,wEAAwE;AAExE,MAAM,QAAQ,GAAG,wBAAwB,CAAC;AAE1C,MAAM,UAAU,YAAY,CAAC,YAA+B,EAAE;IAC5D,OAAO;QACL,EAAE,EAAE,YAAY;QAChB,IAAI,EAAE,eAAe;QACrB,WAAW,EAAE,iBAAiB;QAC9B,YAAY,EAAE,WAAW;QACzB,KAAK,EAAE,mBAAmB;QAC1B,MAAM,EAAE,QAAQ;QAChB,cAAc,EAAE,GAAG;QACnB,SAAS,EAAE,0BAA0B;QACrC,SAAS,EAAE,0BAA0B;QACrC,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,YAA+B,EAAE;IAC5D,OAAO;QACL,EAAE,EAAE,YAAY;QAChB,IAAI,EAAE,eAAe;QACrB,QAAQ,EAAE,UAAU;QACpB,MAAM,EAAE,cAAc;QACtB,YAAY,EAAE,EAAE;QAChB,YAAY,EAAE,EAAE;QAChB,cAAc,EAAE,MAAM;QACtB,oBAAoB,EAAE,EAAE;QACxB,iBAAiB,EAAE,EAAE;QACrB,iBAAiB,EAAE,EAAE;QACrB,SAAS,EAAE,0BAA0B;QACrC,SAAS,EAAE,0BAA0B;QACrC,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,OAAO,CACrB,qBAA6C,EAC7C,UAAmB;IAEnB,MAAM,SAAS,GACb,OAAO,qBAAqB,KAAK,QAAQ;QACvC,CAAC,CAAC,EAAE,UAAU,EAAE,qBAAqB,EAAE,UAAU,EAAE,UAAU,IAAI,YAAY,EAAE;QAC/E,CAAC,CAAC,qBAAqB,IAAI,EAAE,CAAC;IAClC,OAAO;QACL,EAAE,EAAE,OAAO;QACX,UAAU,EAAE,YAAY;QACxB,UAAU,EAAE,YAAY;QACxB,MAAM,EAAE,WAAW;QACnB,gBAAgB,EAAE,YAAY,EAAE;QAChC,gBAAgB,EAAE,YAAY,EAAE;QAChC,QAAQ,EAAE,EAAE;QACZ,UAAU,EAAE,MAAM;QAClB,YAAY,EAAE,IAAI;QAClB,UAAU,EAAE,IAAI;QAChB,QAAQ,EAAE,CAAC;QACX,SAAS,EAAE,0BAA0B;QACrC,SAAS,EAAE,0BAA0B;QACrC,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,YAAiC,EAAE;IAChE,OAAO;QACL,EAAE,EAAE,QAAQ;QACZ,KAAK,EAAE,OAAO;QACd,MAAM,EAAE,WAAW;QACnB,UAAU,EAAE,EAAE;QACd,MAAM,EAAE,EAAE;QACV,gBAAgB,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,MAAM,EAAE,UAAU,EAAE,GAAG,EAAE;QACzE,eAAe,EAAE,EAAE;QACnB,eAAe,EAAE;YACf,qBAAqB,EAAE;gBACrB,QAAQ,EAAE,EAAE;gBACZ,QAAQ,EAAE,EAAE;gBACZ,aAAa,EAAE,EAAE;gBACjB,iBAAiB,EAAE,CAAC;aACrB;SACF;QACD,SAAS,EAAE;YACT,eAAe,EAAE,EAAE;YACnB,aAAa,EAAE,CAAC;YAChB,UAAU,EAAE,GAAG;YACf,UAAU,EAAE,EAAE;SACf;QACD,MAAM,EAAE,EAAE;QACV,YAAY,EAAE,IAAI;QAClB,SAAS,EAAE,0BAA0B;QACrC,SAAS,EAAE,0BAA0B;QACrC,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AASD,MAAM,UAAU,YAAY;IAC1B,MAAM,KAAK,GAAgB;QACzB,KAAK,EAAE,IAAI,GAAG,EAAE;QAChB,IAAI,EAAE,IAAI,GAAG,EAAE;KAChB,CAAC;IAEF,MAAM,MAAM,GAAc;QACxB,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,QAAgB,EAAE,EAAE;YAChC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACzB,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;QAC3B,CAAC,CAAC;QAEF,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,SAAiB,EAAE,IAAY,EAAE,EAAE;YACnD,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YACjC,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;QAC3B,CAAC,CAAC;QAEF,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,QAAgB,EAAE,EAAE;YACnC,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACvC,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;gBACvB,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,yBAAyB,QAAQ,EAAE,CAAC,CAAC,CAAC;YACxE,CAAC;YACD,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC/B,CAAC,CAAC;QAEF,OAAO,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,OAAe,EAAE,EAAE;YACjC,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,GAAG,GAAG,CAAC;YAC/D,MAAM,OAAO,GAAa,EAAE,CAAC;YAC7B,KAAK,MAAM,GAAG,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC;gBACrC,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC3B,MAAM,IAAI,GAAG,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;oBACtC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;wBACxB,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACrB,CAAC;gBACH,CAAC;YACH,CAAC;YACD,OAAO,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAClC,CAAC,CAAC;QAEF,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,QAAgB,EAAE,EAAE;YACjC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC/B,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,yBAAyB,QAAQ,EAAE,CAAC,CAAC,CAAC;YACxE,CAAC;YACD,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC7B,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;QAC3B,CAAC,CAAC;QAEF,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,OAAe,EAAE,OAAe,EAAE,EAAE;YACjD,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACtC,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;gBACvB,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,yBAAyB,OAAO,EAAE,CAAC,CAAC,CAAC;YACvE,CAAC;YACD,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAC/B,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAC5B,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;QAC3B,CAAC,CAAC;QAEF,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,QAAgB,EAAE,EAAE;YAC/B,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC/B,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,yBAAyB,QAAQ,EAAE,CAAC,CAAC,CAAC;YACxE,CAAC;YACD,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;QACjD,CAAC,CAAC;QAEF,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,QAAgB,EAAE,EAAE;YACjC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC/B,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,yBAAyB,QAAQ,EAAE,CAAC,CAAC,CAAC;YACxE,CAAC;YACD,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;QAC3B,CAAC,CAAC;KACH,CAAC;IAEF,OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;AAC/B,CAAC"}
|
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
import type { IStorage, ProviderFilter, ScenarioFilter, RunFilter, EvaluationFilter } from '../interfaces/storage.js';
|
|
2
|
+
import type { Provider, Scenario, Run, Evaluation } from '../types/index.js';
|
|
3
|
+
import type { FsAdapter } from './fs-adapter.js';
|
|
4
|
+
export declare class JsonFileStorage implements IStorage {
|
|
5
|
+
private readonly basePath;
|
|
6
|
+
private readonly fs;
|
|
7
|
+
private initialized;
|
|
8
|
+
private readonly entities;
|
|
9
|
+
constructor(basePath?: string, fsAdapter?: FsAdapter);
|
|
10
|
+
private ensureInit;
|
|
11
|
+
private entityDir;
|
|
12
|
+
private entityPath;
|
|
13
|
+
private getEntity;
|
|
14
|
+
private listEntities;
|
|
15
|
+
private saveEntity;
|
|
16
|
+
private deleteEntity;
|
|
17
|
+
getProvider(id: string): Promise<Provider | undefined>;
|
|
18
|
+
listProviders(filter?: ProviderFilter): Promise<readonly Provider[]>;
|
|
19
|
+
saveProvider(provider: Provider): Promise<void>;
|
|
20
|
+
deleteProvider(id: string): Promise<boolean>;
|
|
21
|
+
getScenario(id: string): Promise<Scenario | undefined>;
|
|
22
|
+
listScenarios(filter?: ScenarioFilter): Promise<readonly Scenario[]>;
|
|
23
|
+
saveScenario(scenario: Scenario): Promise<void>;
|
|
24
|
+
deleteScenario(id: string): Promise<boolean>;
|
|
25
|
+
private normalizeRun;
|
|
26
|
+
getRun(id: string): Promise<Run | undefined>;
|
|
27
|
+
listRuns(filter?: RunFilter): Promise<readonly Run[]>;
|
|
28
|
+
saveRun(run: Run): Promise<void>;
|
|
29
|
+
deleteRun(id: string): Promise<boolean>;
|
|
30
|
+
getEvaluation(id: string): Promise<Evaluation | undefined>;
|
|
31
|
+
listEvaluations(filter?: EvaluationFilter): Promise<readonly Evaluation[]>;
|
|
32
|
+
saveEvaluation(evaluation: Evaluation): Promise<void>;
|
|
33
|
+
deleteEvaluation(id: string): Promise<boolean>;
|
|
34
|
+
}
|
|
35
|
+
//# sourceMappingURL=storage.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"storage.d.ts","sourceRoot":"","sources":["../../../src/server/services/storage.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,cAAc,EAAE,SAAS,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AACtH,OAAO,KAAK,EACV,QAAQ,EACR,QAAQ,EACR,GAAG,EACH,UAAU,EACX,MAAM,mBAAmB,CAAC;AAE3B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAQjD,qBAAa,eAAgB,YAAW,QAAQ;IAC9C,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAS;IAClC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAY;IAC/B,OAAO,CAAC,WAAW,CAAS;IAE5B,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAKvB;gBAEU,QAAQ,CAAC,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,SAAS;YAgDtC,UAAU;IAwBxB,OAAO,CAAC,SAAS;IAIjB,OAAO,CAAC,UAAU;YAMJ,SAAS;YAWT,YAAY;YA6BZ,UAAU;YAeV,YAAY;IAc1B,WAAW,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,GAAG,SAAS,CAAC;IAItD,aAAa,CAAC,MAAM,CAAC,EAAE,cAAc,GAAG,OAAO,CAAC,SAAS,QAAQ,EAAE,CAAC;IAKpE,YAAY,CAAC,QAAQ,EAAE,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC;IAK/C,cAAc,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAM5C,WAAW,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,GAAG,SAAS,CAAC;IAItD,aAAa,CAAC,MAAM,CAAC,EAAE,cAAc,GAAG,OAAO,CAAC,SAAS,QAAQ,EAAE,CAAC;IAKpE,YAAY,CAAC,QAAQ,EAAE,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC;IAK/C,cAAc,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAK5C,OAAO,CAAC,YAAY;IAed,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,GAAG,SAAS,CAAC;IAI5C,QAAQ,CAAC,MAAM,CAAC,EAAE,SAAS,GAAG,OAAO,CAAC,SAAS,GAAG,EAAE,CAAC;IAI3D,OAAO,CAAC,GAAG,EAAE,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC;IAChC,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAGvC,aAAa,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,GAAG,SAAS,CAAC;IAC1D,eAAe,CAAC,MAAM,CAAC,EAAE,gBAAgB,GAAG,OAAO,CAAC,SAAS,UAAU,EAAE,CAAC;IAC1E,cAAc,CAAC,UAAU,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IACrD,gBAAgB,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;CAC/C"}
|
|
@@ -0,0 +1,219 @@
|
|
|
1
|
+
import path from 'node:path';
|
|
2
|
+
import { randomUUID } from 'node:crypto';
|
|
3
|
+
import { defaultFs } from './fs-adapter.js';
|
|
4
|
+
export class JsonFileStorage {
|
|
5
|
+
basePath;
|
|
6
|
+
fs;
|
|
7
|
+
initialized = false;
|
|
8
|
+
entities;
|
|
9
|
+
constructor(basePath, fsAdapter) {
|
|
10
|
+
this.basePath = basePath ?? path.join(process.cwd(), '.model-test-bench');
|
|
11
|
+
this.fs = fsAdapter ?? defaultFs;
|
|
12
|
+
this.entities = {
|
|
13
|
+
providers: {
|
|
14
|
+
subdir: 'providers',
|
|
15
|
+
sensitive: true,
|
|
16
|
+
matchesFilter: (s, f) => {
|
|
17
|
+
if (!f)
|
|
18
|
+
return true;
|
|
19
|
+
if (f.provider && s.providerName !== f.provider)
|
|
20
|
+
return false;
|
|
21
|
+
if (f.model && s.model !== f.model)
|
|
22
|
+
return false;
|
|
23
|
+
return true;
|
|
24
|
+
},
|
|
25
|
+
},
|
|
26
|
+
scenarios: {
|
|
27
|
+
subdir: 'scenarios/custom',
|
|
28
|
+
sensitive: false,
|
|
29
|
+
matchesFilter: (s, f) => {
|
|
30
|
+
if (!f)
|
|
31
|
+
return true;
|
|
32
|
+
if (f.category && s.category !== f.category)
|
|
33
|
+
return false;
|
|
34
|
+
return true;
|
|
35
|
+
},
|
|
36
|
+
},
|
|
37
|
+
runs: {
|
|
38
|
+
subdir: 'runs',
|
|
39
|
+
sensitive: false,
|
|
40
|
+
matchesFilter: (r, f) => {
|
|
41
|
+
if (!f)
|
|
42
|
+
return true;
|
|
43
|
+
if (f.providerId && r.providerId !== f.providerId)
|
|
44
|
+
return false;
|
|
45
|
+
if (f.scenarioId && r.scenarioId !== f.scenarioId)
|
|
46
|
+
return false;
|
|
47
|
+
if (f.status && r.status !== f.status)
|
|
48
|
+
return false;
|
|
49
|
+
return true;
|
|
50
|
+
},
|
|
51
|
+
},
|
|
52
|
+
evaluations: {
|
|
53
|
+
subdir: 'evaluations',
|
|
54
|
+
sensitive: false,
|
|
55
|
+
matchesFilter: (e, f) => {
|
|
56
|
+
if (!f)
|
|
57
|
+
return true;
|
|
58
|
+
if (f.runId && e.runId !== f.runId)
|
|
59
|
+
return false;
|
|
60
|
+
if (f.status && e.status !== f.status)
|
|
61
|
+
return false;
|
|
62
|
+
return true;
|
|
63
|
+
},
|
|
64
|
+
},
|
|
65
|
+
};
|
|
66
|
+
}
|
|
67
|
+
async ensureInit() {
|
|
68
|
+
if (this.initialized)
|
|
69
|
+
return;
|
|
70
|
+
// Migrate legacy 'setups' directory to 'providers'
|
|
71
|
+
const legacyDir = path.join(this.basePath, 'setups');
|
|
72
|
+
const newDir = path.join(this.basePath, 'providers');
|
|
73
|
+
try {
|
|
74
|
+
await this.fs.access(legacyDir);
|
|
75
|
+
try {
|
|
76
|
+
await this.fs.access(newDir);
|
|
77
|
+
}
|
|
78
|
+
catch {
|
|
79
|
+
// New dir doesn't exist — rename legacy dir
|
|
80
|
+
await this.fs.rename(legacyDir, newDir);
|
|
81
|
+
}
|
|
82
|
+
}
|
|
83
|
+
catch {
|
|
84
|
+
// Legacy dir doesn't exist — nothing to migrate
|
|
85
|
+
}
|
|
86
|
+
for (const cfg of Object.values(this.entities)) {
|
|
87
|
+
await this.fs.mkdir(path.join(this.basePath, cfg.subdir), { recursive: true });
|
|
88
|
+
}
|
|
89
|
+
this.initialized = true;
|
|
90
|
+
}
|
|
91
|
+
entityDir(subdir) {
|
|
92
|
+
return path.join(this.basePath, subdir);
|
|
93
|
+
}
|
|
94
|
+
entityPath(subdir, id) {
|
|
95
|
+
return path.join(this.basePath, subdir, `${id}.json`);
|
|
96
|
+
}
|
|
97
|
+
// ─── Generic CRUD ──────────────────────────────────────────────────
|
|
98
|
+
async getEntity(subdir, id) {
|
|
99
|
+
await this.ensureInit();
|
|
100
|
+
const filePath = this.entityPath(subdir, id);
|
|
101
|
+
try {
|
|
102
|
+
const raw = await this.fs.readFile(filePath, 'utf-8');
|
|
103
|
+
return JSON.parse(raw);
|
|
104
|
+
}
|
|
105
|
+
catch {
|
|
106
|
+
return undefined;
|
|
107
|
+
}
|
|
108
|
+
}
|
|
109
|
+
async listEntities(subdir, matchesFilter, filter) {
|
|
110
|
+
await this.ensureInit();
|
|
111
|
+
const dir = this.entityDir(subdir);
|
|
112
|
+
let files;
|
|
113
|
+
try {
|
|
114
|
+
files = await this.fs.readdir(dir);
|
|
115
|
+
}
|
|
116
|
+
catch {
|
|
117
|
+
return [];
|
|
118
|
+
}
|
|
119
|
+
const results = [];
|
|
120
|
+
for (const file of files) {
|
|
121
|
+
if (!file.endsWith('.json'))
|
|
122
|
+
continue;
|
|
123
|
+
try {
|
|
124
|
+
const raw = await this.fs.readFile(path.join(dir, file), 'utf-8');
|
|
125
|
+
const entity = JSON.parse(raw);
|
|
126
|
+
if (matchesFilter(entity, filter)) {
|
|
127
|
+
results.push(entity);
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
catch {
|
|
131
|
+
// Skip corrupt files
|
|
132
|
+
}
|
|
133
|
+
}
|
|
134
|
+
return results;
|
|
135
|
+
}
|
|
136
|
+
async saveEntity(subdir, entity, sensitive) {
|
|
137
|
+
await this.ensureInit();
|
|
138
|
+
const dir = this.entityDir(subdir);
|
|
139
|
+
const tmpPath = path.join(dir, `.tmp-${randomUUID()}.json`);
|
|
140
|
+
const finalPath = this.entityPath(subdir, entity.id);
|
|
141
|
+
const data = JSON.stringify(entity, null, 2);
|
|
142
|
+
const opts = sensitive ? { mode: 0o600 } : undefined;
|
|
143
|
+
await this.fs.writeFile(tmpPath, data, opts);
|
|
144
|
+
await this.fs.rename(tmpPath, finalPath);
|
|
145
|
+
}
|
|
146
|
+
async deleteEntity(subdir, id) {
|
|
147
|
+
await this.ensureInit();
|
|
148
|
+
const filePath = this.entityPath(subdir, id);
|
|
149
|
+
try {
|
|
150
|
+
await this.fs.access(filePath);
|
|
151
|
+
await this.fs.unlink(filePath);
|
|
152
|
+
return true;
|
|
153
|
+
}
|
|
154
|
+
catch {
|
|
155
|
+
return false;
|
|
156
|
+
}
|
|
157
|
+
}
|
|
158
|
+
// ─── Providers ──────────────────────────────────────────────────────
|
|
159
|
+
getProvider(id) {
|
|
160
|
+
return this.getEntity(this.entities.providers.subdir, id);
|
|
161
|
+
}
|
|
162
|
+
listProviders(filter) {
|
|
163
|
+
const cfg = this.entities.providers;
|
|
164
|
+
return this.listEntities(cfg.subdir, cfg.matchesFilter, filter);
|
|
165
|
+
}
|
|
166
|
+
saveProvider(provider) {
|
|
167
|
+
const cfg = this.entities.providers;
|
|
168
|
+
return this.saveEntity(cfg.subdir, provider, cfg.sensitive);
|
|
169
|
+
}
|
|
170
|
+
deleteProvider(id) {
|
|
171
|
+
return this.deleteEntity(this.entities.providers.subdir, id);
|
|
172
|
+
}
|
|
173
|
+
// ─── Scenarios ─────────────────────────────────────────────────────
|
|
174
|
+
getScenario(id) {
|
|
175
|
+
return this.getEntity(this.entities.scenarios.subdir, id);
|
|
176
|
+
}
|
|
177
|
+
listScenarios(filter) {
|
|
178
|
+
const cfg = this.entities.scenarios;
|
|
179
|
+
return this.listEntities(cfg.subdir, cfg.matchesFilter, filter);
|
|
180
|
+
}
|
|
181
|
+
saveScenario(scenario) {
|
|
182
|
+
const cfg = this.entities.scenarios;
|
|
183
|
+
return this.saveEntity(cfg.subdir, scenario, cfg.sensitive);
|
|
184
|
+
}
|
|
185
|
+
deleteScenario(id) {
|
|
186
|
+
return this.deleteEntity(this.entities.scenarios.subdir, id);
|
|
187
|
+
}
|
|
188
|
+
// ─── Runs ──────────────────────────────────────────────────────────
|
|
189
|
+
normalizeRun(raw) {
|
|
190
|
+
// Accept both old and new field names for backward compat
|
|
191
|
+
const r = raw;
|
|
192
|
+
if ('setupId' in r && !('providerId' in r)) {
|
|
193
|
+
return {
|
|
194
|
+
...raw,
|
|
195
|
+
providerId: r['setupId'],
|
|
196
|
+
providerSnapshot: (r['setupSnapshot'] ?? raw.providerSnapshot),
|
|
197
|
+
reviewerProviderIds: (r['reviewerSetupIds'] ?? raw.reviewerProviderIds),
|
|
198
|
+
reviewerProviderSnapshots: (r['reviewerSetupSnapshots'] ?? raw.reviewerProviderSnapshots),
|
|
199
|
+
};
|
|
200
|
+
}
|
|
201
|
+
return raw;
|
|
202
|
+
}
|
|
203
|
+
async getRun(id) {
|
|
204
|
+
const raw = await this.getEntity(this.entities.runs.subdir, id);
|
|
205
|
+
return raw ? this.normalizeRun(raw) : undefined;
|
|
206
|
+
}
|
|
207
|
+
async listRuns(filter) {
|
|
208
|
+
const results = await this.listEntities(this.entities.runs.subdir, this.entities.runs.matchesFilter, filter);
|
|
209
|
+
return results.map((r) => this.normalizeRun(r));
|
|
210
|
+
}
|
|
211
|
+
saveRun(run) { return this.saveEntity(this.entities.runs.subdir, run, this.entities.runs.sensitive); }
|
|
212
|
+
deleteRun(id) { return this.deleteEntity(this.entities.runs.subdir, id); }
|
|
213
|
+
// ─── Evaluations ───────────────────────────────────────────────────
|
|
214
|
+
getEvaluation(id) { return this.getEntity(this.entities.evaluations.subdir, id); }
|
|
215
|
+
listEvaluations(filter) { return this.listEntities(this.entities.evaluations.subdir, this.entities.evaluations.matchesFilter, filter); }
|
|
216
|
+
saveEvaluation(evaluation) { return this.saveEntity(this.entities.evaluations.subdir, evaluation, this.entities.evaluations.sensitive); }
|
|
217
|
+
deleteEvaluation(id) { return this.deleteEntity(this.entities.evaluations.subdir, id); }
|
|
218
|
+
}
|
|
219
|
+
//# sourceMappingURL=storage.js.map
|