@@ -11,7 +11,7 @@ export async function sendImage<T>(
 	const HeaderDataType =
 		dataForSend instanceof FormData ? 'multipart/form-data' : 'application/json'
 
-	return await axios.post<Message[]>(
+	return await axios.post<Message[] | string>(
 		API_URL + `/media/image/${model}`,
 		dataForSend,
 		{
@@ -27,28 +27,30 @@ export function useImageBotCreateImage(
 
 		const dataSending = file ? formDataHelper(file, dataForSend) : dataForSend
 
-		const response = await sendImage(type, dataSending, data?.access)
+		const { data: messages } = await sendImage(type, dataSending, data?.access)
 
-		if (response.status >= 400) showError('Ошибка отправки сообщения')
-
-		setLoading(false)
+		if (typeof messages === 'string') {
+			showError(messages)
+			setLoading(false)
+			return
+		}
 
 		dispatch(getUserBalance(data?.access))
 
 		setMessages((prev: Message[]) => {
-			if (!prev || !prev.length) return response.data
+			if (!prev || !prev.length) return messages
 
 			if (device === 'desktop') {
-				return [...response.data, ...prev]
+				return [...messages, ...prev]
 			}
-			return [...prev, ...response.data]
+			return [...prev, ...messages]
 		})
 
 		setIsComplete(true)
 	}
 
-  return {
-    createImage,
-    isComplete
-  }
+	return {
+		createImage,
+		isComplete,
+	}
 }