mitchell/lmc
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: renderStream to use switch

Browse source
This commit is contained in:
mitchell 2026-04-17 03:28:51 -04:00
parent 4716712a92
commit ebbdda2853
1 changed files with 43 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

70
src/render/index.ts
View file

@ -60,33 +60,49 @@ export default class StreamRenderer {
this.resetCounts();
for await (const part of stream) {
if (part.type === "reasoning-start") {
if (supportsColorStderr)
this.write(`${styles.dim.open}${styles.italic.open}`, process.stderr);
this.write("\n", process.stderr);
} else if (part.type === "reasoning-delta") {
this.write(part.text, process.stderr);
reasoningWriter.write(part.text);
this.checkContainment(part.text, process.stderr);
} else if (part.type === "reasoning-end") {
if (supportsColorStderr)
this.write(
`${styles.italic.close}${styles.dim.close}`,
process.stderr,
);
this.moveCursor(-this.lineCount - 1, process.stderr);
this.clearScreenDown(process.stderr);
this.resetCounts();
reasoningWriter.end();
} else if (part.type === "text-start") {
this.write("\n");
} else if (part.type === "text-delta") {
this.write(part.text);
this.checkContainment(part.text);
} else if (part.type === "text-end") {
this.moveCursor(-this.lineCount - 1);
this.clearScreenDown(process.stderr);
this.resetCounts();
switch (part.type) {
case "reasoning-start": {
if (supportsColorStderr)
this.write(
`${styles.dim.open}${styles.italic.open}`,
process.stderr,
);
this.write("\n", process.stderr);
break;
}
case "reasoning-delta": {
this.write(part.text, process.stderr);
this.checkContainment(part.text, process.stderr);
reasoningWriter.write(part.text);
break;
}
case "reasoning-end": {
if (supportsColorStderr)
this.write(
`${styles.italic.close}${styles.dim.close}`,
process.stderr,
);
this.moveCursor(-this.lineCount - 1, process.stderr);
this.clearScreenDown(process.stderr);
this.resetCounts();
reasoningWriter.end();
break;
}
case "text-start": {
this.write("\n");
break;
}
case "text-delta": {
this.write(part.text);
this.checkContainment(part.text);
break;
}
case "text-end": {
this.moveCursor(-this.lineCount - 1);
this.clearScreenDown(process.stderr);
this.resetCounts();
break;
}
}
while (!this.memOk) await Bun.sleep(10);