OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-05-31 11:00:49 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

chore: format

Browse Source
This commit is contained in:
Timothy J. Baek 2024-08-14 16:39:02 +02:00
parent 9f0c9d973c
commit 04e2b6e2bd
2 changed files with 96 additions and 89 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/lib/components/chat/Messages/ResponseMessage.svelte
View File

@ -159,7 +159,7 @@
const res = await synthesizeOpenAISpeech( const res = await synthesizeOpenAISpeech(
localStorage.token, localStorage.token,
$settings?.audio?.tts?.defaultVoice === $config.audio.tts.voice $settings?.audio?.tts?.defaultVoice === $config.audio.tts.voice
? $settings?.audio?.tts?.voice ?? $config?.audio?.tts?.voice ? ($settings?.audio?.tts?.voice ?? $config?.audio?.tts?.voice)
: $config?.audio?.tts?.voice, : $config?.audio?.tts?.voice,
sentence sentence
).catch((error) => { ).catch((error) => {
@ -766,7 +766,7 @@
100 100
) / 100 ) / 100
} tokens` ?? 'N/A' } tokens` ?? 'N/A'
}<br/> }<br/>
prompt_token/s: ${ prompt_token/s: ${
Math.round( Math.round(
((message.info.prompt_eval_count ?? 0) / ((message.info.prompt_eval_count ?? 0) /

181
src/lib/utils/katex-extension.ts
View File

@ -1,15 +1,15 @@
import katex from 'katex'; import katex from 'katex';
const DELIMITER_LIST = [ const DELIMITER_LIST = [
{ left: '$$', right: '$$', display: false }, { left: '$$', right: '$$', display: false },
{ left: '$', right: '$', display: false }, { left: '$', right: '$', display: false },
{ left: '\\pu{', right: '}', display: false }, { left: '\\pu{', right: '}', display: false },
{ left: '\\ce{', right: '}', display: false }, { left: '\\ce{', right: '}', display: false },
{ left: '\\(', right: '\\)', display: false }, { left: '\\(', right: '\\)', display: false },
{ left: '( ', right: ' )', display: false }, { left: '( ', right: ' )', display: false },
{ left: '\\[', right: '\\]', display: true }, { left: '\\[', right: '\\]', display: true },
{ left: '[', right: ']', display: true } { left: '[', right: ']', display: true }
] ];
// const DELIMITER_LIST = [ // const DELIMITER_LIST = [
// { left: '$$', right: '$$', display: false }, // { left: '$$', right: '$$', display: false },
@ -23,109 +23,116 @@ let inlinePatterns = [];
let blockPatterns = []; let blockPatterns = [];
function escapeRegex(string) { function escapeRegex(string) {
return string.replace(/[-\/\\^$*+?.()|[\]{}]/g, '\\$&'); return string.replace(/[-\/\\^$*+?.()|[\]{}]/g, '\\$&');
} }
function generateRegexRules(delimiters) { function generateRegexRules(delimiters) {
delimiters.forEach(delimiter => { delimiters.forEach((delimiter) => {
const { left, right } = delimiter; const { left, right } = delimiter;
// Ensure regex-safe delimiters // Ensure regex-safe delimiters
const escapedLeft = escapeRegex(left); const escapedLeft = escapeRegex(left);
const escapedRight = escapeRegex(right); const escapedRight = escapeRegex(right);
// Inline pattern - Capture group $1, token content, followed by end delimiter and normal punctuation marks. // Inline pattern - Capture group $1, token content, followed by end delimiter and normal punctuation marks.
// Example: $text$ // Example: $text$
inlinePatterns.push(`${escapedLeft}((?:\\\\.|[^\\\\\\n])*?(?:\\\\.|[^\\\\\\n${escapedRight}]))${escapedRight}`); inlinePatterns.push(
`${escapedLeft}((?:\\\\.|[^\\\\\\n])*?(?:\\\\.|[^\\\\\\n${escapedRight}]))${escapedRight}`
);
// Block pattern - Starts and ends with the delimiter on new lines. Example: // Block pattern - Starts and ends with the delimiter on new lines. Example:
// $$\ncontent here\n$$ // $$\ncontent here\n$$
blockPatterns.push(`${escapedLeft}\n((?:\\\\[^]|[^\\\\])+?)\n${escapedRight}`); blockPatterns.push(`${escapedLeft}\n((?:\\\\[^]|[^\\\\])+?)\n${escapedRight}`);
}); });
const inlineRule = new RegExp(`^(${inlinePatterns.join('|')})(?=[\\s?!.,:？！。，：]|$)`, 'u'); const inlineRule = new RegExp(`^(${inlinePatterns.join('|')})(?=[\\s?!.,:？！。，：]|$)`, 'u');
const blockRule = new RegExp(`^(${blockPatterns.join('|')})(?:\n|$)`, 'u'); const blockRule = new RegExp(`^(${blockPatterns.join('|')})(?:\n|$)`, 'u');
return { inlineRule, blockRule }; return { inlineRule, blockRule };
} }
const { inlineRule, blockRule } = generateRegexRules(DELIMITER_LIST); const { inlineRule, blockRule } = generateRegexRules(DELIMITER_LIST);
export default function (options = {}) {
return {
export default function(options = {}) { extensions: [
return { inlineKatex(options, createRenderer(options, false)),
extensions: [ blockKatex(options, createRenderer(options, true))
inlineKatex(options, createRenderer(options, false)), ]
blockKatex(options, createRenderer(options, true)), };
],
};
} }
function createRenderer(options, newlineAfter) { function createRenderer(options, newlineAfter) {
return (token) => katex.renderToString(token.text, { ...options, displayMode: token.displayMode }) + (newlineAfter ? '\n' : ''); return (token) =>
katex.renderToString(token.text, { ...options, displayMode: token.displayMode }) +
(newlineAfter ? '\n' : '');
} }
function inlineKatex(options, renderer) { function inlineKatex(options, renderer) {
const ruleReg = inlineRule; const ruleReg = inlineRule;
return { return {
name: 'inlineKatex', name: 'inlineKatex',
level: 'inline', level: 'inline',
start(src) { start(src) {
let index; let index;
let indexSrc = src; let indexSrc = src;
while (indexSrc) { while (indexSrc) {
index = indexSrc.indexOf('$'); index = indexSrc.indexOf('$');
if (index === -1) { if (index === -1) {
return; return;
} }
const f = index === 0 || indexSrc.charAt(index - 1) === ' '; const f = index === 0 || indexSrc.charAt(index - 1) === ' ';
if (f) { if (f) {
const possibleKatex = indexSrc.substring(index); const possibleKatex = indexSrc.substring(index);
if (possibleKatex.match(ruleReg)) { if (possibleKatex.match(ruleReg)) {
return index; return index;
} }
} }
indexSrc = indexSrc.substring(index + 1).replace(/^\$+/, ''); indexSrc = indexSrc.substring(index + 1).replace(/^\$+/, '');
} }
}, },
tokenizer(src, tokens) { tokenizer(src, tokens) {
const match = src.match(ruleReg); const match = src.match(ruleReg);
if (match) { if (match) {
const text = match.slice(2).filter((item) => item).find((item) => item.trim()); const text = match
.slice(2)
.filter((item) => item)
.find((item) => item.trim());
return { return {
type: 'inlineKatex', type: 'inlineKatex',
raw: match[0], raw: match[0],
text: text, text: text
}; };
} }
}, },
renderer, renderer
}; };
} }
function blockKatex(options, renderer) { function blockKatex(options, renderer) {
return { return {
name: 'blockKatex', name: 'blockKatex',
level: 'block', level: 'block',
tokenizer(src, tokens) { tokenizer(src, tokens) {
const match = src.match(blockRule); const match = src.match(blockRule);
if (match) { if (match) {
const text = match.slice(2).filter((item) => item).find((item) => item.trim()); const text = match
.slice(2)
.filter((item) => item)
.find((item) => item.trim());
return { return {
type: 'blockKatex', type: 'blockKatex',
raw: match[0], raw: match[0],
text: text, text: text
}; };
} }
}, },
renderer, renderer
}; };
} }