From 38c76b34f428a38d7613a42e9ec1bf2d65c1fa90 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Thu, 23 Apr 2026 00:58:13 +0100
Subject: [PATCH] test(agents): stabilize context lookup warmup

---
 src/agents/context.lookup.test.ts | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/agents/context.lookup.test.ts b/src/agents/context.lookup.test.ts
index 32863184e27..8402f074fc6 100644
--- a/src/agents/context.lookup.test.ts
+++ b/src/agents/context.lookup.test.ts
@@ -20,7 +20,7 @@ vi.mock("../config/config.js", () => ({
   loadConfig: () => contextTestState.loadConfigImpl(),
 }));
 
-vi.mock("./models-config.js", () => ({
+vi.mock("./models-config.runtime.js", () => ({
   ensureOpenClawModelsJson: contextTestState.ensureOpenClawModelsJson,
 }));
 
@@ -75,7 +75,9 @@ async function flushAsyncWarmup() {
     await vi.advanceTimersByTimeAsync(0);
     return;
   }
+  await Promise.resolve();
   await new Promise((r) => setTimeout(r, 0));
+  await Promise.resolve();
 }
 
 let contextModule: ContextModule;