diff --git a/packages/llm/llm-retry/tests/transport-recovery.spec.ts b/packages/llm/llm-retry/tests/transport-recovery.spec.ts index d9a7d4d756..8fd91ece9e 100644 --- a/packages/llm/llm-retry/tests/transport-recovery.spec.ts +++ b/packages/llm/llm-retry/tests/transport-recovery.spec.ts @@ -14,7 +14,7 @@ import type { SessionEvent } from '@deepseek-ai/dsh-session' import * as Retry from '../src/index.ts' let context: Context | undefined -const servers: MockLlmServer[] = [] +const servers: Promise[] = [] afterEach(async () => { const ownedContext = context @@ -22,18 +22,18 @@ afterEach(async () => { // A failed transport can still await a stalled socket while its context disposes. const results = await Promise.allSettled([ ownedContext?.fiber.dispose(), - ...servers.splice(0).map(server => server.close()), + ...servers.splice(0).map(async server => (await server).close()), ]) vi.unstubAllEnvs() const failure = results.find(result => result.status === 'rejected') if (failure?.status === 'rejected') throw failure.reason }) -async function start( +function start( sequence: readonly MockLlmBehavior[], options: Omit[0], 'sequence'> = {}, ): Promise { - const server = await startMockLlmServer({ sequence, ...options }) + const server = startMockLlmServer({ sequence, ...options }) servers.push(server) return server }