@hef2024/llmasaservice-ui 0.20.2 → 0.20.3

package/dist/index.js

@@ -5698,9 +5698,13 @@ var AIAgentPanel = import_react14.default.forwardRef(({
   const [currentAgentId, setCurrentAgentId] = (0, import_react14.useState)(
     defaultAgent || agentIds[0] || ""
   );
+  const [pendingFollowOnPrompt, setPendingFollowOnPrompt] = (0, import_react14.useState)(null);
+  const [agentSwitchSettled, setAgentSwitchSettled] = (0, import_react14.useState)(true);
+  const lastProcessedFollowOnRef = (0, import_react14.useRef)("");
   (0, import_react14.useEffect)(() => {
     if (selectedAgent && selectedAgent !== currentAgentId) {
       const oldAgentId = currentAgentId;
+      setAgentSwitchSettled(false);
       setCurrentAgentId(selectedAgent);
       if (onAgentSwitch) {
         onAgentSwitch(oldAgentId, selectedAgent);
@@ -5720,6 +5724,37 @@ var AIAgentPanel = import_react14.default.forwardRef(({
       }
     }
   }, [selectedAgent]);
+  (0, import_react14.useEffect)(() => {
+    if (!agentSwitchSettled && followOnPrompt && followOnPrompt !== "" && followOnPrompt !== lastProcessedFollowOnRef.current) {
+      setPendingFollowOnPrompt(followOnPrompt);
+      lastProcessedFollowOnRef.current = followOnPrompt;
+    }
+  }, [followOnPrompt, agentSwitchSettled]);
+  (0, import_react14.useEffect)(() => {
+    if (!agentSwitchSettled) {
+      const timer = setTimeout(() => {
+        setAgentSwitchSettled(true);
+      }, 100);
+      return () => clearTimeout(timer);
+    }
+  }, [agentSwitchSettled]);
+  const effectiveFollowOnPrompt = (0, import_react14.useMemo)(() => {
+    if (!agentSwitchSettled) {
+      return "";
+    }
+    if (pendingFollowOnPrompt) {
+      return pendingFollowOnPrompt;
+    }
+    return followOnPrompt;
+  }, [followOnPrompt, pendingFollowOnPrompt, agentSwitchSettled]);
+  (0, import_react14.useEffect)(() => {
+    if (agentSwitchSettled && pendingFollowOnPrompt) {
+      const timer = setTimeout(() => {
+        setPendingFollowOnPrompt(null);
+      }, 100);
+      return () => clearTimeout(timer);
+    }
+  }, [agentSwitchSettled, pendingFollowOnPrompt]);
   const [apiConversations, setApiConversations] = (0, import_react14.useState)([]);
   const [conversationsLoading, setConversationsLoading] = (0, import_react14.useState)(false);
   const [conversationsError, setConversationsError] = (0, import_react14.useState)(null);
@@ -6766,7 +6801,7 @@ var AIAgentPanel = import_react14.default.forwardRef(({
         initialMessage,
         hideInitialPrompt,
         followOnQuestions,
-        followOnPrompt,
+        followOnPrompt: effectiveFollowOnPrompt,
         agentOptions,
         currentAgentId,
         handleAgentSwitch,

package/dist/index.mjs

@@ -5665,9 +5665,13 @@ var AIAgentPanel = React13.forwardRef(({
   const [currentAgentId, setCurrentAgentId] = useState8(
     defaultAgent || agentIds[0] || ""
   );
+  const [pendingFollowOnPrompt, setPendingFollowOnPrompt] = useState8(null);
+  const [agentSwitchSettled, setAgentSwitchSettled] = useState8(true);
+  const lastProcessedFollowOnRef = useRef6("");
   useEffect9(() => {
     if (selectedAgent && selectedAgent !== currentAgentId) {
       const oldAgentId = currentAgentId;
+      setAgentSwitchSettled(false);
       setCurrentAgentId(selectedAgent);
       if (onAgentSwitch) {
         onAgentSwitch(oldAgentId, selectedAgent);
@@ -5687,6 +5691,37 @@ var AIAgentPanel = React13.forwardRef(({
       }
     }
   }, [selectedAgent]);
+  useEffect9(() => {
+    if (!agentSwitchSettled && followOnPrompt && followOnPrompt !== "" && followOnPrompt !== lastProcessedFollowOnRef.current) {
+      setPendingFollowOnPrompt(followOnPrompt);
+      lastProcessedFollowOnRef.current = followOnPrompt;
+    }
+  }, [followOnPrompt, agentSwitchSettled]);
+  useEffect9(() => {
+    if (!agentSwitchSettled) {
+      const timer = setTimeout(() => {
+        setAgentSwitchSettled(true);
+      }, 100);
+      return () => clearTimeout(timer);
+    }
+  }, [agentSwitchSettled]);
+  const effectiveFollowOnPrompt = useMemo4(() => {
+    if (!agentSwitchSettled) {
+      return "";
+    }
+    if (pendingFollowOnPrompt) {
+      return pendingFollowOnPrompt;
+    }
+    return followOnPrompt;
+  }, [followOnPrompt, pendingFollowOnPrompt, agentSwitchSettled]);
+  useEffect9(() => {
+    if (agentSwitchSettled && pendingFollowOnPrompt) {
+      const timer = setTimeout(() => {
+        setPendingFollowOnPrompt(null);
+      }, 100);
+      return () => clearTimeout(timer);
+    }
+  }, [agentSwitchSettled, pendingFollowOnPrompt]);
   const [apiConversations, setApiConversations] = useState8([]);
   const [conversationsLoading, setConversationsLoading] = useState8(false);
   const [conversationsError, setConversationsError] = useState8(null);
@@ -6733,7 +6768,7 @@ var AIAgentPanel = React13.forwardRef(({
         initialMessage,
         hideInitialPrompt,
         followOnQuestions,
-        followOnPrompt,
+        followOnPrompt: effectiveFollowOnPrompt,
         agentOptions,
         currentAgentId,
         handleAgentSwitch,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hef2024/llmasaservice-ui",
-  "version": "0.20.2",
+  "version": "0.20.3",
   "description": "Prebuilt UI components for LLMAsAService.io",
   "main": "dist/index.js",
   "module": "dist/index.mjs",

package/src/AIAgentPanel.tsx

@@ -716,11 +716,20 @@ const AIAgentPanel = React.forwardRef<AIAgentPanelHandle, AIAgentPanelProps>(({
     defaultAgent || agentIds[0] || ''
   );
 
+  // Track pending follow-on prompt after agent switch (to handle timing issues)
+  const [pendingFollowOnPrompt, setPendingFollowOnPrompt] = useState<string | null>(null);
+  const [agentSwitchSettled, setAgentSwitchSettled] = useState(true);
+  const lastProcessedFollowOnRef = useRef<string>('');
+
   // Sync with controlled selectedAgent prop when it changes
   useEffect(() => {
     // Only sync if selectedAgent is provided (controlled mode) and differs from current
     if (selectedAgent && selectedAgent !== currentAgentId) {
       const oldAgentId = currentAgentId;
+      
+      // Mark that an agent switch is in progress (not yet settled)
+      setAgentSwitchSettled(false);
+      
       setCurrentAgentId(selectedAgent);
       
       // Fire onAgentSwitch callback for programmatic changes too
@@ -744,7 +753,54 @@ const AIAgentPanel = React.forwardRef<AIAgentPanelHandle, AIAgentPanelProps>(({
         });
       }
     }
-  }, [selectedAgent]); // Note: intentionally excluding currentAgentId and onAgentSwitch to only trigger on prop change
+  }, [selectedAgent]); // Note: intentionally excluding other deps to only trigger on prop change
+
+  // Queue followOnPrompt if it arrives during an agent switch
+  useEffect(() => {
+    if (!agentSwitchSettled && followOnPrompt && followOnPrompt !== '' && followOnPrompt !== lastProcessedFollowOnRef.current) {
+      // Agent switch in progress - queue the prompt
+      setPendingFollowOnPrompt(followOnPrompt);
+      lastProcessedFollowOnRef.current = followOnPrompt;
+    }
+  }, [followOnPrompt, agentSwitchSettled]);
+
+  // Mark agent switch as settled after a short delay (allows ChatPanelWrapper to re-render)
+  useEffect(() => {
+    if (!agentSwitchSettled) {
+      const timer = setTimeout(() => {
+        setAgentSwitchSettled(true);
+      }, 100); // Small delay to ensure ChatPanelWrapper has re-rendered with new agent
+      
+      return () => clearTimeout(timer);
+    }
+  }, [agentSwitchSettled]);
+
+  // Compute effective follow-on prompt (handles timing with agent switch)
+  const effectiveFollowOnPrompt = useMemo(() => {
+    // If agent switch hasn't settled yet, don't send any prompt
+    if (!agentSwitchSettled) {
+      return '';
+    }
+    
+    // If there's a pending prompt from agent switch, use it
+    if (pendingFollowOnPrompt) {
+      return pendingFollowOnPrompt;
+    }
+    
+    // Otherwise use the prop directly
+    return followOnPrompt;
+  }, [followOnPrompt, pendingFollowOnPrompt, agentSwitchSettled]);
+
+  // Clear pending prompt after it's been passed down
+  useEffect(() => {
+    if (agentSwitchSettled && pendingFollowOnPrompt) {
+      // Clear after a tick to ensure it was processed
+      const timer = setTimeout(() => {
+        setPendingFollowOnPrompt(null);
+      }, 100);
+      return () => clearTimeout(timer);
+    }
+  }, [agentSwitchSettled, pendingFollowOnPrompt]);
 
   // API-based conversation state
   const [apiConversations, setApiConversations] = useState<APIConversationSummary[]>([]);
@@ -2267,7 +2323,7 @@ const AIAgentPanel = React.forwardRef<AIAgentPanelHandle, AIAgentPanelProps>(({
             initialMessage={initialMessage}
             hideInitialPrompt={hideInitialPrompt}
             followOnQuestions={followOnQuestions}
-            followOnPrompt={followOnPrompt}
+            followOnPrompt={effectiveFollowOnPrompt}
             agentOptions={agentOptions}
             currentAgentId={currentAgentId}
             handleAgentSwitch={handleAgentSwitch}