test kill app

block · Jan 12, 2025 · 0c286c7 · 0c286c7
1 parent a77a770
commit 0c286c7
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/crates/goose-server/src/routes/reply.rs b/crates/goose-server/src/routes/reply.rs
@@ -307,6 +307,7 @@ async fn handler(
                     let _ = tx
                         .send(ProtocolFormatter::format_moderation_error(moderation_error))
                         .await;
+                    // Kill the stream since we encountered a moderation error
                 } else {
                     // Send a generic error message
                     let _ = tx

diff --git a/ui/desktop/src/ChatWindow.tsx b/ui/desktop/src/ChatWindow.tsx
@@ -89,6 +89,16 @@ function ChatContent({
       if (!response.ok) {
         setProgressMessage('An error occurred while receiving the response.');
         updateWorking(Working.Idle);
+
+        // Check if this is a moderation error
+        if (response.status === 400) {
+          response.json().then(data => {
+            if (data.error && data.error.includes('moderation')) {
+              // Kill the goosed process for moderation violations
+              window.electron.stopGooseProcess();
+            }
+          });
+        }
       } else {
         setProgressMessage('thinking...');
         updateWorking(Working.Working);
@@ -462,4 +472,4 @@ export default function ChatWindow() {
       )}
     </div>
   );
-}
+}