@meetsmore-oss/use-ai-client 1.12.1 → 1.13.0

package/dist/bundled.js

@@ -1371,6 +1371,8 @@ var defaultStrings = {
     API_OVERLOADED: "The AI service is currently experiencing high demand. Please try again in a moment.",
     /** Error when rate limited */
     RATE_LIMITED: "Too many requests. Please wait a moment before trying again.",
+    /** Error when the connection was lost while a response was being generated */
+    CONNECTION_LOST: "The connection was lost. Please send your message again.",
     /** Error for unknown/unexpected errors */
     UNKNOWN_ERROR: "An unexpected error occurred. Please try again."
   },
@@ -1631,6 +1633,17 @@ function mergeAssistantMessagesForDisplay(messages) {
   return result;
 }
 
+// src/utils/keyboard.ts
+function shouldSubmitOnEnter(e, mode) {
+  if (e.key !== "Enter" || e.nativeEvent.isComposing || e.keyCode === 229) {
+    return false;
+  }
+  if (mode === "enter") {
+    return !e.shiftKey;
+  }
+  return e.metaKey || e.ctrlKey;
+}
+
 // ../../node_modules/.bun/react-markdown@8.0.7+32264e8fb3466d46/node_modules/react-markdown/lib/uri-transformer.js
 var protocols = ["http", "https", "mailto", "tel"];
 function uriTransformer(uri) {
@@ -14678,7 +14691,8 @@ function UseAIChatPanel({
   onFeedback,
   pendingApprovals = [],
   onApproveToolCall,
-  onRejectToolCall
+  onRejectToolCall,
+  submitMode = "enter"
 }) {
   const strings = useStrings();
   const theme = useTheme();
@@ -14753,7 +14767,7 @@ function UseAIChatPanel({
     if (slashCommands.handleKeyDown(e)) {
       return;
     }
-    if (e.key === "Enter" && !e.shiftKey && !e.nativeEvent.isComposing && !(e.keyCode === 229)) {
+    if (shouldSubmitOnEnter(e, submitMode)) {
       e.preventDefault();
       handleSend();
     }
@@ -15773,9 +15787,10 @@ function useChatUIContext() {
   }
   return context;
 }
-function UseAIChat({ floating = false }) {
+function UseAIChat({ floating = false, submitMode }) {
   const ctx = useChatUIContext();
   const chatPanelProps = {
+    submitMode: submitMode ?? ctx.submitMode,
     onSendMessage: ctx.sendMessage,
     messages: ctx.messages,
     loading: ctx.loading,
@@ -23647,6 +23662,7 @@ var ErrorCode;
 ((ErrorCode2) => {
   ErrorCode2["API_OVERLOADED"] = "API_OVERLOADED";
   ErrorCode2["RATE_LIMITED"] = "RATE_LIMITED";
+  ErrorCode2["CONNECTION_LOST"] = "CONNECTION_LOST";
   ErrorCode2["UNKNOWN_ERROR"] = "UNKNOWN_ERROR";
 })(ErrorCode ||= {});
 var TOOL_APPROVAL_REQUEST = "TOOL_APPROVAL_REQUEST";
@@ -23663,9 +23679,12 @@ var UseAIClient = class {
   }
   socket = null;
   eventHandlers = /* @__PURE__ */ new Map();
-  reconnectAttempts = 0;
-  maxReconnectAttempts = 5;
+  // Reconnect indefinitely so clients recover after extended outages (mobile
+  // app backgrounded long enough for server pingTimeout, airplane mode, etc.).
+  // Socket.IO applies exponential backoff capped at reconnectionDelayMax,
+  // so steady-state retry frequency is ~one attempt per 10s.
   reconnectDelay = 1e3;
+  reconnectDelayMax = 1e4;
   // Session state
   _threadId = null;
   _tools = [];
@@ -23703,14 +23722,14 @@ var UseAIClient = class {
     this.socket = lookup2(this.serverUrl, {
       transports: ["polling", "websocket"],
       reconnection: true,
-      reconnectionAttempts: this.maxReconnectAttempts,
+      reconnectionAttempts: Infinity,
       reconnectionDelay: this.reconnectDelay,
+      reconnectionDelayMax: this.reconnectDelayMax,
       withCredentials: true
     });
     this.socket.on("connect", () => {
       console.log("[UseAI] Connected to server");
       console.log("[UseAI] Transport:", this.socket?.io?.engine?.transport?.name);
-      this.reconnectAttempts = 0;
       const engine = this.socket?.io?.engine;
       if (engine) {
         engine.on("upgrade", (transport) => {
@@ -39298,6 +39317,8 @@ function useServerEvents({
   const [streamingText, setStreamingText] = useState13("");
   const [streamingReasoning, setStreamingReasoning] = useState13("");
   const streamingChatIdRef = useRef10(null);
+  const loadingRef = useRef10(loading);
+  loadingRef.current = loading;
   const messageCountAtRunStartRef = useRef10(0);
   const hasTextFromPriorStepRef = useRef10(false);
   const [executingToolRaw, setExecutingTool] = useState13(null);
@@ -39393,6 +39414,17 @@ function useServerEvents({
       setLoading(false);
     }
   }, []);
+  const handleDisconnect = useCallback11(() => {
+    if (!loadingRef.current) return;
+    const strs = stringsRef.current;
+    const message = strs.errors[ErrorCode.CONNECTION_LOST] || strs.errors[ErrorCode.UNKNOWN_ERROR];
+    saveAIResponseRef.current(message, "error");
+    setStreamingText("");
+    setStreamingReasoning("");
+    streamingChatIdRef.current = null;
+    setExecutingTool(null);
+    setLoading(false);
+  }, []);
   const executingTool = executingToolRaw ? {
     displayText: executingToolRaw.title ?? executingToolFallbackRef.current ?? strings.toolExecution.fallbackMessages[0]
   } : null;
@@ -39404,7 +39436,8 @@ function useServerEvents({
     executingTool,
     streamingChatIdRef,
     streamingReasoning,
-    handleServerEvent
+    handleServerEvent,
+    handleDisconnect
   };
 }
 
@@ -39570,7 +39603,8 @@ function UseAIProvider({
   theme: customTheme,
   strings: customStrings,
   visibleAgentIds,
-  onOpenChange
+  onOpenChange,
+  submitMode = "enter"
 }) {
   const fileUploadConfig = fileUploadConfigProp === false ? void 0 : fileUploadConfigProp ?? DEFAULT_FILE_UPLOAD_CONFIG;
   const theme = { ...defaultTheme, ...customTheme };
@@ -39629,12 +39663,17 @@ function UseAIProvider({
   } = useCommandManagement({ repository: commandRepository });
   const handleServerEventRef = useRef12(serverEvents.handleServerEvent);
   handleServerEventRef.current = serverEvents.handleServerEvent;
+  const handleDisconnectRef = useRef12(serverEvents.handleDisconnect);
+  handleDisconnectRef.current = serverEvents.handleDisconnect;
   useEffect11(() => {
     console.log("[UseAIProvider] Initializing client with serverUrl:", serverUrl);
     const client = new UseAIClient(serverUrl);
     const unsubscribeConnection = client.onConnectionStateChange((isConnected) => {
       console.log("[UseAIProvider] Connection state changed:", isConnected);
       setConnected(isConnected);
+      if (!isConnected) {
+        handleDisconnectRef.current();
+      }
     });
     console.log("[UseAIProvider] Connecting...");
     client.connect();
@@ -39822,7 +39861,8 @@ function UseAIProvider({
     feedback: {
       enabled: feedback.enabled,
       submit: feedback.submitFeedback
-    }
+    },
+    submitMode
   };
   const isUIDisabled = CustomButton === null || CustomChat === null;
   const ButtonComponent = isUIDisabled ? null : CustomButton || UseAIFloatingButton;
@@ -39855,7 +39895,8 @@ function UseAIProvider({
     onFeedback: feedback.submitFeedback,
     pendingApprovals: toolSystem.pendingApprovals,
     onApproveToolCall: toolSystem.pendingApprovals.length > 0 ? toolSystem.approveAll : void 0,
-    onRejectToolCall: toolSystem.pendingApprovals.length > 0 ? toolSystem.rejectAll : void 0
+    onRejectToolCall: toolSystem.pendingApprovals.length > 0 ? toolSystem.rejectAll : void 0,
+    submitMode
   };
   const renderDefaultChat = () => {
     if (isUIDisabled) return null;