@bytexbyte/nxtlinq-ai-agent-sdk 1.1.9 → 1.2.1

package/README.md

@@ -13,15 +13,17 @@ A powerful SDK for building intelligent conversation applications with Nxtlinq A
 - 🔌 Easy integration with React applications
 - 🔒 Secure authentication and API key management
 - 👛 MetaMask wallet integration
-- 🔐 AIT (AI Token) management
+- 🔐 AIT (AI Identity Token) management
 - 🔑 Permission-based access control
+- 🔍 Enhanced error handling and debugging
+- ⚡ Improved async operation handling
 
 ## Installation
 
 ```bash
-npm install @bytexbyte/nxtlinq-ai-agent-sdk
+npm install @bytexbyte/nxtlinq-ai-agent-sdk@1.1.9
 # or
-yarn add @bytexbyte/nxtlinq-ai-agent-sdk
+yarn add @bytexbyte/nxtlinq-ai-agent-sdk@1.1.9
 ```
 
 ## Quick Start
@@ -59,8 +61,8 @@ function App() {
       onMessage={handleMessage}
       onToolUse={handleToolUse}
       presetMessages={presetMessages}
-      onVerifyWallet={async (address) => {
-        // Implement your wallet verification logic here
+      onVerifyWallet={async () => {
+        // Implement Berify.me verification
         return { token: 'verification-token' };
       }}
     />
@@ -68,47 +70,47 @@ function App() {
 }
 ```
 
-## API Reference
-
-### NxtlinqAITSDK
-
-| Method | Description |
-|--------|-------------|
-| `connectWallet()` | Connect to MetaMask wallet |
-| `verifyWallet(token: string, method: string)` | Verify wallet ownership |
-| `signInWithWallet()` | Sign in using wallet |
-| `generateAndRegisterAIT(permissions: string[])` | Generate and register a new AIT |
-| `getAIT()` | Get AIT information |
-| `getWalletInfo()` | Get wallet information |
-
-### NxtlinqAIAgent
-
-| Method | Description |
-|--------|-------------|
-| `setAIT(ait: AIT, signer?: ethers.Signer)` | Set AIT for the agent |
-| `generateAIT(options: GenerateAITOptions)` | Generate a new AIT |
-| `getAITInfo(serviceId: string, controller: string, signer?: ethers.Signer)` | Get AIT information |
-| `sendMessage(message: string, toolName?: string)` | Send a message to the agent |
-
-### API Endpoints
+## Berify.me Integration
 
-#### AIT API
-- `getAITByServiceIdAndController`: Get AIT by service ID and controller
-- `createAIT`: Create a new AIT
+The SDK supports Berify.me wallet verification for enhanced security:
 
-#### Wallet API
-- `verifyWallet`: Verify wallet ownership
-- `getWallet`: Get wallet information
+```tsx
+import { ChatBot } from '@bytexbyte/nxtlinq-ai-agent-sdk';
 
-#### Metadata API
-- `createMetadata`: Create metadata for AIT
+function App() {
+  const handleVerifyWallet = async () => {
+    // Open Berify.me verification modal
+    const modal = window.open(
+      'https://berify.me/verify',
+      'BerifyMeModal',
+      'width=500,height=600'
+    );
+
+    return new Promise((resolve) => {
+      const handleMessage = (event) => {
+        if (event.origin === 'https://berify.me' && event.data.type === 'VERIFICATION_SUCCESS') {
+          window.removeEventListener('message', handleMessage);
+          modal?.close();
+          resolve({ token: event.data.token });
+        }
+      };
+
+      window.addEventListener('message', handleMessage);
+    });
+  };
 
-#### Auth API
-- `getNonce`: Get nonce for wallet sign-in
-- `signIn`: Sign in with wallet
+  return (
+    <ChatBot
+      serviceId="your-service-id"
+      apiKey="your-api-key"
+      apiSecret="your-api-secret"
+      onVerifyWallet={handleVerifyWallet}
+    />
+  );
+}
+```
 
-#### Agent API
-- `sendMessage`: Send message to AI agent
+## API Reference
 
 ### ChatBot Component Props
 
@@ -120,12 +122,13 @@ function App() {
 | onMessage | (message: Message) => void | No | Callback when a new message is received |
 | onError | (error: Error) => void | No | Callback when an error occurs |
 | onToolUse | (toolUse: ToolUse) => Promise<Message \| void> | No | Callback for handling tool usage |
-| onVerifyWallet | (address: string) => Promise<{ token: string }> | Yes | Callback for wallet verification |
+| onVerifyWallet | () => Promise<{ token: string } \| undefined> | Yes | Callback for wallet verification |
 | presetMessages | PresetMessage[] | No | Array of preset messages to display |
 | placeholder | string | No | Input placeholder text (default: "Type a message...") |
 | className | string | No | Additional CSS class name |
 | maxRetries | number | No | Maximum number of retry attempts (default: 3) |
 | retryDelay | number | No | Delay between retries in milliseconds (default: 1000) |
+| permissionGroup | string | No | Permission group name for filtering permissions |
 
 ## Types
 
@@ -136,7 +139,7 @@ interface Message {
   content: string;
   role: 'user' | 'assistant';
   timestamp: string;
-  button?: boolean;
+  button?: string;
   error?: string;
 }
 ```
@@ -165,7 +168,6 @@ interface AIT {
   metadata: AITMetadata;
   metadataHash: string;
   metadataCid: string;
-  signature: string;
 }
 
 interface AITMetadata {
@@ -174,39 +176,15 @@ interface AITMetadata {
   issuedBy: string;
   serviceId?: string;
 }
+```
 
-interface AITInfo {
-  aitId: string;
-  controller: string;
-  metadata: AITMetadata;
-  metadataHash: string;
-  metadataCid: string;
-  signature: string;
-}
-
-interface AITPermission {
-  hasPermission: boolean;
-  reason?: string;
-  permissions?: string[];
-}
-
-interface GenerateAITOptions {
-  hitAddress: string;
-  signer: ethers.Signer;
-  permissions: string[];
-  serviceId: string;
-}
-
-interface WalletInfo {
+### ServicePermission
+```typescript
+interface ServicePermission {
   id: string;
-  address: string;
-  verified: boolean;
-  method?: string;
-}
-
-interface MessageContext {
-  aitId?: string;
-  walletAddress?: string | null;
+  label: string;
+  description: string;
+  groups: string[];
 }
 ```
 
@@ -214,30 +192,32 @@ interface MessageContext {
 
 ### Wallet Integration
 - MetaMask wallet connection
-- Wallet verification
+- Berify.me wallet verification
 - Wallet-based authentication
-- AIT (AI Token) generation and management
-
-### Available Permissions
-- Set User Name
-- Navigate To Page
-- Add Member
+- AIT (AI Identity Token) generation and management
 
 ### Authentication Flow
 1. Connect MetaMask wallet
-2. Verify wallet ownership
+2. Verify wallet ownership (optional, via Berify.me)
 3. Sign in with wallet
 4. Generate and register AIT with required permissions
 5. Use AIT for authenticated operations
 
+### Enhanced Error Handling
+- Improved async operation handling
+- Better state management
+- Detailed error messages
+- Comprehensive debugging logs
+
 ## Error Handling
 
-The SDK includes built-in error handling with:
+The SDK includes enhanced error handling with:
 - Automatic retry mechanism for failed requests
 - Error callback for custom error handling
 - User-friendly error messages
 - Wallet connection error handling
 - Authentication error handling
+- Detailed console logging for debugging
 
 ## Best Practices
 
@@ -248,6 +228,32 @@ The SDK includes built-in error handling with:
 5. Use context management to maintain conversation coherence
 6. Handle wallet connection errors gracefully
 7. Implement proper error handling for authentication flow
+8. Monitor console logs for debugging
+
+## Troubleshooting
+
+Having issues? Check our [Troubleshooting Guide](./TROUBLESHOOTING.md) which includes:
+
+- Common problems and solutions
+- Authentication debugging steps
+- Permission issue resolution
+- Best practice recommendations
+
+### Common Issues
+
+**Q: Still need to sign in after connecting wallet?**
+A: This is usually due to expired authentication tokens or address mismatches. Check browser console logs and refer to the [Troubleshooting Guide](./TROUBLESHOOTING.md) for detailed solutions.
+
+**Q: Cannot use specific AI tool features?**
+A: Check AIT permission settings to ensure required tool permissions are included. See [Troubleshooting Guide](./TROUBLESHOOTING.md) for details.
+
+## Changelog
+
+### v1.1.9
+- 🔧 Fixed async operation timing issues in wallet connection
+- 🐛 Improved state management for AIT validation
+- 📝 Enhanced error messages and debugging logs
+- ⚡ Better handling of authentication flow
 
 ## License
 

package/dist/components/ChatBot.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ChatBot.d.ts","sourceRoot":"","sources":["../../src/components/ChatBot.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,KAAK,MAAM,OAAO,CAAC;AAM/B,OAAO,EAAE,OAAO,EAAkC,MAAM,kBAAkB,CAAC;AAE3E,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;AAED,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;CAC1B;AA0VD,eAAO,MAAM,OAAO,EAAE,KAAK,CAAC,EAAE,CAAC,YAAY,CAk8B1C,CAAC"}
+{"version":3,"file":"ChatBot.d.ts","sourceRoot":"","sources":["../../src/components/ChatBot.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,KAAK,MAAM,OAAO,CAAC;AAM/B,OAAO,EAAE,OAAO,EAAkC,MAAM,kBAAkB,CAAC;AAE3E,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;AAED,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;CAC1B;AA2cD,eAAO,MAAM,OAAO,EAAE,KAAK,CAAC,EAAE,CAAC,YAAY,CAogC1C,CAAC"}

package/dist/components/ChatBot.js

@@ -5,8 +5,9 @@ import { createNxtlinqApi } from '../api/nxtlinq-api';
 import stringify from 'fast-json-stable-stringify';
 import metakeepClient from '../core/metakeepClient';
 import useLocalStorage from '../core/lib/useLocalStorage';
-const PermissionForm = ({ hitAddress, permissions, setPermissions, setIsDisabled, onClose, onSave, onConnectWallet, onSignIn, isNeedSignInWithWallet, walletInfo, onVerifyWallet, serviceId, nxtlinqApi, permissionGroup }) => {
+const PermissionForm = ({ hitAddress, permissions, setPermissions, setIsDisabled, onClose, onSave, onConnectWallet, onSignIn, isNeedSignInWithWallet, walletInfo, onVerifyWallet, serviceId, nxtlinqApi, permissionGroup, isAITLoading, isWalletLoading = false }) => {
     const [availablePermissions, setAvailablePermissions] = React.useState([]);
+    const [isSaving, setIsSaving] = React.useState(false);
     const fetchAvailablePermissions = async () => {
         if (!serviceId)
             return;
@@ -28,7 +29,72 @@ const PermissionForm = ({ hitAddress, permissions, setPermissions, setIsDisabled
     React.useEffect(() => {
         fetchAvailablePermissions();
     }, [serviceId, nxtlinqApi, permissionGroup]);
-    const isWalletVerified = walletInfo?.id;
+    const isWalletVerified = Boolean(walletInfo?.id);
+    const handleSave = async () => {
+        setIsSaving(true);
+        try {
+            await onSave();
+        }
+        finally {
+            setIsSaving(false);
+        }
+    };
+    // Show loading state while checking wallet status
+    if (isWalletLoading) {
+        return (_jsxs("div", { style: {
+                backgroundColor: 'white',
+                padding: '24px',
+                borderRadius: '12px',
+                width: '480px',
+                maxWidth: '90%',
+                boxShadow: '0 4px 12px rgba(0, 0, 0, 0.1)'
+            }, children: [_jsxs("div", { style: {
+                        display: 'flex',
+                        justifyContent: 'space-between',
+                        alignItems: 'center',
+                        marginBottom: '24px'
+                    }, children: [_jsx("h3", { style: {
+                                margin: 0,
+                                fontSize: '20px',
+                                fontWeight: '600',
+                                color: '#1a1a1a'
+                            }, children: "AIT Settings" }), _jsx("button", { onClick: onClose, style: {
+                                background: 'none',
+                                border: 'none',
+                                fontSize: '24px',
+                                cursor: 'pointer',
+                                color: '#666',
+                                padding: '4px',
+                                display: 'flex',
+                                alignItems: 'center',
+                                justifyContent: 'center'
+                            }, children: "\u00D7" })] }), _jsxs("div", { style: { textAlign: 'center', padding: '32px 0' }, children: [_jsx("div", { style: {
+                                width: '64px',
+                                height: '64px',
+                                margin: '0 auto 16px',
+                                backgroundColor: '#f5f5f5',
+                                borderRadius: '50%',
+                                display: 'flex',
+                                alignItems: 'center',
+                                justifyContent: 'center'
+                            }, children: _jsx("div", { style: {
+                                    width: '32px',
+                                    height: '32px',
+                                    border: '3px solid #e3e3e3',
+                                    borderTop: '3px solid #007bff',
+                                    borderRadius: '50%',
+                                    animation: 'spin 1s linear infinite'
+                                } }) }), _jsx("p", { style: {
+                                marginBottom: '24px',
+                                fontSize: '16px',
+                                color: '#666'
+                            }, children: "Checking wallet status..." })] }), _jsx("style", { children: `
+          @keyframes spin {
+            0% { transform: rotate(0deg); }
+            100% { transform: rotate(360deg); }
+          }
+        ` })] }));
+    }
     return (_jsxs("div", { style: {
             backgroundColor: 'white',
             padding: '24px',
@@ -147,7 +213,14 @@ const PermissionForm = ({ hitAddress, permissions, setPermissions, setIsDisabled
                                     marginBottom: '12px',
                                     fontSize: '16px',
                                     color: '#666'
-                                }, children: "Permissions" }), _jsx("div", { style: {
+                                }, children: "Permissions" }), isAITLoading ? (_jsx("div", { style: {
+                                    backgroundColor: '#f8f9fa',
+                                    padding: '16px',
+                                    borderRadius: '8px',
+                                    border: '1px solid #e9ecef',
+                                    textAlign: 'center',
+                                    color: '#666'
+                                }, children: "Loading permissions..." })) : (_jsx("div", { style: {
                                     backgroundColor: '#f8f9fa',
                                     padding: '16px',
                                     borderRadius: '8px',
@@ -156,25 +229,36 @@ const PermissionForm = ({ hitAddress, permissions, setPermissions, setIsDisabled
                                             display: 'flex',
                                             alignItems: 'center',
                                             gap: '12px',
-                                            cursor: 'pointer',
+                                            cursor: isAITLoading ? 'not-allowed' : 'pointer',
                                             padding: '8px',
                                             borderRadius: '6px',
-                                            transition: 'background-color 0.2s'
-                                        }, onMouseOver: (e) => e.currentTarget.style.backgroundColor = '#e9ecef', onMouseOut: (e) => e.currentTarget.style.backgroundColor = 'transparent', children: [_jsx("input", { type: "checkbox", checked: permissions.includes(permission.label), onChange: () => {
-                                                    const newPermissions = permissions.includes(permission.label)
-                                                        ? permissions.filter(p => p !== permission.label)
-                                                        : [...permissions, permission.label].sort();
-                                                    setPermissions(newPermissions);
-                                                    setIsDisabled(false);
-                                                }, style: {
+                                            transition: 'background-color 0.2s',
+                                            opacity: isAITLoading ? 0.6 : 1
+                                        }, onMouseOver: (e) => {
+                                            if (!isAITLoading) {
+                                                e.currentTarget.style.backgroundColor = '#e9ecef';
+                                            }
+                                        }, onMouseOut: (e) => {
+                                            if (!isAITLoading) {
+                                                e.currentTarget.style.backgroundColor = 'transparent';
+                                            }
+                                        }, children: [_jsx("input", { type: "checkbox", checked: permissions.includes(permission.label), onChange: () => {
+                                                    if (!isAITLoading) {
+                                                        const newPermissions = permissions.includes(permission.label)
+                                                            ? permissions.filter(p => p !== permission.label)
+                                                            : [...permissions, permission.label].sort();
+                                                        setPermissions(newPermissions);
+                                                        setIsDisabled(false);
+                                                    }
+                                                }, disabled: isAITLoading, style: {
                                                     margin: 0,
                                                     width: '18px',
                                                     height: '18px',
-                                                    cursor: 'pointer'
+                                                    cursor: isAITLoading ? 'not-allowed' : 'pointer'
                                                 } }), _jsx("span", { style: {
                                                     fontSize: '14px',
                                                     color: '#333'
-                                                }, children: permission.label })] }) }, permission.id))) })] }), _jsxs("div", { style: {
+                                                }, children: permission.label })] }) }, permission.id))) }))] }), _jsxs("div", { style: {
                             display: 'flex',
                             justifyContent: 'flex-end',
                             gap: '12px',
@@ -196,25 +280,25 @@ const PermissionForm = ({ hitAddress, permissions, setPermissions, setIsDisabled
                                 }, onMouseOut: (e) => {
                                     e.currentTarget.style.backgroundColor = '#f8f9fa';
                                     e.currentTarget.style.borderColor = '#dee2e6';
-                                }, children: "Cancel" }), _jsx("button", { onClick: onSave, disabled: permissions.length === 0, style: {
+                                }, children: "Cancel" }), _jsx("button", { onClick: handleSave, disabled: permissions.length === 0 || isSaving || isAITLoading, style: {
                                     padding: '10px 20px',
-                                    backgroundColor: permissions.length === 0 ? '#e9ecef' : '#007bff',
-                                    color: 'white',
+                                    backgroundColor: permissions.length === 0 || isSaving || isAITLoading ? '#e9ecef' : '#007bff',
+                                    color: permissions.length === 0 || isSaving || isAITLoading ? '#6c757d' : 'white',
                                     border: 'none',
                                     borderRadius: '8px',
-                                    cursor: permissions.length === 0 ? 'not-allowed' : 'pointer',
+                                    cursor: permissions.length === 0 || isSaving || isAITLoading ? 'not-allowed' : 'pointer',
                                     fontSize: '14px',
                                     fontWeight: '500',
                                     transition: 'background-color 0.2s'
                                 }, onMouseOver: (e) => {
-                                    if (permissions.length > 0) {
+                                    if (permissions.length > 0 && !isSaving && !isAITLoading) {
                                         e.currentTarget.style.backgroundColor = '#0056b3';
                                     }
                                 }, onMouseOut: (e) => {
-                                    if (permissions.length > 0) {
+                                    if (permissions.length > 0 && !isSaving && !isAITLoading) {
                                         e.currentTarget.style.backgroundColor = '#007bff';
                                     }
-                                }, children: "Save" })] })] }))] }));
+                                }, children: isSaving ? 'Saving...' : 'Save' })] })] }))] }));
 };
 export const ChatBot = ({ onMessage, onError, onToolUse, presetMessages = [], placeholder = 'Type a message...', className = '', maxRetries = 3, retryDelay = 1000, serviceId, apiKey, apiSecret, onVerifyWallet, permissionGroup }) => {
     const [messages, setMessages] = React.useState([]);
@@ -227,12 +311,15 @@ export const ChatBot = ({ onMessage, onError, onToolUse, presetMessages = [], pl
     const [availablePermissions, setAvailablePermissions] = React.useState([]);
     const [showPermissionForm, setShowPermissionForm] = React.useState(false);
     const [success, setSuccess] = React.useState(false);
+    const [isPermissionFormOpen, setIsPermissionFormOpen] = React.useState(false);
+    const [isAITLoading, setIsAITLoading] = React.useState(false);
     const messagesEndRef = React.useRef(null);
     const [isDisabled, setIsDisabled] = React.useState(true);
     const [signer, setSigner] = React.useState(null);
     const [walletInfo, setWalletInfo] = React.useState(null);
     const verifyWalletBtnRef = React.useRef(null);
     const [nxtlinqAITServiceAccessToken, setNxtlinqAITServiceAccessToken] = useLocalStorage('nxtlinqAITServiceAccessToken', '');
+    const [isWalletLoading, setIsWalletLoading] = React.useState(false);
     const nxtlinqApi = React.useMemo(() => createNxtlinqApi(apiKey, apiSecret), [apiKey, apiSecret]);
     const fetchAvailablePermissions = async () => {
         if (!serviceId)
@@ -252,7 +339,7 @@ export const ChatBot = ({ onMessage, onError, onToolUse, presetMessages = [], pl
             console.error('Error fetching permissions:', error);
         }
     };
-    const refreshAIT = async () => {
+    const refreshAIT = async (forceUpdatePermissions = false) => {
         console.log('refreshAIT called');
         console.log('hitAddress:', hitAddress);
         console.log('nxtlinqAITServiceAccessToken:', nxtlinqAITServiceAccessToken);
@@ -261,6 +348,7 @@ export const ChatBot = ({ onMessage, onError, onToolUse, presetMessages = [], pl
             setPermissions([]);
             return;
         }
+        setIsAITLoading(true);
         try {
             const response = await nxtlinqApi.ait.getAITByServiceIdAndController({
                 serviceId,
@@ -274,13 +362,19 @@ export const ChatBot = ({ onMessage, onError, onToolUse, presetMessages = [], pl
             }
             console.log('AIT response:', response);
             setAit(response);
-            setPermissions(response.metadata?.permissions || []);
+            // Update permissions if form is not open OR if force update is requested
+            if (!isPermissionFormOpen || forceUpdatePermissions) {
+                setPermissions(response.metadata?.permissions || []);
+            }
         }
         catch (error) {
             console.error('Failed to fetch AIT:', error);
             setAit(null);
             setPermissions([]);
         }
+        finally {
+            setIsAITLoading(false);
+        }
     };
     React.useEffect(() => {
         console.log('hitAddress:', hitAddress);
@@ -313,20 +407,19 @@ export const ChatBot = ({ onMessage, onError, onToolUse, presetMessages = [], pl
             if (address !== hitAddress) {
                 return true;
             }
-            // Check if AIT exists - if not, user needs to sign in to generate AIT
-            if (!ait) {
-                return true;
-            }
+            // If user has valid token but no AIT, they don't need to sign in again
+            // They just need to generate AIT through the permission form
             return false;
         }
         catch (error) {
             console.error('Error parsing token:', error);
             return true;
         }
-    }, [hitAddress, nxtlinqAITServiceAccessToken, ait]);
+    }, [hitAddress, nxtlinqAITServiceAccessToken]);
     const handleVerifySuccess = (address) => {
         localStorage.setItem(`wallet_verified_${address}`, 'true');
         const getWalletInfo = async () => {
+            setIsWalletLoading(true);
             try {
                 const token = JSON.parse(localStorage.getItem('nxtlinqAITServiceAccessToken') || '');
                 const walletResponse = await nxtlinqApi.wallet.getWallet({ address }, token);
@@ -346,6 +439,9 @@ export const ChatBot = ({ onMessage, onError, onToolUse, presetMessages = [], pl
             catch (error) {
                 console.error('Failed to update wallet info after verification:', error);
             }
+            finally {
+                setIsWalletLoading(false);
+            }
         };
         getWalletInfo();
     };
@@ -357,16 +453,25 @@ export const ChatBot = ({ onMessage, onError, onToolUse, presetMessages = [], pl
             if (isNeedSignInWithWallet) {
                 return;
             }
-            const getWalletResponse = await nxtlinqApi.wallet.getWallet({ address: hitAddress }, nxtlinqAITServiceAccessToken);
-            if ('error' in getWalletResponse) {
-                if (getWalletResponse.error === 'Wallet not found') {
-                    console.log('Wallet not found - this is expected for new users');
+            setIsWalletLoading(true);
+            try {
+                const getWalletResponse = await nxtlinqApi.wallet.getWallet({ address: hitAddress }, nxtlinqAITServiceAccessToken);
+                if ('error' in getWalletResponse) {
+                    if (getWalletResponse.error === 'Wallet not found') {
+                        console.log('Wallet not found - this is expected for new users');
+                        return;
+                    }
+                    console.error(getWalletResponse.error);
                     return;
                 }
-                console.error(getWalletResponse.error);
-                return;
+                setWalletInfo(getWalletResponse);
+            }
+            catch (error) {
+                console.error('Error getting wallet info:', error);
+            }
+            finally {
+                setIsWalletLoading(false);
             }
-            setWalletInfo(getWalletResponse);
         };
         getWalletInfo();
     }, [hitAddress, isNeedSignInWithWallet]);
@@ -405,19 +510,29 @@ export const ChatBot = ({ onMessage, onError, onToolUse, presetMessages = [], pl
                         if ('error' in verifyWalletResponse) {
                             if (verifyWalletResponse.error === 'Wallet already exists') {
                                 // If wallet exists, get wallet info directly
-                                const walletResponse = await nxtlinqApi.wallet.getWallet({ address }, token);
-                                if (!('error' in walletResponse)) {
-                                    setWalletInfo(walletResponse);
-                                    const aitResponse = await nxtlinqApi.ait.getAITByServiceIdAndController({ serviceId, controller: address }, token);
-                                    if (!('error' in aitResponse)) {
-                                        setAit(aitResponse);
+                                setIsWalletLoading(true);
+                                try {
+                                    const walletResponse = await nxtlinqApi.wallet.getWallet({ address }, token);
+                                    if (!('error' in walletResponse)) {
+                                        setWalletInfo(walletResponse);
+                                        const aitResponse = await nxtlinqApi.ait.getAITByServiceIdAndController({ serviceId, controller: address }, token);
+                                        if (!('error' in aitResponse)) {
+                                            setAit(aitResponse);
+                                        }
                                     }
                                 }
-                                // Clean up token from URL
+                                finally {
+                                    setIsWalletLoading(false);
+                                }
+                                // Clean up URL parameters after wallet verification
                                 if (typeof window !== 'undefined') {
                                     try {
                                         const url = new URL(window.location.href);
+                                        // Clean up all Berify.me related parameters
                                         url.searchParams.delete('token');
+                                        url.searchParams.delete('isAutoConnect');
+                                        url.searchParams.delete('method');
+                                        url.searchParams.delete('returnUrl');
                                         window.history.replaceState({}, '', url.toString());
                                     }
                                     catch (e) {
@@ -434,19 +549,29 @@ export const ChatBot = ({ onMessage, onError, onToolUse, presetMessages = [], pl
                             return;
                         }
                         // Verification successful, get wallet info
-                        const walletResponse = await nxtlinqApi.wallet.getWallet({ address }, token);
-                        if (!('error' in walletResponse)) {
-                            setWalletInfo(walletResponse);
-                            const aitResponse = await nxtlinqApi.ait.getAITByServiceIdAndController({ serviceId, controller: address }, token);
-                            if (!('error' in aitResponse)) {
-                                setAit(aitResponse);
+                        setIsWalletLoading(true);
+                        try {
+                            const walletResponse = await nxtlinqApi.wallet.getWallet({ address }, token);
+                            if (!('error' in walletResponse)) {
+                                setWalletInfo(walletResponse);
+                                const aitResponse = await nxtlinqApi.ait.getAITByServiceIdAndController({ serviceId, controller: address }, token);
+                                if (!('error' in aitResponse)) {
+                                    setAit(aitResponse);
+                                }
                             }
                         }
-                        // Clean up token from URL
+                        finally {
+                            setIsWalletLoading(false);
+                        }
+                        // Clean up URL parameters after wallet verification
                         if (typeof window !== 'undefined') {
                             try {
                                 const url = new URL(window.location.href);
+                                // Clean up all Berify.me related parameters
                                 url.searchParams.delete('token');
+                                url.searchParams.delete('isAutoConnect');
+                                url.searchParams.delete('method');
+                                url.searchParams.delete('returnUrl');
                                 window.history.replaceState({}, '', url.toString());
                             }
                             catch (e) {
@@ -639,16 +764,26 @@ export const ChatBot = ({ onMessage, onError, onToolUse, presetMessages = [], pl
                 }]);
             return false;
         }
-        if (!ait) {
+        // Check if user has signed in with wallet
+        if (!nxtlinqAITServiceAccessToken) {
             setMessages(prev => [...prev, {
                     id: Date.now().toString(),
-                    content: 'Please sign in with your wallet to continue.',
+                    content: 'Please sign in with your HIT wallet to continue.',
                     role: 'assistant',
                     timestamp: new Date().toISOString(),
                     button: 'signIn'
                 }]);
             return false;
         }
+        if (!ait) {
+            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;
+        }
         // Check if the tool permission is available for current identity provider
         const availablePermissionLabels = availablePermissions.map(p => p.label);
         if (!availablePermissionLabels.includes(toolName)) {
@@ -664,7 +799,7 @@ export const ChatBot = ({ onMessage, onError, onToolUse, presetMessages = [], pl
         if (!permissions.includes(toolName)) {
             setMessages(prev => [...prev, {
                     id: Date.now().toString(),
-                    content: `You do not have the required AIT permission: ${toolName}`,
+                    content: `You do not have the required AIT permission: ${toolName}. Please click the settings button (⚙️) to update your permissions.`,
                     role: 'assistant',
                     timestamp: new Date().toISOString()
                 }]);
@@ -805,17 +940,15 @@ export const ChatBot = ({ onMessage, onError, onToolUse, presetMessages = [], pl
         setAit(aitInfo);
     };
     const savePermissions = async () => {
-        setIsLoading(true);
         setIsDisabled(true);
         try {
             await generateAndRegisterAIT();
-            setIsLoading(false);
             setSuccess(true);
             setShowPermissionForm(false);
+            setIsPermissionFormOpen(false);
         }
         catch (error) {
             console.error('Failed to generate AIT:', error);
-            setIsLoading(false);
             setIsDisabled(false);
             if (error instanceof Error) {
                 alert(error.message);
@@ -838,6 +971,15 @@ export const ChatBot = ({ onMessage, onError, onToolUse, presetMessages = [], pl
             onMessage?.(lastMessage);
         }
     }, [messages, onMessage]);
+    // Auto hide success message after 3 seconds
+    React.useEffect(() => {
+        if (success) {
+            const timer = setTimeout(() => {
+                setSuccess(false);
+            }, 3000);
+            return () => clearTimeout(timer);
+        }
+    }, [success]);
     return (_jsxs("div", { style: {
             position: 'fixed',
             bottom: '20px',
@@ -864,7 +1006,12 @@ export const ChatBot = ({ onMessage, onError, onToolUse, presetMessages = [], pl
                             display: 'flex',
                             justifyContent: 'space-between',
                             alignItems: 'center'
-                        }, children: [_jsx("h3", { style: { margin: 0 }, children: "AI Agent" }), _jsxs("div", { style: { display: 'flex', gap: '10px' }, children: [_jsx("button", { onClick: () => setShowPermissionForm(true), style: {
+                        }, children: [_jsx("h3", { style: { margin: 0 }, children: "AI Agent" }), _jsxs("div", { style: { display: 'flex', gap: '10px' }, children: [_jsx("button", { onClick: async () => {
+                                            setShowPermissionForm(true);
+                                            setIsPermissionFormOpen(true);
+                                            // Force refresh AIT data when opening the form to show latest permissions
+                                            await refreshAIT(true);
+                                        }, style: {
                                             background: 'none',
                                             border: 'none',
                                             color: 'white',
@@ -973,7 +1120,10 @@ export const ChatBot = ({ onMessage, onError, onToolUse, presetMessages = [], pl
                     alignItems: 'center',
                     justifyContent: 'center',
                     zIndex: 2000
-                }, children: _jsx(PermissionForm, { hitAddress: hitAddress, permissions: permissions, setPermissions: setPermissions, setIsDisabled: setIsDisabled, onClose: () => setShowPermissionForm(false), onConnectWallet: () => connectWallet(false), onSignIn: () => signInWallet(false), onSave: savePermissions, isNeedSignInWithWallet: isNeedSignInWithWallet, walletInfo: walletInfo, onVerifyWallet: handleVerifyWalletClick, serviceId: serviceId, nxtlinqApi: nxtlinqApi, permissionGroup: permissionGroup }) })), success && (_jsxs("div", { style: {
+                }, children: _jsx(PermissionForm, { hitAddress: hitAddress, permissions: permissions, setPermissions: setPermissions, setIsDisabled: setIsDisabled, onClose: () => {
+                        setShowPermissionForm(false);
+                        setIsPermissionFormOpen(false);
+                    }, onConnectWallet: () => connectWallet(false), onSignIn: () => signInWallet(false), onSave: savePermissions, isNeedSignInWithWallet: isNeedSignInWithWallet, walletInfo: walletInfo, onVerifyWallet: handleVerifyWalletClick, serviceId: serviceId, nxtlinqApi: nxtlinqApi, permissionGroup: permissionGroup, isAITLoading: isAITLoading, isWalletLoading: isWalletLoading }) })), success && (_jsxs("div", { style: {
                     position: 'fixed',
                     bottom: 100,
                     right: 40,

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@bytexbyte/nxtlinq-ai-agent-sdk",
-  "version": "1.1.9",
-  "description": "Nxtlinq AI Agent SDK - Proprietary Software",
+  "version": "1.2.1",
+  "description": "Nxtlinq AI Agent SDK - Proprietary Software with enhanced async operation handling",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "files": [
@@ -16,7 +16,10 @@
     "nxtlinq",
     "ai",
     "agent",
-    "sdk"
+    "sdk",
+    "wallet",
+    "authentication",
+    "berify"
   ],
   "author": "ByteXByte",
   "license": "MIT",