@cryptiklemur/lattice 1.46.7 → 2.0.0

package/src/client/components/ui/SaveFooter.tsx

@@ -0,0 +1,63 @@
+import { useEffect, useRef } from "react";
+import { Loader2, Check, X } from "lucide-react";
+import type { SaveState } from "../../hooks/useSaveState";
+import { showToast } from "./Toast";
+
+interface SaveFooterProps {
+  dirty: boolean;
+  saving: boolean;
+  saveState: SaveState;
+  onSave: () => void;
+  extraStatus?: string;
+}
+
+export function SaveFooter({ dirty, saving, saveState, onSave, extraStatus }: SaveFooterProps) {
+  var disabled = saving || (!dirty && saveState !== "error");
+  var prevStateRef = useRef<SaveState>("idle");
+
+  useEffect(function () {
+    if (prevStateRef.current !== saveState) {
+      if (saveState === "saved") {
+        showToast("Settings saved", "info");
+      } else if (saveState === "error") {
+        showToast("Failed to save settings — server did not respond within 5 seconds", "error");
+      }
+      prevStateRef.current = saveState;
+    }
+  }, [saveState]);
+
+  return (
+    <div className="flex items-center justify-end gap-3">
+      {extraStatus && (
+        <div className="text-[11px] text-warning/70">{extraStatus}</div>
+      )}
+      {!extraStatus && dirty && saveState === "idle" && !saving && (
+        <div className="text-[11px] text-warning/70">Unsaved changes</div>
+      )}
+      {saveState === "error" && (
+        <div className="flex items-center gap-1.5 text-[11px] text-error">
+          <X size={11} />
+          Save failed — server did not respond
+        </div>
+      )}
+      {saveState === "saved" && (
+        <div className="flex items-center gap-1.5 text-[11px] text-success/70">
+          <Check size={11} />
+          Saved
+        </div>
+      )}
+      <button
+        onClick={onSave}
+        disabled={disabled}
+        className={
+          "btn btn-sm gap-1.5 " +
+          (saveState === "saved" ? "btn-success" : saveState === "error" ? "btn-error" : "btn-primary") +
+          (disabled ? " opacity-50 cursor-not-allowed" : "")
+        }
+      >
+        {saving && <Loader2 size={13} className="animate-spin" />}
+        {saving ? "Saving..." : saveState === "error" ? "Retry" : "Save Changes"}
+      </button>
+    </div>
+  );
+}

package/{client/src → src/client}/components/workspace/TabBar.tsx

@@ -1,22 +1,16 @@
-import { useState, useEffect, useRef } from "react";
 import { X, Columns2, Rows2, MessageSquare, FolderOpen, TerminalSquare, StickyNote, Calendar, Bookmark, BarChart3 } from "lucide-react";
 import { useWorkspace } from "../../hooks/useWorkspace";
 import { useSession } from "../../hooks/useSession";
 import type { Tab, TabType } from "../../stores/workspace";
 import { pinTab } from "../../stores/workspace";
 import { formatSessionTitle } from "../../utils/formatSessionTitle";
+import { ContextMenu, useContextMenu } from "../ui/ContextMenu";
 
 interface TabBarProps {
   paneId?: string;
   isActivePane?: boolean;
 }
 
