e009be51e7
* model config * feat: model config ui * perf: rename variable * feat: custom request url * perf: model buffer * perf: init model * feat: json model config * auto login * fix: ts * update packages * package * fix: dockerfile
37 lines
880 B
TypeScript
37 lines
880 B
TypeScript
import { findAIModel } from '../../../core/ai/model';
|
|
import { ModelTypeEnum } from '@fastgpt/global/core/ai/model';
|
|
|
|
export const formatModelChars2Points = ({
|
|
model,
|
|
inputTokens = 0,
|
|
outputTokens = 0,
|
|
modelType,
|
|
multiple = 1000
|
|
}: {
|
|
model: string;
|
|
inputTokens?: number;
|
|
outputTokens?: number;
|
|
modelType: `${ModelTypeEnum}`;
|
|
multiple?: number;
|
|
}) => {
|
|
const modelData = findAIModel(model);
|
|
if (!modelData) {
|
|
return {
|
|
totalPoints: 0,
|
|
modelName: ''
|
|
};
|
|
}
|
|
|
|
const isIOPriceType = typeof modelData.inputPrice === 'number';
|
|
|
|
const totalPoints = isIOPriceType
|
|
? (modelData.inputPrice || 0) * (inputTokens / multiple) +
|
|
(modelData.outputPrice || 0) * (outputTokens / multiple)
|
|
: (modelData.charsPointsPrice || 0) * ((inputTokens + outputTokens) / multiple);
|
|
|
|
return {
|
|
modelName: modelData.name,
|
|
totalPoints
|
|
};
|
|
};
|