Add Claude context window selection support

apps/server/src/git/Layers/ClaudeTextGeneration.ts

@@ -11,6 +11,7 @@ import { Effect, Layer, Option, Schema, Stream } from "effect";
 import { ChildProcess, ChildProcessSpawner } from "effect/unstable/process";
 
 import { ClaudeModelSelection } from "@t3tools/contracts";
+import { resolveApiModelId } from "@t3tools/shared/model";
 import { sanitizeBranchFragment, sanitizeFeatureBranchName } from "@t3tools/shared/git";
 
 import { TextGenerationError } from "../Errors.ts";
@@ -103,7 +104,7 @@ const makeClaudeTextGeneration = Effect.gen(function* () {
             "--json-schema",
             jsonSchemaStr,
             "--model",
-            modelSelection.model,
+            resolveApiModelId(modelSelection),
             ...(normalizedOptions?.effort ? ["--effort", normalizedOptions.effort] : []),
             ...(Object.keys(settings).length > 0 ? ["--settings", JSON.stringify(settings)] : []),
             "--dangerously-skip-permissions",

apps/server/src/provider/Layers/ClaudeAdapter.ts

@@ -40,7 +40,12 @@ import {
   type UserInputQuestion,
   ClaudeCodeEffort,
 } from "@t3tools/contracts";
-import { hasEffortLevel, applyClaudePromptEffortPrefix, trimOrNull } from "@t3tools/shared/model";
+import {
+  hasEffortLevel,
+  applyClaudePromptEffortPrefix,
+  resolveApiModelId,
+  trimOrNull,
+} from "@t3tools/shared/model";
 import {
   Cause,
   DateTime,
@@ -2727,6 +2732,7 @@ function makeClaudeAdapter(options?: ClaudeAdapterLiveOptions) {
         const claudeBinaryPath = claudeSettings.binaryPath;
         const modelSelection =
           input.modelSelection?.provider === "claudeAgent" ? input.modelSelection : undefined;
+        const apiModelId = modelSelection ? resolveApiModelId(modelSelection) : undefined;
         const requestedEffort = trimOrNull(modelSelection?.options?.effort ?? null);
         const caps = getClaudeModelCapabilities(modelSelection?.model);
         const effort =
@@ -2746,7 +2752,7 @@ function makeClaudeAdapter(options?: ClaudeAdapterLiveOptions) {
 
         const queryOptions: ClaudeQueryOptions = {
           ...(input.cwd ? { cwd: input.cwd } : {}),
-          ...(modelSelection?.model ? { model: modelSelection.model } : {}),
+          ...(apiModelId ? { model: apiModelId } : {}),
           pathToClaudeCodeExecutable: claudeBinaryPath,
           settingSources: [...CLAUDE_SETTING_SOURCES],
           ...(effectiveEffort ? { effort: effectiveEffort } : {}),
@@ -2840,7 +2846,7 @@ function makeClaudeAdapter(options?: ClaudeAdapterLiveOptions) {
           threadId,
           payload: {
             config: {
-              ...(modelSelection?.model ? { model: modelSelection.model } : {}),
+              ...(apiModelId ? { model: apiModelId } : {}),
               ...(input.cwd ? { cwd: input.cwd } : {}),
               ...(effectiveEffort ? { effort: effectiveEffort } : {}),
               ...(permissionMode ? { permissionMode } : {}),
@@ -2893,8 +2899,9 @@ function makeClaudeAdapter(options?: ClaudeAdapterLiveOptions) {
         }
 
         if (modelSelection?.model) {
+          const apiModelId = resolveApiModelId(modelSelection);
           yield* Effect.tryPromise({
-            try: () => context.query.setModel(modelSelection.model),
+            try: () => context.query.setModel(apiModelId),
             catch: (cause) => toRequestError(input.threadId, "turn/setModel", cause),
           });
         }

apps/server/src/provider/Layers/ClaudeProvider.ts

@@ -42,6 +42,10 @@ const BUILT_IN_MODELS: ReadonlyArray<ServerProviderModel> = [
       ],
       supportsFastMode: true,
       supportsThinkingToggle: false,
+      contextWindowOptions: [
+        { value: "200k", label: "200k", isDefault: true },
+        { value: "1m", label: "1M" },
+      ],
       promptInjectedEffortLevels: ["ultrathink"],
     } satisfies ModelCapabilities,
   },
@@ -58,6 +62,10 @@ const BUILT_IN_MODELS: ReadonlyArray<ServerProviderModel> = [
       ],
       supportsFastMode: false,
       supportsThinkingToggle: false,
+      contextWindowOptions: [
+        { value: "200k", label: "200k", isDefault: true },
+        { value: "1m", label: "1M" },
+      ],
       promptInjectedEffortLevels: ["ultrathink"],
     } satisfies ModelCapabilities,
   },
@@ -69,6 +77,7 @@ const BUILT_IN_MODELS: ReadonlyArray<ServerProviderModel> = [
       reasoningEffortLevels: [],
       supportsFastMode: false,
       supportsThinkingToggle: true,
+      contextWindowOptions: [],
       promptInjectedEffortLevels: [],
     } satisfies ModelCapabilities,
   },
@@ -81,6 +90,7 @@ export function getClaudeModelCapabilities(model: string | null | undefined): Mo
       reasoningEffortLevels: [],
       supportsFastMode: false,
       supportsThinkingToggle: false,
+      contextWindowOptions: [],
       promptInjectedEffortLevels: [],
     }
   );

apps/server/src/provider/Layers/CodexProvider.ts

@@ -48,6 +48,7 @@ const BUILT_IN_MODELS: ReadonlyArray<ServerProviderModel> = [
       ],
       supportsFastMode: true,
       supportsThinkingToggle: false,
+      contextWindowOptions: [],
       promptInjectedEffortLevels: [],
     },
   },
