fix(publish): generate correct chat history and usage (#2412)

This commit is contained in:
Finley Ge
2024-08-16 15:59:17 +08:00
committed by GitHub
parent bebbba399a
commit 9b74437417
6 changed files with 71 additions and 5 deletions

View File

@@ -2,6 +2,7 @@
import { PRICE_SCALE } from '../constants';
import { UsageSourceEnum } from './constants';
import { AuthUserTypeEnum } from '../../permission/constant';
import { PublishChannelEnum } from '../../outLink/constant';
/**
* dataset price / PRICE_SCALE = real price
@@ -21,3 +22,22 @@ export const getUsageSourceByAuthType = ({
if (authType === AuthUserTypeEnum.apikey) return UsageSourceEnum.api;
return UsageSourceEnum.fastgpt;
};
export const getUsageSourceByPublishChannel = (publishchannel: PublishChannelEnum) => {
switch (publishchannel) {
case PublishChannelEnum.share:
return UsageSourceEnum.share;
case PublishChannelEnum.iframe:
return UsageSourceEnum.shareLink;
case PublishChannelEnum.apikey:
return UsageSourceEnum.api;
case PublishChannelEnum.feishu:
return UsageSourceEnum.feishu;
case PublishChannelEnum.wecom:
return UsageSourceEnum.wecom;
case PublishChannelEnum.officialAccount:
return UsageSourceEnum.official_account;
default:
return UsageSourceEnum.fastgpt;
}
};