Enable regex stages by default

This commit is contained in:
Youzini-afk
2026-04-07 01:50:11 +08:00
parent 4fb45b4ce2
commit 949fdff78e
3 changed files with 58 additions and 48 deletions

View File

@@ -189,12 +189,12 @@ export const DEFAULT_TASK_PROFILE_TEMPLATES = {
"character": true "character": true
}, },
"stages": { "stages": {
"input.userMessage": false, "input.userMessage": true,
"input.recentMessages": false, "input.recentMessages": true,
"input.candidateText": false, "input.candidateText": true,
"input.finalPrompt": false, "input.finalPrompt": true,
"output.rawResponse": false, "output.rawResponse": true,
"output.beforeParse": false, "output.beforeParse": true,
"input": true, "input": true,
"output": true "output": true
}, },
@@ -393,12 +393,12 @@ export const DEFAULT_TASK_PROFILE_TEMPLATES = {
"character": true "character": true
}, },
"stages": { "stages": {
"input.userMessage": false, "input.userMessage": true,
"input.recentMessages": false, "input.recentMessages": true,
"input.candidateText": false, "input.candidateText": true,
"input.finalPrompt": false, "input.finalPrompt": true,
"output.rawResponse": false, "output.rawResponse": true,
"output.beforeParse": false, "output.beforeParse": true,
"input": true, "input": true,
"output": true "output": true
}, },
@@ -573,12 +573,12 @@ export const DEFAULT_TASK_PROFILE_TEMPLATES = {
"character": true "character": true
}, },
"stages": { "stages": {
"input.userMessage": false, "input.userMessage": true,
"input.recentMessages": false, "input.recentMessages": true,
"input.candidateText": false, "input.candidateText": true,
"input.finalPrompt": false, "input.finalPrompt": true,
"output.rawResponse": false, "output.rawResponse": true,
"output.beforeParse": false, "output.beforeParse": true,
"input": true, "input": true,
"output": true "output": true
}, },
@@ -765,12 +765,12 @@ export const DEFAULT_TASK_PROFILE_TEMPLATES = {
"character": true "character": true
}, },
"stages": { "stages": {
"input.userMessage": false, "input.userMessage": true,
"input.recentMessages": false, "input.recentMessages": true,
"input.candidateText": false, "input.candidateText": true,
"input.finalPrompt": false, "input.finalPrompt": true,
"output.rawResponse": false, "output.rawResponse": true,
"output.beforeParse": false, "output.beforeParse": true,
"input": true, "input": true,
"output": true "output": true
}, },
@@ -969,12 +969,12 @@ export const DEFAULT_TASK_PROFILE_TEMPLATES = {
"character": true "character": true
}, },
"stages": { "stages": {
"input.userMessage": false, "input.userMessage": true,
"input.recentMessages": false, "input.recentMessages": true,
"input.candidateText": false, "input.candidateText": true,
"input.finalPrompt": false, "input.finalPrompt": true,
"output.rawResponse": false, "output.rawResponse": true,
"output.beforeParse": false, "output.beforeParse": true,
"input": true, "input": true,
"output": true "output": true
}, },
@@ -1185,12 +1185,12 @@ export const DEFAULT_TASK_PROFILE_TEMPLATES = {
"character": true "character": true
}, },
"stages": { "stages": {
"input.userMessage": false, "input.userMessage": true,
"input.recentMessages": false, "input.recentMessages": true,
"input.candidateText": false, "input.candidateText": true,
"input.finalPrompt": false, "input.finalPrompt": true,
"output.rawResponse": false, "output.rawResponse": true,
"output.beforeParse": false, "output.beforeParse": true,
"input": true, "input": true,
"output": true "output": true
}, },

View File

@@ -617,6 +617,15 @@ const TASK_REGEX_STAGE_GROUPS = Object.freeze({
]), ]),
}); });
const DEFAULT_TASK_REGEX_STAGES = Object.freeze({
"input.userMessage": true,
"input.recentMessages": true,
"input.candidateText": true,
"input.finalPrompt": true,
"output.rawResponse": true,
"output.beforeParse": true,
});
function normalizeRegexStageKey(stageKey = "") { function normalizeRegexStageKey(stageKey = "") {
const normalized = String(stageKey || "").trim(); const normalized = String(stageKey || "").trim();
return TASK_REGEX_STAGE_ALIAS_MAP[normalized] || normalized; return TASK_REGEX_STAGE_ALIAS_MAP[normalized] || normalized;
@@ -890,14 +899,7 @@ function createFallbackDefaultTaskProfile(taskType) {
preset: true, preset: true,
character: true, character: true,
}, },
stages: normalizeTaskRegexStages({ stages: normalizeTaskRegexStages(DEFAULT_TASK_REGEX_STAGES),
"input.userMessage": false,
"input.recentMessages": false,
"input.candidateText": false,
"input.finalPrompt": false,
"output.rawResponse": false,
"output.beforeParse": false,
}),
localRules: [], localRules: [],
}, },
metadata: { metadata: {

View File

@@ -199,19 +199,27 @@ try {
defaultProfiles.extract?.profiles?.[0]?.regex?.stages || {}; defaultProfiles.extract?.profiles?.[0]?.regex?.stages || {};
assert.equal( assert.equal(
isTaskRegexStageEnabled(defaultExtractStages, "input.finalPrompt"), isTaskRegexStageEnabled(defaultExtractStages, "input.finalPrompt"),
false, true,
); );
assert.equal( assert.equal(
isTaskRegexStageEnabled(defaultExtractStages, "input.userMessage"), isTaskRegexStageEnabled(defaultExtractStages, "input.userMessage"),
false, true,
); );
assert.equal( assert.equal(
isTaskRegexStageEnabled(defaultExtractStages, "input.recentMessages"), isTaskRegexStageEnabled(defaultExtractStages, "input.recentMessages"),
false, true,
); );
assert.equal( assert.equal(
isTaskRegexStageEnabled(defaultExtractStages, "input.candidateText"), isTaskRegexStageEnabled(defaultExtractStages, "input.candidateText"),
false, true,
);
assert.equal(
isTaskRegexStageEnabled(defaultExtractStages, "output.rawResponse"),
true,
);
assert.equal(
isTaskRegexStageEnabled(defaultExtractStages, "output.beforeParse"),
true,
); );
const normalizedLegacyOnlyProfile = normalizeTaskProfile( const normalizedLegacyOnlyProfile = normalizeTaskProfile(