[shared-ui] Teach ChatController about errors.

- **Teach `ChatController` about errors.** - **docs(changeset): Teach `ChatController` about errors.** Fixes #4204
breadboard-ai · Jan 25, 2025 · 976928d · 976928d
1 parent b17362c
commit 976928d
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 2 deletions.
diff --git a/.changeset/blue-peaches-ring.md b/.changeset/blue-peaches-ring.md
@@ -0,0 +1,5 @@
+---
+"@breadboard-ai/shared-ui": patch
+---
+
+Teach `ChatController` about errors.
diff --git a/packages/shared-ui/src/elements/chat/chat.ts b/packages/shared-ui/src/elements/chat/chat.ts
@@ -122,6 +122,8 @@ export class Chat extends LitElement {
         } else {
           items.push(html` <p>${entry.text}</p>`);
         }
+      } else if ("error" in entry) {
+        items.push(html` <p style="color:red">${entry.error}</p>`);
       }
     }
 

diff --git a/packages/shared-ui/src/state/chat-controller.ts b/packages/shared-ui/src/state/chat-controller.ts
@@ -26,6 +26,7 @@ import {
   ChatStatus,
   ChatSystemTurnState,
 } from "./types";
+import { formatError } from "../utils/format-error";
 
 export { ChatController };
 
@@ -169,8 +170,18 @@ class ChatController {
     this.#appendTurn(turn);
   }
 
-  #onError(_event: RunErrorEvent) {
+  #onError(event: RunErrorEvent) {
     this.#currentInput = null;
+    this.#status = "stopped";
+    this.#appendTurn(
+      this.#createSystemTurn([
+        {
+          title: "Stopping due to the following error",
+          error: formatError(event.data.error),
+        },
+      ])
+    );
+    this.#stale = true;
   }
 
   #createSystemTurn(content: ChatContent[]): ChatSystemTurnState {

diff --git a/packages/shared-ui/src/state/types.ts b/packages/shared-ui/src/state/types.ts
@@ -29,7 +29,16 @@ export type ChatObjectContent = {
   object: NodeValue;
 };
 
-export type ChatContent = ChatTextContent | ChatLLMContent | ChatObjectContent;
+export type ChatError = {
+  title: string;
+  error: string;
+};
+
+export type ChatContent =
+  | ChatTextContent
+  | ChatLLMContent
+  | ChatObjectContent
+  | ChatError;
 
 /**
  * Represents the system entry in the chat conversation between the