diff --git a/kipina-codebench/benchmark.mjs b/kipina-codebench/benchmark.mjs index 274849d..27af48f 100644 --- a/kipina-codebench/benchmark.mjs +++ b/kipina-codebench/benchmark.mjs @@ -398,6 +398,7 @@ async function runPipeline(model, scenario, round = 1) { let code = fileResp.text .replace(/^```(?:go|golang)?\s*\n?/m, '').replace(/\n?```\s*$/m, '') + .replace(/^===\s*\S+\s*===\s*\n?/m, '') .replace(/^(?:Here|Sure|Below|This|The|I )[\s\S]*?(?=package\s)/m, '') .trim(); @@ -449,6 +450,7 @@ async function runPipeline(model, scenario, round = 1) { timings.push(fixResp); let fixed = fixResp.text .replace(/^```(?:go|golang)?\s*\n?/m, '').replace(/\n?```\s*$/m, '') + .replace(/^===\s*\S+\s*===\s*\n?/m, '') .replace(/^(?:Here|Sure|Below|This|The|I )[\s\S]*?(?=package\s)/m, '') .trim(); if (fixed) {