Commit 13439c51 by archer

perf: read file token error

parent 93030afe
...@@ -173,9 +173,16 @@ export const splitText2Chunks = ({ text, maxLen }: { text: string; maxLen: numbe ...@@ -173,9 +173,16 @@ export const splitText2Chunks = ({ text, maxLen }: { text: string; maxLen: numbe
chunks.push(chunk); chunks.push(chunk);
} }
const enc = getOpenAiEncMap(); const tokens = (() => {
const encodeText = enc.encode(chunks.join('')); try {
const tokens = encodeText.length; const enc = getOpenAiEncMap();
const encodeText = enc.encode(chunks.join(''));
const tokens = encodeText.length;
return tokens;
} catch (error) {
return chunks.join('').length;
}
})();
return { return {
chunks, chunks,
...@@ -274,5 +281,6 @@ export const simpleText = (text: string) => { ...@@ -274,5 +281,6 @@ export const simpleText = (text: string) => {
text = text.replace(/([\u4e00-\u9fa5])\s+([\u4e00-\u9fa5])/g, '$1$2'); text = text.replace(/([\u4e00-\u9fa5])\s+([\u4e00-\u9fa5])/g, '$1$2');
text = text.replace(/\n{2,}/g, '\n'); text = text.replace(/\n{2,}/g, '\n');
text = text.replace(/\s{2,}/g, ' '); text = text.replace(/\s{2,}/g, ' ');
text = text.replace(/[^\x00-\x7F]/g, ' ');
return text; return text;
}; };
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment