🤖 Make 1M context setting global instead of per-workspace (#120)

## Summary

Make the 1M context checkbox setting global instead of per-workspace.

## Changes

- Updated `use1MContext` hook to use global localStorage key
- Removed `workspaceId` parameter from hook signature
- Updated all call sites: Context1MCheckbox, CostsTab, ChatInput
- Removed `workspaceId` from ChatContext (no longer needed)
- **Bonus:** Added closed PR detection to `wait_pr_checks.sh` script

## Rationale

Users typically want 1M context either on or off everywhere.
Per-workspace state caused confusion when switching between workspaces.

## Testing

- ✅ make typecheck
- ✅ make test (273 tests passing)

_Generated with `cmux`_

Author: ammar-agent

scripts/wait_pr_checks.sh

@@ -30,6 +30,12 @@ while true; do
     exit 0
   fi
 
+  # Check if PR is closed without merging
+  if [ "$PR_STATE" = "CLOSED" ]; then
+    echo "❌ PR #$PR_NUMBER is closed (not merged)!"
+    exit 1
+  fi
+
   MERGEABLE=$(echo "$STATUS" | jq -r '.mergeable')
   MERGE_STATE=$(echo "$STATUS" | jq -r '.mergeStateStatus')
 

src/components/AIView.tsx

@@ -320,12 +320,7 @@ const AIViewInner: React.FC<AIViewProps> = ({
   }
 
   return (
-    <ChatProvider
-      messages={messages}
-      cmuxMessages={cmuxMessages}
-      model={currentModel}
-      workspaceId={workspaceId}
-    >
+    <ChatProvider messages={messages} cmuxMessages={cmuxMessages} model={currentModel}>
       <ViewContainer className={className}>
         <ChatArea>
           <ViewHeader>

src/components/ChatInput.tsx

@@ -306,7 +306,7 @@ export const ChatInput: React.FC<ChatInputProps> = ({
   const modelSelectorRef = useRef<ModelSelectorRef>(null);
   const [thinkingLevel] = useThinkingLevel();
   const [mode, setMode] = useMode();
-  const [use1M] = use1MContext(workspaceId);
+  const [use1M] = use1MContext();
   const { recentModels } = useModelLRU();
 
   const focusMessageInput = useCallback(() => {
@@ -763,7 +763,7 @@ export const ChatInput: React.FC<ChatInputProps> = ({
           </EditingIndicator>
         )}
         <ModeTogglesRow>
-          <ChatToggles workspaceId={workspaceId} modelString={preferredModel}>
+          <ChatToggles modelString={preferredModel}>
             <ModelDisplayWrapper>
               <ModelSelector
                 ref={modelSelectorRef}

src/components/ChatMetaSidebar/CostsTab.tsx

@@ -280,9 +280,9 @@ const VIEW_MODE_OPTIONS: Array<ToggleOption<ViewMode>> = [
 ];
 
 export const CostsTab: React.FC = () => {
-  const { stats, isCalculating, workspaceId } = useChatContext();
+  const { stats, isCalculating } = useChatContext();
   const [viewMode, setViewMode] = usePersistedState<ViewMode>("costsTab:viewMode", "last-request");
-  const [use1M] = use1MContext(workspaceId);
+  const [use1M] = use1MContext();
 
   // Only show loading if we don't have any stats yet
   if (isCalculating && !stats) {

src/components/ChatToggles.tsx

@@ -10,17 +10,16 @@ export const TogglesContainer = styled.div`
 `;
 
 interface ChatTogglesProps {
-  workspaceId: string;
   modelString: string;
   children: React.ReactNode;
 }
 
-export const ChatToggles: React.FC<ChatTogglesProps> = ({ workspaceId, modelString, children }) => {
+export const ChatToggles: React.FC<ChatTogglesProps> = ({ modelString, children }) => {
   return (
     <TogglesContainer>
       {children}
       <ThinkingSliderComponent />
-      <Context1MCheckbox workspaceId={workspaceId} modelString={modelString} />
+      <Context1MCheckbox modelString={modelString} />
     </TogglesContainer>
   );
 };

src/components/Context1MCheckbox.tsx

@@ -59,15 +59,11 @@ const Checkbox = styled.input`
 `;
 
 interface Context1MCheckboxProps {
-  workspaceId: string;
   modelString: string;
 }
 
-export const Context1MCheckbox: React.FC<Context1MCheckboxProps> = ({
-  workspaceId,
-  modelString,
-}) => {
-  const [use1M, setUse1M] = use1MContext(workspaceId);
+export const Context1MCheckbox: React.FC<Context1MCheckboxProps> = ({ modelString }) => {
+  const [use1M, setUse1M] = use1MContext();
   const isSupported = supports1MContext(modelString);
 
   if (!isSupported) {

src/contexts/ChatContext.tsx

@@ -8,7 +8,6 @@ interface ChatContextType {
   messages: DisplayedMessage[];
   stats: ChatStats | null;
   isCalculating: boolean;
-  workspaceId: string;
 }
 
 const ChatContext = createContext<ChatContextType | undefined>(undefined);
@@ -18,15 +17,13 @@ interface ChatProviderProps {
   messages: DisplayedMessage[];
   cmuxMessages: CmuxMessage[];
   model: string;
-  workspaceId: string;
 }
 
 export const ChatProvider: React.FC<ChatProviderProps> = ({
   children,
   messages,
   cmuxMessages,
   model,
-  workspaceId,
 }) => {
   const [stats, setStats] = useState<ChatStats | null>(null);
   const [isCalculating, setIsCalculating] = useState(false);
@@ -91,7 +88,7 @@ export const ChatProvider: React.FC<ChatProviderProps> = ({
   }, [cmuxMessages, model]);
 
   return (
-    <ChatContext.Provider value={{ messages, stats, isCalculating, workspaceId }}>
+    <ChatContext.Provider value={{ messages, stats, isCalculating }}>
       {children}
     </ChatContext.Provider>
   );

src/hooks/use1MContext.ts

@@ -2,11 +2,10 @@ import { usePersistedState } from "@/hooks/usePersistedState";
 
 /**
  * Custom hook for 1M context state.
- * Persists state per workspace in localStorage.
+ * Persists state globally in localStorage (applies to all workspaces).
  *
- * @param workspaceId - Unique identifier for the workspace
  * @returns [use1MContext, setUse1MContext] tuple
  */
-export function use1MContext(workspaceId: string): [boolean, (value: boolean) => void] {
-  return usePersistedState<boolean>(`use1MContext:${workspaceId}`, false);
+export function use1MContext(): [boolean, (value: boolean) => void] {
+  return usePersistedState<boolean>("use1MContext", false);
 }