From c3192ef1b4f005dbb8c5b8ea49077e7e8d800bf6 Mon Sep 17 00:00:00 2001 From: archer <545436317@qq.com> Date: Sat, 25 Jan 2025 12:02:20 +0800 Subject: [PATCH] reload buffer --- packages/service/common/system/config/controller.ts | 10 ++++++++++ packages/service/core/ai/config/utils.ts | 4 ++++ projects/app/src/pages/api/core/ai/model/update.ts | 1 - 3 files changed, 14 insertions(+), 1 deletion(-) diff --git a/packages/service/common/system/config/controller.ts b/packages/service/common/system/config/controller.ts index a44ec63bb547..5eb6d1bc1345 100644 --- a/packages/service/common/system/config/controller.ts +++ b/packages/service/common/system/config/controller.ts @@ -39,3 +39,13 @@ export const updateFastGPTConfigBuffer = async () => { global.systemInitBufferId = res.createTime.getTime().toString(); }; + +export const reloadFastGPTConfigBuffer = async () => { + const res = await MongoSystemConfigs.findOne({ + type: SystemConfigsTypeEnum.fastgpt + }).sort({ + createTime: -1 + }); + if (!res) return; + global.systemInitBufferId = res.createTime.getTime().toString(); +}; diff --git a/packages/service/core/ai/config/utils.ts b/packages/service/core/ai/config/utils.ts index f6b997813797..68252c368d78 100644 --- a/packages/service/core/ai/config/utils.ts +++ b/packages/service/core/ai/config/utils.ts @@ -13,6 +13,7 @@ import { import { debounce } from 'lodash'; import { ModelProviderType } from '@fastgpt/global/core/ai/provider'; import { findModelFromAlldata } from '../model'; +import { reloadFastGPTConfigBuffer } from '../../../common/system/config/controller'; /* TODO: 分优先级读取: @@ -170,7 +171,10 @@ export const watchSystemModelUpdate = () => { 'change', debounce(async () => { try { + // Main node will reload twice await loadSystemModels(true); + // All node reaload buffer + await reloadFastGPTConfigBuffer(); } catch (error) {} }, 500) ); diff --git a/projects/app/src/pages/api/core/ai/model/update.ts b/projects/app/src/pages/api/core/ai/model/update.ts index 6fd14e60eae8..5fd6568ce5ea 100644 --- a/projects/app/src/pages/api/core/ai/model/update.ts +++ b/projects/app/src/pages/api/core/ai/model/update.ts @@ -2,7 +2,6 @@ import type { ApiRequestProps, ApiResponseType } from '@fastgpt/service/type/nex import { NextAPI } from '@/service/middleware/entry'; import { authSystemAdmin } from '@fastgpt/service/support/permission/user/auth'; import { MongoSystemModel } from '@fastgpt/service/core/ai/config/schema'; -import { delay } from '@fastgpt/global/common/system/utils'; import { updateFastGPTConfigBuffer } from '@fastgpt/service/common/system/config/controller'; import { findModelFromAlldata } from '@fastgpt/service/core/ai/model'; import { loadSystemModels } from '@fastgpt/service/core/ai/config/utils';