bs-agent 0.0.10 → 0.0.12

package/dist/react/index.js

@@ -197,6 +197,20 @@ var createDebugHandlers = (setDebugData) => {
         }));
         break;
       }
+      case "run_error": {
+        setDebugData((prev) => ({
+          ...prev,
+          [executionId]: [
+            ...prev[executionId] || [],
+            {
+              itemType: "run_error",
+              message: event.data.message,
+              code: event.data.code
+            }
+          ]
+        }));
+        break;
+      }
     }
   };
   return {
@@ -293,6 +307,16 @@ function useClientToolHelpers(agentId, toolContext) {
 }
 
 // src/core/stream.ts
+function tryParseJSON2(value) {
+  if (typeof value === "string") {
+    try {
+      return JSON.parse(value);
+    } catch {
+      return value;
+    }
+  }
+  return value;
+}
 var FatalStreamError = class extends Error {
   constructor(message, statusCode) {
     super(message);
@@ -327,7 +351,8 @@ async function executeStream(options) {
   });
   const sessionId = response.headers.get("X-BuildShip-Agent-Session-ID");
   if (sessionId && onSessionId) {
-    onSessionId(sessionId);
+    const sessionName = response.headers.get("X-BuildShip-Agent-Session-Name") || void 0;
+    onSessionId(sessionId, sessionName);
   }
   onResponse?.(response);
   if (!response.ok) {
@@ -344,9 +369,8 @@ async function executeStream(options) {
     throw error;
   }
   if (!response.body) {
-    const error = new Error("Response body is null \u2014 streaming not supported");
-    callbacks.onError?.(error);
-    throw error;
+    callbacks.onComplete?.(fullText);
+    return;
   }
   const reader = response.body.getReader();
   const decoder = new TextDecoder();
@@ -359,9 +383,17 @@ async function executeStream(options) {
       const parts = buffer.split("\n\n");
       buffer = parts.pop() || "";
       for (const part of parts) {
-        const dataLine = part.split("\n").find((line) => line.startsWith("data:"));
-        if (!dataLine) continue;
-        const jsonStr = dataLine.slice(5).trim();
+        const trimmedPart = part.trim();
+        if (!trimmedPart) continue;
+        let jsonStr = "";
+        for (const line of trimmedPart.split("\n")) {
+          if (line.startsWith("data: ")) {
+            jsonStr += line.slice(6);
+          } else if (line.startsWith("data:")) {
+            jsonStr += line.slice(5);
+          }
+        }
+        jsonStr = jsonStr.trim();
         if (!jsonStr) continue;
         let raw;
         try {
@@ -369,9 +401,9 @@ async function executeStream(options) {
         } catch {
           continue;
         }
-        const event = normalizeEvent(raw);
+        const streamEvent = normalizeEvent(raw);
         handleEvent(
-          event,
+          streamEvent,
           fullText,
           callbacks,
           clientTools,
@@ -379,39 +411,41 @@ async function executeStream(options) {
           onAutoResume,
           pendingAutoResumes
         );
-        if (event.type === "text_delta") {
-          fullText += event.data;
+        if (streamEvent.type === "text_delta") {
+          fullText += streamEvent.data;
         }
       }
     }
     if (buffer.trim()) {
-      const dataLine = buffer.split("\n").find((line) => line.startsWith("data:"));
-      if (dataLine) {
-        const jsonStr = dataLine.slice(5).trim();
-        if (jsonStr) {
-          try {
-            const event = normalizeEvent(JSON.parse(jsonStr));
-            handleEvent(
-              event,
-              fullText,
-              callbacks,
-              clientTools,
-              onPaused,
-              onAutoResume,
-              pendingAutoResumes
-            );
-            if (event.type === "text_delta") {
-              fullText += event.data;
-            }
-          } catch {
+      let jsonStr = "";
+      for (const line of buffer.trim().split("\n")) {
+        if (line.startsWith("data: ")) {
+          jsonStr += line.slice(6);
+        } else if (line.startsWith("data:")) {
+          jsonStr += line.slice(5);
+        }
+      }
+      jsonStr = jsonStr.trim();
+      if (jsonStr) {
+        try {
+          const raw = JSON.parse(jsonStr);
+          const streamEvent = normalizeEvent(raw);
+          handleEvent(
+            streamEvent,
+            fullText,
+            callbacks,
+            clientTools,
+            onPaused,
+            onAutoResume,
+            pendingAutoResumes
+          );
+          if (streamEvent.type === "text_delta") {
+            fullText += streamEvent.data;
           }
+        } catch {
         }
       }
     }
-    if (pendingAutoResumes.length > 0) {
-      await Promise.allSettled(pendingAutoResumes);
-    }
-    callbacks.onComplete?.(fullText);
   } catch (err) {
     if (err instanceof Error && err.name === "AbortError") {
       throw err;
@@ -420,6 +454,10 @@ async function executeStream(options) {
     callbacks.onError?.(error);
     throw error;
   }
+  if (pendingAutoResumes.length > 0) {
+    await Promise.allSettled(pendingAutoResumes);
+  }
+  callbacks.onComplete?.(fullText);
 }
 function normalizeEvent(raw) {
   const typeMap = {
@@ -447,7 +485,8 @@ function handleEvent(event, _fullText, callbacks, clientTools, onPaused, onAutoR
       break;
     }
     case "tool_call_start": {
-      const { callId, toolName, toolType, inputs, paused } = event.data;
+      const { callId, toolName, toolType, inputs: rawInputs, paused } = event.data;
+      const inputs = tryParseJSON2(rawInputs);
       callbacks.onToolStart?.(toolName, toolType);
       if (toolType === "client") {
         const tool = clientTools.get(toolName);
@@ -484,6 +523,13 @@ function handleEvent(event, _fullText, callbacks, clientTools, onPaused, onAutoR
       callbacks.onToolEnd?.(toolName, result, error);
       break;
     }
+    case "run_error": {
+      const { message, code } = event.data;
+      const error = new Error(message);
+      if (code != null) error.code = code;
+      callbacks.onError?.(error);
+      break;
+    }
   }
 }
 
@@ -575,10 +621,14 @@ async function executeAgentStream(deps, body, headers, callbacks, debugKey) {
     callbacks,
     clientTools: deps.getClientToolsMap(),
     signal,
-    onSessionId: (sessionId) => {
+    onSessionId: (sessionId, sessionName) => {
       lastRequestContextRef.current.sessionId = sessionId;
       if (!currentSessionId || currentSessionId === TEMPORARY_SESSION_ID2) {
-        createSessionFromResponse(sessionId, DEFAULT_SESSION_NAME2, messagesRef.current);
+        createSessionFromResponse(
+          sessionId,
+          sessionName || DEFAULT_SESSION_NAME2,
+          messagesRef.current
+        );
         deps.setCurrentSessionId(sessionId);
       }
     },