File size: 2,712 Bytes
494c9e4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 | import { tr } from '../lang/i18n-lite';
/** 归一后的展示类别（内部用，与 API 字符串无强耦合） */
export type GenerationEndDisplayKind = 'eos' | 'length_limit' | 'stopped' | 'error';
export function generationEndReasonLabel(kind: GenerationEndDisplayKind): string {
switch (kind) {
case 'eos':
return tr('EOS reached');
case 'length_limit':
return tr('Maximum length reached');
case 'stopped':
return tr('Stopped');
case 'error':
return tr('Error');
default: {
const _exhaustive: never = kind;
return _exhaustive;
}
}
}
/**
* 产生方统一采用 OpenAI completions `finish_reason` _subset_ + `error`：
* Chat 服务端返回值、Causal Flow 循环结束、缓存 `completionReason` 均使用同一套字符串。
*/
export const COMPLETION_FINISH_REASONS = ['stop', 'length', 'abort', 'error'] as const;
export type CompletionFinishReason = (typeof COMPLETION_FINISH_REASONS)[number];
/** 旧版 Causal Flow 曾写入的别名（仅读写兼容，新写入均为 {@link COMPLETION_FINISH_REASONS}） */
const LEGACY_FINISH_TO_CANONICAL: Record<string, CompletionFinishReason> = {
eos: 'stop',
max_tokens: 'length',
aborted: 'abort',
};
const FINISH_REASON_TO_DISPLAY: Record<CompletionFinishReason, GenerationEndDisplayKind> = {
stop: 'eos',
length: 'length_limit',
abort: 'stopped',
error: 'error',
};
export function isCompletionFinishReason(s: string): s is CompletionFinishReason {
return (COMPLETION_FINISH_REASONS as readonly string[]).includes(s);
}
/** demo JSON / IndexedDB 里可能出现的 completionReason 值（含旧别名） */
export function isKnownPersistedCompletionReason(s: string): boolean {
return isCompletionFinishReason(s) || s in LEGACY_FINISH_TO_CANONICAL;
}
/** 将旧别名转为规范词；已是规范词或非别名则原样返回（如 OpenAI `content_filter`）。 */
export function canonicalizeCompletionFinishReason(raw: string): string {
const r = raw.trim();
if (!r) return r;
if (isCompletionFinishReason(r)) return r;
return LEGACY_FINISH_TO_CANONICAL[r] ?? r;
}
/**
* Completions `finish_reason` 与 Causal Flow 结束原因 → 展示文案。
* 未识别的非空值原样返回（不翻译）。
*/
export function completionFinishReasonLabel(finishReason: string | null | undefined): string {
const r = (finishReason ?? '').trim();
if (!r) return '';
const c = canonicalizeCompletionFinishReason(r);
if (isCompletionFinishReason(c)) {
return generationEndReasonLabel(FINISH_REASON_TO_DISPLAY[c]);
}
return r;
}
|