✨ feat: Use current model to generate summary title

src/chains/summaryTitle.ts

@@ -1,8 +1,9 @@
-import { chatHelpers } from '@/store/chat/helpers';
 import { globalHelpers } from '@/store/global/helpers';
+import { LobeAgentConfig } from '@/types/agent';
 import { ChatStreamPayload, OpenAIChatMessage } from '@/types/openai/chat';
 
 export const chainSummaryTitle = async (
+  agentConfig: LobeAgentConfig,
   messages: OpenAIChatMessage[],
 ): Promise<Partial<ChatStreamPayload>> => {
   const lang = globalHelpers.getCurrentLanguage();
@@ -19,15 +20,9 @@ export const chainSummaryTitle = async (
       role: 'user',
     },
   ];
-  // 如果超过 16k，则使用 GPT-4-turbo 模型
-  const tokens = await chatHelpers.getMessagesTokenCount(finalMessages);
-  let model: string | undefined = undefined;
-  if (tokens > 16_000) {
-    model = 'gpt-4-turbo-preview';
-  }
 
   return {
     messages: finalMessages,
-    model,
+    ...agentConfig,
   };
 };

src/store/chat/slices/topic/action.ts

@@ -14,6 +14,8 @@ import { chatService } from '@/services/chat';
 import { messageService } from '@/services/message';
 import { topicService } from '@/services/topic';
 import { ChatStore } from '@/store/chat';
+import { useSessionStore } from '@/store/session';
+import { agentSelectors } from '@/store/session/selectors';
 import { ChatMessage } from '@/types/message';
 import { ChatTopic } from '@/types/topic';
 import { setNamespace } from '@/utils/storeDebug';
@@ -46,6 +48,8 @@ export interface ChatTopicAction {
   useSearchTopics: (keywords?: string) => SWRResponse<ChatTopic[]>;
 }
 
+const getAgentConfig = () => agentSelectors.currentAgentConfig(useSessionStore.getState());
+
 export const chatTopic: StateCreator<
   ChatStore,
   [['zustand/devtools', never]],
@@ -100,6 +104,7 @@ export const chatTopic: StateCreator<
   },
   // update
   summaryTopicTitle: async (topicId, messages) => {
+    const config = getAgentConfig();
     const { updateTopicTitleInSummary, updateTopicLoading, refreshTopic } = get();
     const topic = topicSelectors.getTopicById(topicId)(get());
     if (!topic) return;
@@ -123,7 +128,7 @@ export const chatTopic: StateCreator<
         output += x;
         updateTopicTitleInSummary(topicId, output);
       },
-      params: await chainSummaryTitle(messages),
+      params: await chainSummaryTitle(config, messages),
       trace: get().getCurrentTracePayload({ traceName: TraceNameMap.SummaryTopicTitle, topicId }),
     });
     await refreshTopic();