@ash-cloud/ash-ui 0.2.4 → 0.2.5

package/dist/index.d.cts

@@ -1059,6 +1059,7 @@ interface StreamEvent {
     text?: string;
     /** Message content array (for 'message' events) */
     content?: unknown;
+    requestId?: string;
     toolId?: string;
     toolName?: string;
     input?: unknown;
@@ -1071,6 +1072,8 @@ interface StreamEvent {
     result?: string;
     totalCost?: number;
     totalTokens?: number;
+    stop_reason?: string;
+    stop_sequence?: string;
     /** SSE event ID (sequence number from relay) for reconnection */
     id?: string;
     entry?: {
@@ -1081,6 +1084,12 @@ interface StreamEvent {
         data?: Record<string, unknown>;
     };
 }
+interface ToolPermissionRequest {
+    requestId: string;
+    sessionId: string;
+    toolName: string;
+    input?: unknown;
+}
 /**
  * Async generator that yields StreamEvent objects
  */
@@ -1204,6 +1213,15 @@ interface UseAgentChatOptions {
      * Callback for sandbox log events
      */
     onSandboxLog?: (entry: NonNullable<StreamEvent['entry']>) => void;
+    /**
+     * Callback to approve/deny tool usage requests.
+     * When provided, tool permission events will be resolved automatically.
+     */
+    canUseTool?: (request: ToolPermissionRequest) => boolean | Promise<boolean>;
+    /**
+     * Function to resolve a tool permission request (sends the decision to the server).
+     */
+    resolveToolPermission?: (request: ToolPermissionRequest, allow: boolean) => Promise<void>;
     /**
      * Callback when reconnection is attempted
      */
@@ -1409,6 +1427,15 @@ interface UseChatOptions {
      * Callback when a tool is called
      */
     onToolCall?: (handler: ToolCallHandler) => void;
+    /**
+     * Callback to approve/deny tool usage requests.
+     * When provided, tool permission events will be resolved automatically.
+     */
+    canUseTool?: (request: ToolPermissionRequest) => boolean | Promise<boolean>;
+    /**
+     * Function to resolve a tool permission request (sends the decision to the server).
+     */
+    resolveToolPermission?: (request: ToolPermissionRequest, allow: boolean) => Promise<void>;
     /**
      * Callback when streaming finishes
      */

package/dist/index.d.ts

@@ -1059,6 +1059,7 @@ interface StreamEvent {
     text?: string;
     /** Message content array (for 'message' events) */
     content?: unknown;
+    requestId?: string;
     toolId?: string;
     toolName?: string;
     input?: unknown;
@@ -1071,6 +1072,8 @@ interface StreamEvent {
     result?: string;
     totalCost?: number;
     totalTokens?: number;
+    stop_reason?: string;
+    stop_sequence?: string;
     /** SSE event ID (sequence number from relay) for reconnection */
     id?: string;
     entry?: {
@@ -1081,6 +1084,12 @@ interface StreamEvent {
         data?: Record<string, unknown>;
     };
 }
+interface ToolPermissionRequest {
+    requestId: string;
+    sessionId: string;
+    toolName: string;
+    input?: unknown;
+}
 /**
  * Async generator that yields StreamEvent objects
  */
@@ -1204,6 +1213,15 @@ interface UseAgentChatOptions {
      * Callback for sandbox log events
      */
     onSandboxLog?: (entry: NonNullable<StreamEvent['entry']>) => void;
+    /**
+     * Callback to approve/deny tool usage requests.
+     * When provided, tool permission events will be resolved automatically.
+     */
+    canUseTool?: (request: ToolPermissionRequest) => boolean | Promise<boolean>;
+    /**
+     * Function to resolve a tool permission request (sends the decision to the server).
+     */
+    resolveToolPermission?: (request: ToolPermissionRequest, allow: boolean) => Promise<void>;
     /**
      * Callback when reconnection is attempted
      */
@@ -1409,6 +1427,15 @@ interface UseChatOptions {
      * Callback when a tool is called
      */
     onToolCall?: (handler: ToolCallHandler) => void;
+    /**
+     * Callback to approve/deny tool usage requests.
+     * When provided, tool permission events will be resolved automatically.
+     */
+    canUseTool?: (request: ToolPermissionRequest) => boolean | Promise<boolean>;
+    /**
+     * Function to resolve a tool permission request (sends the decision to the server).
+     */
+    resolveToolPermission?: (request: ToolPermissionRequest, allow: boolean) => Promise<void>;
     /**
      * Callback when streaming finishes
      */

package/dist/index.js

@@ -3664,6 +3664,8 @@ function useAgentChat(options) {
     onSessionEnd,
     onError,
     onSandboxLog,
+    canUseTool,
+    resolveToolPermission,
     onReconnect,
     maxReconnectAttempts = 3,
     reconnectBaseDelay = 1e3,
@@ -3697,6 +3699,23 @@ function useAgentChat(options) {
   const emitStreamingEntries = useCallback((newEntries) => {
     setStreamingEntries([...newEntries]);
   }, []);
+  const handleToolPermission = 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 = useCallback((id, content) => ({
     id,
     timestamp: (/* @__PURE__ */ new Date()).toISOString(),
@@ -3855,6 +3874,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") {
@@ -3888,7 +3908,7 @@ function useAgentChat(options) {
       }
       return false;
     }
-  }, [subscribeToSession, maxReconnectAttempts, reconnectBaseDelay, onReconnect, processEvent, emitStreamingEntries]);
+  }, [subscribeToSession, maxReconnectAttempts, reconnectBaseDelay, onReconnect, processEvent, emitStreamingEntries, handleToolPermission]);
   const send = useCallback(async (prompt) => {
     if (isStreaming) return;
     let finalPrompt = prompt;
@@ -3967,6 +3987,7 @@ function useAgentChat(options) {
             console.error("[useAgentChat] onEvent error:", err);
           }
         }
+        await handleToolPermission(event);
         let processedEvent = event;
         if (middleware?.length) {
           processedEvent = await applyEventMiddleware(middleware, event);
@@ -4016,7 +4037,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 = useCallback(() => {
     reconnectAttemptsRef.current = maxReconnectAttempts + 1;
     setIsReconnecting(false);
@@ -4067,6 +4088,8 @@ function useChat(options) {
     initialSessionId,
     initialMessages = [],
     onToolCall,
+    canUseTool,
+    resolveToolPermission,
     onFinish,
     onError,
     onSessionStart,
@@ -4101,6 +4124,23 @@ function useChat(options) {
       return prev;
     });
   }, []);
+  const handleToolPermission = 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 = useCallback((event) => {
     switch (event.type) {
       case "session_start":
@@ -4220,6 +4260,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);
@@ -4249,7 +4290,7 @@ function useChat(options) {
       abortControllerRef.current = null;
       currentAssistantMessageRef.current = null;
     }
-  }, [isLoading, createStream, processEvent, middleware, onError]);
+  }, [isLoading, createStream, processEvent, middleware, onError, handleToolPermission]);
   const stop = useCallback(() => {
     reconnectAttemptsRef.current = maxReconnectAttempts + 1;
     setIsReconnecting(false);