@@ -64,6 +65,7 @@ const BUILT_IN_MODELS: ReadonlyArray<ServerProviderModel> = [
       ],
       supportsFastMode: true,
       supportsThinkingToggle: false,
+      contextWindowOptions: [],
       promptInjectedEffortLevels: [],
     },
   },
@@ -80,6 +82,7 @@ const BUILT_IN_MODELS: ReadonlyArray<ServerProviderModel> = [
       ],
       supportsFastMode: true,
       supportsThinkingToggle: false,
+      contextWindowOptions: [],
       promptInjectedEffortLevels: [],
     },
   },
@@ -96,6 +99,7 @@ const BUILT_IN_MODELS: ReadonlyArray<ServerProviderModel> = [
       ],
       supportsFastMode: true,
       supportsThinkingToggle: false,
+      contextWindowOptions: [],
       promptInjectedEffortLevels: [],
     },
   },
@@ -112,6 +116,7 @@ const BUILT_IN_MODELS: ReadonlyArray<ServerProviderModel> = [
       ],
       supportsFastMode: true,
       supportsThinkingToggle: false,
+      contextWindowOptions: [],
       promptInjectedEffortLevels: [],
     },
   },
@@ -128,6 +133,7 @@ const BUILT_IN_MODELS: ReadonlyArray<ServerProviderModel> = [
       ],
       supportsFastMode: true,
       supportsThinkingToggle: false,
+      contextWindowOptions: [],
       promptInjectedEffortLevels: [],
     },
   },
@@ -140,6 +146,7 @@ export function getCodexModelCapabilities(model: string | null | undefined): Mod
       reasoningEffortLevels: [],
       supportsFastMode: false,
       supportsThinkingToggle: false,
+      contextWindowOptions: [],
       promptInjectedEffortLevels: [],
     }
   );

apps/web/src/components/chat/CompactComposerControlsMenu.browser.tsx

@@ -60,6 +60,7 @@ async function mountMenu(props?: { modelSelection?: ModelSelection; prompt?: str
               ],
               supportsFastMode: true,
               supportsThinkingToggle: false,
+              contextWindowOptions: [],
               promptInjectedEffortLevels: ["ultrathink"],
             },
           },
@@ -71,6 +72,7 @@ async function mountMenu(props?: { modelSelection?: ModelSelection; prompt?: str
               reasoningEffortLevels: [],
               supportsFastMode: false,
               supportsThinkingToggle: true,
+              contextWindowOptions: [],
               promptInjectedEffortLevels: [],
             },
           },
