From 023223270ffcf515e5d3adf193fbbb2e547f4ef1 Mon Sep 17 00:00:00 2001
From: ramnique <30795890+ramnique@users.noreply.github.com>
Date: Tue, 4 Feb 2025 17:22:37 +0530
Subject: [PATCH] fix playground chat effects

---
 apps/rowboat/app/projects/[projectId]/playground/chat.tsx  | 6 +++++-
 apps/rowboat/app/projects/[projectId]/workflow/copilot.tsx | 5 ++++-
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/apps/rowboat/app/projects/[projectId]/playground/chat.tsx b/apps/rowboat/app/projects/[projectId]/playground/chat.tsx
index 05319536..f1ac6c30 100644
--- a/apps/rowboat/app/projects/[projectId]/playground/chat.tsx
+++ b/apps/rowboat/app/projects/[projectId]/playground/chat.tsx
@@ -53,6 +53,7 @@ export function Chat({
             createdAt: new Date().toISOString(),
         }];
         setMessages(updatedMessages);
+        setFetchResponseError(null);
     }
 
     function handleToolCallResults(results: z.infer<typeof apiV1.ToolMessage>[]) {
@@ -126,7 +127,9 @@ export function Chat({
                     setFetchResponseError(`Failed to get assistant response: ${err instanceof Error ? err.message : 'Unknown error'}`);
                 }
             } finally {
-                setLoadingAssistantResponse(false);
+                if (!ignore) {
+                    setLoadingAssistantResponse(false);
+                }
             }
         }
 
@@ -183,6 +186,7 @@ export function Chat({
                     chatId: chatId ?? '',
                     createdAt: new Date().toISOString(),
                 }]);
+                setFetchResponseError(null);
             } catch (err) {
                 setFetchResponseError(`Failed to simulate user response: ${err instanceof Error ? err.message : 'Unknown error'}`);
             } finally {
diff --git a/apps/rowboat/app/projects/[projectId]/workflow/copilot.tsx b/apps/rowboat/app/projects/[projectId]/workflow/copilot.tsx
index 7e95af2a..9fc55039 100644
--- a/apps/rowboat/app/projects/[projectId]/workflow/copilot.tsx
+++ b/apps/rowboat/app/projects/[projectId]/workflow/copilot.tsx
@@ -220,6 +220,7 @@ function App({
             role: 'user',
             content: prompt,
         }]);
+        setResponseError(null);
     }
 
     const handleApplyChange = useCallback((
@@ -351,7 +352,9 @@ function App({
                     setResponseError(`Failed to get copilot response: ${err instanceof Error ? err.message : 'Unknown error'}`);
                 }
             } finally {
-                setLoadingResponse(false);
+                if (!ignore) {
+                    setLoadingResponse(false);
+                }
             }
         }