diff --git a/src/mitm/handlers/base.ts b/src/mitm/handlers/base.ts
index 6b1e45023e..bb98a76706 100644
--- a/src/mitm/handlers/base.ts
+++ b/src/mitm/handlers/base.ts
@@ -53,6 +53,7 @@ async function loadAgentBridgeHook(): Promise<{
     body: Buffer;
     agentId: AgentId;
     mappedModel: string;
+    sourceModel?: string | null;
   }) => Promise<InterceptedRequest>;
   recordRequestComplete?: (
     intercepted: InterceptedRequest,