@@ -87,6 +89,7 @@ async function mountMenu(props?: { modelSelection?: ModelSelection; prompt?: str
               ],
               supportsFastMode: false,
               supportsThinkingToggle: false,
+              contextWindowOptions: [],
               promptInjectedEffortLevels: ["ultrathink"],
             },
           },
@@ -103,6 +106,7 @@ async function mountMenu(props?: { modelSelection?: ModelSelection; prompt?: str
               ],
               supportsFastMode: true,
               supportsThinkingToggle: false,
+              contextWindowOptions: [],
               promptInjectedEffortLevels: [],
             },
           },

apps/web/src/components/chat/ProviderModelPicker.browser.tsx

@@ -33,6 +33,7 @@ const TEST_PROVIDERS: ReadonlyArray<ServerProvider> = [
           reasoningEffortLevels: [effort("low"), effort("medium", true), effort("high")],
           supportsFastMode: true,
           supportsThinkingToggle: false,
+          contextWindowOptions: [],
           promptInjectedEffortLevels: [],
         },
       },
@@ -44,6 +45,7 @@ const TEST_PROVIDERS: ReadonlyArray<ServerProvider> = [
           reasoningEffortLevels: [effort("low"), effort("medium", true), effort("high")],
           supportsFastMode: true,
           supportsThinkingToggle: false,
+          contextWindowOptions: [],
           promptInjectedEffortLevels: [],
         },
       },
@@ -71,6 +73,7 @@ const TEST_PROVIDERS: ReadonlyArray<ServerProvider> = [
           ],
           supportsFastMode: false,
           supportsThinkingToggle: true,
+          contextWindowOptions: [],
           promptInjectedEffortLevels: [],
         },
       },
@@ -87,6 +90,7 @@ const TEST_PROVIDERS: ReadonlyArray<ServerProvider> = [
           ],
           supportsFastMode: false,
           supportsThinkingToggle: true,
+          contextWindowOptions: [],
           promptInjectedEffortLevels: [],
         },
       },
@@ -98,6 +102,7 @@ const TEST_PROVIDERS: ReadonlyArray<ServerProvider> = [
           reasoningEffortLevels: [effort("low"), effort("medium", true), effort("high")],
           supportsFastMode: false,
           supportsThinkingToggle: true,
+          contextWindowOptions: [],
           promptInjectedEffortLevels: [],
         },
       },

apps/web/src/components/chat/TraitsPicker.browser.tsx

@@ -49,6 +49,7 @@ const TEST_PROVIDERS: ReadonlyArray<ServerProvider> = [
           ],
           supportsFastMode: true,
           supportsThinkingToggle: false,
+          contextWindowOptions: [],
           promptInjectedEffortLevels: [],
         },
       },
@@ -77,6 +78,7 @@ const TEST_PROVIDERS: ReadonlyArray<ServerProvider> = [
           ],
           supportsFastMode: true,
           supportsThinkingToggle: false,
+          contextWindowOptions: [],
           promptInjectedEffortLevels: ["ultrathink"],
         },
       },
@@ -93,6 +95,7 @@ const TEST_PROVIDERS: ReadonlyArray<ServerProvider> = [
           ],
           supportsFastMode: false,
           supportsThinkingToggle: false,
+          contextWindowOptions: [],
           promptInjectedEffortLevels: ["ultrathink"],
         },
       },
@@ -104,6 +107,7 @@ const TEST_PROVIDERS: ReadonlyArray<ServerProvider> = [
           reasoningEffortLevels: [],
           supportsFastMode: false,
           supportsThinkingToggle: true,
+          contextWindowOptions: [],
           promptInjectedEffortLevels: [],
         },
       },

apps/web/src/components/chat/TraitsPicker.tsx

@@ -11,7 +11,9 @@ import {
   isClaudeUltrathinkPrompt,
   trimOrNull,
   getDefaultEffort,
+  getDefaultContextWindow,
   hasEffortLevel,
+  hasContextWindowOption,
 } from "@t3tools/shared/model";
 import { memo, useCallback, useState } from "react";
 import type { VariantProps } from "class-variance-authority";
@@ -53,6 +55,16 @@ function getRawEffort(
   return trimOrNull((modelOptions as ClaudeModelOptions | undefined)?.effort);
 }
 
