@kmugalkhod/lightcode 0.2.0 → 0.3.0

package/cli.js

@@ -124,7 +124,7 @@ var init_logger = __esm(() => {
 });
 
 // packages/shared/src/version.ts
-var lightcodeVersion = "0.2.0";
+var lightcodeVersion = "0.3.0";
 
 // packages/shared/src/index.ts
 var productName = "Lightcode";
@@ -4173,6 +4173,43 @@ var init_build_provider_view = __esm(() => {
   init_tier1_prune();
 });
 
+// packages/ai/src/context/normalize-provider-messages.ts
+import { isToolUIPart } from "ai";
+function hasDanglingToolParts(messages) {
+  return messages.some((message) => message.parts.some((part) => isToolUIPart(part) && !terminalToolStates.has(part.state)));
+}
+function resolveDanglingToolParts(messages, errorText = interruptedToolErrorText) {
+  return messages.map((message) => {
+    if (message.role !== "assistant") {
+      return message;
+    }
+    let changed = false;
+    const parts = message.parts.map((part) => {
+      if (!isToolUIPart(part) || terminalToolStates.has(part.state)) {
+        return part;
+      }
+      changed = true;
+      return {
+        ...part,
+        state: "output-error",
+        errorText
+      };
+    });
+    return changed ? { ...message, parts } : message;
+  });
+}
+function normalizeProviderMessages(messages) {
+  return resolveDanglingToolParts(messages);
+}
+var terminalToolStates, interruptedToolErrorText = "Tool call was interrupted by a connection retry. Re-issue the tool call if it is still needed.";
+var init_normalize_provider_messages = __esm(() => {
+  terminalToolStates = new Set([
+    "output-available",
+    "output-error",
+    "output-denied"
+  ]);
+});
+
 // packages/ai/src/context/index.ts
 var init_context = __esm(() => {
   init_build_provider_view();
@@ -4181,6 +4218,7 @@ var init_context = __esm(() => {
   init_estimate();
   init_heuristic_summary();
   init_message_parts();
+  init_normalize_provider_messages();
   init_tier1_prune();
 });
 
@@ -4279,6 +4317,12 @@ function readEnvConfig(env) {
     summaryMaxChars: parseNumberEnv(env.LIGHTCODE_CONTEXT_SUMMARY_MAX_CHARS, "LIGHTCODE_CONTEXT_SUMMARY_MAX_CHARS")
   };
   const compactContextConfig = Object.fromEntries(Object.entries(contextConfig).filter(([, value]) => value !== undefined));
+  const headroomConfig = {
+    enabled: parseBooleanEnv(env.LIGHTCODE_HEADROOM_ENABLED, "LIGHTCODE_HEADROOM_ENABLED"),
+    proxyUrl: env.LIGHTCODE_HEADROOM_PROXY_URL?.trim() || undefined,
+    failOpen: parseBooleanEnv(env.LIGHTCODE_HEADROOM_FAIL_OPEN, "LIGHTCODE_HEADROOM_FAIL_OPEN")
+  };
+  const compactHeadroomConfig = Object.fromEntries(Object.entries(headroomConfig).filter(([, value]) => value !== undefined));
   const rawConfig = {
     provider: env.LIGHTCODE_PROVIDER,
     model: env.LIGHTCODE_CHAT_MODEL,
@@ -4288,7 +4332,8 @@ function readEnvConfig(env) {
     maxOutputTokens: parseNumberEnv(env.LIGHTCODE_MAX_OUTPUT_TOKENS, "LIGHTCODE_MAX_OUTPUT_TOKENS"),
     maxSteps: parseNumberEnv(env.LIGHTCODE_MAX_STEPS, "LIGHTCODE_MAX_STEPS"),
     maxRetries: parseNumberEnv(env.LIGHTCODE_MAX_RETRIES, "LIGHTCODE_MAX_RETRIES"),
-    context: Object.keys(compactContextConfig).length > 0 ? compactContextConfig : undefined
+    context: Object.keys(compactContextConfig).length > 0 ? compactContextConfig : undefined,
+    headroom: Object.keys(compactHeadroomConfig).length > 0 ? compactHeadroomConfig : undefined
   };
   const compactConfig = Object.fromEntries(Object.entries(rawConfig).filter(([, value]) => value !== undefined));
   const parsedConfig = lightcodeConfigSchema.safeParse(compactConfig);
@@ -4312,6 +4357,10 @@ function mergeConfig(...configs) {
       ...mergedConfig.autoContinue,
       ...config2.autoContinue
     } : mergedConfig.autoContinue,
+    headroom: config2.headroom ? {
+      ...mergedConfig.headroom,
+      ...config2.headroom
+    } : mergedConfig.headroom,
     allowedTools: config2.allowedTools ?? mergedConfig.allowedTools
   }), {});
 }