-interface ContextMenuState {
-  tabId: string;
-  x: number;
-  y: number;
-}
-
 var TAB_ICONS: Record<TabType, typeof MessageSquare> = {
   chat: MessageSquare,
   files: FolderOpen,
@@ -30,31 +24,7 @@ var TAB_ICONS: Record<TabType, typeof MessageSquare> = {
 export function TabBar({ paneId, isActivePane }: TabBarProps) {
   var workspace = useWorkspace();
   var session = useSession();
-  var [contextMenu, setContextMenu] = useState<ContextMenuState | null>(null);
-  var menuRef = useRef<HTMLDivElement>(null);
-
-  useEffect(function () {
-    if (!contextMenu) return;
-
-    function handleClickOutside(e: MouseEvent) {
-      if (menuRef.current && !menuRef.current.contains(e.target as Node)) {
-        setContextMenu(null);
-      }
-    }
-
-    function handleEscape(e: KeyboardEvent) {
-      if (e.key === "Escape") {
-        setContextMenu(null);
-      }
-    }
-
-    document.addEventListener("mousedown", handleClickOutside);
-    document.addEventListener("keydown", handleEscape);
-    return function () {
-      document.removeEventListener("mousedown", handleClickOutside);
-      document.removeEventListener("keydown", handleEscape);
-    };
-  }, [contextMenu]);
+  var ctxMenu = useContextMenu<string>();
 
   var paneTabs: Tab[];
   var activeTabId: string;
@@ -102,25 +72,18 @@ export function TabBar({ paneId, isActivePane }: TabBarProps) {
   }
 
   function handleContextMenu(e: React.MouseEvent, tabId: string) {
-    e.preventDefault();
     if (workspace.panes.length >= 2) return;
     var contextPane = paneId
       ? workspace.panes.find(function (p) { return p.id === paneId; })
       : workspace.panes[0];
     if (!contextPane || contextPane.tabIds.length < 2) return;
-    var menuWidth = 160;
-    var menuHeight = 80;
-    var x = e.clientX;
-    var y = e.clientY;
-    if (x + menuWidth > window.innerWidth) x = window.innerWidth - menuWidth - 8;
-    if (y + menuHeight > window.innerHeight) y = window.innerHeight - menuHeight - 8;
-    setContextMenu({ tabId, x, y });
+    ctxMenu.open(e, tabId);
   }
 
   function handleSplit(direction: "horizontal" | "vertical") {
-    if (!contextMenu) return;
-    workspace.splitPane(contextMenu.tabId, direction);
-    setContextMenu(null);
+    if (!ctxMenu.state) return;
+    workspace.splitPane(ctxMenu.state.data, direction);
+    ctxMenu.close();
   }
 
   function handleMiddleClick(e: React.MouseEvent, tab: Tab) {
@@ -190,27 +153,17 @@ export function TabBar({ paneId, isActivePane }: TabBarProps) {
           );
         })}
       </div>
-      {contextMenu && (
-        <div
-          ref={menuRef}
-          className="fixed z-50 bg-base-300 border border-base-content/15 rounded-lg shadow-lg py-1 min-w-[160px]"
-          style={{ left: contextMenu.x, top: contextMenu.y }}
-        >
-          <button
-            onClick={function () { handleSplit("horizontal"); }}
-            className="flex items-center gap-2 w-full px-3 py-1.5 text-[12px] font-mono text-base-content/80 hover:bg-base-content/15 hover:text-base-content transition-colors outline-none focus-visible:ring-2 focus-visible:ring-primary/50"
-          >
-            <Columns2 size={14} />
-            Split Right
-          </button>
-          <button
-            onClick={function () { handleSplit("vertical"); }}
-            className="flex items-center gap-2 w-full px-3 py-1.5 text-[12px] font-mono text-base-content/80 hover:bg-base-content/15 hover:text-base-content transition-colors outline-none focus-visible:ring-2 focus-visible:ring-primary/50"
-          >
-            <Rows2 size={14} />
-            Split Down
-          </button>
-        </div>
+      {ctxMenu.state !== null && (
+        <ContextMenu
+          x={ctxMenu.state.x}
+          y={ctxMenu.state.y}
+          items={[
+            { label: "Split Right", icon: <Columns2 size={14} />, onClick: function () { handleSplit("horizontal"); } },
+            { label: "Split Down", icon: <Rows2 size={14} />, onClick: function () { handleSplit("vertical"); } },
+          ]}
+          onClose={ctxMenu.close}
+          label="Tab actions"
+        />
       )}
     </>
   );

package/{client/src → src/client}/hooks/useAnalytics.ts

@@ -2,10 +2,12 @@ import { useEffect, useRef } from "react";
 import { useStore } from "@tanstack/react-store";
 import { useWebSocket } from "./useWebSocket";
 import type { ServerMessage } from "@lattice/shared";
-import type { AnalyticsPeriod, AnalyticsScope } from "@lattice/shared";
+import type { AnalyticsPeriod, AnalyticsScope, AnalyticsSectionName } from "@lattice/shared";
+import type { AnalyticsPayload } from "@lattice/shared";
 import {
   getAnalyticsStore,
-  setAnalyticsData,
+  mergeAnalyticsSection,
+  clearAnalyticsForRequest,
   setAnalyticsLoading,
   setAnalyticsError,
   setAnalyticsPeriod,
@@ -26,7 +28,7 @@ export function useAnalytics(): AnalyticsState & {
 
   function requestAnalytics(forceRefresh?: boolean) {
     var s = getAnalyticsStore().state;
-    setAnalyticsLoading(true);
+    clearAnalyticsForRequest();
     sendRef.current({
       type: "analytics:request",
       requestId: crypto.randomUUID(),
@@ -38,9 +40,13 @@ export function useAnalytics(): AnalyticsState & {
   }
 
   useEffect(function () {
-    function handleData(msg: ServerMessage) {
-      var m = msg as { type: string; data: any };
-      setAnalyticsData(m.data);
+    function handleSection(msg: ServerMessage) {
+      var m = msg as { type: string; section: AnalyticsSectionName; data: Partial<AnalyticsPayload> };
+      mergeAnalyticsSection(m.section, m.data);
+    }
+
+    function handleComplete() {
+      setAnalyticsLoading(false);
     }
 
     function handleError(msg: ServerMessage) {
@@ -48,11 +54,13 @@ export function useAnalytics(): AnalyticsState & {
       setAnalyticsError(m.message);
     }
 
-    subscribe("analytics:data", handleData);
+    subscribe("analytics:section", handleSection);
+    subscribe("analytics:complete", handleComplete);
     subscribe("analytics:error", handleError);
 
     return function () {
-      unsubscribe("analytics:data", handleData);
+      unsubscribe("analytics:section", handleSection);
+      unsubscribe("analytics:complete", handleComplete);
       unsubscribe("analytics:error", handleError);
     };
   }, [subscribe, unsubscribe]);
@@ -63,6 +71,7 @@ export function useAnalytics(): AnalyticsState & {
 
   return {
     data: state.data,
+    loadedSections: state.loadedSections,
     loading: state.loading,
     error: state.error,
     period: state.period,

package/src/client/hooks/usePushNotifications.ts

@@ -0,0 +1,92 @@
+import { useEffect, useState } from "react";
+
+function urlBase64ToUint8Array(base64String: string): Uint8Array {
+  var padding = "=".repeat((4 - (base64String.length % 4)) % 4);
+  var base64 = (base64String + padding).replace(/-/g, "+").replace(/_/g, "/");
+  var rawData = window.atob(base64);
+  var outputArray = new Uint8Array(rawData.length);
+  for (var i = 0; i < rawData.length; ++i) {
+    outputArray[i] = rawData.charCodeAt(i);
+  }
+  return outputArray;
+}
+
+export type PushStatus = "unsupported" | "denied" | "prompt" | "subscribed" | "unsubscribed";
+
+export function usePushNotifications() {
+  var [status, setStatus] = useState<PushStatus>("unsupported");
+
+  useEffect(function () {
+    if (!("serviceWorker" in navigator) || !("PushManager" in window)) {
+      setStatus("unsupported");
+      return;
+    }
+
+    if (Notification.permission === "denied") {
+      setStatus("denied");
+      return;
+    }
+
+    navigator.serviceWorker.getRegistration("/sw-push.js").then(function (reg) {
+      if (!reg) {
+        setStatus("unsubscribed");
+        return;
+      }
+      reg.pushManager.getSubscription().then(function (sub) {
+        setStatus(sub ? "subscribed" : "unsubscribed");
+      });
+    });
+  }, []);
+
+  async function subscribe(): Promise<boolean> {
+    if (!("serviceWorker" in navigator) || !("PushManager" in window)) return false;
+
+    try {
+      var permission = await Notification.requestPermission();
+      if (permission !== "granted") {
+        setStatus("denied");
+        return false;
+      }
+
+      var reg = await navigator.serviceWorker.register("/sw-push.js", { scope: "/" });
+      await navigator.serviceWorker.ready;
+
+      var response = await fetch("/api/vapid-public-key");
+      var data = await response.json() as { publicKey: string };
+
+      var subscription = await reg.pushManager.subscribe({
+        userVisibleOnly: true,
+        applicationServerKey: urlBase64ToUint8Array(data.publicKey) as BufferSource,
+      });
+
+      await fetch("/api/push-subscribe", {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify(subscription.toJSON()),
+      });
+
+      setStatus("subscribed");
+      return true;
+    } catch (err) {
+      console.error("Push subscription failed:", err);
+      return false;
+    }
+  }
+
+  async function unsubscribe(): Promise<void> {
+    try {
+      var reg = await navigator.serviceWorker.getRegistration("/sw-push.js");
+      if (reg) {
+        var sub = await reg.pushManager.getSubscription();
+        if (sub) {
+          await sub.unsubscribe();
+        }
+      }
+      setStatus("unsubscribed");
+    } catch (err) {
+      console.error("Push unsubscribe failed:", err);
+    }
+  }
+
+  return { status, subscribe, unsubscribe };
+}

package/{client/src → src/client}/hooks/useSession.ts

@@ -13,6 +13,7 @@ import type {
   ChatContextUsageMessage,
   ChatContextBreakdownMessage,
   ChatPromptSuggestionMessage,
+  ChatElicitationRequestMessage,
   SessionHistoryMessage,
   ServerMessage,
 } from "@lattice/shared";
@@ -49,12 +50,12 @@ import {
   removeQueuedMessage,
   updateQueuedMessage,
   clearMessageQueue,
-  setSessionBusy,
   addPromptQuestion,
   addTodoUpdate,
   setIsPlanMode,
   setBudgetStatus,
   setBudgetExceeded,
+  updateRateLimit,
 } from "../stores/session";
 import type { SessionState, BudgetStatus } from "../stores/session";
 
@@ -80,6 +81,7 @@ export interface UseSessionReturn extends SessionState {
   dismissBudgetExceeded: () => void;
   loadMoreHistory: () => void;
   historyHasMore: boolean;
+  historyLoadingFileSize: number | null;
 }
 
 export function useSession(): UseSessionReturn {
@@ -119,7 +121,12 @@ export function useSession(): UseSessionReturn {
     setFailedInput(null);
     setPromptSuggestion(null);
     setIsProcessing(true);
-    setSessionBusy(false);
+    addSessionMessage({
+      type: "user",
+      uuid: "optimistic-" + Date.now(),
+      text: text,
+      timestamp: Date.now(),
+    } as HistoryMessage);
     pinTab("chat-" + currentSessionId);
     sendRef.current(msg as ChatSendMessage);
   }
@@ -143,6 +150,16 @@ export function useSession(): UseSessionReturn {
       if (isStaleStream()) return;
       var m = msg as ChatUserMessage;
       setCurrentAssistantUuid(null);
+      var messages = getSessionStore().state.messages;
+      var last = messages.length > 0 ? messages[messages.length - 1] : null;
+      if (last && last.type === "user" && last.uuid && last.uuid.startsWith("optimistic-") && last.text === m.text) {
+        getSessionStore().setState(function (s) {
+          var updated = s.messages.slice();
+          updated[updated.length - 1] = { ...updated[updated.length - 1], uuid: m.uuid };
+          return { ...s, messages: updated };
+        });
+        return;
+      }
       addSessionMessage({
         type: "user",
         uuid: m.uuid,
@@ -174,13 +191,24 @@ export function useSession(): UseSessionReturn {
       if (isStaleStream()) return;
       var m = msg as ChatToolStartMessage;
       setCurrentAssistantUuid(null);
-      addSessionMessage({
-        type: "tool_start",
-        toolId: m.toolId,
-        name: m.name,
-        args: m.args,
-        timestamp: Date.now(),
-      } as HistoryMessage);
+      var existing = getSessionStore().state.messages.findLastIndex(function (msg) {
+        return msg.toolId === m.toolId && msg.type === "tool_start";
+      });
+      if (existing >= 0) {
+        getSessionStore().setState(function (s) {
+          var updated = s.messages.slice();
+          updated[existing] = { ...updated[existing], args: m.args };
+          return { ...s, messages: updated };
+        });
+      } else {
+        addSessionMessage({
+          type: "tool_start",
+          toolId: m.toolId,
+          name: m.name,
+          args: m.args,
+          timestamp: Date.now(),
+        } as HistoryMessage);
+      }
     }
 
     function handleToolResult(msg: ServerMessage) {
@@ -279,6 +307,22 @@ export function useSession(): UseSessionReturn {
       updatePermissionStatus(m.requestId, m.status);
     }
 
+    function handleElicitationRequest(msg: ServerMessage) {
+      var m = msg as ChatElicitationRequestMessage;
+      setCurrentAssistantUuid(null);
+      addSessionMessage({
+        type: "elicitation",
+        toolId: m.requestId,
+        elicitationMode: m.mode,
+        elicitationServerName: m.serverName,
+        elicitationMessage: m.message,
+        elicitationUrl: m.url,
+        elicitationSchema: m.requestedSchema,
+        elicitationStatus: "pending",
+        timestamp: Date.now(),
+      } as HistoryMessage);
+    }
+
     function handleHistoryPage(msg: ServerMessage) {
       var m = msg as { type: string; sessionId: string; messages: HistoryMessage[]; hasMore: boolean };
       var state = getSessionStore().state;
@@ -292,6 +336,12 @@ export function useSession(): UseSessionReturn {
       });
     }
 
+    function handleLoadingProgress(msg: ServerMessage) {
+      var m = msg as { type: string; sessionId: string; fileSize: number | null };
+      if (m.sessionId !== getSessionStore().state.activeSessionId) return;
+      getSessionStore().setState(function (s) { return { ...s, historyLoadingFileSize: m.fileSize }; });
+    }
+
     function handleHistory(msg: ServerMessage) {
       var m = msg as SessionHistoryMessage;
       if (m.sessionId && m.messages && m.messages.length === 0 && m.title) {
@@ -303,37 +353,53 @@ export function useSession(): UseSessionReturn {
       }
       setCurrentAssistantUuid(null);
       if (m.sessionId) {
+        if (m.sessionId !== getSessionStore().state.activeSessionId) {
+          if (m.title) updateSessionTabTitle(m.sessionId, m.title);
+          return;
+        }
         var projectSlug = m.projectSlug || getSessionStore().state.activeProjectSlug;
         setSidebarSessionId(m.sessionId);
         streamSessionId = m.sessionId;
-        if (m.busy) {
-          activeStreamGeneration = getStreamGeneration();
-        }
         if (m.title) {
           updateSessionTabTitle(m.sessionId, m.title);
         }
-        getSessionStore().setState(function (state) {
-          return {
-            ...state,
-            activeProjectSlug: projectSlug,
-            activeSessionId: m.sessionId,
-            activeSessionTitle: m.title ?? null,
-            messages: mergeToolResults(m.messages),
-            isProcessing: false,
-            currentStatus: null,
-            pendingPermissionCount: 0,
-            lastResponseCost: null,
-            lastResponseDuration: null,
-            lastReadIndex: null,
-            historyLoading: false,
-            historyHasMore: m.hasMore || false,
-            historyTotalMessages: m.totalMessages || m.messages.length,
-            wasInterrupted: m.interrupted || false,
-            isBusy: m.busy || false,
-            busyOwner: m.busyOwner ?? null,
-            isPlanMode: false,
-          };
-        });
+        var currentState = getSessionStore().state;
+        var alreadyCached = currentState.activeSessionId === m.sessionId
+          && !currentState.historyLoading
+          && currentState.messages.length > 0;
+
+        if (alreadyCached) {
+          getSessionStore().setState(function (state) {
+            return {
+              ...state,
+              activeSessionTitle: m.title ?? state.activeSessionTitle,
+              historyHasMore: m.hasMore || state.historyHasMore,
+              historyTotalMessages: m.totalMessages || state.historyTotalMessages,
+              wasInterrupted: m.interrupted || false,
+            };
+          });
+        } else {
+          getSessionStore().setState(function (state) {
+            return {
+              ...state,
+              activeProjectSlug: projectSlug,
+              activeSessionId: m.sessionId,
+              activeSessionTitle: m.title ?? null,
+              messages: mergeToolResults(m.messages),
+              isProcessing: false,
+              currentStatus: null,
+              pendingPermissionCount: 0,
+              lastResponseCost: null,
+              lastResponseDuration: null,
+              lastReadIndex: null,
+              historyLoading: false,
+              historyHasMore: m.hasMore || false,
+              historyTotalMessages: m.totalMessages || m.messages.length,
+              wasInterrupted: m.interrupted || false,
+              isPlanMode: false,
+            };
+          });
+        }
         var storedIndex = getLastReadIndex(m.sessionId);
         if (storedIndex >= 0 && storedIndex < m.messages.length) {
           setLastReadIndex(storedIndex);
@@ -360,17 +426,6 @@ export function useSession(): UseSessionReturn {
       setPromptSuggestion(m.suggestion);
     }
 
-    function handleSessionBusy(msg: ServerMessage) {
-      var m = msg as { type: string; sessionId: string; busy: boolean; busyOwner?: "cli" | "lattice" };
-      var sessionState = getSessionStore().state;
-      if (m.sessionId === sessionState.activeSessionId) {
-        if (m.busy && sessionState.isProcessing) {
-          return;
-        }
-        setSessionBusy(m.busy, m.busyOwner);
-      }
-    }
-
     function handlePromptRequest(msg: ServerMessage) {
       if (isStaleStream()) return;
       var m = msg as { type: string; requestId: string; questions: Array<{ question: string; header: string; options: Array<{ label: string; description: string; preview?: string }>; multiSelect: boolean }> };
@@ -397,12 +452,28 @@ export function useSession(): UseSessionReturn {
       setBudgetStatus({ dailySpend: m.dailySpend, dailyLimit: m.dailyLimit, enforcement: m.enforcement });
     }
 
+    function handleRateLimit(msg: ServerMessage) {
+      var m = msg as { type: string; status: "allowed" | "allowed_warning" | "rejected"; utilization?: number; resetsAt?: number; rateLimitType?: string; overageStatus?: string; overageResetsAt?: number; isUsingOverage?: boolean };
+      if (!m.rateLimitType) return;
+      updateRateLimit({
+        status: m.status,
+        utilization: m.utilization,
+        resetsAt: m.resetsAt,
+        rateLimitType: m.rateLimitType,
+        overageStatus: m.overageStatus,
+        overageResetsAt: m.overageResetsAt,
+        isUsingOverage: m.isUsingOverage,
+        updatedAt: Date.now(),
+      });
+    }
+
     function handleBudgetExceeded(msg: ServerMessage) {
       setBudgetExceeded(true);
       setIsProcessing(false);
       setCurrentStatus(null);
     }
 
+    subscribe("session:loading_progress", handleLoadingProgress);
     subscribe("chat:user_message", handleUserMessage);
     subscribe("chat:delta", handleDelta);
     subscribe("chat:tool_start", handleToolStart);
@@ -417,16 +488,18 @@ export function useSession(): UseSessionReturn {
     subscribe("session:history", handleHistory);
     subscribe("session:history_page_result", handleHistoryPage);
     subscribe("chat:prompt_suggestion", handlePromptSuggestion);
-    subscribe("session:busy", handleSessionBusy);
     subscribe("chat:prompt_request", handlePromptRequest);
     subscribe("chat:prompt_resolved", handlePromptResolved);
     subscribe("chat:todo_update", handleTodoUpdate);
     subscribe("chat:plan_mode", handlePlanMode);
     subscribe("budget:status", handleBudgetStatus);
     subscribe("budget:exceeded", handleBudgetExceeded);
+    subscribe("chat:elicitation_request", handleElicitationRequest);
+    subscribe("chat:rate_limit", handleRateLimit);
 
     return function () {
       subscriptionsActive--;
+      unsubscribe("session:loading_progress", handleLoadingProgress);
       unsubscribe("chat:user_message", handleUserMessage);
       unsubscribe("chat:delta", handleDelta);
       unsubscribe("chat:tool_start", handleToolStart);
@@ -441,13 +514,14 @@ export function useSession(): UseSessionReturn {
       unsubscribe("session:history", handleHistory);
       unsubscribe("session:history_page_result", handleHistoryPage);
       unsubscribe("chat:prompt_suggestion", handlePromptSuggestion);
-      unsubscribe("session:busy", handleSessionBusy);
       unsubscribe("chat:prompt_request", handlePromptRequest);
       unsubscribe("chat:prompt_resolved", handlePromptResolved);
       unsubscribe("chat:todo_update", handleTodoUpdate);
       unsubscribe("chat:plan_mode", handlePlanMode);
       unsubscribe("budget:status", handleBudgetStatus);
       unsubscribe("budget:exceeded", handleBudgetExceeded);
+      unsubscribe("chat:elicitation_request", handleElicitationRequest);
+      unsubscribe("chat:rate_limit", handleRateLimit);
     };
   }, [subscribe, unsubscribe]);
 
@@ -468,6 +542,7 @@ export function useSession(): UseSessionReturn {
     lastResponseDuration: state.lastResponseDuration,
     lastReadIndex: state.lastReadIndex,
     historyLoading: state.historyLoading,
+    historyLoadingFileSize: state.historyLoadingFileSize,
     historyHasMore: state.historyHasMore,
     historyTotalMessages: state.historyTotalMessages,
     loadMoreHistory: function () {
@@ -481,8 +556,6 @@ export function useSession(): UseSessionReturn {
     promptSuggestion: state.promptSuggestion,
     failedInput: state.failedInput,
     messageQueue: state.messageQueue,
-    isBusy: state.isBusy,
-    busyOwner: state.busyOwner,
     isPlanMode: state.isPlanMode,
     pendingPrefill: state.pendingPrefill,
     budgetStatus: state.budgetStatus,
@@ -503,5 +576,6 @@ export function useSession(): UseSessionReturn {
         lastSentText = null;
       }
     },
+    rateLimits: state.rateLimits,
   };
 }

package/{client/src → src/client}/hooks/useWebSocket.ts

@@ -22,9 +22,6 @@ export function useWebSocket(): WebSocketContextValue {
 }
 
 export function getWebSocketUrl(): string {
-  if (import.meta.env.DEV) {
-    return "ws://" + window.location.hostname + ":7654/ws";
-  }
   var protocol = window.location.protocol === "https:" ? "wss:" : "ws:";
   return protocol + "//" + window.location.host + "/ws";
 }