@bytexbyte/nxtlinq-ai-agent-sdk 1.2.5 → 1.2.6

package/dist/components/context/ChatBotContext.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ChatBotContext.d.ts","sourceRoot":"","sources":["../../../src/components/context/ChatBotContext.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,KAAK,MAAM,OAAO,CAAC;AAO/B,OAAO,EACL,kBAAkB,EAClB,YAAY,EAOb,MAAM,uBAAuB,CAAC;AAI/B,eAAO,MAAM,UAAU,0BAMtB,CAAC;AAEF,eAAO,MAAM,eAAe,EAAE,KAAK,CAAC,EAAE,CAAC,YAAY,CAg7BlD,CAAC"}
+{"version":3,"file":"ChatBotContext.d.ts","sourceRoot":"","sources":["../../../src/components/context/ChatBotContext.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,KAAK,MAAM,OAAO,CAAC;AAO/B,OAAO,EACL,kBAAkB,EAClB,YAAY,EAOb,MAAM,uBAAuB,CAAC;AAI/B,eAAO,MAAM,UAAU,0BAMtB,CAAC;AAEF,eAAO,MAAM,eAAe,EAAE,KAAK,CAAC,EAAE,CAAC,YAAY,CAqpClD,CAAC"}

package/dist/components/context/ChatBotContext.js

@@ -33,7 +33,27 @@ availableModels = DEFAULT_AI_MODELS, defaultModelIndex = 0, showModelSelector =
     const [signer, setSigner] = React.useState(null);
     const [walletInfo, setWalletInfo] = React.useState(null);
     const [isWalletLoading, setIsWalletLoading] = React.useState(false);
+    const [isAutoConnecting, setIsAutoConnecting] = React.useState(false);
+    const [justAutoConnected, setJustAutoConnected] = React.useState(false);
     const [nxtlinqAITServiceAccessToken, setNxtlinqAITServiceAccessToken] = useLocalStorage('nxtlinqAITServiceAccessToken', '');
+    // Use refs to get latest state values in hasPermission function
+    const hitAddressRef = React.useRef(hitAddress);
+    const aitRef = React.useRef(ait);
+    const permissionsRef = React.useRef(permissions);
+    const nxtlinqAITServiceAccessTokenRef = React.useRef(nxtlinqAITServiceAccessToken);
+    // Update refs when state changes
+    React.useEffect(() => {
+        hitAddressRef.current = hitAddress;
+    }, [hitAddress]);
+    React.useEffect(() => {
+        aitRef.current = ait;
+    }, [ait]);
+    React.useEffect(() => {
+        permissionsRef.current = permissions;
+    }, [permissions]);
+    React.useEffect(() => {
+        nxtlinqAITServiceAccessTokenRef.current = nxtlinqAITServiceAccessToken;
+    }, [nxtlinqAITServiceAccessToken]);
     const [notification, setNotification] = React.useState({
         show: false,
         type: 'info',
@@ -170,6 +190,7 @@ availableModels = DEFAULT_AI_MODELS, defaultModelIndex = 0, showModelSelector =
             localStorage.setItem('walletAddress', userAddress);
             setHitAddress(userAddress);
             setSigner(userSigner);
+            // Wait for AIT to be loaded if we have a token
             if (nxtlinqAITServiceAccessToken) {
                 await refreshAIT();
             }
@@ -230,10 +251,11 @@ availableModels = DEFAULT_AI_MODELS, defaultModelIndex = 0, showModelSelector =
             }
             const { accessToken } = response;
             setNxtlinqAITServiceAccessToken(accessToken);
+            // Wait for AIT to be loaded before showing success message
+            await refreshAIT();
             if (autoShowSuccessMessage) {
                 showSuccess('Successfully signed in with your HIT wallet. You can now use the AI agent.');
             }
-            refreshAIT();
         }
         catch (error) {
             console.error('Failed to sign in:', error);
@@ -241,70 +263,164 @@ availableModels = DEFAULT_AI_MODELS, defaultModelIndex = 0, showModelSelector =
         }
     };
     // Check permissions
