Просмотр исходного кода

fix: update assistant model configurations to use 'gpt-4.1-mini'

Yuki Takei 11 месяцев назад
Родитель
Сommit
9fc6dd59df

+ 3 - 10
apps/app/src/features/openai/server/services/assistant/assistant.ts

@@ -11,27 +11,20 @@ const AssistantType = {
   EDIT: 'Edit',
   EDIT: 'Edit',
 } as const;
 } as const;
 
 
-const AssistantDefaultModelMap: Record<AssistantType, OpenAI.Chat.ChatModel> = {
-  [AssistantType.SEARCH]: 'gpt-4o-mini',
-  [AssistantType.CHAT]: 'gpt-4o-mini',
-  [AssistantType.EDIT]: 'gpt-4o-mini',
-};
-
 const getAssistantModelByType = (type: AssistantType): OpenAI.Chat.ChatModel => {
 const getAssistantModelByType = (type: AssistantType): OpenAI.Chat.ChatModel => {
   const configValue = (() => {
   const configValue = (() => {
     switch (type) {
     switch (type) {
       case AssistantType.SEARCH:
       case AssistantType.SEARCH:
         // return configManager.getConfig('openai:assistantModel:search');
         // return configManager.getConfig('openai:assistantModel:search');
-        return undefined;
+        return 'gpt-4.1-mini';
       case AssistantType.CHAT:
       case AssistantType.CHAT:
         return configManager.getConfig('openai:assistantModel:chat');
         return configManager.getConfig('openai:assistantModel:chat');
       case AssistantType.EDIT:
       case AssistantType.EDIT:
-        // return configManager.getConfig('openai:assistantModel:edit');
-        return undefined;
+        return configManager.getConfig('openai:assistantModel:edit');
     }
     }
   })();
   })();
 
 
-  return configValue ?? AssistantDefaultModelMap[type];
+  return configValue;
 };
 };
 
 
 type AssistantType = typeof AssistantType[keyof typeof AssistantType];
 type AssistantType = typeof AssistantType[keyof typeof AssistantType];

+ 6 - 1
apps/app/src/server/service/config-manager/config-definition.ts

@@ -254,6 +254,7 @@ export const CONFIG_KEYS = [
   'openai:apiKey',
   'openai:apiKey',
   'openai:chatAssistantInstructions',
   'openai:chatAssistantInstructions',
   'openai:assistantModel:chat',
   'openai:assistantModel:chat',
+  'openai:assistantModel:edit',
   'openai:threadDeletionCronExpression',
   'openai:threadDeletionCronExpression',
   'openai:threadDeletionBarchSize',
   'openai:threadDeletionBarchSize',
   'openai:threadDeletionApiCallInterval',
   'openai:threadDeletionApiCallInterval',
@@ -1107,7 +1108,11 @@ Guideline as a RAG:
   /* eslint-enable max-len */
   /* eslint-enable max-len */
   'openai:assistantModel:chat': defineConfig<OpenAI.Chat.ChatModel>({
   'openai:assistantModel:chat': defineConfig<OpenAI.Chat.ChatModel>({
     envVarName: 'OPENAI_CHAT_ASSISTANT_MODEL',
     envVarName: 'OPENAI_CHAT_ASSISTANT_MODEL',
-    defaultValue: 'gpt-4o-mini',
+    defaultValue: 'gpt-4.1-mini',
+  }),
+  'openai:assistantModel:edit': defineConfig<OpenAI.Chat.ChatModel>({
+    envVarName: 'OPENAI_EDITOR_ASSISTANT_MODEL',
+    defaultValue: 'gpt-4.1-mini',
   }),
   }),
   'openai:threadDeletionCronExpression': defineConfig<string>({
   'openai:threadDeletionCronExpression': defineConfig<string>({
     envVarName: 'OPENAI_THREAD_DELETION_CRON_EXPRESSION',
     envVarName: 'OPENAI_THREAD_DELETION_CRON_EXPRESSION',