fix(ui): use model input limit for avail tokens

Upgrade @opencode-ai/sdk to 1.2.6 and prefer v2 model limit.input when present for the session AVAIL chip; otherwise keep the existing context-window-based estimate.

packages/ui/src/stores/message-v2/session-info.ts

@@ -63,9 +63,14 @@ export function updateSessionInfo(instanceId: string, sessionId: string): void {
     resolveSelectedModel(instanceProviders, latestProviderId, latestModelId)
 
   let modelOutputLimit = DEFAULT_MODEL_OUTPUT_LIMIT
+  let modelInputLimit: number | null = null
 
   if (selectedModel) {
     contextWindow = selectedModel.limit?.context ?? 0
+    const inputLimit = selectedModel.limit?.input
+    if (typeof inputLimit === "number" && inputLimit > 0) {
+      modelInputLimit = inputLimit
+    }
     const outputLimit = selectedModel.limit?.output
     if (typeof outputLimit === "number" && outputLimit > 0) {
       modelOutputLimit = Math.min(outputLimit, DEFAULT_MODEL_OUTPUT_LIMIT)
@@ -107,7 +112,13 @@ export function updateSessionInfo(instanceId: string, sessionId: string): void {
 
   const outputBudget = Math.min(modelOutputLimit, DEFAULT_MODEL_OUTPUT_LIMIT)
 
-  if (!contextAvailableFromPrevious) {
+  if (modelInputLimit !== null) {
+    // Prefer explicit input limits when provided by the API.
+    // This is used by the UI "Avail" chip.
+    contextAvailableTokens = modelInputLimit
+  }
+
+  if (!contextAvailableFromPrevious && contextAvailableTokens === null) {
     if (contextWindow > 0) {
       if (latestHasContextUsage && actualUsageTokens > 0) {
         contextAvailableTokens = Math.max(contextWindow - (actualUsageTokens + outputBudget), 0)

packages/ui/src/stores/session-api.ts

@@ -291,12 +291,13 @@ async function createSession(instanceId: string, agent?: string): Promise<Sessio
     const initialProvider = instanceProviders.find((p) => p.id === session.model.providerId)
     const initialModel = initialProvider?.models.find((m) => m.id === session.model.modelId)
     const initialContextWindow = initialModel?.limit?.context ?? 0
+    const initialInputLimit = initialModel?.limit?.input ?? 0
     const initialSubscriptionModel = initialModel?.cost?.input === 0 && initialModel?.cost?.output === 0
     const initialOutputLimit =
       initialModel?.limit?.output && initialModel.limit.output > 0
         ? initialModel.limit.output
         : DEFAULT_MODEL_OUTPUT_LIMIT
-    const initialContextAvailable = initialContextWindow > 0 ? initialContextWindow : null
+    const initialContextAvailable = initialInputLimit > 0 ? initialInputLimit : initialContextWindow > 0 ? initialContextWindow : null
 
     setSessionInfoByInstance((prev) => {
       const next = new Map(prev)
@@ -398,10 +399,11 @@ async function forkSession(
   const forkProvider = instanceProviders.find((p) => p.id === forkedSession.model.providerId)
   const forkModel = forkProvider?.models.find((m) => m.id === forkedSession.model.modelId)
   const forkContextWindow = forkModel?.limit?.context ?? 0
+  const forkInputLimit = forkModel?.limit?.input ?? 0
   const forkSubscriptionModel = forkModel?.cost?.input === 0 && forkModel?.cost?.output === 0
   const forkOutputLimit =
     forkModel?.limit?.output && forkModel.limit.output > 0 ? forkModel.limit.output : DEFAULT_MODEL_OUTPUT_LIMIT
-  const forkContextAvailable = forkContextWindow > 0 ? forkContextWindow : null
+  const forkContextAvailable = forkInputLimit > 0 ? forkInputLimit : forkContextWindow > 0 ? forkContextWindow : null
 
   setSessionInfoByInstance((prev) => {
     const next = new Map(prev)

packages/ui/src/types/session.ts

@@ -90,6 +90,7 @@ export interface Model {
   variantKeys?: string[]
   limit?: {
     context?: number
+    input?: number
     output?: number
   }
   cost?: {