From a7f62adad40560deff133bd94ad0dd0628da068f Mon Sep 17 00:00:00 2001 From: suyx Date: Wed, 5 Aug 2026 13:26:44 +0800 Subject: [PATCH] =?UTF-8?q?fix(POSTV1-02):=20=E7=A1=AE=E4=BF=9DAI=E6=8E=A2?= =?UTF-8?q?=E6=B5=8B=E8=BE=93=E5=87=BA=E5=90=8E=E9=80=80=E5=87=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/worker/src/worker.ts | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/apps/worker/src/worker.ts b/apps/worker/src/worker.ts index 194cb0e..bb03306 100644 --- a/apps/worker/src/worker.ts +++ b/apps/worker/src/worker.ts @@ -27,18 +27,20 @@ if (workerPort) { if (!workerPort && process.argv.includes("--dada-ai-probe")) { const credentialClient = initializeWorkerCredentialClient(await receiveWorkerCredentials()); let adapter: OneApiGenerationAdapter | undefined; + let probeResult: Awaited> | { code: "ai_probe_failed"; error_category: "upstream_failed"; real_calls: 0; success: false }; try { adapter = new OneApiGenerationAdapter({ credential: credentialClient.aiGatewayCredential }); - const result = await runAiRuntimeProbe(adapter); - process.stdout.write(JSON.stringify(result)); - if (!result.success) process.exitCode = 2; + probeResult = await runAiRuntimeProbe(adapter); } catch { - process.stdout.write(JSON.stringify({ code: "ai_probe_failed", error_category: "upstream_failed", real_calls: 0, success: false })); - process.exitCode = 2; + probeResult = { code: "ai_probe_failed", error_category: "upstream_failed", real_calls: 0, success: false }; } finally { credentialClient.aiGatewayCredential.fill(0); adapter?.dispose(); } + await new Promise((resolveWrite, rejectWrite) => { + process.stdout.write(JSON.stringify(probeResult), (error) => error ? rejectWrite(error) : resolveWrite()); + }); + process.exit(probeResult.success ? 0 : 2); } else if (!workerPort && process.argv.includes("--dada-credential-stdin")) { const credentialClient = initializeWorkerCredentialClient(await receiveWorkerCredentials()); const controlPipeIndex = process.argv.indexOf("--dada-control-pipe");