@dbx-tools/ui-mastra 0.9.118 → 0.9.120

package/src/support/mastra-client.ts

@@ -33,6 +33,17 @@ export interface MastraStreamResponse {
   stream: ReadableStream<UIMessageChunk>;
 }
 
+/** Terminal HTTP failure returned before a chat stream was established. */
+export class MastraChatRequestError extends Error {
+  constructor(
+    message: string,
+    readonly status: number,
+  ) {
+    super(message);
+    this.name = "MastraChatRequestError";
+  }
+}
+
 /** One memory thread returned by the native Mastra client. */
 export type MastraMemoryThread = ListMemoryThreadsResponse["threads"][number];
 
@@ -130,6 +141,7 @@ export class MastraPluginClient extends MastraClient {
       fetch: async (input, init) => {
         const response = await fetch(input, init);
         headers = new Headers(response.headers);
+        await assertChatResponse(response);
         return response;
       },
     });
@@ -160,6 +172,7 @@ export class MastraPluginClient extends MastraClient {
       fetch: async (input, init) => {
         const response = await fetch(input, init);
         headers = new Headers(response.headers);
+        await assertChatResponse(response);
         return response;
       },
     });
@@ -620,6 +633,38 @@ export class MastraPluginClient extends MastraClient {
   }
 }
 
+async function assertChatResponse(response: Response): Promise<void> {
+  if (response.ok) return;
+  const body = await response
+    .clone()
+    .json()
+    .catch(() => undefined);
+  const message =
+    responseErrorMessage(body) ??
+    (await response
+      .clone()
+      .text()
+      .catch(() => ""));
+  throw new MastraChatRequestError(
+    message.trim() || `Chat request failed (${response.status})`,
+    response.status,
+  );
+}
+
+function responseErrorMessage(body: unknown): string | undefined {
+  if (!body || typeof body !== "object") return undefined;
+  const record = body as Record<string, unknown>;
+  for (const field of ["error", "message"] as const) {
+    const value = record[field];
+    if (typeof value === "string" && value.trim()) return value.trim();
+    if (value && typeof value === "object") {
+      const nested = (value as Record<string, unknown>).message;
+      if (typeof nested === "string" && nested.trim()) return nested.trim();
+    }
+  }
+  return undefined;
+}
+
 /**
  * Read the Mastra plugin's `clientConfig()` payload (`basePath`,
  * default agent, registered agent list). One call per render; values

package/src/support/thread-sessions.ts

@@ -42,6 +42,7 @@ export type ThreadSession = {
 /** Map key for the classic single-thread chat (no explicit thread id). */
 export const DEFAULT_THREAD_SESSION_KEY = "__session__";
 
+/** Create the empty transcript and run state used for a new conversation thread. */
 export function createThreadSession(): ThreadSession {
   return {
     messages: [],
@@ -61,6 +62,7 @@ export function createThreadSession(): ThreadSession {
   };
 }
 
+/** Return whether a thread has a submitted or actively streaming run. */
 export function isSessionRunning(session: ThreadSession): boolean {
   return session.status === "submitted" || session.status === "streaming";
 }
@@ -98,6 +100,7 @@ export function reorderSteers<T extends QueuedSteer>(queue: T[], orderedIds: str
   return next;
 }
 
+/** Resolve an explicit thread id or the shared key used by single-thread chat. */
 export function sessionKey(activeThreadId: string | undefined): string {
   return activeThreadId ?? DEFAULT_THREAD_SESSION_KEY;
 }