bs-agent 0.0.24 → 0.0.25

package/dist/react/index.cjs

@@ -372,23 +372,17 @@ function buildStreamCallbacks(deps, debugKey) {
   return {
     onComplete: () => {
       console.log("Agent closed");
-      setMessages((current) => {
-        if (syncSessionRef.current) {
-          syncSessionRef.current(current);
-        }
-        return current;
-      });
       setInProgress(false);
+      if (syncSessionRef.current) {
+        syncSessionRef.current(messagesRef.current);
+      }
     },
     onError: (error) => {
       console.log("Agent error", error);
-      setMessages((current) => {
-        if (syncSessionRef.current) {
-          syncSessionRef.current(current);
-        }
-        return current;
-      });
       setInProgress(false);
+      if (syncSessionRef.current) {
+        syncSessionRef.current(messagesRef.current);
+      }
     },
     onEvent: (event) => {
       if (event.type === "text_delta") {
@@ -515,25 +509,7 @@ function handleTextDelta(event, setMessages, syncSessionRef, modifier, fullTextM
 }
 function handleClientToolCall(event, setMessages, syncSessionRef, toolContext, agentId) {
   const tool = toolContext?.getTool(agentId, event.data.toolName);
-  console.log(
-    "[DEBUG] handleClientToolCall:",
-    event.data.toolName,
-    "paused:",
-    event.data.paused,
-    "toolFound:",
-    !!tool,
-    "hasRender:",
-    !!tool?.render
-  );
-  if (!tool?.render && !event.data.paused) {
-    console.log("[DEBUG] handleClientToolCall: SKIPPING part creation \u2014 not paused and no render");
-    return;
-  }
-  console.log(
-    "[DEBUG] handleClientToolCall: CREATING widget part for",
-    event.data.toolName,
-    event.data.callId
-  );
+  if (!tool?.render && !event.data.paused) return;
   setMessages((prev) => {
     const lastMessage = prev[prev.length - 1];
     const newPart = {
@@ -821,34 +797,19 @@ function useAgent(agent, options) {
       sessionUtils.syncSessionRef.current(messagesRef.current);
     }
   }, [sessionUtils.syncSessionRef]);
-  return (0, import_react3.useMemo)(
-    () => ({
-      inProgress,
-      messages,
-      handleSend,
-      resumeTool,
-      addOptimisticMessage,
-      abort,
-      sessionId: currentSessionId,
-      switchSession: sessionUtils.switchSession,
-      deleteSession: sessionUtils.deleteSession,
-      sessions: sessionUtils.sessionsList,
-      debugData
-    }),
-    [
-      inProgress,
-      messages,
-      handleSend,
-      resumeTool,
-      addOptimisticMessage,
-      abort,
-      currentSessionId,
-      sessionUtils.switchSession,
-      sessionUtils.deleteSession,
-      sessionUtils.sessionsList,
-      debugData
-    ]
-  );
+  return {
+    inProgress,
+    messages,
+    handleSend,
+    resumeTool,
+    addOptimisticMessage,
+    abort,
+    sessionId: currentSessionId,
+    switchSession: sessionUtils.switchSession,
+    deleteSession: sessionUtils.deleteSession,
+    sessions: sessionUtils.sessionsList,
+    debugData
+  };
 }
 
 // src/core/stream.ts
@@ -1667,19 +1628,6 @@ function ToolRenderer({ agentId, part }) {
     throw new Error("ToolRenderer must be used within <AgentContextProvider>");
   }
   const tool = context.getTool(agentId, part.toolName);
-  console.log(
-    "[DEBUG] ToolRenderer:",
-    part.toolName,
-    part.callId,
-    "toolFound:",
-    !!tool,
-    "hasRender:",
-    !!tool?.render,
-    "paused:",
-    part.paused,
-    "status:",
-    localStatus
-  );
   const handleSubmit = (0, import_react6.useCallback)(
     (result) => {
       if (localStatus === "submitted") return;