@bytexbyte/nxtlinq-ai-agent-ui-react-development 0.4.1 → 0.4.3

package/src/index.ts

@@ -32,6 +32,8 @@ export type {
   Message,
   SendMessageOptions,
   NxtlinqAgentSnapshot,
+  TextToolAuthorizationSnapshot,
+  TextToolAuthorizationOutcome,
   VoiceSession,
   VoiceStatus,
 } from '@bytexbyte/nxtlinq-ai-agent-core-development';
@@ -41,6 +43,9 @@ export {
   setApiHosts,
   VoiceNotSupportedError,
   STORAGE_KEYS,
+  authorizeTextFrontendTool,
+  hasRecordedWalletVerification,
+  validateWalletSession,
 } from '@bytexbyte/nxtlinq-ai-agent-core-development';
 
 export {

package/src/pendingTextToolRetry.ts

@@ -0,0 +1,39 @@
+import type { Attachment } from '@bytexbyte/nxtlinq-ai-agent-core-development';
+
+export type PendingTextToolRequest = {
+  content: string;
+  isPresetMessage: boolean;
+  attachments?: Attachment[];
+};
+
+/** Keeps one interrupted text-tool request and serializes all resume attempts. */
+export class PendingTextToolRetryController {
+  private pending: PendingTextToolRequest | null = null;
+  private inFlight: Promise<boolean> | null = null;
+
+  set(request: PendingTextToolRequest): void {
+    this.pending = request;
+  }
+
+  clear(): void {
+    this.pending = null;
+  }
+
+  hasPending(): boolean {
+    return this.pending !== null;
+  }
+
+  retry(execute: (request: PendingTextToolRequest) => Promise<void>): Promise<boolean> {
+    if (this.inFlight) return this.inFlight;
+    if (!this.pending) return Promise.resolve(false);
+
+    const request = this.pending;
+    this.inFlight = (async () => {
+      await execute(request);
+      return true;
+    })().finally(() => {
+      this.inFlight = null;
+    });
+    return this.inFlight;
+  }
+}

package/src/types/ChatBotTypes.ts

@@ -163,8 +163,9 @@ export interface ChatBotContextType {
 
   // Functions
   connectWallet: (autoShowSignInMessage?: boolean) => Promise<string | false | undefined>;
-  signInWallet: (autoShowSuccessMessage?: boolean) => Promise<void>;
-  sendMessage: (content: string, retryCount?: number) => Promise<void>;
+  signInWallet: (autoShowSuccessMessage?: boolean) => Promise<boolean>;
+  retryPendingTextTool: () => Promise<boolean>;
+  sendMessage: (content: string, retryCount?: number, isPresetMessage?: boolean, attachments?: Attachment[]) => Promise<void>;
   handleSubmit: (e: React.FormEvent, attachments?: Attachment[]) => Promise<void>;
   handlePresetMessage: (message: PresetMessage) => void;
   savePermissions: (newPermissions?: string[]) => Promise<void>;
@@ -194,7 +195,7 @@ export interface ChatBotContextType {
   // Additional properties for PermissionForm
   onSave: (newPermissions?: string[]) => Promise<void>;
   onConnectWallet: () => Promise<string | false | undefined>;
-  onSignIn: () => Promise<void>;
+  onSignIn: () => Promise<boolean>;
   isNeedSignInWithWallet: boolean;
   onVerifyWallet: (method: 'berifyme' | 'custom') => Promise<void>;
   serviceId: string;
@@ -203,4 +204,4 @@ export interface ChatBotContextType {
   // Props
   props: ChatBotProps;
   nxtlinqApi: AITApi;
-} 
+}

package/src/ui/ChatBotUI.tsx

@@ -1,6 +1,7 @@
 /** @jsxImportSource @emotion/react */
 import { css } from '@emotion/react';
 import * as React from 'react';
+import { hasRecordedWalletVerification } from '@bytexbyte/nxtlinq-ai-agent-core-development';
 import { useDraggable, useLocalStorage, useResizable, walletTextUtils } from '@bytexbyte/nxtlinq-ai-agent-web-development';
 import { useChatBot } from '../context/ChatBotContext';
 import { ChatBotHeader } from './ChatBotHeader';
@@ -161,7 +162,8 @@ export const ChatBotUI: React.FC = () => {
   const urlParams = new URLSearchParams(window.location.search);
   const hasBerifymeToken = urlParams.get('token') && urlParams.get('method') === 'berifyme';
 
-  const isWalletVerifiedWithBerifyme = walletInfo?.id && walletInfo?.method === 'berifyme';
+  const hasRecordedVerification = Boolean(walletInfo?.id)
+    && hasRecordedWalletVerification(walletInfo);
 
   // Helper function to update IDV suggestion state based on storage value
   const updateIDVSuggestionState = React.useCallback((dismissedValue: string | null) => {
@@ -221,7 +223,7 @@ export const ChatBotUI: React.FC = () => {
     const shouldShowBanner = hitAddress &&
       !props.requireWalletIDVVerification &&
       !hasBerifymeToken &&
-      !isWalletVerifiedWithBerifyme &&
+      !hasRecordedVerification &&
       !isNeedSignInWithWallet;
 
     if (shouldShowBanner) {
@@ -229,7 +231,7 @@ export const ChatBotUI: React.FC = () => {
         const shouldShowBannerAfterDelay = hitAddress &&
           !props.requireWalletIDVVerification &&
           !hasBerifymeToken &&
-          !isWalletVerifiedWithBerifyme &&
+          !hasRecordedVerification &&
           !isNeedSignInWithWallet;
 
         if (shouldShowBannerAfterDelay) {
@@ -728,7 +730,7 @@ export const ChatBotUI: React.FC = () => {
           onClose={handleClose}
         />
 
-        {showIDVSuggestion && hitAddress && !props.requireWalletIDVVerification && !hasBerifymeToken && !isWalletVerifiedWithBerifyme && (
+        {showIDVSuggestion && hitAddress && !props.requireWalletIDVVerification && !hasBerifymeToken && !hasRecordedVerification && (
           <div
             data-idv-banner
             css={idvBanner}>

package/src/ui/MessageList.tsx

@@ -319,6 +319,7 @@ export const MessageList: React.FC = () => {
     retryTtsWithGesture,
     connectWallet,
     signInWallet,
+    retryPendingTextTool,
     hitAddress,
     isAutoConnecting,
     isNeedSignInWithWallet,
@@ -330,6 +331,8 @@ export const MessageList: React.FC = () => {
     availableModels,
     serviceId,
     piiDisplayMode,
+    setShowPermissionForm,
+    setIsPermissionFormOpen,
   } = useChatBot();
   const messagesEndRef = React.useRef<HTMLDivElement>(null);
 
@@ -351,22 +354,27 @@ export const MessageList: React.FC = () => {
     }
 
     if (buttonType === 'connectWallet') {
-      connectWallet(true);
+      const connected = await connectWallet(false);
+      if (connected) {
+        // Continue the interrupted flow immediately. This opens the wallet
+        // signature UI instead of leaving the user on a stale connect prompt.
+        const signedIn = await signInWallet(true);
+        if (signedIn) await retryPendingTextTool();
+      }
     } else if (buttonType === 'signIn') {
-      signInWallet(true);
+      const signedIn = await signInWallet(true);
+      if (signedIn) await retryPendingTextTool();
     } else if (buttonType === 'enableAIT') {
       const requiredPermission = message.metadata?.requiredPermission;
       if (requiredPermission) {
         const success = await enableAIT(requiredPermission);
         if (success) {
-          // Find the last user message
-          const lastUserMsg = [...messages].reverse().find(m => m.role === 'user');
-          if (lastUserMsg && lastUserMsg.content) {
-            // Re-execute the previous user message (retryCount=1 to skip duplicate PII log)
-            await sendMessage(lastUserMsg.content, 1);
-          }
+          await retryPendingTextTool();
         }
       }
+    } else if (buttonType === 'verifyWallet') {
+      setIsPermissionFormOpen(true);
+      setShowPermissionForm(true);
     } else if (buttonType === 'continue') {
       const lastUserMsg = [...messages].reverse().find(m => m.role === 'user');
       if (lastUserMsg && lastUserMsg.content) {
@@ -672,6 +680,7 @@ export const MessageList: React.FC = () => {
                   {isAutoConnecting ? 'Connecting...' :
                     message.button === 'connectWallet' ? (Boolean(hitAddress) ? 'Connected' : walletTextUtils.getWalletText('Connect Wallet', serviceId)) :
                       message.button === 'signIn' ? (!isNeedSignInWithWallet ? 'Signed In' : 'Sign In') :
+                        message.button === 'verifyWallet' ? 'Verify Wallet' :
                         message.button === 'continue' ? 'Continue' :
                           message.button === 'enableAIT' ?
                             ((isAITLoading || isAITEnabling) ? 'Enabling...' :

package/src/ui/PermissionForm.tsx

@@ -3,6 +3,7 @@ import * as React from 'react';
 import { css } from '@emotion/react';
 import { useChatBot } from '../context/ChatBotContext';
 import { walletTextUtils } from '@bytexbyte/nxtlinq-ai-agent-web-development';
+import { hasRecordedWalletVerification } from '@bytexbyte/nxtlinq-ai-agent-core-development';
 import { actionButton, modalOverlay } from './styles/isolatedStyles';
 
 interface PermissionFormProps {
@@ -103,13 +104,10 @@ export const PermissionForm: React.FC<PermissionFormProps> = ({ onClose }) => {
 
   // Check if wallet IDV verification is required based on configuration
   const requireWalletIDVVerification = props.requireWalletIDVVerification ?? true; // Default to true
-  const isWalletVerified = Boolean(walletInfo?.id);
-  const isWalletVerifiedWithBerifyme = walletInfo?.method === 'berifyme';
-
-  // Show verification prompt if:
-  // 1. IDV verification is required AND wallet is not verified, OR
-  // 2. IDV verification is required AND wallet is verified but not with Berifyme
-  const shouldShowVerificationPrompt = requireWalletIDVVerification && (!isWalletVerified || !isWalletVerifiedWithBerifyme);
+  // Any recorded verification source is valid. Custom additionally requires
+  // its explicit username evidence so synthetic AIT-provisioned rows do not count.
+  const shouldShowVerificationPrompt = requireWalletIDVVerification
+    && (!walletInfo?.id || !hasRecordedWalletVerification(walletInfo));
 
   const handleSave = async () => {
     setIsSaving(true);
@@ -415,10 +413,7 @@ export const PermissionForm: React.FC<PermissionFormProps> = ({ onClose }) => {
             font-size: 16px !important;
             color: #666 !important;
           `}>
-            {isWalletVerified && !isWalletVerifiedWithBerifyme
-              ? walletTextUtils.getWalletText('Your wallet is verified with custom method, but Berify.me verification is required to continue.', serviceId)
-              : walletTextUtils.getWalletText('Please verify your wallet with Berify.me to continue', serviceId)
-            }
+            {walletTextUtils.getWalletText('Please verify your wallet identity to continue', serviceId)}
           </p>
           <button
             onClick={() => onVerifyWallet('berifyme')}
@@ -629,4 +624,4 @@ export const PermissionForm: React.FC<PermissionFormProps> = ({ onClose }) => {
       )}
     </div>
   );
-}; 
+};