diff --git a/README.md b/README.md index 21cec0c..20df797 100644 --- a/README.md +++ b/README.md @@ -14,7 +14,7 @@ RoomId、UserId 以及申请的 AppID、BusinessID、Token、ASR AppID、TTS AppID,修改 `src/config/index.ts` 文件中的配置信息**。 - **根据你申请到的 AK、SK、SessionToken, 修改 `Server/app.js` 文件中的配置信息。** - 如果您已经自己完成了服务端的逻辑,可以修改前端代码文件 `src/config/index.ts` 中的 `AIGC_PROXY_HOST` 修改请求的域名,并在 `src/app/api.ts` 中修改接口的参数配置。 -- **您需要在 [火山方舟-在线推理](https://console.volcengine.com/ark/region:ark+cn-beijing/endpoint?config=%7B%7D) 中创建接入点, 并将模型对应的接入点 ID 填入 `src/config/config.ts` 文件中的 `ARK_V2_MODEL_ID`**。 +- **您需要在 [火山方舟-在线推理](https://console.volcengine.com/ark/region:ark+cn-beijing/endpoint?config=%7B%7D) 中创建接入点, 并将模型对应的接入点 ID 填入 `src/config/config.ts` 文件中的 `ARK_V3_MODEL_ID`**。 ## 服务端 进到项目根目录 diff --git a/src/config/config.ts b/src/config/config.ts index 9a5acc1..13ea02f 100644 --- a/src/config/config.ts +++ b/src/config/config.ts @@ -79,8 +79,7 @@ export const TTS_CLUSTER_MAP = { * @brief 模型 Prompt */ export enum PROMPT { - ARK_V2 = '你是小宁,性格幽默又善解人意。你在表达时要像朋友间聊天的口气,表达简明扼要,有自己的观点,面对争议性问题时可以客观表达。你健谈且有好奇心,会适当地推动话题的发展,在回答时,如果场景合适可以向用户进行一些问询或提出新话题。你有很强的共情能力,在我分享感受经历时会给予很好的情绪反馈。你的人生态度比较积极,兴趣广泛,对主流价值观认可的人或事都比较喜好。', - ARK_V3 = '', + ARK_V3 = '你是小宁,性格幽默又善解人意。你在表达时要像朋友间聊天的口气,表达简明扼要,有自己的观点,面对争议性问题时可以客观表达。你健谈且有好奇心,会适当地推动话题的发展,在回答时,如果场景合适可以向用户进行一些问询或提出新话题。你有很强的共情能力,在我分享感受经历时会给予很好的情绪反馈。你的人生态度比较积极,兴趣广泛,对主流价值观认可的人或事都比较喜好。', DEFAULT = '所有回复控制在100字以内。', } @@ -109,34 +108,23 @@ export enum AI_MODEL { */ export enum AI_MODEL_MODE { CUSTOM = 'CustomLLM', - ARK_V2 = 'ArkV2', ARK_V3 = 'ArkV3', } -/** - * @brief 各模型对应的模式 - */ -export const AI_MODE_MAP = { - [AI_MODEL.DOUBAO_LITE_4K]: AI_MODEL_MODE.ARK_V2, - [AI_MODEL.DOUBAO_PRO_4K]: AI_MODEL_MODE.ARK_V2, - [AI_MODEL.DOUBAO_PRO_32K]: AI_MODEL_MODE.ARK_V3, - [AI_MODEL.DOUBAO_PRO_128K]: AI_MODEL_MODE.ARK_V3, -}; - /** * @brief 各模型对应的 Prompt */ export const AI_MODE_PROMPT = { - [AI_MODEL.DOUBAO_LITE_4K]: PROMPT.ARK_V2, - [AI_MODEL.DOUBAO_PRO_4K]: PROMPT.ARK_V2, - [AI_MODEL.DOUBAO_PRO_32K]: PROMPT.ARK_V2, - [AI_MODEL.DOUBAO_PRO_128K]: PROMPT.ARK_V2, + [AI_MODEL.DOUBAO_LITE_4K]: PROMPT.ARK_V3, + [AI_MODEL.DOUBAO_PRO_4K]: PROMPT.ARK_V3, + [AI_MODEL.DOUBAO_PRO_32K]: PROMPT.ARK_V3, + [AI_MODEL.DOUBAO_PRO_128K]: PROMPT.ARK_V3, }; /** * @brief 豆包模型的 ID */ -export const ARK_V2_MODEL_ID: Record = { +export const ARK_V3_MODEL_ID: Record = { [AI_MODEL.DOUBAO_LITE_4K]: '', [AI_MODEL.DOUBAO_PRO_4K]: '', [AI_MODEL.DOUBAO_PRO_32K]: '', @@ -149,4 +137,4 @@ export const ARK_V2_MODEL_ID: Record = { */ export const LLM_BOT_ID = { // ... 可根据所开通的模型进行扩充 -}; +}; \ No newline at end of file diff --git a/src/config/index.ts b/src/config/index.ts index c6aa831..98f55f1 100644 --- a/src/config/index.ts +++ b/src/config/index.ts @@ -4,9 +4,9 @@ */ import { AI_MODEL, - AI_MODE_MAP, + AI_MODEL_MODE, AI_MODE_PROMPT, - ARK_V2_MODEL_ID, + ARK_V3_MODEL_ID, ModelSourceType, TTS_CLUSTER, VOICE_TYPE, @@ -41,11 +41,11 @@ export class Config { LLMConfig = { ModelName: AI_MODEL.DOUBAO_LITE_4K, - Mode: AI_MODE_MAP[AI_MODEL.DOUBAO_LITE_4K], + Mode: AI_MODEL_MODE.ARK_V3, ModelVersion: '1.0', WelcomeSpeech: '欢迎使用火山引擎视频云 RTC 驱动的虚拟人大模型', SystemMessages: [AI_MODE_PROMPT[AI_MODEL.DOUBAO_LITE_4K]], - EndPointId: ARK_V2_MODEL_ID[AI_MODEL.DOUBAO_LITE_4K], + EndPointId: ARK_V3_MODEL_ID[AI_MODEL.DOUBAO_LITE_4K], ModeSourceType: ModelSourceType.Available, APIKey: '', diff --git a/src/lib/RtcClient.ts b/src/lib/RtcClient.ts index 18f6baa..a3618c5 100644 --- a/src/lib/RtcClient.ts +++ b/src/lib/RtcClient.ts @@ -21,8 +21,7 @@ import RTCAIAnsExtension from '@volcengine/rtc/extension-ainr'; import openAPIs from '@/app/api'; import aigcConfig, { AI_MODEL, - AI_MODE_MAP, - ARK_V2_MODEL_ID, + ARK_V3_MODEL_ID, ModelSourceType, AI_MODEL_MODE, LLM_BOT_ID, @@ -269,8 +268,8 @@ export class RTCClient { LLMConfig: { ...mergedConfigs.LLMConfig, Mode: - modeSourceType === ModelSourceType.Custom ? AI_MODEL_MODE.CUSTOM : AI_MODE_MAP[model], - EndPointId: ARK_V2_MODEL_ID[model], + modeSourceType === ModelSourceType.Custom ? AI_MODEL_MODE.CUSTOM : AI_MODEL_MODE.ARK_V3, + EndPointId: ARK_V3_MODEL_ID[model], BotId: (LLM_BOT_ID as Record)[model], }, }, diff --git a/src/pages/View/Menu/components/AISettingDrawerButton/index.tsx b/src/pages/View/Menu/components/AISettingDrawerButton/index.tsx index d36f319..11649d0 100644 --- a/src/pages/View/Menu/components/AISettingDrawerButton/index.tsx +++ b/src/pages/View/Menu/components/AISettingDrawerButton/index.tsx @@ -64,7 +64,7 @@ function AISettingDrawerButton() { LLMConfig: { ModelName: formValues.modelName, WelcomeSpeech: formValues.welcomeSpeech, - Mode: AI_MODEL_MODE.ARK_V2, + Mode: AI_MODEL_MODE.ARK_V3, EndPointId: '', SystemMessages: [formValues.prompt], ModeSourceType: ModelSourceType.Available,