fix(web): preserve slash commands on reconnect and add initialize retry logic

CHANGELOG.md

@@ -12,6 +12,7 @@ Only write entries that are worth mentioning to users.
 ## Unreleased
 
 - Core: Pass session ID as `user_id` metadata to Anthropic API
+- Web: Preserve slash commands on WebSocket reconnect and add automatic retry logic for session initialization
 
 ## 1.17.0 (2026-03-03)
 

docs/en/reference/kimi-info.md

@@ -27,15 +27,15 @@ kimi info [--json]
 
 ```sh
 $ kimi info
-kimi-cli version: 0.71
+kimi-cli version: 1.17.0
 agent spec versions: 1
-wire protocol: 1
-python version: 3.14.0
+wire protocol: 1.4
+python version: 3.13.1
 ```
 
 **JSON output**
 
 ```sh
 $ kimi info --json
-{"kimi_cli_version": "0.71", "agent_spec_versions": ["1"], "wire_protocol_version": "1", "python_version": "3.13.1"}
+{"kimi_cli_version": "1.17.0", "agent_spec_versions": ["1"], "wire_protocol_version": "1.4", "python_version": "3.13.1"}
 ```

docs/en/reference/kimi-web.md

@@ -92,7 +92,7 @@ Use `--auth-token` to set an access token. Clients need to include `Authorizatio
 kimi web --network --auth-token my-secret-token
 ```
 
-::: tip Tip
+::: tip
 The access token should be a randomly generated string with at least 32 characters. You can use `openssl rand -hex 32` to generate a random token.
 :::
 
@@ -104,7 +104,7 @@ Use `--allowed-origins` to restrict the origin domains that can access Web UI:
 kimi web --network --allowed-origins "https://example.com,https://app.example.com"
 ```
 
-::: tip Tip
+::: tip
 When using `--network` or `--host` to enable network access, it is recommended to configure `--allowed-origins` to prevent Cross-Site Request Forgery (CSRF) attacks.
 :::
 
@@ -134,7 +134,7 @@ kimi web --network --restrict-sensitive-apis
 
 In `--public` mode, `--restrict-sensitive-apis` is enabled by default; in `--lan-only` mode (default), it is not enabled.
 
-::: tip Tip
+::: tip
 When you need to expose Web UI to untrusted network environments, it is recommended to enable the `--restrict-sensitive-apis` option.
 :::
 

docs/en/release-notes/changelog.md

@@ -5,6 +5,7 @@ This page documents the changes in each Kimi Code CLI release.
 ## Unreleased
 
 - Core: Pass session ID as `user_id` metadata to Anthropic API
+- Web: Preserve slash commands on WebSocket reconnect and add automatic retry logic for session initialization
 
 ## 1.17.0 (2026-03-03)
 

docs/zh/reference/kimi-info.md

@@ -27,15 +27,15 @@ kimi info [--json]
 
 ```sh
 $ kimi info
-kimi-cli version: 0.71
+kimi-cli version: 1.17.0
 agent spec versions: 1
-wire protocol: 1
-python version: 3.14.0
+wire protocol: 1.4
+python version: 3.13.1
 ```
 
 **JSON 输出**
 
 ```sh
 $ kimi info --json
