diff --git a/web/app/components/workflow/store/workflow/tool-slice.ts b/web/app/components/workflow/store/workflow/tool-slice.ts
index 45cce1286b..d5ff7743be 100644
--- a/web/app/components/workflow/store/workflow/tool-slice.ts
+++ b/web/app/components/workflow/store/workflow/tool-slice.ts
@@ -1,10 +1,15 @@
 import type { StateCreator } from 'zustand'
+import type { ToolWithProvider } from '../../types'
 
 export type ToolSliceShape = {
   toolPublished: boolean
   setToolPublished: (toolPublished: boolean) => void
   lastPublishedHasUserInput: boolean
   setLastPublishedHasUserInput: (hasUserInput: boolean) => void
+  buildInTools?: ToolWithProvider[]
+  customTools?: ToolWithProvider[]
+  workflowTools?: ToolWithProvider[]
+  mcpTools?: ToolWithProvider[]
 }
 
 export const createToolSlice: StateCreator<ToolSliceShape> = set => ({
@@ -12,4 +17,8 @@ export const createToolSlice: StateCreator<ToolSliceShape> = set => ({
   setToolPublished: toolPublished => set(() => ({ toolPublished })),
   lastPublishedHasUserInput: false,
   setLastPublishedHasUserInput: hasUserInput => set(() => ({ lastPublishedHasUserInput: hasUserInput })),
+  buildInTools: undefined,
+  customTools: undefined,
+  workflowTools: undefined,
+  mcpTools: undefined,
 })