From 182986bb1d602b04465bbb1ccd53e2e731581c0d Mon Sep 17 00:00:00 2001
From: Saxon Fletcher <saxonafletcher@gmail.com>
Date: Tue, 8 Jul 2025 18:02:49 +1000
Subject: [PATCH] Assistant error state (#36907)

* ui refinements

* rename chat

* copy

* prose message styles

* add icon back

* fix message save

* simplify empty state

* update suggestions

* pass through props

* button styles

* onboarding icons

* name button

* remove results

* current chat name

* use type

* fix down arrow

* error and retry

---------

Co-authored-by: Alaister Young <a@alaisteryoung.com>
---
 .../components/ui/AIAssistantPanel/AIAssistant.tsx  | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/apps/studio/components/ui/AIAssistantPanel/AIAssistant.tsx b/apps/studio/components/ui/AIAssistantPanel/AIAssistant.tsx
index c54c899f46c..55889cafcbe 100644
--- a/apps/studio/components/ui/AIAssistantPanel/AIAssistant.tsx
+++ b/apps/studio/components/ui/AIAssistantPanel/AIAssistant.tsx
@@ -149,6 +149,8 @@ export const AIAssistant = ({ className }: AIAssistantProps) => {
     isLoading: isChatLoading,
     append,
     setMessages,
+    error,
+    reload,
   } = useChat({
     id: snap.activeChatId,
     api: useBedrockAssistant
@@ -439,6 +441,17 @@ export const AIAssistant = ({ className }: AIAssistantProps) => {
           {hasMessages ? (
             <div className="w-full px-7 py-8 space-y-6">
               {renderedMessages}
+              {error && (
+                <div className="border rounded-md pl-2 pr-1 py-1 flex items-center justify-between">
+                  <div className="flex items-center gap-2 text-foreground-light text-sm">
+                    <Info size={16} />
+                    <p>Sorry, I'm having trouble responding right now</p>
+                  </div>
+                  <Button type="text" size="tiny" onClick={() => reload()} className="text-xs">
+                    Retry
+                  </Button>
+                </div>
+              )}
               <AnimatePresence>
                 {isChatLoading && (
                   <motion.div