feat: 更新模型映射

Change-Id: I4230181b6d2bc8326f8165c3f71e45ef7a59f7b2
visual-ddd · Jan 18, 2024 · 98b61ba · 98b61ba
1 parent f1ec050
commit 98b61ba
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/modules/ai/chat.ts b/modules/ai/chat.ts
@@ -6,7 +6,7 @@ import { createFailResponse } from '../backend-node';
 import { handleRequestError } from '../session/api-helper';
 
 import { countToken, countTokenForString } from './encoding';
-import { ChatModel, DEFAULT_MAX_TOKEN, MAX_TOKENS, ChatOptions, ErrorResponse } from './constants';
+import { ChatModel, DEFAULT_MAX_TOKEN, MAX_TOKENS, ChatOptions, ErrorResponse, ModelPointer } from './constants';
 import { ChatCompletion, ChatCompletionInStream } from './types';
 import { getChatCompletionSupport } from './platform';
 import { checkRequest, RequestControlError } from './request-control';
@@ -68,7 +68,7 @@ export async function chat(options: ChatOptions) {
 
     const content = {
       stream: true,
-      model,
+      model: ModelPointer[model] ?? model,
       user: support.user,
       // max_tokens: maxToken - token,
       ...other,

diff --git a/modules/ai/constants.ts b/modules/ai/constants.ts
@@ -98,6 +98,14 @@ export enum ChatModel {
   GPT_4_32K = 'gpt-4-32k',
 }
 
+/**
+ * 模型指针
+ */
+export const ModelPointer: Record<string, string> = {
+  [ChatModel.GPT_4]: 'gpt-4-1106-preview',
+  [ChatModel.GPT_4_32K]: 'gpt-4-1106-preview',
+};
+
 export const ALL_SUPPORTED_CHAT_MODEL = [
   ChatModel.GPT3_5_TURBO,
   ChatModel.GPT3_5_TURBO_16K,