diff --git a/src/components/Message/AiMessage.tsx b/src/components/Message/AiMessage.tsx index 41500576..61db1582 100644 --- a/src/components/Message/AiMessage.tsx +++ b/src/components/Message/AiMessage.tsx @@ -89,6 +89,7 @@ function AiMessage(props: AiMessageProps) { const [message, setMessage] = useState(props.message); const [retrying, setRetrying] = useState(false); const { settings } = useSettings(); + const { error } = useAlert(); useEffect(() => { setMessage(props.message); @@ -127,14 +128,17 @@ function AiMessage(props: AiMessageProps) { message.addVersion(version); message.switchVersion(version.id); await message.save(chat.id); - } catch (err) { - // TODO: UI error handling + } catch (err: any) { + error({ + title: `Retry Error`, + message: err.message, + }); console.warn("Unable to retry message", { model, err }); } finally { setRetrying(false); } }, - [props.chatId, settings.currentProvider.apiKey, message, callChatApi] + [props.chatId, settings.currentProvider.apiKey, message, callChatApi, error] ); // While we're streaming in a new version, use a different display