diff --git a/src/main/worker/entry/types.ts b/src/main/worker/entry/types.ts
index ced960a4..c354b93a 100644
--- a/src/main/worker/entry/types.ts
+++ b/src/main/worker/entry/types.ts
@@ -53,6 +53,7 @@ export const workerMessageSchema = z.union([
       "beforeToolCall",
       "afterToolCall",
       "transformContext",
+      "invokeInternalTool",
     ]),
     payload: z.unknown(),
   }),