From 61bf7b8536c509bb870dadb92e41886c3fb82b7e Mon Sep 17 00:00:00 2001
From: Darshil <ddhameliya@mail.sfsu.edu>
Date: Fri, 13 Mar 2026 00:25:27 -0700
Subject: [PATCH] fix: annotate shared failover mocks (openclaw#39820) thanks
 @lupuletic

---
 .../run.overflow-compaction.mocks.shared.ts         | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/src/agents/pi-embedded-runner/run.overflow-compaction.mocks.shared.ts b/src/agents/pi-embedded-runner/run.overflow-compaction.mocks.shared.ts
index 5276bd1c0d6..53e73e6246d 100644
--- a/src/agents/pi-embedded-runner/run.overflow-compaction.mocks.shared.ts
+++ b/src/agents/pi-embedded-runner/run.overflow-compaction.mocks.shared.ts
@@ -209,7 +209,6 @@ vi.mock("../defaults.js", () => ({
   DEFAULT_PROVIDER: "anthropic",
 }));
 
-export const mockedCoerceToFailoverError = vi.fn();
 type MockFailoverErrorDescription = {
   message: string;
   reason: string | undefined;
@@ -217,7 +216,15 @@ type MockFailoverErrorDescription = {
   code: string | undefined;
 };
 
-export const mockedDescribeFailoverError = vi.fn(
+type MockCoerceToFailoverError = (
+  err: unknown,
+  params?: { provider?: string; model?: string; profileId?: string },
+) => unknown;
+type MockDescribeFailoverError = (err: unknown) => MockFailoverErrorDescription;
+type MockResolveFailoverStatus = (reason: string) => number | undefined;
+
+export const mockedCoerceToFailoverError = vi.fn<MockCoerceToFailoverError>();
+export const mockedDescribeFailoverError = vi.fn<MockDescribeFailoverError>(
   (err: unknown): MockFailoverErrorDescription => ({
     message: err instanceof Error ? err.message : String(err),
     reason: undefined,
@@ -225,7 +232,7 @@ export const mockedDescribeFailoverError = vi.fn(
     code: undefined,
   }),
 );
-export const mockedResolveFailoverStatus = vi.fn();
+export const mockedResolveFailoverStatus = vi.fn<MockResolveFailoverStatus>();
 
 vi.mock("../failover-error.js", () => ({
   FailoverError: class extends Error {},