Skip to content

Commit

Permalink
refactor
Browse files Browse the repository at this point in the history
  • Loading branch information
mdjastrzebski committed Dec 7, 2024
1 parent 423459a commit dd33a7f
Showing 1 changed file with 9 additions and 9 deletions.
18 changes: 9 additions & 9 deletions src/commands/chat/streaming.ts
Original file line number Diff line number Diff line change
Expand Up @@ -12,21 +12,21 @@ let frameIndex = 0;

export function streamingStart(text: string) {
const lines = text.trimEnd().split('\n');
const readyLines = lines.slice(0, -1);
outputtedLines = readyLines;
if (readyLines.length > 0) {
process.stdout.write(`${CLEAR_LINE}${readyLines.join('\n')}\n`);
const linesToPrint = lines.slice(0, -1);
if (linesToPrint.length > 0) {
process.stdout.write(`${CLEAR_LINE}${linesToPrint.join('\n')}\n`);
}

outputtedLines = linesToPrint;
currentLine = lines[lines.length - 1];
startSpinner();
}

export function streamingUpdate(text: string) {
const lines = text.trimEnd().split('\n');
const readyLines = lines.slice(outputtedLines.length, -1);
if (readyLines.length > 0) {
process.stdout.write(`${CLEAR_LINE}${readyLines.join('\n')}\n`);
const linesToPrint = lines.slice(outputtedLines.length, -1);
if (linesToPrint.length > 0) {
process.stdout.write(`${CLEAR_LINE}${linesToPrint.join('\n')}\n`);
}

outputtedLines = lines.slice(0, -1);
Expand All @@ -37,8 +37,8 @@ export function streamingUpdate(text: string) {
export function streamingFinish(text: string) {
clearInterval(intervalRef);
const lines = text.trimEnd().split('\n');
const readyLines = lines.slice(outputtedLines.length);
process.stdout.write(`${CLEAR_LINE}${readyLines.join('\n')}\n`);
const linesToPrint = lines.slice(outputtedLines.length);
process.stdout.write(`${CLEAR_LINE}${linesToPrint.join('\n')}\n`);
}

export function streamingClear() {
Expand Down

0 comments on commit dd33a7f

Please sign in to comment.