+function getRawContextWindow(
+  provider: ProviderKind,
+  modelOptions: ProviderOptions | null | undefined,
+): string | null {
+  if (provider === "claudeAgent") {
+    return trimOrNull((modelOptions as ClaudeModelOptions | undefined)?.contextWindow);
+  }
+  return null;
+}
+
 function buildNextOptions(
   provider: ProviderKind,
   modelOptions: ProviderOptions | null | undefined,
@@ -104,6 +116,15 @@ function getSelectedTraits(
     caps.supportsFastMode &&
     (modelOptions as { fastMode?: boolean } | undefined)?.fastMode === true;
 
+  // Context window
+  const contextWindowOptions = caps.contextWindowOptions;
+  const rawContextWindow = getRawContextWindow(provider, modelOptions);
+  const defaultContextWindow = getDefaultContextWindow(caps);
+  const contextWindow =
+    rawContextWindow && hasContextWindowOption(caps, rawContextWindow)
+      ? rawContextWindow
+      : defaultContextWindow;
+
   // Prompt-controlled effort (e.g. ultrathink in prompt text)
   const ultrathinkPromptControlled =
     allowPromptInjectedEffort &&
@@ -116,6 +137,9 @@ function getSelectedTraits(
     effortLevels,
     thinkingEnabled,
     fastModeEnabled,
+    contextWindowOptions,
+    contextWindow,
+    defaultContextWindow,
     ultrathinkPromptControlled,
   };
 }
@@ -159,6 +183,9 @@ export const TraitsMenuContent = memo(function TraitsMenuContentImpl({
     effortLevels,
     thinkingEnabled,
     fastModeEnabled,
+    contextWindowOptions,
+    contextWindow,
+    defaultContextWindow,
     ultrathinkPromptControlled,
   } = getSelectedTraits(provider, models, model, prompt, modelOptions, allowPromptInjectedEffort);
   const defaultEffort = getDefaultEffort(caps);
@@ -258,6 +285,33 @@ export const TraitsMenuContent = memo(function TraitsMenuContentImpl({
           </MenuGroup>
         </>
       ) : null}
+      {contextWindowOptions.length > 1 ? (
+        <>
+          <MenuDivider />
+          <MenuGroup>
+            <div className="px-2 py-1.5 font-medium text-muted-foreground text-xs">
+              Context Window
+            </div>
+            <MenuRadioGroup
+              value={contextWindow ?? defaultContextWindow ?? ""}
+              onValueChange={(value) => {
+                updateModelOptions(
+                  buildNextOptions(provider, modelOptions, {
+                    contextWindow: value === defaultContextWindow ? undefined : value,
+                  }),
+                );
+              }}
+            >
+              {contextWindowOptions.map((option) => (
+                <MenuRadioItem key={option.value} value={option.value}>
+                  {option.label}
+                  {option.value === defaultContextWindow ? " (default)" : ""}
+                </MenuRadioItem>
+              ))}
+            </MenuRadioGroup>
+          </MenuGroup>
+        </>
+      ) : null}
     </>
   );
 });
@@ -281,12 +335,19 @@ export const TraitsPicker = memo(function TraitsPicker({
     effortLevels,
     thinkingEnabled,
     fastModeEnabled,
+    contextWindowOptions,
+    contextWindow,
+    defaultContextWindow,
     ultrathinkPromptControlled,
   } = getSelectedTraits(provider, models, model, prompt, modelOptions, allowPromptInjectedEffort);
 
   const effortLabel = effort
     ? (effortLevels.find((l) => l.value === effort)?.label ?? effort)
     : null;
+  const contextWindowLabel =
+    contextWindowOptions.length > 1 && contextWindow !== defaultContextWindow
+      ? (contextWindowOptions.find((o) => o.value === contextWindow)?.label ?? null)
+      : null;
   const triggerLabel = [
     ultrathinkPromptControlled
       ? "Ultrathink"
@@ -296,6 +357,7 @@ export const TraitsPicker = memo(function TraitsPicker({
           ? null
           : `Thinking ${thinkingEnabled ? "On" : "Off"}`,
     ...(caps.supportsFastMode && fastModeEnabled ? ["Fast"] : []),
+    ...(contextWindowLabel ? [contextWindowLabel] : []),
   ]
     .filter(Boolean)
     .join(" · ");