-{"kimi_cli_version": "0.71", "agent_spec_versions": ["1"], "wire_protocol_version": "1", "python_version": "3.13.1"}
+{"kimi_cli_version": "1.17.0", "agent_spec_versions": ["1"], "wire_protocol_version": "1.4", "python_version": "3.13.1"}
 ```

docs/zh/release-notes/changelog.md

@@ -5,6 +5,7 @@
 ## 未发布
 
 - Core：将会话 ID 作为 `user_id` 元数据传递给 Anthropic API
+- Web：修复 WebSocket 重连时斜杠命令丢失的问题，并为会话初始化添加自动重试逻辑
 
 ## 1.17.0 (2026-03-03)
 

web/src/hooks/useSessionStream.ts

@@ -324,6 +324,9 @@ export function useSessionStream(
 
   // Initialize message tracking
   const initializeIdRef = useRef<string | null>(null);
+  const initializeRetryCountRef = useRef(0); // Track retry attempts for initialize
+  const MAX_INITIALIZE_RETRIES = 5; // Maximum retry attempts
+  const usingCachedCommandsRef = useRef(false); // Track if using cached slash commands
 
   // Current state accumulators
   const currentThinkingRef = useRef("");
@@ -776,7 +779,7 @@ export function useSessionStream(
   }, []);
 
   // Reset all state
-  const resetState = useCallback(() => {
+  const resetState = useCallback((preserveSlashCommands = false) => {
     resetStepState();
     currentToolCallsRef.current.clear();
     currentToolCallIdRef.current = null;
@@ -792,7 +795,6 @@ export function useSessionStream(
     isReplayingRef.current = true;
     setIsReplayingHistory(true);
     setAwaitingFirstResponse(false);
-    setSlashCommands([]);
     // Reset first turn tracking
     hasTurnStartedRef.current = false;
     firstTurnCompleteCalledRef.current = false;
@@ -803,7 +805,14 @@ export function useSessionStream(
       window.clearTimeout(historyCompleteTimeoutRef.current);
       historyCompleteTimeoutRef.current = null;
     }
-  }, [resetStepState, setAwaitingFirstResponse]);
+    // Handle slashCommands: preserve or clear
+    if (!preserveSlashCommands) {
+      setSlashCommands([]);
+      usingCachedCommandsRef.current = false;
+    } else if (slashCommands.length > 0) {
+      usingCachedCommandsRef.current = true;
+    }
+  }, [resetStepState, setAwaitingFirstResponse, slashCommands.length]);
 
   // Process a SubagentEvent: accumulate inner events into parent Task tool's subagentSteps
   const processSubagentEvent = useCallback(
@@ -1805,20 +1814,56 @@ export function useSessionStream(
     ],
   );
 
+  // Helper to send initialize message
+  const sendInitialize = useCallback((ws: WebSocket) => {
+    const id = uuidV4();
+    initializeIdRef.current = id;
+    const message = {
+      jsonrpc: "2.0",
+      method: "initialize",
+      id,
+      params: {
+        protocol_version: "1.3",
+        client: {
+          name: "kiwi",
+          version: kimiCliVersion,
+        },
+        capabilities: {
+          supports_question: true,
+        },
+      },
+    };
+    ws.send(JSON.stringify(message));
+    console.log("[SessionStream] Sent initialize message");
+  }, []);
+
   // Handle incoming WebSocket message
   const handleMessage = useCallback(
     (data: string) => {
       try {
-        console.log("[SessionStream] Received raw message:", data);
         const message: WireMessage = JSON.parse(data);
-        console.log("[SessionStream] Parsed message:", message);
 
         // Check for JSON-RPC error response
         if (message.error) {
-          // Initialize failure during busy session is non-fatal - just skip
+          // Initialize failure during busy session is non-fatal - retry after delay
           if (message.id === initializeIdRef.current) {
-            console.warn("[SessionStream] Initialize rejected (session busy), continuing...");
+            initializeRetryCountRef.current += 1;
+
+            if (initializeRetryCountRef.current > MAX_INITIALIZE_RETRIES) {
+              initializeIdRef.current = null;
+              initializeRetryCountRef.current = 0;
+              return;
+            }
+
             initializeIdRef.current = null;
+
+            // Auto-retry initialize after 2 seconds
+            setTimeout(() => {
+              if (wsRef.current?.readyState === WebSocket.OPEN) {
+                sendInitialize(wsRef.current);
+              }
+            }, 2000);
+
             return;
           }
 
@@ -1897,13 +1942,14 @@ export function useSessionStream(
 
         // Handle initialize response
         if (message.id && message.id === initializeIdRef.current && message.result) {
-          console.log("[SessionStream] Initialize response received:", message.result);
           initializeIdRef.current = null;
+          initializeRetryCountRef.current = 0;
 
-          // Extract slash commands
           const { slash_commands } = message.result;
-          if (slash_commands) {
+
+          if (slash_commands && slash_commands.length > 0) {
             setSlashCommands(slash_commands);
+            usingCachedCommandsRef.current = false;
           }
           return;
         }
@@ -1961,6 +2007,7 @@ export function useSessionStream(
       setAwaitingFirstResponse,
       applySessionStatus,
       completeStreamingMessages,
+      sendInitialize,
     ],
   );
 
@@ -2010,29 +2057,6 @@ export function useSessionStream(
     [setAwaitingFirstResponse],
   );
 
-  // Helper to send initialize message
-  const sendInitialize = useCallback((ws: WebSocket) => {
-    const id = uuidV4();
-    initializeIdRef.current = id;
-    const message = {
-      jsonrpc: "2.0",
-      method: "initialize",
-      id,
-      params: {
-        protocol_version: "1.3",
-        client: {
-          name: "kiwi",
-          version: kimiCliVersion,
-        },
-        capabilities: {
-          supports_question: true,
-        },
-      },
-    };
-    ws.send(JSON.stringify(message));
-    console.log("[SessionStream] Sent initialize message");
-  }, []);
-
   const respondToApproval = useCallback(
     async (
       requestId: string,
@@ -2196,8 +2220,11 @@ export function useSessionStream(
   const connect = useCallback(() => {
     if (!sessionId) return;
 
+    initializeRetryCountRef.current = 0; // Reset retry count for new connection
+
     // Close existing connection
     if (wsRef.current) {
+      console.log("[SessionStream] Closing existing WebSocket");
       wsRef.current.close();
       wsRef.current = null;
     }
@@ -2207,7 +2234,7 @@ export function useSessionStream(
     }
 
     awaitingIdleRef.current = false;
-    resetState();
+    resetState(true);  // preserve slashCommands on reconnect
     setMessages([]);
     setStatus("submitted");
     setAwaitingFirstResponse(Boolean(pendingMessageRef.current));