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

packages/global/support/wallet/usage/constants.ts

@@ -6,7 +6,10 @@ export enum UsageSourceEnum {
   shareLink = 'shareLink',
   training = 'training',
   cronJob = 'cronJob',
-  feishu = 'feishu'
+  share = 'share',
+  wecom = 'wecom',
+  feishu = 'feishu',
+  official_account = 'official_account'
 }
 
 export const UsageSourceMap = {
@@ -27,5 +30,14 @@ export const UsageSourceMap = {
   },
   [UsageSourceEnum.feishu]: {
     label: i18nT('user:usage.feishu')
+  },
+  [UsageSourceEnum.official_account]: {
+    label: i18nT('user:usage.official_account')
+  },
+  [UsageSourceEnum.share]: {
+    label: i18nT('user:usage.share')
+  },
+  [UsageSourceEnum.wecom]: {
+    label: i18nT('user:usage.wecom')
   }
 };

packages/global/support/wallet/usage/tools.ts

@@ -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;
+  }
+};