server : remove array check of completion_probabilities in messages

This commit is contained in:
jhen 2023-08-13 07:07:40 +08:00
parent 3409735cff
commit bffd3cde10
2 changed files with 988 additions and 998 deletions

File diff suppressed because it is too large Load diff

View file

@ -317,11 +317,7 @@
const chatLine = ([user, data], index) => { const chatLine = ([user, data], index) => {
let message let message
const isArrayMessage = Array.isArray(data) const isArrayMessage = Array.isArray(data)
if ( if (params.value.n_probs > 0 && isArrayMessage) {
isArrayMessage &&
params.value.n_probs > 0 &&
data.every(msg => Array.isArray(msg.completion_probabilities))
) {
message = html`<${Probabilites} data=${data} />` message = html`<${Probabilites} data=${data} />`
} else { } else {
const text = isArrayMessage ? const text = isArrayMessage ?
@ -440,9 +436,10 @@
const Probabilites = (params) => { const Probabilites = (params) => {
return params.data.map(msg => { return params.data.map(msg => {
const { completion_probabilities } = msg; const { completion_probabilities } = msg;
if (
// Skip byte pair !completion_probabilities ||
if (completion_probabilities.length > 1) return msg.content completion_probabilities.length > 1 // Skip byte pair
) return msg.content
const { probs } = completion_probabilities[0] const { probs } = completion_probabilities[0]
const found = probs.find(p => p.tok_str === msg.content) const found = probs.find(p => p.tok_str === msg.content)