-    const hasPermission = async (toolName) => {
-        if (!hitAddress) {
-            setMessages(prev => [...prev, {
-                    id: Date.now().toString(),
-                    content: 'Please connect your HIT wallet to continue.',
-                    role: 'assistant',
-                    timestamp: new Date().toISOString(),
-                    button: 'connect'
-                }]);
+    const hasPermission = async (toolName, autoRetry = true, onAutoConnect) => {
+        // Use refs to get latest state values
+        const currentHitAddress = hitAddressRef.current;
+        const currentAit = aitRef.current;
+        const currentPermissions = permissionsRef.current;
+        const currentToken = nxtlinqAITServiceAccessTokenRef.current;
+        if (!currentHitAddress) {
+            if (autoRetry) {
+                setIsLoading(false); // Stop thinking before showing message
+                setMessages(prev => [...prev, {
+                        id: Date.now().toString(),
+                        content: 'Please connect your HIT wallet to continue.',
+                        role: 'assistant',
+                        timestamp: new Date().toISOString(),
+                        button: 'connect'
+                    }]);
+                try {
+                    setIsAutoConnecting(true); // Mark as auto-connecting
+                    await connectWallet(false); // Don't show sign-in message yet
+                    setIsAutoConnecting(false); // Clear auto-connecting state
+                    setJustAutoConnected(true); // Mark that we just auto-connected
+                    console.log('Set justAutoConnected to true after auto-connect');
+                    onAutoConnect?.(); // Call callback if provided
+                    // If connection successful, continue with permission check
+                    const result = await hasPermission(toolName, false);
+                    return result;
+                }
+                catch (error) {
+                    console.error('Failed to auto-connect wallet:', error);
+                    setIsAutoConnecting(false); // Clear auto-connecting state on error
+                    return false;
+                }
+            }
+            // If autoRetry is false, don't show message again, just return false
             return false;
         }
-        if (!nxtlinqAITServiceAccessToken) {
-            setMessages(prev => [...prev, {
-                    id: Date.now().toString(),
-                    content: 'Please sign in with your HIT wallet to continue.',
-                    role: 'assistant',
-                    timestamp: new Date().toISOString(),
-                    button: 'signIn'
-                }]);
+        if (!currentToken) {
+            if (autoRetry) {
+                setIsLoading(false); // Stop thinking before showing message
+                setMessages(prev => [...prev, {
+                        id: Date.now().toString(),
+                        content: 'Please sign in with your HIT wallet to continue.',
+                        role: 'assistant',
+                        timestamp: new Date().toISOString(),
+                        button: 'signIn'
+                    }]);
+                try {
+                    setIsAutoConnecting(true); // Mark as auto-signing
+                    await signInWallet(false); // Don't show success message yet
+                    setIsAutoConnecting(false); // Clear auto-signing state
+                    // If sign-in successful, continue with permission check
+                    const result = await hasPermission(toolName, false);
+                    return result;
+                }
+                catch (error) {
+                    console.error('Failed to auto-sign-in wallet:', error);
+                    setIsAutoConnecting(false); // Clear auto-signing state on error
+                    return false;
+                }
+            }
+            // If autoRetry is false, don't show message again, just return false
             return false;
         }
         try {
-            const payload = JSON.parse(atob(nxtlinqAITServiceAccessToken.split('.')[1]));
+            const payload = JSON.parse(atob(currentToken.split('.')[1]));
             const exp = payload.exp * 1000;
             const now = Date.now();
             if (exp < now) {
                 setNxtlinqAITServiceAccessToken('');
-                setMessages(prev => [...prev, {
-                        id: Date.now().toString(),
-                        content: 'Your wallet session has expired. Please sign in again.',
-                        role: 'assistant',
-                        timestamp: new Date().toISOString(),
-                        button: 'signIn'
-                    }]);
+                if (autoRetry) {
+                    setIsLoading(false); // Stop thinking before showing message
+                    setMessages(prev => [...prev, {
+                            id: Date.now().toString(),
+                            content: 'Your wallet session has expired. Please sign in again.',
+                            role: 'assistant',
+                            timestamp: new Date().toISOString(),
+                            button: 'signIn'
+                        }]);
+                    try {
+                        setIsAutoConnecting(true); // Mark as auto-signing
+                        await signInWallet(false); // Don't show success message yet
+                        setIsAutoConnecting(false); // Clear auto-signing state
+                        // If sign-in successful, continue with permission check
+                        const result = await hasPermission(toolName, false);
+                        return result;
+                    }
+                    catch (error) {
+                        console.error('Failed to auto-sign-in after session expiry:', error);
+                        setIsAutoConnecting(false); // Clear auto-signing state on error
+                        return false;
+                    }
+                }
+                // If autoRetry is false, don't show message again, just return false
                 return false;
             }
             const address = payload.address;
-            if (address !== hitAddress) {
+            if (address !== currentHitAddress) {
                 setNxtlinqAITServiceAccessToken('');
+                if (autoRetry) {
+                    setIsLoading(false); // Stop thinking before showing message
+                    setMessages(prev => [...prev, {
+                            id: Date.now().toString(),
+                            content: 'Wallet address mismatch. Please sign in with the correct wallet.',
+                            role: 'assistant',
+                            timestamp: new Date().toISOString(),
+                            button: 'signIn'
+                        }]);
+                    try {
+                        setIsAutoConnecting(true); // Mark as auto-signing
+                        await signInWallet(false); // Don't show success message yet
+                        setIsAutoConnecting(false); // Clear auto-signing state
+                        // If sign-in successful, continue with permission check
+                        const result = await hasPermission(toolName, false);
+                        return result;
+                    }
+                    catch (error) {
+                        console.error('Failed to auto-sign-in after address mismatch:', error);
+                        setIsAutoConnecting(false); // Clear auto-signing state on error
+                        return false;
+                    }
+                }
+                // If autoRetry is false, don't show message again, just return false
+                return false;
+            }
+        }
+        catch (error) {
+            console.error('Error parsing token:', error);
+            setNxtlinqAITServiceAccessToken('');
+            if (autoRetry) {
+                setIsLoading(false); // Stop thinking before showing message
                 setMessages(prev => [...prev, {
                         id: Date.now().toString(),
-                        content: 'Wallet address mismatch. Please sign in with the correct wallet.',
+                        content: 'Invalid wallet session. Please sign in again.',
                         role: 'assistant',
                         timestamp: new Date().toISOString(),
                         button: 'signIn'
                     }]);
-                return false;
+                try {
+                    setIsAutoConnecting(true); // Mark as auto-signing
+                    await signInWallet(false); // Don't show success message yet
+                    setIsAutoConnecting(false); // Clear auto-signing state
+                    // If sign-in successful, continue with permission check
+                    const result = await hasPermission(toolName, false);
+                    return result;
+                }
+                catch (signInError) {
+                    console.error('Failed to auto-sign-in after token parse error:', signInError);
+                    setIsAutoConnecting(false); // Clear auto-signing state on error
+                    return false;
+                }
             }
-        }
-        catch (error) {
-            console.error('Error parsing token:', error);
-            setNxtlinqAITServiceAccessToken('');
-            setMessages(prev => [...prev, {
-                    id: Date.now().toString(),
-                    content: 'Invalid wallet session. Please sign in again.',
-                    role: 'assistant',
-                    timestamp: new Date().toISOString(),
-                    button: 'signIn'
-                }]);
+            // If autoRetry is false, don't show message again, just return false
             return false;
         }
-        if (!ait) {
+        if (!currentAit) {
             // Show loading message if AIT is still loading
             if (isAITLoading) {
+                setIsLoading(false); // Stop thinking before showing message
                 setMessages(prev => [...prev, {
                         id: Date.now().toString(),
                         content: 'Loading your wallet configuration... Please wait a moment.',
@@ -313,16 +429,55 @@ availableModels = DEFAULT_AI_MODELS, defaultModelIndex = 0, showModelSelector =
                     }]);
                 return false;
             }
-            setMessages(prev => [...prev, {
-                    id: Date.now().toString(),
-                    content: 'No AIT found for your wallet. Please click the settings button (⚙️) to configure your AIT permissions.',
-                    role: 'assistant',
-                    timestamp: new Date().toISOString()
-                }]);
-            return false;
+            // If AIT is not loaded but we have a token, try to refresh it once
+            if (currentToken && !isAITLoading) {
+                try {
+                    await refreshAIT();
+                    // Wait for AIT to be loaded with polling
+                    let attempts = 0;
+                    const maxAttempts = 10; // Wait up to 5 seconds
+                    while (!aitRef.current && attempts < maxAttempts) {
+                        await new Promise(resolve => setTimeout(resolve, 500));
+                        attempts++;
+                    }
+                    // Check again after refresh
+                    if (!aitRef.current) {
+                        setIsLoading(false); // Stop thinking before showing message
+                        setMessages(prev => [...prev, {
+                                id: Date.now().toString(),
+                                content: 'No AIT found for your wallet. Please click the settings button (⚙️) to configure your AIT permissions.',
+                                role: 'assistant',
+                                timestamp: new Date().toISOString()
+                            }]);
+                        return false;
+                    }
+                }
+                catch (error) {
+                    console.error('Failed to refresh AIT during permission check:', error);
+                    setIsLoading(false); // Stop thinking before showing message
+                    setMessages(prev => [...prev, {
+                            id: Date.now().toString(),
+                            content: 'No AIT found for your wallet. Please click the settings button (⚙️) to configure your AIT permissions.',
+                            role: 'assistant',
+                            timestamp: new Date().toISOString()
+                        }]);
+                    return false;
+                }
+            }
+            else {
+                setIsLoading(false); // Stop thinking before showing message
+                setMessages(prev => [...prev, {
+                        id: Date.now().toString(),
+                        content: 'No AIT found for your wallet. Please click the settings button (⚙️) to configure your AIT permissions.',
+                        role: 'assistant',
+                        timestamp: new Date().toISOString()
+                    }]);
+                return false;
+            }
         }
         const availablePermissionLabels = availablePermissions.map(p => p.label);
         if (availablePermissions.length === 0) {
+            setIsLoading(false); // Stop thinking before showing message
             setMessages(prev => [...prev, {
                     id: Date.now().toString(),
                     content: `No permissions available for your current identity provider. Please check your service configuration or contact support. Service ID: ${serviceId}, Permission Group: ${permissionGroup || 'None'}`,
@@ -332,6 +487,7 @@ availableModels = DEFAULT_AI_MODELS, defaultModelIndex = 0, showModelSelector =
             return false;
         }
         if (!availablePermissionLabels.includes(toolName)) {
+            setIsLoading(false); // Stop thinking before showing message
             setMessages(prev => [...prev, {
                     id: Date.now().toString(),
                     content: `This tool (${toolName}) is not available for your current identity provider.`,
@@ -340,12 +496,20 @@ availableModels = DEFAULT_AI_MODELS, defaultModelIndex = 0, showModelSelector =
                 }]);
             return false;
         }
-        if (!permissions.includes(toolName)) {
+        if (!currentPermissions.includes(toolName)) {
+            setIsLoading(false); // Stop thinking before showing message
             setMessages(prev => [...prev, {
                     id: Date.now().toString(),
-                    content: `You do not have the required AIT permission: ${toolName}. Please click the settings button (⚙️) to update your permissions.`,
+                    content: `You don't have the required AIT permission: ${toolName}. Would you like to enable AIT permissions?`,
                     role: 'assistant',
-                    timestamp: new Date().toISOString()
+                    timestamp: new Date().toISOString(),
+                    button: 'enableAIT',
+                    metadata: {
+                        model: 'gpt-4',
+                        permissions: currentPermissions,
+                        issuedBy: currentHitAddress || '',
+                        toolName: toolName
+                    }
                 }]);
             return false;
         }
@@ -361,23 +525,27 @@ availableModels = DEFAULT_AI_MODELS, defaultModelIndex = 0, showModelSelector =
         return availableModels[selectedModelIndex];
     }, [availableModels, selectedModelIndex]);
     // Updated sendMessage function to support different AI models
-    const sendMessage = async (content, retryCount = 0) => {
+    const sendMessage = async (content, retryCount = 0, isPresetMessage = false) => {
         if (!content.trim() || isLoading)
             return;
         const currentModel = getCurrentModel();
-        const userMessage = {
-            id: Date.now().toString(),
-            content,
-            role: 'user',
-            timestamp: new Date().toISOString(),
-            metadata: {
-                model: currentModel.value,
-                permissions: permissions,
-                issuedBy: hitAddress || ''
-            }
-        };
-        setMessages(prev => [...prev, userMessage]);
-        setInputValue('');
+        // Only add user message on first attempt, not on retries
+        // Also check if this is not a preset message (preset messages are added separately)
+        if (retryCount === 0 && !isPresetMessage) {
+            const userMessage = {
+                id: Date.now().toString(),
+                content,
+                role: 'user',
+                timestamp: new Date().toISOString(),
+                metadata: {
+                    model: currentModel.value,
+                    permissions: permissions,
+                    issuedBy: hitAddress || ''
+                }
+            };
+            setMessages(prev => [...prev, userMessage]);
+            setInputValue('');
+        }
         setIsLoading(true);
         try {
             // Build context messages
@@ -400,8 +568,21 @@ availableModels = DEFAULT_AI_MODELS, defaultModelIndex = 0, showModelSelector =
                 // Handle tool calls
                 const toolUse = response.toolCall.toolUse;
                 if (onToolUse) {
-                    const isToolAllowed = await hasPermission(toolUse.name);
+                    let wasAutoConnected = false;
+                    const isToolAllowed = await hasPermission(toolUse.name, true, () => {
+                        wasAutoConnected = true;
+                    });
                     if (!isToolAllowed) {
+                        // If we just auto-connected, retry the message once
+                        if (wasAutoConnected && retryCount < 1) {
+                            console.log('Auto-connected, retrying message...');
+                            setTimeout(() => {
+                                sendMessage(content, retryCount + 1, isPresetMessage);
+                            }, 2000);
+                        }
+                        else {
+                            console.log('Permission denied, not retrying. wasAutoConnected:', wasAutoConnected, 'retryCount:', retryCount);
+                        }
                         return;
                     }
                     const toolResult = await onToolUse(toolUse);
@@ -486,8 +667,24 @@ availableModels = DEFAULT_AI_MODELS, defaultModelIndex = 0, showModelSelector =
         catch (error) {
             console.error('Failed to send message:', error);
             if (retryCount < maxRetries) {
+                // Show retry indicator to user
+                const retryMessage = {
+                    id: `retry-${Date.now()}-${retryCount}`,
+                    content: `Retrying... (Attempt ${retryCount + 1}/${maxRetries})`,
+                    role: 'assistant',
+                    timestamp: new Date().toISOString(),
+                    metadata: {
+                        model: currentModel.value,
+                        permissions: permissions,
+                        issuedBy: hitAddress || '',
+                        isRetry: true
+                    }
+                };
+                setMessages(prev => [...prev, retryMessage]);
                 setTimeout(() => {
-                    sendMessage(content, retryCount + 1);
+                    // Remove retry message before next attempt
+                    setMessages(prev => prev.filter(msg => msg.id !== retryMessage.id));
+                    sendMessage(content, retryCount + 1, isPresetMessage);
                 }, retryDelay);
                 return;
             }
@@ -552,12 +749,12 @@ availableModels = DEFAULT_AI_MODELS, defaultModelIndex = 0, showModelSelector =
     };
     // Handle preset message
     const handlePresetMessage = (message) => {
-        // Show loading message if AIT is still loading
-        if (isAITLoading) {
-            const loadingMessage = {
+        if (message.autoSend) {
+            // For preset messages, we need to add the user message first since sendMessage won't add it on retries
+            const userMessage = {
                 id: Date.now().toString(),
-                content: 'Loading your wallet configuration... Please wait a moment.',
-                role: 'assistant',
+                content: message.text,
+                role: 'user',
                 timestamp: new Date().toISOString(),
                 metadata: {
                     model: getCurrentModel().value,
@@ -565,22 +762,9 @@ availableModels = DEFAULT_AI_MODELS, defaultModelIndex = 0, showModelSelector =
                     issuedBy: hitAddress || ''
                 }
             };
-            setMessages(prev => [...prev, loadingMessage]);
-            return;
-        }
-        if (message.autoSend) {
-            setMessages(prev => [...prev, {
-                    id: Date.now().toString(),
-                    content: message.text,
-                    role: 'user',
-                    timestamp: new Date().toISOString(),
-                    metadata: {
-                        model: getCurrentModel().value,
-                        permissions: permissions,
-                        issuedBy: hitAddress || ''
-                    }
-                }]);
-            sendMessage(message.text);
+            setMessages(prev => [...prev, userMessage]);
+            // Pass a flag to indicate this is a preset message so sendMessage won't add user message again
+            sendMessage(message.text, 0, true);
         }
         else {
             setInputValue(message.text);
@@ -625,6 +809,41 @@ availableModels = DEFAULT_AI_MODELS, defaultModelIndex = 0, showModelSelector =
         setAit(aitInfo);
         setPermissions(permissions);
     };
+    // Auto enable AIT permissions
+    const autoEnableAIT = async (toolName) => {
+        if (!signer || !hitAddress) {
+            showError('Please connect your wallet first');
+            return false;
+        }
+        try {
+            // Get available permissions to find the tool
+            const availablePermissionLabels = availablePermissions.map(p => p.label);
+            if (!availablePermissionLabels.includes(toolName)) {
+                showError(`Tool ${toolName} is not available for your current identity provider`);
+                return false;
+            }
+            // Add the required permission to current permissions
+            const newPermissions = [...permissions];
+            if (!newPermissions.includes(toolName)) {
+                newPermissions.push(toolName);
+            }
+            // Generate and register AIT with new permissions
+            await generateAndRegisterAIT(newPermissions);
+            showSuccess('AIT permissions enabled successfully! You can now use the AI agent.');
+            await refreshAIT(true);
+            return true;
+        }
+        catch (error) {
+            console.error('Failed to auto-enable AIT:', error);
+            if (error instanceof Error) {
+                showError(error.message);
+            }
+            else {
+                showError('Failed to enable AIT permissions. Please try again.');
+            }
+            return false;
+        }
+    };
     // Save permissions
     const savePermissions = async (newPermissions) => {
         setIsDisabled(true);
@@ -811,6 +1030,7 @@ availableModels = DEFAULT_AI_MODELS, defaultModelIndex = 0, showModelSelector =
         isDisabled,
         walletInfo,
         isWalletLoading,
+        isAutoConnecting,
         notification,
         // AI Model related state
         availableModels,
@@ -835,6 +1055,7 @@ availableModels = DEFAULT_AI_MODELS, defaultModelIndex = 0, showModelSelector =
         handleSubmit,
         handlePresetMessage,
         savePermissions,
+        autoEnableAIT,
         handleVerifyWalletClick,
         showSuccess,
         showError,

package/dist/components/types/ChatBotTypes.d.ts

@@ -68,6 +68,7 @@ export interface ChatBotContextType {
     isDisabled: boolean;
     walletInfo: any;
     isWalletLoading: boolean;
+    isAutoConnecting: boolean;
     notification: {
         show: boolean;
         type: 'success' | 'error' | 'warning' | 'info';
@@ -94,6 +95,7 @@ export interface ChatBotContextType {
     handleSubmit: (e: React.FormEvent) => Promise<void>;
     handlePresetMessage: (message: PresetMessage) => void;
     savePermissions: (newPermissions?: string[]) => Promise<void>;
+    autoEnableAIT: (toolName: string) => Promise<boolean>;
     handleVerifyWalletClick: () => Promise<void>;
     showSuccess: (message: string) => void;
     showError: (message: string) => void;

package/dist/components/types/ChatBotTypes.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ChatBotTypes.d.ts","sourceRoot":"","sources":["../../../src/components/types/ChatBotTypes.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,GAAG,EAAE,iBAAiB,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAE9E,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,OAAO,CAAC;CACpB;AAED,MAAM,WAAW,OAAO;IACtB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC5B;AAED,MAAM,WAAW,QAAQ;IACvB,OAAO,EAAE,OAAO,CAAC;CAClB;AAED,MAAM,WAAW,YAAY;IAC3B,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC/B,QAAQ,CAAC,EAAE,QAAQ,CAAC;CACrB;AAED,MAAM,WAAW,SAAS;IACxB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,WAAW;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,EAAE,CAAC;IACtB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAGD,MAAM,WAAW,OAAO;IACtB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;CACf;AAED,eAAO,MAAM,iBAAiB,EAAE,OAAO,EAMtC,CAAC;AAEF,eAAO,MAAM,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAM/C,CAAC;AAEF,MAAM,WAAW,YAAY;IAC3B,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;IACvC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACjC,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC,CAAC;IAC1D,cAAc,CAAC,EAAE,aAAa,EAAE,CAAC;IACjC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,cAAc,CAAC,EAAE,MAAM,OAAO,CAAC;QAC7B,KAAK,EAAE,MAAM,CAAC;KACf,GAAG,SAAS,CAAC,CAAC;IACf,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,EAAE,KAAK,CAAC,SAAS,CAAC;IAE3B,eAAe,CAAC,EAAE,OAAO,EAAE,CAAC;IAC5B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;CAC1C;AAED,MAAM,WAAW,kBAAkB;IAEjC,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,OAAO,CAAC;IACnB,MAAM,EAAE,OAAO,CAAC;IAChB,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,GAAG,EAAE,GAAG,GAAG,IAAI,CAAC;IAChB,WAAW,EAAE,MAAM,EAAE,CAAC;IACtB,oBAAoB,EAAE,iBAAiB,EAAE,CAAC;IAC1C,kBAAkB,EAAE,OAAO,CAAC;IAC5B,oBAAoB,EAAE,OAAO,CAAC;IAC9B,YAAY,EAAE,OAAO,CAAC;IACtB,UAAU,EAAE,OAAO,CAAC;IACpB,UAAU,EAAE,GAAG,CAAC;IAChB,eAAe,EAAE,OAAO,CAAC;IACzB,YAAY,EAAE;QACZ,IAAI,EAAE,OAAO,CAAC;QACd,IAAI,EAAE,SAAS,GAAG,OAAO,GAAG,SAAS,GAAG,MAAM,CAAC;QAC/C,OAAO,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,OAAO,CAAC;QACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB,CAAC;IAEF,eAAe,EAAE,OAAO,EAAE,CAAC;IAC3B,kBAAkB,EAAE,MAAM,CAAC;IAC3B,iBAAiB,EAAE,OAAO,CAAC;IAG3B,aAAa,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IACvC,SAAS,EAAE,CAAC,IAAI,EAAE,OAAO,KAAK,IAAI,CAAC;IACnC,qBAAqB,EAAE,CAAC,IAAI,EAAE,OAAO,KAAK,IAAI,CAAC;IAC/C,uBAAuB,EAAE,CAAC,IAAI,EAAE,OAAO,KAAK,IAAI,CAAC;IACjD,cAAc,EAAE,CAAC,WAAW,EAAE,MAAM,EAAE,KAAK,IAAI,CAAC;IAChD,aAAa,EAAE,CAAC,QAAQ,EAAE,OAAO,KAAK,IAAI,CAAC;IAC3C,kBAAkB,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;IAC/C,eAAe,EAAE,CAAC,YAAY,EAAE,GAAG,KAAK,IAAI,CAAC;IAE7C,qBAAqB,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IAC/C,oBAAoB,EAAE,CAAC,IAAI,EAAE,OAAO,KAAK,IAAI,CAAC;IAG9C,aAAa,EAAE,CAAC,qBAAqB,CAAC,EAAE,OAAO,KAAK,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC;IAChF,YAAY,EAAE,CAAC,sBAAsB,CAAC,EAAE,OAAO,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAClE,WAAW,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACrE,YAAY,EAAE,CAAC,CAAC,EAAE,KAAK,CAAC,SAAS,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACpD,mBAAmB,EAAE,CAAC,OAAO,EAAE,aAAa,KAAK,IAAI,CAAC;IACtD,eAAe,EAAE,CAAC,cAAc,CAAC,EAAE,MAAM,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9D,uBAAuB,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7C,WAAW,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IACvC,SAAS,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IACrC,WAAW,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IACvC,QAAQ,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IACpC,UAAU,EAAE,CAAC,sBAAsB,CAAC,EAAE,OAAO,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAEhE,iBAAiB,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,IAAI,CAAC;IAChD,eAAe,EAAE,MAAM,OAAO,CAAC;IAG/B,MAAM,EAAE,CAAC,cAAc,CAAC,EAAE,MAAM,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACrD,eAAe,EAAE,MAAM,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC;IACnD,QAAQ,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9B,sBAAsB,EAAE,OAAO,CAAC;IAChC,cAAc,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;IACpC,SAAS,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IAGzB,KAAK,EAAE,YAAY,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;CACpB"}
+{"version":3,"file":"ChatBotTypes.d.ts","sourceRoot":"","sources":["../../../src/components/types/ChatBotTypes.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,GAAG,EAAE,iBAAiB,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAE9E,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,OAAO,CAAC;CACpB;AAED,MAAM,WAAW,OAAO;IACtB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC5B;AAED,MAAM,WAAW,QAAQ;IACvB,OAAO,EAAE,OAAO,CAAC;CAClB;AAED,MAAM,WAAW,YAAY;IAC3B,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC/B,QAAQ,CAAC,EAAE,QAAQ,CAAC;CACrB;AAED,MAAM,WAAW,SAAS;IACxB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,WAAW;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,EAAE,CAAC;IACtB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAGD,MAAM,WAAW,OAAO;IACtB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;CACf;AAED,eAAO,MAAM,iBAAiB,EAAE,OAAO,EAMtC,CAAC;AAEF,eAAO,MAAM,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAM/C,CAAC;AAEF,MAAM,WAAW,YAAY;IAC3B,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;IACvC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACjC,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC,CAAC;IAC1D,cAAc,CAAC,EAAE,aAAa,EAAE,CAAC;IACjC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,cAAc,CAAC,EAAE,MAAM,OAAO,CAAC;QAC7B,KAAK,EAAE,MAAM,CAAC;KACf,GAAG,SAAS,CAAC,CAAC;IACf,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,EAAE,KAAK,CAAC,SAAS,CAAC;IAE3B,eAAe,CAAC,EAAE,OAAO,EAAE,CAAC;IAC5B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;CAC1C;AAED,MAAM,WAAW,kBAAkB;IAEjC,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,OAAO,CAAC;IACnB,MAAM,EAAE,OAAO,CAAC;IAChB,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,GAAG,EAAE,GAAG,GAAG,IAAI,CAAC;IAChB,WAAW,EAAE,MAAM,EAAE,CAAC;IACtB,oBAAoB,EAAE,iBAAiB,EAAE,CAAC;IAC1C,kBAAkB,EAAE,OAAO,CAAC;IAC5B,oBAAoB,EAAE,OAAO,CAAC;IAC9B,YAAY,EAAE,OAAO,CAAC;IACtB,UAAU,EAAE,OAAO,CAAC;IACpB,UAAU,EAAE,GAAG,CAAC;IAChB,eAAe,EAAE,OAAO,CAAC;IACzB,gBAAgB,EAAE,OAAO,CAAC;IAC1B,YAAY,EAAE;QACZ,IAAI,EAAE,OAAO,CAAC;QACd,IAAI,EAAE,SAAS,GAAG,OAAO,GAAG,SAAS,GAAG,MAAM,CAAC;QAC/C,OAAO,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,OAAO,CAAC;QACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB,CAAC;IAEF,eAAe,EAAE,OAAO,EAAE,CAAC;IAC3B,kBAAkB,EAAE,MAAM,CAAC;IAC3B,iBAAiB,EAAE,OAAO,CAAC;IAG3B,aAAa,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IACvC,SAAS,EAAE,CAAC,IAAI,EAAE,OAAO,KAAK,IAAI,CAAC;IACnC,qBAAqB,EAAE,CAAC,IAAI,EAAE,OAAO,KAAK,IAAI,CAAC;IAC/C,uBAAuB,EAAE,CAAC,IAAI,EAAE,OAAO,KAAK,IAAI,CAAC;IACjD,cAAc,EAAE,CAAC,WAAW,EAAE,MAAM,EAAE,KAAK,IAAI,CAAC;IAChD,aAAa,EAAE,CAAC,QAAQ,EAAE,OAAO,KAAK,IAAI,CAAC;IAC3C,kBAAkB,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;IAC/C,eAAe,EAAE,CAAC,YAAY,EAAE,GAAG,KAAK,IAAI,CAAC;IAE7C,qBAAqB,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IAC/C,oBAAoB,EAAE,CAAC,IAAI,EAAE,OAAO,KAAK,IAAI,CAAC;IAG9C,aAAa,EAAE,CAAC,qBAAqB,CAAC,EAAE,OAAO,KAAK,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC;IAChF,YAAY,EAAE,CAAC,sBAAsB,CAAC,EAAE,OAAO,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAClE,WAAW,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACrE,YAAY,EAAE,CAAC,CAAC,EAAE,KAAK,CAAC,SAAS,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACpD,mBAAmB,EAAE,CAAC,OAAO,EAAE,aAAa,KAAK,IAAI,CAAC;IACtD,eAAe,EAAE,CAAC,cAAc,CAAC,EAAE,MAAM,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9D,aAAa,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;IACtD,uBAAuB,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7C,WAAW,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IACvC,SAAS,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IACrC,WAAW,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IACvC,QAAQ,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IACpC,UAAU,EAAE,CAAC,sBAAsB,CAAC,EAAE,OAAO,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAEhE,iBAAiB,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,IAAI,CAAC;IAChD,eAAe,EAAE,MAAM,OAAO,CAAC;IAG/B,MAAM,EAAE,CAAC,cAAc,CAAC,EAAE,MAAM,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACrD,eAAe,EAAE,MAAM,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC;IACnD,QAAQ,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9B,sBAAsB,EAAE,OAAO,CAAC;IAChC,cAAc,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;IACpC,SAAS,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IAGzB,KAAK,EAAE,YAAY,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;CACpB"}

package/dist/components/ui/MessageList.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"MessageList.d.ts","sourceRoot":"","sources":["../../../src/components/ui/MessageList.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,KAAK,MAAM,OAAO,CAAC;AAI/B,eAAO,MAAM,WAAW,EAAE,KAAK,CAAC,EAiJ/B,CAAC"}
+{"version":3,"file":"MessageList.d.ts","sourceRoot":"","sources":["../../../src/components/ui/MessageList.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,KAAK,MAAM,OAAO,CAAC;AAI/B,eAAO,MAAM,WAAW,EAAE,KAAK,CAAC,EA6K/B,CAAC"}

package/dist/components/ui/MessageList.js

@@ -3,7 +3,7 @@ import * as React from 'react';
 import { useChatBot } from '../context/ChatBotContext';
 import { AI_MODEL_MAP } from '../types/ChatBotTypes';
 export const MessageList = () => {
-    const { messages, isLoading, connectWallet, signInWallet, hitAddress, ait, setShowPermissionForm, isWalletLoading, isNeedSignInWithWallet } = useChatBot();
+    const { messages, isLoading, connectWallet, signInWallet, hitAddress, ait, setShowPermissionForm, isWalletLoading, isAutoConnecting, isNeedSignInWithWallet, autoEnableAIT, isAITLoading } = useChatBot();
     const messagesEndRef = React.useRef(null);
     const scrollToBottom = () => {
         messagesEndRef.current?.scrollIntoView({ behavior: 'smooth' });
@@ -11,13 +11,23 @@ export const MessageList = () => {
     React.useEffect(() => {
         scrollToBottom();
     }, [messages]);
-    const handleButtonClick = (buttonType) => {
+    const handleButtonClick = async (buttonType, message) => {
         if (buttonType === 'connect') {
             connectWallet(true);
         }
         else if (buttonType === 'signIn') {
             signInWallet(true);
         }
+        else if (buttonType === 'enableAIT') {
+            const toolName = message.metadata?.toolName;
+            if (toolName) {
+                const success = await autoEnableAIT(toolName);
+                if (success) {
+                    // If AIT is successfully enabled, you can choose to resend the user's last message
+                    // Here you can add logic to reprocess the user's request
+                }
+            }
+        }
     };
     const getModelDisplayName = (modelValue) => {
         if (!modelValue)
@@ -36,28 +46,41 @@ export const MessageList = () => {
                     maxWidth: '80%',
                     marginBottom: '10px'
                 }, children: [_jsxs("div", { style: {
-                            backgroundColor: message.role === 'user' ? '#007bff' : '#f1f1f1',
-                            color: message.role === 'user' ? 'white' : '#333',
+                            backgroundColor: message.role === 'user' ? '#007bff' :
+                                message.metadata?.isRetry ? '#fff3cd' : '#f1f1f1',
+                            color: message.role === 'user' ? 'white' :
+                                message.metadata?.isRetry ? '#856404' : '#333',
                             padding: '10px 15px',
                             borderRadius: '15px',
                             wordWrap: 'break-word',
-                            position: 'relative'
-                        }, children: [message.content, message.button && (_jsx("div", { style: { marginTop: '10px' }, children: _jsx("button", { onClick: () => handleButtonClick(message.button), disabled: (message.button === 'connect' && Boolean(hitAddress)) ||
-                                        (message.button === 'signIn' && !isNeedSignInWithWallet), style: {
+                            position: 'relative',
+                            border: message.metadata?.isRetry ? '1px solid #ffeaa7' : 'none'
+                        }, children: [message.metadata?.isRetry && (_jsx("span", { style: { marginRight: '8px', fontSize: '14px' }, children: "\uD83D\uDD04" })), message.content, message.button && (_jsx("div", { style: { marginTop: '10px' }, children: _jsx("button", { onClick: () => handleButtonClick(message.button, message), disabled: isAutoConnecting ||
+                                        (message.button === 'connect' && Boolean(hitAddress)) ||
+                                        (message.button === 'signIn' && !isNeedSignInWithWallet) ||
+                                        (message.button === 'enableAIT' && isAITLoading), style: {
                                         padding: '8px 16px',
                                         backgroundColor: message.role === 'user' ? 'rgba(255, 255, 255, 0.2)' :
-                                            (message.button === 'connect' && Boolean(hitAddress)) ? '#28a745' :
-                                                (message.button === 'signIn' && !isNeedSignInWithWallet) ? '#28a745' : '#007bff',
+                                            isAutoConnecting ? '#6c757d' :
+                                                (message.button === 'connect' && Boolean(hitAddress)) ? '#28a745' :
+                                                    (message.button === 'signIn' && !isNeedSignInWithWallet) ? '#28a745' :
+                                                        (message.button === 'enableAIT' && isAITLoading) ? '#6c757d' : '#007bff',
                                         color: message.role === 'user' ? 'white' : 'white',
                                         border: 'none',
                                         borderRadius: '5px',
-                                        cursor: ((message.button === 'connect' && Boolean(hitAddress)) ||
-                                            (message.button === 'signIn' && !isNeedSignInWithWallet)) ? 'not-allowed' : 'pointer',
+                                        cursor: (isAutoConnecting ||
+                                            (message.button === 'connect' && Boolean(hitAddress)) ||
+                                            (message.button === 'signIn' && !isNeedSignInWithWallet) ||
+                                            (message.button === 'enableAIT' && isAITLoading)) ? 'not-allowed' : 'pointer',
                                         fontSize: '14px',
-                                        opacity: ((message.button === 'connect' && Boolean(hitAddress)) ||
-                                            (message.button === 'signIn' && !isNeedSignInWithWallet)) ? 0.8 : 1
-                                    }, children: message.button === 'connect' ? (Boolean(hitAddress) ? 'Connected' : 'Connect Wallet') :
-                                        message.button === 'signIn' ? (!isNeedSignInWithWallet ? 'Signed In' : 'Sign In') : message.button }) }))] }), message.role === 'assistant' && message.metadata?.model && (_jsx("div", { style: {
+                                        opacity: (isAutoConnecting ||
+                                            (message.button === 'connect' && Boolean(hitAddress)) ||
+                                            (message.button === 'signIn' && !isNeedSignInWithWallet) ||
+                                            (message.button === 'enableAIT' && isAITLoading)) ? 0.8 : 1
+                                    }, children: isAutoConnecting ? 'Connecting...' :
+                                        message.button === 'connect' ? (Boolean(hitAddress) ? 'Connected' : 'Connect Wallet') :
+                                            message.button === 'signIn' ? (!isNeedSignInWithWallet ? 'Signed In' : 'Sign In') :
+                                                message.button === 'enableAIT' ? (isAITLoading ? 'Enabling...' : 'Enable AIT Permissions') : message.button }) }))] }), message.role === 'assistant' && message.metadata?.model && (_jsx("div", { style: {
                             display: 'flex',
                             alignItems: 'center',
                             marginTop: '4px',

package/dist/types/ait-api.d.ts

@@ -13,6 +13,8 @@ export interface Message {
             name: string;
             input: Record<string, any>;
         };
+        isRetry?: boolean;
+        toolName?: string;
     };
 }
 export interface AIT {

package/dist/types/ait-api.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ait-api.d.ts","sourceRoot":"","sources":["../../src/types/ait-api.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,OAAO;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE;QACT,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;QACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,OAAO,CAAC,EAAE;YACR,IAAI,EAAE,MAAM,CAAC;YACb,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;SAC5B,CAAC;KACH,CAAC;CACH;AAED,MAAM,WAAW,GAAG;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,WAAW,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,WAAW;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,EAAE,CAAC;IACtB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,OAAO;IACtB,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,WAAW,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,UAAU;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,OAAO,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,eAAe;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,gBAAgB,EAAE,CAAC;CAC7B;AAED,MAAM,WAAW,gBAAgB;IAC/B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,OAAO,CAAC;CACpB;AAED,MAAM,WAAW,eAAe;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,oBAAqB,SAAQ,WAAW;CACxD;AAED,MAAM,WAAW,kBAAkB;IACjC,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,YAAY;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,cAAc;IAC7B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CAC/B;AAED,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,cAAc,CAAC;CACzB;AAED,MAAM,WAAW,iBAAiB;IAChC,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB;AAED,MAAM,WAAW,MAAM;IACrB,GAAG,EAAE;QACH,8BAA8B,EAAE,CAAC,MAAM,EAAE;YAAE,SAAS,EAAE,MAAM,CAAC;YAAC,UAAU,EAAE,MAAM,CAAA;SAAE,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,GAAG;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;QAC3I,SAAS,EAAE,CAAC,MAAM,EAAE,eAAe,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,GAAG;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KAC7F,CAAC;IACF,MAAM,EAAE;QACN,YAAY,EAAE,CAAC,MAAM,EAAE,kBAAkB,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,UAAU,GAAG;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;QACrG,SAAS,EAAE,CAAC,MAAM,EAAE;YAAE,OAAO,EAAE,MAAM,CAAA;SAAE,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,UAAU,GAAG;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KACpG,CAAC;IACF,QAAQ,EAAE;QACR,cAAc,EAAE,CAAC,QAAQ,EAAE,oBAAoB,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC;YAAE,WAAW,EAAE,MAAM,CAAA;SAAE,GAAG;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KACzH,CAAC;IACF,IAAI,EAAE;QACJ,QAAQ,EAAE,CAAC,MAAM,EAAE;YAAE,OAAO,EAAE,MAAM,CAAA;SAAE,KAAK,OAAO,CAAC;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,SAAS,EAAE,MAAM,CAAA;SAAE,GAAG;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;QAC5G,MAAM,EAAE,CAAC,MAAM,EAAE,YAAY,KAAK,OAAO,CAAC;YAAE,WAAW,EAAE,MAAM,CAAA;SAAE,GAAG;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KACxF,CAAC;IACF,KAAK,EAAE;QACL,WAAW,EAAE,CAAC,MAAM,EAAE;YACpB,OAAO,EAAE,MAAM,CAAC;YAChB,MAAM,EAAE,MAAM,CAAC;YACf,SAAS,EAAE,MAAM,CAAC;YAClB,KAAK,CAAC,EAAE,MAAM,CAAC;YACf,OAAO,CAAC,EAAE,KAAK,CAAC;gBAAE,IAAI,EAAE,MAAM,CAAC;gBAAC,IAAI,EAAE,MAAM,CAAA;aAAE,CAAC,CAAC;SACjD,KAAK,OAAO,CAAC;YACZ,KAAK,EAAE,MAAM,GAAG,KAAK,CAAC;gBAAE,IAAI,EAAE,MAAM,CAAA;aAAE,CAAC,CAAC;YACxC,QAAQ,CAAC,EAAE;gBACT,OAAO,EAAE;oBACP,IAAI,EAAE,MAAM,CAAC;oBACb,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;iBAC5B,CAAC;aACH,CAAC;YACF,YAAY,CAAC,EAAE;gBACb,MAAM,CAAC,EAAE;oBACP,OAAO,CAAC,EAAE;wBACR,OAAO,CAAC,EAAE,KAAK,CAAC;4BAAE,IAAI,EAAE,MAAM,CAAA;yBAAE,CAAC,CAAC;qBACnC,CAAC;iBACH,CAAC;aACH,CAAC;SACH,GAAG;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KACxB,CAAC;IACF,WAAW,EAAE;QACX,qBAAqB,EAAE,CAAC,MAAM,EAAE;YAAE,SAAS,EAAE,MAAM,CAAC;YAAC,SAAS,CAAC,EAAE,MAAM,CAAA;SAAE,KAAK,OAAO,CAAC;YAAE,WAAW,EAAE,iBAAiB,EAAE,CAAA;SAAE,GAAG;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KACjJ,CAAC;CACH"}
+{"version":3,"file":"ait-api.d.ts","sourceRoot":"","sources":["../../src/types/ait-api.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,OAAO;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE;QACT,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;QACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,OAAO,CAAC,EAAE;YACR,IAAI,EAAE,MAAM,CAAC;YACb,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;SAC5B,CAAC;QACF,OAAO,CAAC,EAAE,OAAO,CAAC;QAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB,CAAC;CACH;AAED,MAAM,WAAW,GAAG;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,WAAW,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,WAAW;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,EAAE,CAAC;IACtB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,OAAO;IACtB,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,WAAW,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,UAAU;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,OAAO,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,eAAe;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,gBAAgB,EAAE,CAAC;CAC7B;AAED,MAAM,WAAW,gBAAgB;IAC/B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,OAAO,CAAC;CACpB;AAED,MAAM,WAAW,eAAe;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,oBAAqB,SAAQ,WAAW;CACxD;AAED,MAAM,WAAW,kBAAkB;IACjC,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,YAAY;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,cAAc;IAC7B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CAC/B;AAED,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,cAAc,CAAC;CACzB;AAED,MAAM,WAAW,iBAAiB;IAChC,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB;AAED,MAAM,WAAW,MAAM;IACrB,GAAG,EAAE;QACH,8BAA8B,EAAE,CAAC,MAAM,EAAE;YAAE,SAAS,EAAE,MAAM,CAAC;YAAC,UAAU,EAAE,MAAM,CAAA;SAAE,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,GAAG;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;QAC3I,SAAS,EAAE,CAAC,MAAM,EAAE,eAAe,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,GAAG;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KAC7F,CAAC;IACF,MAAM,EAAE;QACN,YAAY,EAAE,CAAC,MAAM,EAAE,kBAAkB,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,UAAU,GAAG;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;QACrG,SAAS,EAAE,CAAC,MAAM,EAAE;YAAE,OAAO,EAAE,MAAM,CAAA;SAAE,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,UAAU,GAAG;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KACpG,CAAC;IACF,QAAQ,EAAE;QACR,cAAc,EAAE,CAAC,QAAQ,EAAE,oBAAoB,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC;YAAE,WAAW,EAAE,MAAM,CAAA;SAAE,GAAG;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KACzH,CAAC;IACF,IAAI,EAAE;QACJ,QAAQ,EAAE,CAAC,MAAM,EAAE;YAAE,OAAO,EAAE,MAAM,CAAA;SAAE,KAAK,OAAO,CAAC;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,SAAS,EAAE,MAAM,CAAA;SAAE,GAAG;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;QAC5G,MAAM,EAAE,CAAC,MAAM,EAAE,YAAY,KAAK,OAAO,CAAC;YAAE,WAAW,EAAE,MAAM,CAAA;SAAE,GAAG;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KACxF,CAAC;IACF,KAAK,EAAE;QACL,WAAW,EAAE,CAAC,MAAM,EAAE;YACpB,OAAO,EAAE,MAAM,CAAC;YAChB,MAAM,EAAE,MAAM,CAAC;YACf,SAAS,EAAE,MAAM,CAAC;YAClB,KAAK,CAAC,EAAE,MAAM,CAAC;YACf,OAAO,CAAC,EAAE,KAAK,CAAC;gBAAE,IAAI,EAAE,MAAM,CAAC;gBAAC,IAAI,EAAE,MAAM,CAAA;aAAE,CAAC,CAAC;SACjD,KAAK,OAAO,CAAC;YACZ,KAAK,EAAE,MAAM,GAAG,KAAK,CAAC;gBAAE,IAAI,EAAE,MAAM,CAAA;aAAE,CAAC,CAAC;YACxC,QAAQ,CAAC,EAAE;gBACT,OAAO,EAAE;oBACP,IAAI,EAAE,MAAM,CAAC;oBACb,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;iBAC5B,CAAC;aACH,CAAC;YACF,YAAY,CAAC,EAAE;gBACb,MAAM,CAAC,EAAE;oBACP,OAAO,CAAC,EAAE;wBACR,OAAO,CAAC,EAAE,KAAK,CAAC;4BAAE,IAAI,EAAE,MAAM,CAAA;yBAAE,CAAC,CAAC;qBACnC,CAAC;iBACH,CAAC;aACH,CAAC;SACH,GAAG;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KACxB,CAAC;IACF,WAAW,EAAE;QACX,qBAAqB,EAAE,CAAC,MAAM,EAAE;YAAE,SAAS,EAAE,MAAM,CAAC;YAAC,SAAS,CAAC,EAAE,MAAM,CAAA;SAAE,KAAK,OAAO,CAAC;YAAE,WAAW,EAAE,iBAAiB,EAAE,CAAA;SAAE,GAAG;YAAE,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KACjJ,CAAC;CACH"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bytexbyte/nxtlinq-ai-agent-sdk",
-  "version": "1.2.5",
+  "version": "1.2.6",
   "description": "Nxtlinq AI Agent SDK - Proprietary Software with enhanced async operation handling",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",