PR feedback - don't relax type checks

Author: diarmidmackenzie

src/core/webview/webviewMessageHandler.ts

@@ -779,7 +779,7 @@ export const webviewMessageHandler = async (provider: ClineProvider, message: We
 				const existingPrompts = getGlobalState("customModePrompts") ?? {}
 				const updatedPrompts = { ...existingPrompts, [message.promptMode]: message.customPrompt }
 				await updateGlobalState("customModePrompts", updatedPrompts)
-				const currentState = await provider.getState()
+				const currentState = await provider.getStateToPostToWebview()
 				const stateWithPrompts = { ...currentState, customModePrompts: updatedPrompts }
 				provider.postMessageToWebview({ type: "state", state: stateWithPrompts })
 			}

src/shared/ExtensionMessage.ts

@@ -78,7 +78,7 @@ export interface ExtensionMessage {
 		| "promptsButtonClicked"
 		| "didBecomeVisible"
 	invoke?: "newChat" | "sendMessage" | "primaryButtonClick" | "secondaryButtonClick" | "setChatBoxMessage"
-	state?: Partial<ExtensionState>
+	state?: ExtensionState
 	images?: string[]
 	ollamaModels?: string[]
 	lmStudioModels?: string[]

webview-ui/src/context/ExtensionStateContext.tsx

@@ -88,7 +88,7 @@ export interface ExtensionStateContextType extends ExtensionState {
 
 export const ExtensionStateContext = createContext<ExtensionStateContextType | undefined>(undefined)
 
-export const mergeExtensionState = (prevState: ExtensionState, newState: Partial<ExtensionState>) => {
+export const mergeExtensionState = (prevState: ExtensionState, newState: ExtensionState) => {
 	const {
 		customModePrompts: prevCustomModePrompts,
 		customSupportPrompts: prevCustomSupportPrompts,