rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Auto-complete code blocks during streaming

Browse Source
This commit is contained in:
Cohee
2023-10-20 23:52:23 +03:00
parent e4cb916dff
commit 7f8df9533b
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
public/script.js
View File

@ -2185,11 +2185,12 @@ class StreamingProcessor {
let processedText = cleanUpMessage(text, isImpersonate, isContinue, !isFinal); let processedText = cleanUpMessage(text, isImpersonate, isContinue, !isFinal);
// Predict unbalanced asterisks / quotes during streaming // Predict unbalanced asterisks / quotes during streaming
const charsToBalance = ['*', '"']; const charsToBalance = ['*', '"', '```'];
for (const char of charsToBalance) { for (const char of charsToBalance) {
if (!isFinal && isOdd(countOccurrences(processedText, char))) { if (!isFinal && isOdd(countOccurrences(processedText, char))) {
// Add character at the end to balance it // Add character at the end to balance it
processedText = processedText.trimEnd() + char; const separator = char.length > 1 ? '\n' : '';
processedText = processedText.trimEnd() + separator + char;
} }
} }

2
public/scripts/utils.js
View File

@ -498,7 +498,7 @@ export function countOccurrences(string, character) {
let count = 0; let count = 0;
for (let i = 0; i < string.length; i++) { for (let i = 0; i < string.length; i++) {
if (string[i] === character) { if (string.substring(i, i + character.length) === character) {
count++; count++;
} }
} }