perf: init token worker (#4726)

* perf: init token worker

* init worker

* preload worker

* preload worker

* remove invalid code
This commit is contained in:
Archer
2025-04-29 16:00:35 +08:00
committed by GitHub
parent 5e3ec4d6f3
commit 4b8db293ce
3 changed files with 44 additions and 15 deletions

View File

@@ -0,0 +1,24 @@
import { getWorkerController, WorkerNameEnum } from './utils';
export const preLoadWorker = async () => {
const max = Number(global.systemEnv?.tokenWorkers || 30);
const workerController = getWorkerController({
name: WorkerNameEnum.countGptMessagesTokens,
maxReservedThreads: max
});
for await (const item of new Array(max).fill(0)) {
const worker = workerController.createWorker();
await workerController.run({
workerId: worker.id,
messages: [
{
role: 'user',
content: '1'
}
]
});
console.log(`Preload worker ${workerController.workerQueue.length}`);
}
console.log('Preload worker success');
};