codex-copilot-dx 0.6.5 → 0.6.6

package/src/responses-image-limit.mjs

@@ -54,6 +54,7 @@ function addPartContainer(records, refs, parts, options) {
   };
   let found = false;
   for (let partIndex = 0; partIndex < parts.length; partIndex += 1) {
+    if ((partIndex & 63) === 0) options.assertActive?.();
     const part = parts[partIndex];
     if (!isImagePart(part)) continue;
     found = true;
@@ -68,7 +69,7 @@ function addPartContainer(records, refs, parts, options) {
   if (found) records.push(record);
 }
 
-function collectImages(inputItems) {
+function collectImages(inputItems, assertActive) {
   const records = [];
   const refs = [];
   const topLevelRecord = {
@@ -81,6 +82,7 @@ function collectImages(inputItems) {
   let hasTopLevelImages = false;
 
   for (let itemIndex = 0; itemIndex < inputItems.length; itemIndex += 1) {
+    if ((itemIndex & 63) === 0) assertActive?.();
     const item = inputItems[itemIndex];
     if (!item || typeof item !== "object") continue;
 
@@ -96,6 +98,7 @@ function collectImages(inputItems) {
 
     if (item.type === "message") {
       addPartContainer(records, refs, item.content, {
+        assertActive,
         kind: "message",
         role: item.role,
         itemIndex,
@@ -105,6 +108,7 @@ function collectImages(inputItems) {
     const toolOutput = readResponsesToolOutputParts(item);
     if (!toolOutput) continue;
     addPartContainer(records, refs, toolOutput.parts, {
+      assertActive,
       kind: "tool-output",
       itemIndex,
       commit: toolOutput.commit,
@@ -138,6 +142,7 @@ function applyDrops(records, refs, dropped, maxImages) {
 }
 
 export function enforceResponsesImageLimit(inputItems, {
+  assertActive,
   currentInputStart = 0,
   maxImages = MAX_UPSTREAM_IMAGES,
   beforeMutate = () => {},
@@ -149,7 +154,8 @@ export function enforceResponsesImageLimit(inputItems, {
     return { total: 0, kept: 0, omitted: 0, duplicates: 0, historicalOmitted: 0, currentOmitted: 0 };
   }
 
-  const { records, refs } = collectImages(inputItems);
+  assertActive?.();
+  const { records, refs } = collectImages(inputItems, assertActive);
   if (refs.length <= limit) {
     return { total: refs.length, kept: refs.length, omitted: 0, duplicates: 0, historicalOmitted: 0, currentOmitted: 0 };
   }
@@ -161,6 +167,7 @@ export function enforceResponsesImageLimit(inputItems, {
   let duplicates = 0;
 
   for (let index = refs.length - 1; index >= 0; index -= 1) {
+    if ((index & 63) === 0) assertActive?.();
     const identity = refs[index].identity;
     if (!identity) continue;
     let seen = seenByKind.get(identity.kind);
@@ -177,6 +184,7 @@ export function enforceResponsesImageLimit(inputItems, {
 
   duplicateCandidates.sort((left, right) => left - right);
   for (const index of duplicateCandidates) {
+    if ((kept & 63) === 0) assertActive?.();
     if (kept <= limit) break;
     dropped.add(index);
     kept -= 1;
@@ -184,6 +192,7 @@ export function enforceResponsesImageLimit(inputItems, {
   }
 
   for (let index = 0; index < refs.length && kept > limit; index += 1) {
+    if ((index & 63) === 0) assertActive?.();
     if (dropped.has(index)) continue;
     dropped.add(index);
     kept -= 1;
@@ -191,12 +200,17 @@ export function enforceResponsesImageLimit(inputItems, {
 
   let historicalOmitted = 0;
   let currentOmitted = 0;
+  let omittedIndex = 0;
   for (const index of dropped) {
+    if ((omittedIndex & 63) === 0) assertActive?.();
+    omittedIndex += 1;
     if (refs[index].itemIndex >= currentInputStart) currentOmitted += 1;
     else historicalOmitted += 1;
   }
   beforeMutate({ historicalOmitted, currentOmitted });
+  assertActive?.();
   applyDrops(records, refs, dropped, limit);
+  assertActive?.();
 
   return {
     total: refs.length,

package/src/responses-image-pressure.mjs

@@ -18,12 +18,15 @@ function normalizedPolicy(policy = {}) {
   return { ...DEFAULT_RESPONSES_IMAGE_PRESSURE_POLICY, ...policy };
 }
 
-function bodyPayload(body) {
+function bodyPayload(body, assertActive) {
+  assertActive?.();
   const bodyText = JSON.stringify(body);
+  assertActive?.();
   return { bodyText, bodyBytes: Buffer.byteLength(bodyText) };
 }
 
 export function applyResponsesImagePressure(reqContext, {
+  assertActive,
   mode = "normal",
   policy: policyOverrides,
 } = {}) {
@@ -32,7 +35,8 @@ export function applyResponsesImagePressure(reqContext, {
   const currentInputStart = Number.isFinite(reqContext?.currentInputStart)
     ? reqContext.currentInputStart
     : 0;
-  const initialImages = responsesHistoricalImageStats(body?.input, currentInputStart);
+  assertActive?.();
+  const initialImages = responsesHistoricalImageStats(body?.input, currentInputStart, { assertActive });
   if (initialImages.historicalImages === 0) {
     return {
       mode: "normal",
@@ -49,7 +53,7 @@ export function applyResponsesImagePressure(reqContext, {
     };
   }
 
-  const initial = bodyPayload(body);
+  const initial = bodyPayload(body, assertActive);
   const pressureEligible = initialImages.historicalImages > 0 && (
     initialImages.historicalImages > policy.triggerHistoricalImages
       || initial.bodyBytes > policy.triggerBodyBytes
@@ -80,6 +84,7 @@ export function applyResponsesImagePressure(reqContext, {
   }
 
   const trimmed = trimResponsesHistoricalImages(body, {
+    assertActive,
     currentInputStart,
     maxHistoricalImages,
     targetBytes,
@@ -126,8 +131,9 @@ export function createResponsesImagePressureController({
   };
 
   return Object.freeze({
-    apply(reqContext) {
+    apply(reqContext, { assertActive } = {}) {
       const result = applyResponsesImagePressure(reqContext, {
+        assertActive,
         mode: modeFor(reqContext?.historyRootId),
         policy,
       });

package/src/responses-request.mjs

@@ -13,18 +13,28 @@ function cloneJson(value) {
   return value === undefined ? undefined : structuredClone(value);
 }
 
-function responsesInputItems(input, { clone = true } = {}) {
+function responsesInputItems(input, { assertActive, clone = true } = {}) {
   if (input === undefined || input === null) return [];
   if (typeof input === "string") {
     return [{ type: "message", role: "user", content: [{ type: "input_text", text: input }] }];
   }
-  if (Array.isArray(input)) return clone ? cloneJson(input) : input;
-  return [clone ? cloneJson(input) : input];
+  if (Array.isArray(input)) {
+    assertActive?.();
+    const items = clone ? cloneJson(input) : input;
+    assertActive?.();
+    return items;
+  }
+  assertActive?.();
+  const item = clone ? cloneJson(input) : input;
+  assertActive?.();
+  return [item];
 }
 
-export function stripInternalResponsesInputFields(inputItems) {
+export function stripInternalResponsesInputFields(inputItems, { assertActive } = {}) {
   if (!Array.isArray(inputItems)) return inputItems;
-  for (const item of inputItems) {
+  for (let index = 0; index < inputItems.length; index += 1) {
+    if ((index & 63) === 0) assertActive?.();
+    const item = inputItems[index];
     if (!item || typeof item !== "object") continue;
     for (const key of Object.keys(item)) {
       if (key.startsWith("internal_")) delete item[key];
@@ -158,16 +168,20 @@ export async function openCopilotResponse(reqContext, upstream = copilotResponse
   let imageNamespaceRetried = false;
   let payloadPrepared = false;
   for (let attempt = 0; attempt < 3; attempt += 1) {
+    options.assertPrepareActive?.();
     const resp = await upstream(reqContext.body, {
+      assertActive: options.assertPrepareActive,
       signal: options.signal,
       currentInputStart: reqContext.currentInputStart,
       onUpstreamStart: options.onUpstreamStart,
       payloadPrepared,
     });
+    options.assertPrepareActive?.();
     payloadPrepared = true;
     if (resp.ok) return { resp, reqContext };
 
     const errorText = await resp.text();
+    options.assertPrepareActive?.();
     if (!imageNamespaceRetried && isImageNamespaceCollisionError(resp.status, errorText)) {
       const retryContext = sanitizeImageNamespaceCollisionRequest(reqContext);
       if (retryContext) {
@@ -195,9 +209,11 @@ function isBuiltinImageTool(tool) {
   return isImageNamespaceTool(tool);
 }
 
-export function prepareResponsesRequest(reqBody, { mutate = false } = {}) {
+export function prepareResponsesRequest(reqBody, { assertActive, mutate = false } = {}) {
+  assertActive?.();
   const body = mutate ? reqBody : cloneJson(reqBody);
-  const currentInputItems = responsesInputItems(body.input, { clone: !mutate });
+  assertActive?.();
+  const currentInputItems = responsesInputItems(body.input, { assertActive, clone: !mutate });
   const currentInputRefs = new Set(
     currentInputItems.filter((item) => item && typeof item === "object"),
   );
@@ -206,7 +222,7 @@ export function prepareResponsesRequest(reqBody, { mutate = false } = {}) {
   let historyRootId = null;
 
   if (previousId !== undefined && previousId !== null) {
-    historyItems = materializeResponseHistory(previousId);
+    historyItems = materializeResponseHistory(previousId, { assertActive });
     historyRootId = responseHistoryRootId(previousId);
     body.input = [...historyItems, ...currentInputItems];
   } else {
@@ -215,6 +231,7 @@ export function prepareResponsesRequest(reqBody, { mutate = false } = {}) {
 
   let historyInputItems = currentInputItems;
   const imageLimit = enforceResponsesImageLimit(body.input, {
+    assertActive,
     currentInputStart: historyItems.length,
     beforeMutate: ({ currentOmitted }) => {
       if (currentOmitted > 0) historyInputItems = cloneJson(currentInputItems);
@@ -230,8 +247,9 @@ export function prepareResponsesRequest(reqBody, { mutate = false } = {}) {
     body.tools = body.tools.filter((tool) => !isBuiltinImageTool(tool));
     if (!body.tools.length) delete body.tools;
   }
-  stripInternalResponsesInputFields(body.input);
-  stripInternalResponsesInputFields(historyInputItems);
+  stripInternalResponsesInputFields(body.input, { assertActive });
+  stripInternalResponsesInputFields(historyInputItems, { assertActive });
+  assertActive?.();
   const retainedCurrentInputStart = body.input.findIndex((item) => currentInputRefs.has(item));
 
   return {