manage model

src/hooks/useMessageOption.tsx

@@ -12,6 +12,7 @@ import {
 import { useStoreMessageOption } from "~store/option"
 import { saveHistory, saveMessage } from "~libs/db"
 import { useNavigate } from "react-router-dom"
+import { notification } from "antd"
 
 export type BotResponse = {
   bot: {
@@ -253,22 +254,58 @@ export const useMessageOption = () => {
 
       setIsProcessing(false)
     } catch (e) {
+      console.log(e)
+      
+      if (e?.name === "AbortError") {
+        newMessage[appendingIndex].message = newMessage[
+          appendingIndex
+        ].message.slice(0, -1)
+
+        setHistory([
+          ...history,
+          {
+            role: "user",
+            content: message,
+            image
+          },
+          {
+            role: "assistant",
+            content: newMessage[appendingIndex].message
+          }
+        ])
+
+        if (historyId) {
+          await saveMessage(historyId, selectedModel, "user", message, [image])
+          await saveMessage(
+            historyId,
+            selectedModel,
+            "assistant",
+            newMessage[appendingIndex].message,
+            []
+          )
+        } else {
+          const newHistoryId = await saveHistory(message)
+          await saveMessage(newHistoryId.id, selectedModel, "user", message, [
+            image
+          ])
+          await saveMessage(
+            newHistoryId.id,
+            selectedModel,
+            "assistant",
+            newMessage[appendingIndex].message,
+            []
+          )
+          setHistoryId(newHistoryId.id)
+        }
+      } else {
+        notification.error({
+          message: "Error",
+          description: e?.message || "Something went wrong"
+        })
+      }
+
       setIsProcessing(false)
       setStreaming(false)
-
-      setMessages([
-        ...messages,
-        {
-          isBot: true,
-          name: selectedModel,
-          message: `Something went wrong. Check out the following logs:
-        \`\`\`
-        ${e?.message}
-        \`\`\`
-        `,
-          sources: []
-        }
-      ])
     }
   }