ChatGPT/FastGPT
1
0
Fork 0
mirror of https://github.com/labring/FastGPT.git synced 2025-08-02 04:32:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: variable input and update chat time

Browse Source
This commit is contained in:
archer
2023-08-31 18:09:12 +08:00
parent 3420f677b6
commit b22c878cf9
15 changed files with 30 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
client/src/utils/plugin/openai.ts
View File

@@ -47,21 +47,13 @@ export const adaptChatItem_openAI = ({
}));
};
export function countOpenAIToken({
messages,
model = 'gpt-3.5-turbo'
}: {
messages: ChatItemType[];
model?: string;
}) {
const diffVal = model.startsWith('gpt-3.5-turbo') ? 3 : 2;
export function countOpenAIToken({ messages }: { messages: ChatItemType[] }) {
const adaptMessages = adaptChatItem_openAI({ messages, reserveId: true });
const token = adaptMessages.reduce((sum, item) => {
const text = `${item.role}\n${item.content}`;
const enc = getOpenAiEncMap();
const encodeText = enc.encode(text);
const tokens = encodeText.length + diffVal;
const tokens = encodeText.length + 3; // 补充估算值
return sum + tokens;
}, 0);