diff --git a/Yi.Ai.Vue3/src/api/model/types.ts b/Yi.Ai.Vue3/src/api/model/types.ts index a6eb8560..30a43521 100644 --- a/Yi.Ai.Vue3/src/api/model/types.ts +++ b/Yi.Ai.Vue3/src/api/model/types.ts @@ -11,4 +11,5 @@ export interface GetSessionListVO { apiHost?: string; apiKey?: string; remark?: string; + modelId?: string; } diff --git a/Yi.Ai.Vue3/src/components/ModelSelect/index.vue b/Yi.Ai.Vue3/src/components/ModelSelect/index.vue index e7e3c2fc..1deb1ed3 100644 --- a/Yi.Ai.Vue3/src/components/ModelSelect/index.vue +++ b/Yi.Ai.Vue3/src/components/ModelSelect/index.vue @@ -20,7 +20,7 @@ const userStore = useUserStore(); const modelStore = useModelStore(); // 检查模型是否可用 function isModelAvailable(item: GetSessionListVO) { - return isUserRoleVip.value || item.modelName?.includes('DeepSeek-R1') || userStore.userInfo?.user?.userName === 'cc'; + return isUserRoleVip.value || item.modelId?.includes('DeepSeek-R1-0528') || userStore.userInfo?.user?.userName === 'cc'; } onMounted(async () => { @@ -28,7 +28,7 @@ onMounted(async () => { // 设置默认模型 if ( modelStore.modelList.length > 0 - && (!modelStore.currentModelInfo || !modelStore.currentModelInfo.modelName) + && (!modelStore.currentModelInfo || !modelStore.currentModelInfo.modelId) ) { modelStore.setCurrentModelInfo(modelStore.modelList[0]); } diff --git a/Yi.Ai.Vue3/src/pages/chat/layouts/chatWithId/index.vue b/Yi.Ai.Vue3/src/pages/chat/layouts/chatWithId/index.vue index a4f6d613..08b8d475 100644 --- a/Yi.Ai.Vue3/src/pages/chat/layouts/chatWithId/index.vue +++ b/Yi.Ai.Vue3/src/pages/chat/layouts/chatWithId/index.vue @@ -185,7 +185,7 @@ async function startSSE(chatContent: string) { })), sessionId: route.params?.id !== 'not_login' ? String(route.params?.id) : undefined, userId: userStore.userInfo?.userId, - model: modelStore.currentModelInfo.modelName ?? '', + model: modelStore.currentModelInfo.modelId ?? '', })) { handleDataChunk(chunk.result as AnyObject); }