@@ -457,6 +457,8 @@ export function useChatModel(
 
 				if (response.status === 404) {
 					clearChatStreamSession(chatUid)
+					inputMessageUuidRef.current = null
+					setMessages((prev) => removeStreamingModelMessage(prev, inputMessageUuid))
 					return false
 				}