@@ -4352,6 +4401,10 @@ function loadLightcodeConfig({
     autoContinue: {
       ...defaultAutoContinueConfig,
       ...mergedConfig.autoContinue
+    },
+    headroom: {
+      ...defaultHeadroomConfig,
+      ...mergedConfig.headroom
     }
   };
   return lightcodeConfigLoadResultSchema.parse({
@@ -4359,7 +4412,7 @@ function loadLightcodeConfig({
     loadedFiles: [userConfig.file, projectConfig.file]
   });
 }
-var lightcodeProviderSchema, autoContinueConfigSchema, resolvedAutoContinueConfigSchema, defaultAutoContinueConfig, lightcodeConfigSchema, lightcodeConfigDefaults, loadedConfigFileSchema, lightcodeResolvedConfigSchema, lightcodeConfigLoadResultSchema, lightcodeConfigStatusSchema, LightcodeConfigError;
+var lightcodeProviderSchema, autoContinueConfigSchema, resolvedAutoContinueConfigSchema, defaultAutoContinueConfig, headroomConfigSchema, resolvedHeadroomConfigSchema, defaultHeadroomConfig, lightcodeConfigSchema, lightcodeConfigDefaults, loadedConfigFileSchema, lightcodeResolvedConfigSchema, lightcodeConfigLoadResultSchema, lightcodeConfigStatusSchema, LightcodeConfigError;
 var init_lightcode_config = __esm(() => {
   init_coding_agent_modes();
   init_coding_agent_modes();
@@ -4391,6 +4444,24 @@ var init_lightcode_config = __esm(() => {
     maxErrorRetries: 5,
     stallTimeoutSeconds: 180
   };
+  headroomConfigSchema = z28.object({
+    enabled: z28.boolean().optional(),
+    proxyUrl: z28.string().min(1).max(4096).optional(),
+    providers: z28.array(lightcodeProviderSchema).optional(),
+    failOpen: z28.boolean().optional()
+  }).strict();
+  resolvedHeadroomConfigSchema = z28.object({
+    enabled: z28.boolean(),
+    proxyUrl: z28.string().min(1).max(4096),
+    providers: z28.array(lightcodeProviderSchema).nullable(),
+    failOpen: z28.boolean()
+  });
+  defaultHeadroomConfig = {
+    enabled: false,
+    proxyUrl: "http://127.0.0.1:8787",
+    providers: null,
+    failOpen: true
+  };
   lightcodeConfigSchema = z28.object({
     provider: lightcodeProviderSchema.optional(),
     model: z28.string().min(1).max(240).optional(),
@@ -4402,10 +4473,11 @@ var init_lightcode_config = __esm(() => {
     sandbox: sandboxConfigSchema.optional(),
     mcp: mcpConfigSchema.optional(),
     context: contextOptimizerConfigSchema.optional(),
-    maxOutputTokens: z28.number().int().min(1).max(200000).optional(),
+    maxOutputTokens: z28.number().int().min(1).max(1e6).optional(),
     maxSteps: z28.number().int().min(1).max(100).optional(),
     maxRetries: z28.number().int().min(0).max(10).optional(),
-    autoContinue: autoContinueConfigSchema.optional()
+    autoContinue: autoContinueConfigSchema.optional(),
+    headroom: headroomConfigSchema.optional()
   }).strict();
   lightcodeConfigDefaults = {
     provider: "anthropic",
@@ -4414,7 +4486,8 @@ var init_lightcode_config = __esm(() => {
     maxOutputTokens: 32768,
     maxSteps: 30,
     maxRetries: 5,
-    autoContinue: defaultAutoContinueConfig
+    autoContinue: defaultAutoContinueConfig,
+    headroom: defaultHeadroomConfig
   };
   loadedConfigFileSchema = z28.object({
     scope: z28.enum(["user", "project"]),
@@ -4426,10 +4499,11 @@ var init_lightcode_config = __esm(() => {
     provider: lightcodeProviderSchema,
     defaultMode: codingAgentModeSchema,
     context: resolvedContextOptimizerConfigSchema,
-    maxOutputTokens: z28.number().int().min(1).max(200000),
+    maxOutputTokens: z28.number().int().min(1).max(1e6),
     maxSteps: z28.number().int().min(1).max(100),
     maxRetries: z28.number().int().min(0).max(10),
-    autoContinue: resolvedAutoContinueConfigSchema
+    autoContinue: resolvedAutoContinueConfigSchema,
+    headroom: resolvedHeadroomConfigSchema.optional()
   });
   lightcodeConfigLoadResultSchema = z28.object({
     config: lightcodeResolvedConfigSchema,
@@ -4443,10 +4517,15 @@ var init_lightcode_config = __esm(() => {
     baseUrl: z28.string().nullable(),
     defaultMode: codingAgentModeSchema,
     permissionMode: permissionModeSchema.nullable(),
-    maxOutputTokens: z28.number().int().min(1).max(200000),
+    maxOutputTokens: z28.number().int().min(1).max(1e6),
     maxSteps: z28.number().int().min(1).max(100),
     autoContinue: resolvedAutoContinueConfigSchema,
     contextWindow: z28.number().int().positive(),
+    headroom: z28.object({
+      enabled: z28.boolean(),
+      proxyUrl: z28.string().nullable(),
+      routed: z28.boolean()
+    }),
     missingCredentialHints: z28.array(z28.string())
   });
   LightcodeConfigError = class LightcodeConfigError extends Error {
@@ -4705,6 +4784,21 @@ var init_runtime = __esm(() => {
   init_schema13();
 });
 
+// packages/ai/src/providers/clamp-output-tokens.ts
+function clampMaxOutputTokens(configured, modelMaxCompletionTokens) {
+  if (typeof modelMaxCompletionTokens === "number" && Number.isFinite(modelMaxCompletionTokens) && modelMaxCompletionTokens > 0) {
+    return Math.min(configured, modelMaxCompletionTokens);
+  }
+  return configured;
+}
+function resolveMaxOutputTokens(configured, defaultValue, modelMaxCompletionTokens) {
+  const hasModelCap = typeof modelMaxCompletionTokens === "number" && Number.isFinite(modelMaxCompletionTokens) && modelMaxCompletionTokens > 0;
+  if (configured === defaultValue) {
+    return hasModelCap ? modelMaxCompletionTokens : configured;
+  }
+  return clampMaxOutputTokens(configured, modelMaxCompletionTokens);
+}
+
 // packages/ai/src/providers/model-metadata.ts
 function getModelContextWindow(provider, modelId) {
   if (provider === "anthropic") {
@@ -7515,9 +7609,12 @@ __export(exports_runtime, {
   sandboxConfigSchema: () => sandboxConfigSchema,
   safeStringify: () => safeStringify,
   runPluginHooks: () => runPluginHooks,
+  resolvedHeadroomConfigSchema: () => resolvedHeadroomConfigSchema,
   resolvedContextOptimizerConfigSchema: () => resolvedContextOptimizerConfigSchema,
   resolvedAutoContinueConfigSchema: () => resolvedAutoContinueConfigSchema,
   resolveWithinWorkspace: () => resolveWithinWorkspace,
+  resolveMaxOutputTokens: () => resolveMaxOutputTokens,
+  resolveDanglingToolParts: () => resolveDanglingToolParts,
   resolveContextWindowTokens: () => resolveContextWindowTokens,
   resolveCodingPermissionMode: () => resolveCodingPermissionMode,
   requestUserInputToolProviderInputSchema: () => requestUserInputToolProviderInputSchema,
@@ -7547,6 +7644,7 @@ __export(exports_runtime, {
   parseCodingToolInput: () => parseCodingToolInput,
   parseChatStreamError: () => parseChatStreamError,
   normalizeSandboxConfig: () => normalizeSandboxConfig,
+  normalizeProviderMessages: () => normalizeProviderMessages,
   normalizeContextOptimizerConfig: () => normalizeContextOptimizerConfig,
   messageUsageMetadataSchema: () => messageUsageMetadataSchema,
   mcpStdioServerConfigSchema: () => mcpStdioServerConfigSchema,
@@ -7577,7 +7675,10 @@ __export(exports_runtime, {
   isRecord: () => isRecord3,
   isPermissionDeniedError: () => isPermissionDeniedError,
   isElidedToolOutput: () => isElidedToolOutput,
+  interruptedToolErrorText: () => interruptedToolErrorText,
+  headroomConfigSchema: () => headroomConfigSchema,
   hasUnresolvedToolWork: () => hasUnresolvedToolWork,
+  hasDanglingToolParts: () => hasDanglingToolParts,
   getToolNameFromPart: () => getToolNameFromPart,
   getTextPartContent: () => getTextPartContent,
   getSandboxRuntimeStatus: () => getSandboxRuntimeStatus,
@@ -7628,6 +7729,7 @@ __export(exports_runtime, {
   describeChatStreamError: () => describeChatStreamError,
   defaultSubagentTaskMode: () => defaultSubagentTaskMode,
   defaultPermissionModeForCodingMode: () => defaultPermissionModeForCodingMode,
+  defaultHeadroomConfig: () => defaultHeadroomConfig,
   defaultContextOptimizerConfig: () => defaultContextOptimizerConfig,
   defaultCodingAgentSystemPrompt: () => defaultCodingAgentSystemPrompt,
   defaultCodingAgentMode: () => defaultCodingAgentMode,
@@ -7663,6 +7765,7 @@ __export(exports_runtime, {
   codingAgentModeSchema: () => codingAgentModeSchema,
   codingAgentModeOrder: () => codingAgentModeOrder,
   codingAgentCallOptionsSchema: () => codingAgentCallOptionsSchema,
+  clampMaxOutputTokens: () => clampMaxOutputTokens,
   chatInteractionUserPromptPayloadSchema: () => chatInteractionUserPromptPayloadSchema,
   chatInteractionUpsertRequestSchema: () => chatInteractionUpsertRequestSchema,
   chatInteractionToolCallIdSchema: () => chatInteractionToolCallIdSchema,
@@ -8026,7 +8129,7 @@ var init_auto_continue = __esm(() => {
 import { useChat } from "@ai-sdk/react";
 import {
   DefaultChatTransport,
-  isToolUIPart,
+  isToolUIPart as isToolUIPart2,
   lastAssistantMessageIsCompleteWithToolCalls,
   safeValidateUIMessages
 } from "ai";
@@ -8118,30 +8221,38 @@ function queueToolOutput(outputResult) {
     console.warn("Failed to add tool output.", outputError);
   });
 }
-function isIncompleteAssistantMessage(message) {
-  if (message.role !== "assistant") {
-    return false;
-  }
-  if (message.parts.length === 0) {
+function isIncompletePart(part) {
+  if ((part.type === "text" || part.type === "reasoning") && Reflect.get(part, "state") === "streaming") {
     return true;
   }
-  return message.parts.some((part) => {
-    if ((part.type === "text" || part.type === "reasoning") && Reflect.get(part, "state") === "streaming") {
-      return true;
-    }
-    return isToolUIPart(part) && part.state === "input-streaming";
-  });
+  return isToolUIPart2(part) && part.state === "input-streaming";
 }
-function trimIncompleteTrailingAssistantMessages(messages) {
-  let endIndex = messages.length;
-  while (endIndex > 0 && isIncompleteAssistantMessage(messages[endIndex - 1])) {
-    endIndex -= 1;
+function pruneIncompleteTrailingAssistantParts(messages) {
+  let result = messages;
+  while (result.length > 0) {
+    const lastIndex = result.length - 1;
+    const last = result[lastIndex];
+    if (last.role !== "assistant") {
+      break;
+    }
+    const completeParts = last.parts.filter((part) => !isIncompletePart(part));
+    if (completeParts.length === last.parts.length) {
+      break;
+    }
+    if (completeParts.length === 0) {
+      result = result.slice(0, lastIndex);
+      continue;
+    }
+    const next = result.slice();
+    next[lastIndex] = { ...last, parts: completeParts };
+    result = next;
+    break;
   }
-  return endIndex === messages.length ? messages : messages.slice(0, endIndex);
+  return result;
 }
 function hasUnresolvedToolParts(messages) {
   return messages.some((message) => message.parts.some((part) => {
-    if (!isToolUIPart(part)) {
+    if (!isToolUIPart2(part)) {
       return false;
     }
     return ![
@@ -8151,26 +8262,21 @@ function hasUnresolvedToolParts(messages) {
     ].includes(part.state);
   }));
 }
-function sanitizeMessagesForRetry(messages) {
-  const trimmed = trimIncompleteTrailingAssistantMessages(messages);
-  return trimmed.map((message) => {
-    if (message.role !== "assistant") {
-      return message;
-    }
-    let changed = false;
-    const parts = message.parts.map((part) => {
-      if (!isToolUIPart(part) || terminalToolPartStates.has(part.state)) {
-        return part;
+function countCompletedWork(messages) {
+  let total = 0;
+  for (const message of messages) {
+    for (const part of message.parts) {
+      if ((part.type === "text" || part.type === "reasoning") && Reflect.get(part, "state") !== "streaming") {
+        total += 1;
+      } else if (isToolUIPart2(part) && terminalToolPartStates.has(part.state)) {
+        total += 1;
       }
-      changed = true;
-      return {
-        ...part,
-        state: "output-error",
-        errorText: interruptedToolRetryErrorText
-      };
-    });
-    return changed ? { ...message, parts } : message;
-  });
+    }
+  }
+  return total;
+}
+function sanitizeMessagesForRetry(messages) {
+  return resolveDanglingToolParts(pruneIncompleteTrailingAssistantParts(messages));
 }
 function hasActiveClientToolWork({
   messages,
@@ -8180,7 +8286,7 @@ function hasActiveClientToolWork({
   const pendingApprovalIds = new Set(pendingApprovals.map((approval) => approval.toolCallId));
   const pendingPromptIds = new Set(pendingUserPrompts.map((prompt) => prompt.toolCallId));
   return messages.some((message) => message.parts.some((part) => {
-    if (!isToolUIPart(part)) {
+    if (!isToolUIPart2(part)) {
       return false;
     }
     if (part.state === "input-streaming" || part.state === "approval-responded") {
@@ -8308,6 +8414,9 @@ function useCodingSessionChat({
   autoContinueLimitsRef.current = autoContinueLimits;
   const maxErrorRetriesRef = useRef(maxErrorRetries);
   maxErrorRetriesRef.current = maxErrorRetries;
+  const maxNoProgressRetries = maxErrorRetries + 3;
+  const maxNoProgressRetriesRef = useRef(maxNoProgressRetries);
+  maxNoProgressRetriesRef.current = maxNoProgressRetries;
   const stallTimeoutMsRef = useRef(stallTimeoutMs);
   stallTimeoutMsRef.current = stallTimeoutMs;
   const lastStreamActivityRef = useRef(Date.now());
@@ -8316,11 +8425,18 @@ function useCodingSessionChat({
   const errorRetriesRef = useRef(0);
   const errorRetryTimerRef = useRef(null);
   const lastDecidedAssistantIdRef = useRef(null);
+  const lastProgressMarkerRef = useRef(0);
+  const noProgressRetriesRef = useRef(0);
+  const taskStuckRef = useRef(false);
+  const messagesRef = useRef([]);
   const advanceTurnKey = useCallback(() => {
     turnCounterRef.current += 1;
     turnKeyRef.current = `turn-${Date.now()}-${turnCounterRef.current}`;
     autoContinuesRef.current = 0;
     errorRetriesRef.current = 0;
+    lastProgressMarkerRef.current = 0;
+    noProgressRetriesRef.current = 0;
+    taskStuckRef.current = false;
     lastDecidedAssistantIdRef.current = null;
     if (errorRetryTimerRef.current) {
       clearTimeout(errorRetryTimerRef.current);
@@ -8328,6 +8444,20 @@ function useCodingSessionChat({
     }
     setAutoContinueState(null);
   }, []);
+  const registerRetryProgress = useCallback((currentMessages) => {
+    const marker = countCompletedWork(currentMessages);
+    if (marker > lastProgressMarkerRef.current) {
+      lastProgressMarkerRef.current = marker;
+      noProgressRetriesRef.current = 0;
+      return true;
+    }
+    noProgressRetriesRef.current += 1;
+    if (noProgressRetriesRef.current > maxNoProgressRetriesRef.current) {
+      taskStuckRef.current = true;
+      return false;
+    }
+    return true;
+  }, []);
   modeRef.current = mode;
   cwdRef.current = cwd;
   permissionModeRef.current = permissionMode;
@@ -8707,6 +8837,9 @@ function useCodingSessionChat({
       lastStreamActivityRef.current = Date.now();
     }
   }, [messages, status]);
+  useEffect(() => {
+    messagesRef.current = messages;
+  }, [messages]);
   useEffect(() => {
     if (status !== "streaming" && status !== "submitted") {
       return;
@@ -8719,7 +8852,15 @@ function useCodingSessionChat({
       })) {
         return;
       }
-      if (stallRecoveryInFlightRef.current || errorRetriesRef.current >= maxErrorRetriesRef.current) {
+      if (taskStuckRef.current || stallRecoveryInFlightRef.current || errorRetriesRef.current >= maxErrorRetriesRef.current) {
+        return;
+      }
+      if (!registerRetryProgress(messagesRef.current)) {
+        setAutoContinueState({
+          attempt: errorRetriesRef.current,
+          kind: "none",
+          guardTripped: "no-progress"
+        });
         return;
       }
       stallRecoveryInFlightRef.current = true;
@@ -8741,12 +8882,12 @@ function useCodingSessionChat({
       })();
     }, 15000);
     return () => clearInterval(intervalId);
-  }, [clearError, sendMessage, setMessages, status, stop]);
+  }, [clearError, registerRetryProgress, sendMessage, setMessages, status, stop]);
   useEffect(() => {
     if (status !== "ready" || error) {
       return;
     }
-    if (isHistoryLoading || pendingApprovals.length > 0 || pendingUserPrompts.length > 0 || hasUnresolvedToolParts(messages)) {
+    if (taskStuckRef.current || isHistoryLoading || pendingApprovals.length > 0 || pendingUserPrompts.length > 0 || hasUnresolvedToolParts(messages)) {
       return;
     }
     const lastMessage = messages[messages.length - 1];
@@ -8798,7 +8939,7 @@ function useCodingSessionChat({
     if (!error?.message || !isRecoverableChatErrorMessage(error.message)) {
       return;
     }
-    if (pendingApprovals.length > 0 || pendingUserPrompts.length > 0 || errorRetryTimerRef.current || stallRecoveryInFlightRef.current) {
+    if (taskStuckRef.current || pendingApprovals.length > 0 || pendingUserPrompts.length > 0 || errorRetryTimerRef.current || stallRecoveryInFlightRef.current) {
       return;
     }
     if (errorRetriesRef.current >= maxErrorRetriesRef.current) {
@@ -8809,6 +8950,14 @@ function useCodingSessionChat({
     const delay = Math.min(errorRetryBaseDelayMs * 2 ** (attempt - 1), errorRetryMaxDelayMs);
     errorRetryTimerRef.current = setTimeout(() => {
       errorRetryTimerRef.current = null;
+      if (!registerRetryProgress(messagesRef.current)) {
+        setAutoContinueState({
+          attempt: errorRetriesRef.current,
+          kind: "none",
+          guardTripped: "no-progress"
+        });
+        return;
+      }
       errorRetriesRef.current = attempt;
       setAutoContinueState({ attempt, kind: "retry-error" });
       setMessages(retryMessages);
@@ -8828,6 +8977,7 @@ function useCodingSessionChat({
     messages,
     pendingApprovals.length,
     pendingUserPrompts.length,
+    registerRetryProgress,
     sendMessage,
     setMessages
   ]);
@@ -8906,7 +9056,7 @@ function useCodingSessionChat({
         if (cancelled) {
           return;
         }
-        const resumableMessages = restoredApprovals.length === 0 && restoredPrompts.length === 0 ? trimIncompleteTrailingAssistantMessages(validatedMessages) : validatedMessages;
+        const resumableMessages = restoredApprovals.length === 0 && restoredPrompts.length === 0 ? pruneIncompleteTrailingAssistantParts(validatedMessages) : validatedMessages;
         setMessagesRef.current(resumableMessages);
         setPendingApprovals(restoredApprovals);
         setPendingUserPrompts(restoredPrompts);
@@ -9027,7 +9177,7 @@ function useCodingSessionChat({
     isSessionIdValid
   };
 }
-var recoverableDisconnectMessage = "Connection interrupted. Please retry or regenerate your last message.", buildModeAutoPromptResponse = "Proceed with implementation in Build mode. Continue without additional plan questions.", chatMessageUpdateThrottleMs = 50, retryCommandPattern, codingToolNameSet, errorRetryBaseDelayMs = 1000, errorRetryMaxDelayMs = 8000, disconnectErrorPattern, transientProviderErrorPattern, interruptedToolRetryErrorText = "Tool call was interrupted by a connection retry. Re-issue the tool call if it is still needed.", terminalToolPartStates;
+var recoverableDisconnectMessage = "Connection interrupted. Please retry or regenerate your last message.", buildModeAutoPromptResponse = "Proceed with implementation in Build mode. Continue without additional plan questions.", chatMessageUpdateThrottleMs = 50, retryCommandPattern, codingToolNameSet, errorRetryBaseDelayMs = 1000, errorRetryMaxDelayMs = 8000, disconnectErrorPattern, transientProviderErrorPattern, terminalToolPartStates;
 var init_use_coding_session_chat = __esm(() => {
   init_agent_tools();
   init_chat_error();
@@ -9040,6 +9190,7 @@ var init_use_coding_session_chat = __esm(() => {
   init_schema9();
   init_runtime12();
   init_auto_continue();
+  init_normalize_provider_messages();
   retryCommandPattern = /^\/?(retry|regenerate)$/i;
   codingToolNameSet = new Set(Object.keys(codingToolInputSchemas));
   disconnectErrorPattern = /(failed to fetch|networkerror|socket|connection|aborted|terminated|timed?\s?out|timeout|econnreset)/i;
@@ -10156,7 +10307,7 @@ var init_chat_message_tool_invocation_part = __esm(() => {
 });
 
 // apps/cli/src/components/chat/chat-message.tsx
-import { isToolUIPart as isToolUIPart2 } from "ai";
+import { isToolUIPart as isToolUIPart3 } from "ai";
 import { TextAttributes as TextAttributes9 } from "@opentui/core";
 import { jsxDEV as jsxDEV13 } from "@opentui/react/jsx-dev-runtime";
 function ChatMessage({ message, pendingApprovalIds }) {
@@ -10173,7 +10324,7 @@ function ChatMessage({ message, pendingApprovalIds }) {
         part
       }, key, false, undefined, this);
     }
-    if (isToolUIPart2(part)) {
+    if (isToolUIPart3(part)) {
       return /* @__PURE__ */ jsxDEV13(ChatMessageToolInvocationPart, {
         part,
         pendingApprovalIds
@@ -11904,7 +12055,7 @@ function ChatScreen() {
           paddingX: 1,
           children: /* @__PURE__ */ jsxDEV21("text", {
             fg: cliTheme.semantic.warning,
-            children: autoContinueState.guardTripped === "doom-loop" ? "Agent appears stuck repeating itself \u2014 automatic continuation stopped. Send a message to resume." : `Reached the automatic continuation limit (${autoContinueState.attempt}). Send a message to resume.`
+            children: autoContinueState.guardTripped === "doom-loop" ? "Agent appears stuck repeating itself \u2014 automatic continuation stopped. Send a message to resume." : autoContinueState.guardTripped === "no-progress" ? "The connection keeps dropping without making progress \u2014 automatic retry stopped. Send a message to resume." : `Reached the automatic continuation limit (${autoContinueState.attempt}). Send a message to resume.`
           }, undefined, false, undefined, this)
         }, undefined, false, undefined, this) : autoContinueState ? /* @__PURE__ */ jsxDEV21("box", {
           paddingX: 1,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kmugalkhod/lightcode",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "description": "Lightcode — a personal AI coding agent with a terminal UI and local session storage.",
   "type": "module",
   "license": "MIT",

package/server.js

@@ -3201,6 +3201,40 @@ var init_build_provider_view = __esm(() => {
   init_tier1_prune();
 });
 
+// packages/ai/src/context/normalize-provider-messages.ts
+import { isToolUIPart } from "ai";
+function resolveDanglingToolParts(messages, errorText = interruptedToolErrorText) {
+  return messages.map((message) => {
+    if (message.role !== "assistant") {
+      return message;
+    }
+    let changed = false;
+    const parts = message.parts.map((part) => {
+      if (!isToolUIPart(part) || terminalToolStates.has(part.state)) {
+        return part;
+      }
+      changed = true;
+      return {
+        ...part,
+        state: "output-error",
+        errorText
+      };
+    });
+    return changed ? { ...message, parts } : message;
+  });
+}
+function normalizeProviderMessages(messages) {
+  return resolveDanglingToolParts(messages);
+}
+var terminalToolStates, interruptedToolErrorText = "Tool call was interrupted by a connection retry. Re-issue the tool call if it is still needed.";
+var init_normalize_provider_messages = __esm(() => {
+  terminalToolStates = new Set([
+    "output-available",
+    "output-error",
+    "output-denied"
+  ]);
+});
+
 // packages/ai/src/context/index.ts
 var init_context = __esm(() => {
   init_build_provider_view();
@@ -3209,6 +3243,7 @@ var init_context = __esm(() => {
   init_estimate();
   init_heuristic_summary();
   init_message_parts();
+  init_normalize_provider_messages();
   init_tier1_prune();
 });
 
@@ -3307,6 +3342,12 @@ function readEnvConfig(env) {
     summaryMaxChars: parseNumberEnv(env.LIGHTCODE_CONTEXT_SUMMARY_MAX_CHARS, "LIGHTCODE_CONTEXT_SUMMARY_MAX_CHARS")
   };
   const compactContextConfig = Object.fromEntries(Object.entries(contextConfig).filter(([, value]) => value !== undefined));
+  const headroomConfig = {
+    enabled: parseBooleanEnv(env.LIGHTCODE_HEADROOM_ENABLED, "LIGHTCODE_HEADROOM_ENABLED"),
+    proxyUrl: env.LIGHTCODE_HEADROOM_PROXY_URL?.trim() || undefined,
+    failOpen: parseBooleanEnv(env.LIGHTCODE_HEADROOM_FAIL_OPEN, "LIGHTCODE_HEADROOM_FAIL_OPEN")
+  };
+  const compactHeadroomConfig = Object.fromEntries(Object.entries(headroomConfig).filter(([, value]) => value !== undefined));
   const rawConfig = {
     provider: env.LIGHTCODE_PROVIDER,
     model: env.LIGHTCODE_CHAT_MODEL,
@@ -3316,7 +3357,8 @@ function readEnvConfig(env) {
     maxOutputTokens: parseNumberEnv(env.LIGHTCODE_MAX_OUTPUT_TOKENS, "LIGHTCODE_MAX_OUTPUT_TOKENS"),
     maxSteps: parseNumberEnv(env.LIGHTCODE_MAX_STEPS, "LIGHTCODE_MAX_STEPS"),
     maxRetries: parseNumberEnv(env.LIGHTCODE_MAX_RETRIES, "LIGHTCODE_MAX_RETRIES"),
-    context: Object.keys(compactContextConfig).length > 0 ? compactContextConfig : undefined
+    context: Object.keys(compactContextConfig).length > 0 ? compactContextConfig : undefined,
+    headroom: Object.keys(compactHeadroomConfig).length > 0 ? compactHeadroomConfig : undefined
   };
   const compactConfig = Object.fromEntries(Object.entries(rawConfig).filter(([, value]) => value !== undefined));
   const parsedConfig = lightcodeConfigSchema.safeParse(compactConfig);
@@ -3340,6 +3382,10 @@ function mergeConfig(...configs) {
       ...mergedConfig.autoContinue,
       ...config2.autoContinue
     } : mergedConfig.autoContinue,
+    headroom: config2.headroom ? {
+      ...mergedConfig.headroom,
+      ...config2.headroom
+    } : mergedConfig.headroom,
     allowedTools: config2.allowedTools ?? mergedConfig.allowedTools
   }), {});
 }
@@ -3380,6 +3426,10 @@ function loadLightcodeConfig({
     autoContinue: {
       ...defaultAutoContinueConfig,
       ...mergedConfig.autoContinue
+    },
+    headroom: {
+      ...defaultHeadroomConfig,
+      ...mergedConfig.headroom
     }
   };
   return lightcodeConfigLoadResultSchema.parse({
@@ -3387,7 +3437,7 @@ function loadLightcodeConfig({
     loadedFiles: [userConfig.file, projectConfig.file]
   });
 }
-var lightcodeProviderSchema, autoContinueConfigSchema, resolvedAutoContinueConfigSchema, defaultAutoContinueConfig, lightcodeConfigSchema, lightcodeConfigDefaults, loadedConfigFileSchema, lightcodeResolvedConfigSchema, lightcodeConfigLoadResultSchema, lightcodeConfigStatusSchema, LightcodeConfigError;
+var lightcodeProviderSchema, autoContinueConfigSchema, resolvedAutoContinueConfigSchema, defaultAutoContinueConfig, headroomConfigSchema, resolvedHeadroomConfigSchema, defaultHeadroomConfig, lightcodeConfigSchema, lightcodeConfigDefaults, loadedConfigFileSchema, lightcodeResolvedConfigSchema, lightcodeConfigLoadResultSchema, lightcodeConfigStatusSchema, LightcodeConfigError;
 var init_lightcode_config = __esm(() => {
   init_coding_agent_modes();
   init_coding_agent_modes();
@@ -3419,6 +3469,24 @@ var init_lightcode_config = __esm(() => {
     maxErrorRetries: 5,
     stallTimeoutSeconds: 180
   };
+  headroomConfigSchema = z28.object({
+    enabled: z28.boolean().optional(),
+    proxyUrl: z28.string().min(1).max(4096).optional(),
+    providers: z28.array(lightcodeProviderSchema).optional(),
+    failOpen: z28.boolean().optional()
+  }).strict();
+  resolvedHeadroomConfigSchema = z28.object({
+    enabled: z28.boolean(),
+    proxyUrl: z28.string().min(1).max(4096),
+    providers: z28.array(lightcodeProviderSchema).nullable(),
+    failOpen: z28.boolean()
+  });
+  defaultHeadroomConfig = {
+    enabled: false,
+    proxyUrl: "http://127.0.0.1:8787",
+    providers: null,
+    failOpen: true
+  };
   lightcodeConfigSchema = z28.object({
     provider: lightcodeProviderSchema.optional(),
     model: z28.string().min(1).max(240).optional(),
@@ -3430,10 +3498,11 @@ var init_lightcode_config = __esm(() => {
     sandbox: sandboxConfigSchema.optional(),
     mcp: mcpConfigSchema.optional(),
     context: contextOptimizerConfigSchema.optional(),
-    maxOutputTokens: z28.number().int().min(1).max(200000).optional(),
+    maxOutputTokens: z28.number().int().min(1).max(1e6).optional(),
     maxSteps: z28.number().int().min(1).max(100).optional(),
     maxRetries: z28.number().int().min(0).max(10).optional(),
-    autoContinue: autoContinueConfigSchema.optional()
+    autoContinue: autoContinueConfigSchema.optional(),
+    headroom: headroomConfigSchema.optional()
   }).strict();
   lightcodeConfigDefaults = {
     provider: "anthropic",
@@ -3442,7 +3511,8 @@ var init_lightcode_config = __esm(() => {
     maxOutputTokens: 32768,
     maxSteps: 30,
     maxRetries: 5,
-    autoContinue: defaultAutoContinueConfig
+    autoContinue: defaultAutoContinueConfig,
+    headroom: defaultHeadroomConfig
   };
   loadedConfigFileSchema = z28.object({
     scope: z28.enum(["user", "project"]),
@@ -3454,10 +3524,11 @@ var init_lightcode_config = __esm(() => {
     provider: lightcodeProviderSchema,
     defaultMode: codingAgentModeSchema,
     context: resolvedContextOptimizerConfigSchema,
-    maxOutputTokens: z28.number().int().min(1).max(200000),
+    maxOutputTokens: z28.number().int().min(1).max(1e6),
     maxSteps: z28.number().int().min(1).max(100),
     maxRetries: z28.number().int().min(0).max(10),
-    autoContinue: resolvedAutoContinueConfigSchema
+    autoContinue: resolvedAutoContinueConfigSchema,
+    headroom: resolvedHeadroomConfigSchema.optional()
   });
   lightcodeConfigLoadResultSchema = z28.object({
     config: lightcodeResolvedConfigSchema,
@@ -3471,10 +3542,15 @@ var init_lightcode_config = __esm(() => {
     baseUrl: z28.string().nullable(),
     defaultMode: codingAgentModeSchema,
     permissionMode: permissionModeSchema.nullable(),
-    maxOutputTokens: z28.number().int().min(1).max(200000),
+    maxOutputTokens: z28.number().int().min(1).max(1e6),
     maxSteps: z28.number().int().min(1).max(100),
     autoContinue: resolvedAutoContinueConfigSchema,
     contextWindow: z28.number().int().positive(),
+    headroom: z28.object({
+      enabled: z28.boolean(),
+      proxyUrl: z28.string().nullable(),
+      routed: z28.boolean()
+    }),
     missingCredentialHints: z28.array(z28.string())
   });
   LightcodeConfigError = class LightcodeConfigError extends Error {
@@ -3714,6 +3790,21 @@ var init_runtime = __esm(() => {
   init_schema13();
 });
 
+// packages/ai/src/providers/clamp-output-tokens.ts
+function clampMaxOutputTokens(configured, modelMaxCompletionTokens) {
+  if (typeof modelMaxCompletionTokens === "number" && Number.isFinite(modelMaxCompletionTokens) && modelMaxCompletionTokens > 0) {
+    return Math.min(configured, modelMaxCompletionTokens);
+  }
+  return configured;
+}
+function resolveMaxOutputTokens(configured, defaultValue, modelMaxCompletionTokens) {
+  const hasModelCap = typeof modelMaxCompletionTokens === "number" && Number.isFinite(modelMaxCompletionTokens) && modelMaxCompletionTokens > 0;
+  if (configured === defaultValue) {
+    return hasModelCap ? modelMaxCompletionTokens : configured;
+  }
+  return clampMaxOutputTokens(configured, modelMaxCompletionTokens);
+}
+
 // packages/ai/src/providers/model-metadata.ts
 function getModelContextWindow(provider, modelId) {
   if (provider === "anthropic") {
@@ -6551,6 +6642,9 @@ function summarizeRawModel(raw2) {
   }
   const name = Reflect.get(raw2, "name");
   const contextLength = Reflect.get(raw2, "context_length");
+  const topProvider = Reflect.get(raw2, "top_provider");
+  const topProviderMaxCompletion = typeof topProvider === "object" && topProvider !== null ? Reflect.get(topProvider, "max_completion_tokens") : undefined;
+  const rawMaxCompletion = typeof topProviderMaxCompletion === "number" ? topProviderMaxCompletion : Reflect.get(raw2, "max_completion_tokens");
   const supportedParameters = Reflect.get(raw2, "supported_parameters");
   const parameterList = Array.isArray(supportedParameters) ? supportedParameters.filter((entry) => typeof entry === "string") : [];
   const pricingRaw = Reflect.get(raw2, "pricing");
@@ -6562,6 +6656,7 @@ function summarizeRawModel(raw2) {
     id,
     name: typeof name === "string" && name.length > 0 ? name : id,
     contextLength: typeof contextLength === "number" && contextLength > 0 ? Math.round(contextLength) : null,
+    maxCompletionTokens: typeof rawMaxCompletion === "number" && rawMaxCompletion > 0 ? Math.round(rawMaxCompletion) : null,
     supportsTools: parameterList.includes("tools"),
     supportsReasoning: parameterList.includes("reasoning"),
     pricing
@@ -6611,7 +6706,8 @@ async function getOpenRouterModelCapabilities(modelId) {
     }
     return {
       supportsNativeTools: model.supportsTools,
-      contextLength: model.contextLength ?? undefined
+      contextLength: model.contextLength ?? undefined,
+      maxCompletionTokens: model.maxCompletionTokens
     };
   } catch {
     return;
@@ -6626,6 +6722,7 @@ var init_openrouter_models = __esm(() => {
     id: z34.string(),
     name: z34.string(),
     contextLength: z34.number().int().positive().nullable(),
+    maxCompletionTokens: z34.number().int().positive().nullable().default(null),
     supportsTools: z34.boolean(),
     supportsReasoning: z34.boolean(),
     pricing: z34.object({
@@ -6671,6 +6768,16 @@ function getAnthropicProviderOptions(modelId) {
     }
   };
 }
+function resolveHeadroomRouting(config5, realBaseUrl) {
+  const headroom = config5.headroom;
+  if (!headroom?.enabled) {
+    return { effectiveBaseUrl: realBaseUrl, headroomRouted: false };
+  }
+  if (headroom.providers && !headroom.providers.includes(config5.provider)) {
+    return { effectiveBaseUrl: realBaseUrl, headroomRouted: false };
+  }
+  return { effectiveBaseUrl: headroom.proxyUrl, headroomRouted: true };
+}
 function resolveAnthropicModel({
   config: config5,
   env,
@@ -6681,19 +6788,22 @@ function resolveAnthropicModel({
   const apiKey = getEnvValue(env, "ANTHROPIC_API_KEY") ?? storedCredentials.anthropicApiKey;
   const authToken = getEnvValue(env, "ANTHROPIC_AUTH_TOKEN");
   const baseUrl = config5.baseUrl ?? getEnvValue(env, "ANTHROPIC_BASE_URL") ?? null;
+  const { effectiveBaseUrl, headroomRouted } = resolveHeadroomRouting(config5, baseUrl);
   const missingCredentialHints = apiKey || authToken ? [] : [
     "Set ANTHROPIC_API_KEY or ANTHROPIC_AUTH_TOKEN to call Anthropic models."
   ];
   const provider = createAnthropic({
     apiKey,
     authToken,
-    baseURL: baseUrl ?? undefined
+    baseURL: effectiveBaseUrl ?? undefined
   });
   return {
     provider: "anthropic",
     configuredModel,
     resolvedModelId: modelId,
     baseUrl,
+    effectiveBaseUrl,
+    headroomRouted,
     model: provider(modelId),
     providerOptions: getAnthropicProviderOptions(modelId),
     contextWindow: getModelContextWindow("anthropic", modelId),
@@ -6716,6 +6826,7 @@ function resolveOpenAITransportModel({
   if (!baseUrl) {
     throw new Error('OpenAI-compatible provider requires "baseUrl" in config or LIGHTCODE_OPENAI_COMPATIBLE_BASE_URL.');
   }
+  const { effectiveBaseUrl, headroomRouted } = resolveHeadroomRouting(config5, baseUrl);
   const apiKey = isOpenCodeZen ? getEnvValue(env, "OPENCODE_API_KEY", "LIGHTCODE_OPENAI_COMPATIBLE_API_KEY", "OPENAI_API_KEY") ?? storedCredentials.opencodeApiKey : isOpenRouter ? getEnvValue(env, "OPENROUTER_API_KEY", "LIGHTCODE_OPENAI_COMPATIBLE_API_KEY", "OPENAI_API_KEY") ?? storedCredentials.openrouterApiKey : getEnvValue(env, "LIGHTCODE_OPENAI_COMPATIBLE_API_KEY", "OPENAI_API_KEY") ?? storedCredentials.openaiCompatibleApiKey;
   const headers = {};
   if (isOpenRouter) {
@@ -6730,7 +6841,7 @@ function resolveOpenAITransportModel({
   }
   const provider = createOpenAICompatible({
     name: config5.provider,
-    baseURL: baseUrl,
+    baseURL: effectiveBaseUrl ?? baseUrl,
     apiKey,
     headers
   });
@@ -6739,12 +6850,15 @@ function resolveOpenAITransportModel({
     configuredModel,
     resolvedModelId: configuredModel,
     baseUrl,
+    effectiveBaseUrl,
+    headroomRouted,
     model: withXmlToolCallSupport(provider(configuredModel)),
     needsToolCallDiscipline: true,
     contextWindow: capabilities?.contextLength ?? getModelContextWindow(config5.provider, configuredModel),
     missingCredentialHints: apiKey ? [] : isOpenCodeZen ? ["Set OPENCODE_API_KEY to call OpenCode Zen models."] : isOpenRouter ? ["Set OPENROUTER_API_KEY to call OpenRouter models."] : [
       "Set LIGHTCODE_OPENAI_COMPATIBLE_API_KEY or OPENAI_API_KEY if your OpenAI-compatible endpoint requires authentication."
-    ]
+    ],
+    maxCompletionTokens: capabilities?.maxCompletionTokens ?? null
   };
 }
 function resolveConfiguredProviderModel({
@@ -6776,13 +6890,18 @@ function createConfigStatus({
     baseUrl: resolvedProviderModel.baseUrl,
     defaultMode: config5.defaultMode,
     permissionMode: config5.permissionMode ?? null,
-    maxOutputTokens: config5.maxOutputTokens,
+    maxOutputTokens: resolveMaxOutputTokens(config5.maxOutputTokens, lightcodeConfigDefaults.maxOutputTokens, resolvedProviderModel.maxCompletionTokens),
     maxSteps: config5.maxSteps,
     autoContinue: config5.autoContinue,
     contextWindow: resolveContextWindowTokens({
       config: config5.context,
       modelContextWindow: resolvedProviderModel.contextWindow
     }),
+    headroom: {
+      enabled: config5.headroom?.enabled ?? false,
+      proxyUrl: config5.headroom?.proxyUrl ?? null,
+      routed: resolvedProviderModel.headroomRouted
+    },
     missingCredentialHints: resolvedProviderModel.missingCredentialHints
   });
 }
@@ -6802,6 +6921,7 @@ __export(exports_runtime_config, {
   resolvedProviderModel: () => resolvedProviderModel,
   lightcodeConfigResult: () => lightcodeConfigResult,
   initializeModelCapabilities: () => initializeModelCapabilities,
+  disableHeadroomRouting: () => disableHeadroomRouting,
   configStatus: () => configStatus,
   codingAgent: () => codingAgent,
   chatModelId: () => chatModelId,
@@ -6814,7 +6934,7 @@ function buildCodingAgent(config5, model) {
     providerOptions: model.providerOptions,
     maxSteps: config5.maxSteps,
     maxRetries: config5.maxRetries,
-    maxOutputTokens: config5.maxOutputTokens,
+    maxOutputTokens: resolveMaxOutputTokens(config5.maxOutputTokens, lightcodeConfigDefaults.maxOutputTokens, model.maxCompletionTokens),
     includeToolDiscipline: model.needsToolCallDiscipline ?? false
   });
 }
@@ -6859,6 +6979,39 @@ function applyModelSelection({
   });
   return configStatus;
 }
+function disableHeadroomRouting() {
+  if (!lightcodeConfigResult.config.headroom?.enabled) {
+    return;
+  }
+  const nextConfig = {
+    ...lightcodeConfigResult.config,
+    headroom: {
+      ...lightcodeConfigResult.config.headroom ?? lightcodeConfigDefaults.headroom,
+      enabled: false
+    }
+  };
+  const nextResolved = resolveConfiguredProviderModel({
+    config: nextConfig,
+    env: Bun.env
+  });
+  const nextAgent = buildCodingAgent(nextConfig, nextResolved);
+  lightcodeConfigResult = {
+    ...lightcodeConfigResult,
+    config: nextConfig
+  };
+  resolvedProviderModel = nextResolved;
+  chatModelId = nextResolved.resolvedModelId;
+  codingAgent = nextAgent;
+  configStatus = createConfigStatus({
+    config: nextConfig,
+    loadedFiles: lightcodeConfigResult.loadedFiles,
+    resolvedProviderModel: nextResolved
+  });
+  logger5.info("headroom_routing_disabled", {
+    provider: nextResolved.provider,
+    model: nextResolved.resolvedModelId
+  });
+}
 async function initializeModelCapabilities() {
   if (lightcodeConfigResult.config.provider !== "openrouter") {
     return;
@@ -7171,7 +7324,7 @@ async function streamSessionChat(c, sessionIdentifier, messagesPayload, cwd, mod
       });
     }
   }
-  const providerMessages = view.providerMessages;
+  const providerMessages = normalizeProviderMessages(view.providerMessages);
   let overflowNotedThisRequest = false;
   const noteOverflowForSession = () => {
     if (overflowNotedThisRequest) {
@@ -8228,9 +8381,62 @@ var init_app = __esm(() => {
   })).route("/", rootRoutes).route("/config", configRoutes).route("/diagnostics", diagnosticsRoutes).route("/extensions", extensionRoutes).route("/sessions", sessionRoutes);
 });
 
+// apps/server/src/lib/headroom-proxy.ts
+var exports_headroom_proxy = {};
+__export(exports_headroom_proxy, {
+  ensureHeadroomRouting: () => ensureHeadroomRouting,
+  checkHeadroomHealth: () => checkHeadroomHealth
+});
+async function checkHeadroomHealth(proxyUrl) {
+  try {
+    await fetch(new URL("/health", proxyUrl), {
+      signal: AbortSignal.timeout(healthCheckTimeoutMs)
+    });
+    return "healthy";
+  } catch {
+    return "down";
+  }
+}
+async function ensureHeadroomRouting() {
+  const headroom = lightcodeConfigResult.config.headroom;
+  if (!headroom?.enabled) {
+    return;
+  }
+  let health;
+  try {
+    health = await checkHeadroomHealth(headroom.proxyUrl);
+  } catch (error) {
+    health = "down";
+    logger9.warn("headroom_health_probe_failed", {
+      proxyUrl: headroom.proxyUrl,
+      error: getErrorMessage(error)
+    });
+  }
+  if (health === "healthy") {
+    logger9.info("headroom_routing_active", { proxyUrl: headroom.proxyUrl });
+    return;
+  }
+  if (headroom.failOpen) {
+    logger9.warn("headroom_unreachable_failing_open", {
+      proxyUrl: headroom.proxyUrl
+    });
+    disableHeadroomRouting();
+    return;
+  }
+  logger9.error("headroom_unreachable_fail_closed", {
+    proxyUrl: headroom.proxyUrl
+  });
+}
+var logger9, healthCheckTimeoutMs = 1500;
+var init_headroom_proxy = __esm(() => {
+  init_src();
+  init_runtime_config();
+  logger9 = createLogger("headroom-proxy");
+});
+
 // apps/server/src/index.ts
 init_src();
-var logger9 = createLogger("server");
+var logger10 = createLogger("server");
 var httpIdleTimeoutSeconds = 30;
 var shutdownForceExitMs = 5000;
 function isChatStreamingRoute(request) {
@@ -8243,7 +8449,7 @@ function isChatStreamingRoute(request) {
 async function startServer() {
   const logDirectory = enableFileLogSink(Bun.env);
   if (logDirectory) {
-    logger9.info("file_log_sink_enabled", { directory: logDirectory });
+    logger10.info("file_log_sink_enabled", { directory: logDirectory });
   }
   const port = Number(Bun.env.PORT ?? 4983);
   const hostname = Bun.env.LIGHTCODE_HOST ?? "127.0.0.1";
@@ -8264,7 +8470,12 @@ async function startServer() {
     process.exit(1);
   }
   Promise.resolve().then(() => (init_runtime_config(), exports_runtime_config)).then((runtime19) => runtime19.initializeModelCapabilities()).catch((error) => {
-    logger9.warn("model_capability_init_failed", {
+    logger10.warn("model_capability_init_failed", {
+      error: getErrorMessage(error)
+    });
+  });
+  Promise.resolve().then(() => (init_headroom_proxy(), exports_headroom_proxy)).then((headroom) => headroom.ensureHeadroomRouting()).catch((error) => {
+    logger10.warn("headroom_init_failed", {
       error: getErrorMessage(error)
     });
   });
@@ -8300,7 +8511,7 @@ async function startServer() {
       return;
     }
     shuttingDown = true;
-    logger9.info("server_shutdown", { signal });
+    logger10.info("server_shutdown", { signal });
     const forceExitTimer = setTimeout(() => {
       server.stop(true);
       process.exit(0);