@ash-cloud/ash-ui 0.2.4 → 0.2.5

package/dist/index.cjs

@@ -3666,6 +3666,8 @@ function useAgentChat(options) {
     onSessionEnd,
     onError,
     onSandboxLog,
+    canUseTool,
+    resolveToolPermission,
     onReconnect,
     maxReconnectAttempts = 3,
     reconnectBaseDelay = 1e3,
@@ -3699,6 +3701,23 @@ function useAgentChat(options) {
   const emitStreamingEntries = react.useCallback((newEntries) => {
     setStreamingEntries([...newEntries]);
   }, []);
+  const handleToolPermission = react.useCallback(async (event) => {
+    if (event.type !== "tool_permission") return;
+    if (!canUseTool) return;
+    if (!event.requestId || !event.sessionId || !event.toolName) return;
+    const request = {
+      requestId: event.requestId,
+      sessionId: event.sessionId,
+      toolName: event.toolName,
+      input: event.input
+    };
+    const allow = await canUseTool(request);
+    if (!resolveToolPermission) {
+      console.warn("[useAgentChat] resolveToolPermission not provided for tool permission response");
+      return;
+    }
+    await resolveToolPermission(request, allow);
+  }, [canUseTool, resolveToolPermission]);
   const createTextEntry = react.useCallback((id, content) => ({
     id,
     timestamp: (/* @__PURE__ */ new Date()).toISOString(),
@@ -3857,6 +3876,7 @@ function useAgentChat(options) {
       for await (const event of stream) {
         if (controller.signal.aborted) break;
         eventCountRef.current++;
+        await handleToolPermission(event);
         localStreamingEntries = processEvent(event, localStreamingEntries);
         emitStreamingEntries(localStreamingEntries);
         if (event.type === "complete" || event.type === "session_end" || event.type === "error") {
@@ -3890,7 +3910,7 @@ function useAgentChat(options) {
       }
       return false;
     }
-  }, [subscribeToSession, maxReconnectAttempts, reconnectBaseDelay, onReconnect, processEvent, emitStreamingEntries]);
+  }, [subscribeToSession, maxReconnectAttempts, reconnectBaseDelay, onReconnect, processEvent, emitStreamingEntries, handleToolPermission]);
   const send = react.useCallback(async (prompt) => {
     if (isStreaming) return;
     let finalPrompt = prompt;
@@ -3969,6 +3989,7 @@ function useAgentChat(options) {
             console.error("[useAgentChat] onEvent error:", err);
           }
         }
+        await handleToolPermission(event);
         let processedEvent = event;
         if (middleware?.length) {
           processedEvent = await applyEventMiddleware(middleware, event);
@@ -4018,7 +4039,7 @@ function useAgentChat(options) {
       abortControllerRef.current = null;
       resetStreamingState();
     }
-  }, [isStreaming, sessionId, historyEntries, streamingEntries, createStream, subscribeToSession, processEvent, emitStreamingEntries, resetStreamingState, onError, attemptReconnect, onBeforeSend, onEvent, middleware]);
+  }, [isStreaming, sessionId, historyEntries, streamingEntries, createStream, subscribeToSession, processEvent, emitStreamingEntries, resetStreamingState, onError, attemptReconnect, onBeforeSend, onEvent, middleware, handleToolPermission]);
   const stop = react.useCallback(() => {
     reconnectAttemptsRef.current = maxReconnectAttempts + 1;
     setIsReconnecting(false);
@@ -4069,6 +4090,8 @@ function useChat(options) {
     initialSessionId,
     initialMessages = [],
     onToolCall,
+    canUseTool,
+    resolveToolPermission,
     onFinish,
     onError,
     onSessionStart,
@@ -4103,6 +4126,23 @@ function useChat(options) {
       return prev;
     });
   }, []);
+  const handleToolPermission = react.useCallback(async (event) => {
+    if (event.type !== "tool_permission") return;
+    if (!canUseTool) return;
+    if (!event.requestId || !event.sessionId || !event.toolName) return;
+    const request = {
+      requestId: event.requestId,
+      sessionId: event.sessionId,
+      toolName: event.toolName,
+      input: event.input
+    };
+    const allow = await canUseTool(request);
+    if (!resolveToolPermission) {
+      console.warn("[useChat] resolveToolPermission not provided for tool permission response");
+      return;
+    }
+    await resolveToolPermission(request, allow);
+  }, [canUseTool, resolveToolPermission]);
   const processEvent = react.useCallback((event) => {
     switch (event.type) {
       case "session_start":
@@ -4222,6 +4262,7 @@ function useChat(options) {
       const stream = createStream(finalPrompt, sessionIdRef.current || void 0, streamOptions);
       for await (const event of stream) {
         if (controller.signal.aborted) break;
+        await handleToolPermission(event);
         let processedEvent = event;
         if (middleware?.length) {
           processedEvent = await applyEventMiddleware(middleware, event);
@@ -4251,7 +4292,7 @@ function useChat(options) {
       abortControllerRef.current = null;
       currentAssistantMessageRef.current = null;
     }
-  }, [isLoading, createStream, processEvent, middleware, onError]);
+  }, [isLoading, createStream, processEvent, middleware, onError, handleToolPermission]);
   const stop = react.useCallback(() => {
     reconnectAttemptsRef.current = maxReconnectAttempts + 1;
     setIsReconnecting(false);