@bastani/atomic 0.8.31-alpha.2 → 0.8.31-alpha.4

package/dist/core/context-window.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"context-window.d.ts","sourceRoot":"","sources":["../../src/core/context-window.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC;AAExD,OAAO,QAAQ,uBAAuB,CAAC,CAAC;IACvC,UAAU,KAAK,CAAC,IAAI,SAAS,GAAG;QAC/B,oHAAoH;QACpH,oBAAoB,CAAC,EAAE,SAAS,MAAM,EAAE,CAAC;QACzC,wGAAwG;QACxG,oBAAoB,CAAC,EAAE,MAAM,CAAC;KAC9B;CACD;AAED,MAAM,WAAW,wBAAwB;IACxC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,sBAAsB,CAAC,IAAI,SAAS,GAAG,GAAG,GAAG;IAC7D,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,2BAA2B;IAC3C,KAAK,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,6BAA6B;IAC7C;;;;;;OAMG;IACH,+BAA+B,CAAC,EAAE,OAAO,CAAC;CAC1C;AAWD,wBAAgB,0BAA0B,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAE5E;AAED,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,MAAM,GAAG,wBAAwB,CAqB/E;AAED,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAUzD;AAED,wBAAgB,6BAA6B,CAAC,MAAM,EAAE,SAAS,MAAM,EAAE,GAAG,SAAS,GAAG,MAAM,EAAE,CAS7F;AAED,wBAAgB,4BAA4B,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,MAAM,CAEtE;AAED,wBAAgB,0BAA0B,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,MAAM,EAAE,CAEtE;AAED,wBAAgB,wBAAwB,CAAC,IAAI,SAAS,GAAG,EACxD,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,EAClB,oBAAoB,EAAE,SAAS,MAAM,EAAE,GACrC,KAAK,CAAC,IAAI,CAAC,CAMb;AAuBD,wBAAgB,mBAAmB,CAAC,IAAI,SAAS,GAAG,EACnD,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,EAClB,aAAa,EAAE,MAAM,EACrB,OAAO,GAAE,6BAAkC,GACzC,sBAAsB,CAAC,IAAI,CAAC,GAAG,2BAA2B,CAwB5D","sourcesContent":["import type { Api, Model } from \"@earendil-works/pi-ai\";\n\ndeclare module \"@earendil-works/pi-ai\" {\n\tinterface Model<TApi extends Api> {\n\t\t/** Selectable context-window sizes for this model. The scalar contextWindow remains the default/effective value. */\n\t\tcontextWindowOptions?: readonly number[];\n\t\t/** Original/default scalar context window, preserved when contextWindow is overridden for a session. */\n\t\tdefaultContextWindow?: number;\n\t}\n}\n\nexport interface ContextWindowParseResult {\n\tvalue?: number;\n\terror?: string;\n}\n\nexport interface ContextWindowSelection<TApi extends Api = Api> {\n\tmodel: Model<TApi>;\n\tcontextWindow: number;\n}\n\nexport interface ContextWindowSelectionError {\n\terror: string;\n}\n\nexport interface ContextWindowSelectionOptions {\n\t/**\n\t * GitHub Copilot advertises some long-context tiers below their branded 1M size\n\t * (for example 936k or 922k input tokens). When enabled, a request above an\n\t * advertised long tier selects the largest supported Copilot window not\n\t * exceeding the request, but never silently falls back to the model's base\n\t * window.\n\t */\n\tallowCopilotLongContextFallback?: boolean;\n}\n\nconst CONTEXT_WINDOW_UNITS: Record<string, number> = {\n\tk: 1_000,\n\tm: 1_000_000,\n};\n\nfunction isPositiveInteger(value: number): boolean {\n\treturn Number.isFinite(value) && Number.isInteger(value) && value > 0;\n}\n\nexport function validateContextWindowValue(value: number): string | undefined {\n\treturn isPositiveInteger(value) ? undefined : \"Context window must be a positive integer token count\";\n}\n\nexport function parseContextWindowValue(input: string): ContextWindowParseResult {\n\tconst trimmed = input.trim();\n\tif (!trimmed) {\n\t\treturn { error: \"Context window requires a value\" };\n\t}\n\n\tconst match = /^(\\d+(?:\\.\\d+)?)([kKmM])?$/.exec(trimmed);\n\tif (!match) {\n\t\treturn { error: `Invalid context window \"${input}\". Use a positive number, or a compact value like 400k or 1m.` };\n\t}\n\n\tconst numericValue = Number(match[1]);\n\tconst unit = match[2]?.toLowerCase();\n\tconst multiplier = unit ? CONTEXT_WINDOW_UNITS[unit] : 1;\n\tconst tokens = numericValue * multiplier;\n\tconst validationError = validateContextWindowValue(tokens);\n\tif (validationError) {\n\t\treturn { error: `Invalid context window \"${input}\". ${validationError}.` };\n\t}\n\n\treturn { value: tokens };\n}\n\nexport function formatContextWindow(value: number): string {\n\tif (value >= 1_000_000) {\n\t\tconst millions = value / 1_000_000;\n\t\treturn millions % 1 === 0 ? `${millions}m` : `${millions.toFixed(1)}m`;\n\t}\n\tif (value >= 1_000) {\n\t\tconst thousands = value / 1_000;\n\t\treturn thousands % 1 === 0 ? `${thousands}k` : `${thousands.toFixed(1)}k`;\n\t}\n\treturn String(value);\n}\n\nexport function normalizeContextWindowOptions(values: readonly number[] | undefined): number[] {\n\tconst seen = new Set<number>();\n\tconst normalized: number[] = [];\n\tfor (const value of values ?? []) {\n\t\tif (!isPositiveInteger(value) || seen.has(value)) continue;\n\t\tseen.add(value);\n\t\tnormalized.push(value);\n\t}\n\treturn normalized.sort((a, b) => a - b);\n}\n\nexport function getModelDefaultContextWindow(model: Model<Api>): number {\n\treturn isPositiveInteger(model.defaultContextWindow ?? 0) ? model.defaultContextWindow! : model.contextWindow;\n}\n\nexport function getSupportedContextWindows(model: Model<Api>): number[] {\n\treturn normalizeContextWindowOptions([getModelDefaultContextWindow(model), ...(model.contextWindowOptions ?? [])]);\n}\n\nexport function withContextWindowOptions<TApi extends Api>(\n\tmodel: Model<TApi>,\n\tcontextWindowOptions: readonly number[],\n): Model<TApi> {\n\treturn {\n\t\t...model,\n\t\tdefaultContextWindow: getModelDefaultContextWindow(model as Model<Api>),\n\t\tcontextWindowOptions: normalizeContextWindowOptions(contextWindowOptions),\n\t};\n}\n\nfunction resolveSelectableContextWindow(\n\tmodel: Model<Api>,\n\trequestedContextWindow: number,\n\tsupported: readonly number[],\n\toptions: ContextWindowSelectionOptions,\n): number | undefined {\n\tif (supported.includes(requestedContextWindow)) {\n\t\treturn requestedContextWindow;\n\t}\n\n\tif (options.allowCopilotLongContextFallback !== true || model.provider !== \"github-copilot\") {\n\t\treturn undefined;\n\t}\n\n\tconst defaultContextWindow = getModelDefaultContextWindow(model);\n\tconst candidates = supported.filter(\n\t\t(contextWindow) => contextWindow <= requestedContextWindow && contextWindow > defaultContextWindow,\n\t);\n\treturn candidates.length > 0 ? Math.max(...candidates) : undefined;\n}\n\nexport function selectContextWindow<TApi extends Api>(\n\tmodel: Model<TApi>,\n\tcontextWindow: number,\n\toptions: ContextWindowSelectionOptions = {},\n): ContextWindowSelection<TApi> | ContextWindowSelectionError {\n\tconst validationError = validateContextWindowValue(contextWindow);\n\tif (validationError) {\n\t\treturn { error: validationError };\n\t}\n\n\tconst apiModel = model as Model<Api>;\n\tconst supported = getSupportedContextWindows(apiModel);\n\tconst selectedContextWindow = resolveSelectableContextWindow(apiModel, contextWindow, supported, options);\n\tif (selectedContextWindow === undefined) {\n\t\treturn {\n\t\t\terror: `Context window ${formatContextWindow(contextWindow)} is not supported by ${model.provider}/${model.id}. Supported values: ${supported.map(formatContextWindow).join(\", \")}.`,\n\t\t};\n\t}\n\n\treturn {\n\t\tmodel: {\n\t\t\t...model,\n\t\t\tdefaultContextWindow: getModelDefaultContextWindow(apiModel),\n\t\t\tcontextWindow: selectedContextWindow,\n\t\t\tcontextWindowOptions: supported,\n\t\t},\n\t\tcontextWindow: selectedContextWindow,\n\t};\n}\n"]}
+{"version":3,"file":"context-window.d.ts","sourceRoot":"","sources":["../../src/core/context-window.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC;AAExD,OAAO,QAAQ,uBAAuB,CAAC,CAAC;IACvC,UAAU,KAAK,CAAC,IAAI,SAAS,GAAG;QAC/B,oHAAoH;QACpH,oBAAoB,CAAC,EAAE,SAAS,MAAM,EAAE,CAAC;QACzC,wGAAwG;QACxG,oBAAoB,CAAC,EAAE,MAAM,CAAC;QAC9B;;;;;WAKG;QACH,cAAc,CAAC,EAAE,MAAM,CAAC;KACxB;CACD;AAED,MAAM,WAAW,wBAAwB;IACxC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,sBAAsB,CAAC,IAAI,SAAS,GAAG,GAAG,GAAG;IAC7D,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,2BAA2B;IAC3C,KAAK,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,6BAA6B;IAC7C;;;;;;OAMG;IACH,+BAA+B,CAAC,EAAE,OAAO,CAAC;CAC1C;AAWD,wBAAgB,0BAA0B,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAE5E;AAED,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,MAAM,GAAG,wBAAwB,CAqB/E;AAED,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAUzD;AAED,wBAAgB,6BAA6B,CAAC,MAAM,EAAE,SAAS,MAAM,EAAE,GAAG,SAAS,GAAG,MAAM,EAAE,CAS7F;AAED,wBAAgB,4BAA4B,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,MAAM,CAEtE;AAED;;;;;;GAMG;AACH,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,MAAM,CAGjE;AAED,wBAAgB,0BAA0B,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,MAAM,EAAE,CAEtE;AAED,wBAAgB,wBAAwB,CAAC,IAAI,SAAS,GAAG,EACxD,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,EAClB,oBAAoB,EAAE,SAAS,MAAM,EAAE,GACrC,KAAK,CAAC,IAAI,CAAC,CAMb;AAuBD,wBAAgB,mBAAmB,CAAC,IAAI,SAAS,GAAG,EACnD,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,EAClB,aAAa,EAAE,MAAM,EACrB,OAAO,GAAE,6BAAkC,GACzC,sBAAsB,CAAC,IAAI,CAAC,GAAG,2BAA2B,CAwB5D","sourcesContent":["import type { Api, Model } from \"@earendil-works/pi-ai\";\n\ndeclare module \"@earendil-works/pi-ai\" {\n\tinterface Model<TApi extends Api> {\n\t\t/** Selectable context-window sizes for this model. The scalar contextWindow remains the default/effective value. */\n\t\tcontextWindowOptions?: readonly number[];\n\t\t/** Original/default scalar context window, preserved when contextWindow is overridden for a session. */\n\t\tdefaultContextWindow?: number;\n\t\t/**\n\t\t * Hard prompt/input cap for providers (e.g. GitHub Copilot `max_prompt_tokens`) that enforce an\n\t\t * input budget below the displayed context window. When set and below `contextWindow`, it is the\n\t\t * effective input budget for compaction thresholds and overflow recovery; `contextWindow` remains\n\t\t * the displayed/branded window.\n\t\t */\n\t\tmaxInputTokens?: number;\n\t}\n}\n\nexport interface ContextWindowParseResult {\n\tvalue?: number;\n\terror?: string;\n}\n\nexport interface ContextWindowSelection<TApi extends Api = Api> {\n\tmodel: Model<TApi>;\n\tcontextWindow: number;\n}\n\nexport interface ContextWindowSelectionError {\n\terror: string;\n}\n\nexport interface ContextWindowSelectionOptions {\n\t/**\n\t * GitHub Copilot advertises some long-context tiers below their branded 1M size\n\t * (for example 936k or 922k input tokens). When enabled, a request above an\n\t * advertised long tier selects the largest supported Copilot window not\n\t * exceeding the request, but never silently falls back to the model's base\n\t * window.\n\t */\n\tallowCopilotLongContextFallback?: boolean;\n}\n\nconst CONTEXT_WINDOW_UNITS: Record<string, number> = {\n\tk: 1_000,\n\tm: 1_000_000,\n};\n\nfunction isPositiveInteger(value: number): boolean {\n\treturn Number.isFinite(value) && Number.isInteger(value) && value > 0;\n}\n\nexport function validateContextWindowValue(value: number): string | undefined {\n\treturn isPositiveInteger(value) ? undefined : \"Context window must be a positive integer token count\";\n}\n\nexport function parseContextWindowValue(input: string): ContextWindowParseResult {\n\tconst trimmed = input.trim();\n\tif (!trimmed) {\n\t\treturn { error: \"Context window requires a value\" };\n\t}\n\n\tconst match = /^(\\d+(?:\\.\\d+)?)([kKmM])?$/.exec(trimmed);\n\tif (!match) {\n\t\treturn { error: `Invalid context window \"${input}\". Use a positive number, or a compact value like 400k or 1m.` };\n\t}\n\n\tconst numericValue = Number(match[1]);\n\tconst unit = match[2]?.toLowerCase();\n\tconst multiplier = unit ? CONTEXT_WINDOW_UNITS[unit] : 1;\n\tconst tokens = numericValue * multiplier;\n\tconst validationError = validateContextWindowValue(tokens);\n\tif (validationError) {\n\t\treturn { error: `Invalid context window \"${input}\". ${validationError}.` };\n\t}\n\n\treturn { value: tokens };\n}\n\nexport function formatContextWindow(value: number): string {\n\tif (value >= 1_000_000) {\n\t\tconst millions = value / 1_000_000;\n\t\treturn millions % 1 === 0 ? `${millions}m` : `${millions.toFixed(1)}m`;\n\t}\n\tif (value >= 1_000) {\n\t\tconst thousands = value / 1_000;\n\t\treturn thousands % 1 === 0 ? `${thousands}k` : `${thousands.toFixed(1)}k`;\n\t}\n\treturn String(value);\n}\n\nexport function normalizeContextWindowOptions(values: readonly number[] | undefined): number[] {\n\tconst seen = new Set<number>();\n\tconst normalized: number[] = [];\n\tfor (const value of values ?? []) {\n\t\tif (!isPositiveInteger(value) || seen.has(value)) continue;\n\t\tseen.add(value);\n\t\tnormalized.push(value);\n\t}\n\treturn normalized.sort((a, b) => a - b);\n}\n\nexport function getModelDefaultContextWindow(model: Model<Api>): number {\n\treturn isPositiveInteger(model.defaultContextWindow ?? 0) ? model.defaultContextWindow! : model.contextWindow;\n}\n\n/**\n * Effective input-token budget for compaction/overflow decisions. Equals the displayed\n * `contextWindow` unless a smaller hard input cap (`maxInputTokens`, e.g. GitHub Copilot's\n * `max_prompt_tokens`) is advertised, in which case the lower of the two is used. This lets a model\n * display its full/branded window while compaction and overflow recovery respect the real,\n * server-enforced input limit.\n */\nexport function getEffectiveInputBudget(model: Model<Api>): number {\n\tconst cap = model.maxInputTokens;\n\treturn isPositiveInteger(cap ?? 0) ? Math.min(model.contextWindow, cap as number) : model.contextWindow;\n}\n\nexport function getSupportedContextWindows(model: Model<Api>): number[] {\n\treturn normalizeContextWindowOptions([getModelDefaultContextWindow(model), ...(model.contextWindowOptions ?? [])]);\n}\n\nexport function withContextWindowOptions<TApi extends Api>(\n\tmodel: Model<TApi>,\n\tcontextWindowOptions: readonly number[],\n): Model<TApi> {\n\treturn {\n\t\t...model,\n\t\tdefaultContextWindow: getModelDefaultContextWindow(model as Model<Api>),\n\t\tcontextWindowOptions: normalizeContextWindowOptions(contextWindowOptions),\n\t};\n}\n\nfunction resolveSelectableContextWindow(\n\tmodel: Model<Api>,\n\trequestedContextWindow: number,\n\tsupported: readonly number[],\n\toptions: ContextWindowSelectionOptions,\n): number | undefined {\n\tif (supported.includes(requestedContextWindow)) {\n\t\treturn requestedContextWindow;\n\t}\n\n\tif (options.allowCopilotLongContextFallback !== true || model.provider !== \"github-copilot\") {\n\t\treturn undefined;\n\t}\n\n\tconst defaultContextWindow = getModelDefaultContextWindow(model);\n\tconst candidates = supported.filter(\n\t\t(contextWindow) => contextWindow <= requestedContextWindow && contextWindow > defaultContextWindow,\n\t);\n\treturn candidates.length > 0 ? Math.max(...candidates) : undefined;\n}\n\nexport function selectContextWindow<TApi extends Api>(\n\tmodel: Model<TApi>,\n\tcontextWindow: number,\n\toptions: ContextWindowSelectionOptions = {},\n): ContextWindowSelection<TApi> | ContextWindowSelectionError {\n\tconst validationError = validateContextWindowValue(contextWindow);\n\tif (validationError) {\n\t\treturn { error: validationError };\n\t}\n\n\tconst apiModel = model as Model<Api>;\n\tconst supported = getSupportedContextWindows(apiModel);\n\tconst selectedContextWindow = resolveSelectableContextWindow(apiModel, contextWindow, supported, options);\n\tif (selectedContextWindow === undefined) {\n\t\treturn {\n\t\t\terror: `Context window ${formatContextWindow(contextWindow)} is not supported by ${model.provider}/${model.id}. Supported values: ${supported.map(formatContextWindow).join(\", \")}.`,\n\t\t};\n\t}\n\n\treturn {\n\t\tmodel: {\n\t\t\t...model,\n\t\t\tdefaultContextWindow: getModelDefaultContextWindow(apiModel),\n\t\t\tcontextWindow: selectedContextWindow,\n\t\t\tcontextWindowOptions: supported,\n\t\t},\n\t\tcontextWindow: selectedContextWindow,\n\t};\n}\n"]}

package/dist/core/context-window.js

@@ -52,6 +52,17 @@ export function normalizeContextWindowOptions(values) {
 export function getModelDefaultContextWindow(model) {
     return isPositiveInteger(model.defaultContextWindow ?? 0) ? model.defaultContextWindow : model.contextWindow;
 }
+/**
+ * Effective input-token budget for compaction/overflow decisions. Equals the displayed
+ * `contextWindow` unless a smaller hard input cap (`maxInputTokens`, e.g. GitHub Copilot's
+ * `max_prompt_tokens`) is advertised, in which case the lower of the two is used. This lets a model
+ * display its full/branded window while compaction and overflow recovery respect the real,
+ * server-enforced input limit.
+ */
+export function getEffectiveInputBudget(model) {
+    const cap = model.maxInputTokens;
+    return isPositiveInteger(cap ?? 0) ? Math.min(model.contextWindow, cap) : model.contextWindow;
+}
 export function getSupportedContextWindows(model) {
     return normalizeContextWindowOptions([getModelDefaultContextWindow(model), ...(model.contextWindowOptions ?? [])]);
 }

package/dist/core/context-window.js.map

@@ -1 +1 @@
-{"version":3,"file":"context-window.js","sourceRoot":"","sources":["../../src/core/context-window.ts"],"names":[],"mappings":"AAoCA,MAAM,oBAAoB,GAA2B;IACpD,CAAC,EAAE,KAAK;IACR,CAAC,EAAE,SAAS;CACZ,CAAC;AAEF,SAAS,iBAAiB,CAAC,KAAa;IACvC,OAAO,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,CAAC;AACvE,CAAC;AAED,MAAM,UAAU,0BAA0B,CAAC,KAAa;IACvD,OAAO,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,uDAAuD,CAAC;AACvG,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,KAAa;IACpD,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;IAC7B,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,OAAO,EAAE,KAAK,EAAE,iCAAiC,EAAE,CAAC;IACrD,CAAC;IAED,MAAM,KAAK,GAAG,4BAA4B,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACzD,IAAI,CAAC,KAAK,EAAE,CAAC;QACZ,OAAO,EAAE,KAAK,EAAE,2BAA2B,KAAK,+DAA+D,EAAE,CAAC;IACnH,CAAC;IAED,MAAM,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IACtC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC;IACrC,MAAM,UAAU,GAAG,IAAI,CAAC,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACzD,MAAM,MAAM,GAAG,YAAY,GAAG,UAAU,CAAC;IACzC,MAAM,eAAe,GAAG,0BAA0B,CAAC,MAAM,CAAC,CAAC;IAC3D,IAAI,eAAe,EAAE,CAAC;QACrB,OAAO,EAAE,KAAK,EAAE,2BAA2B,KAAK,MAAM,eAAe,GAAG,EAAE,CAAC;IAC5E,CAAC;IAED,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,KAAa;IAChD,IAAI,KAAK,IAAI,SAAS,EAAE,CAAC;QACxB,MAAM,QAAQ,GAAG,KAAK,GAAG,SAAS,CAAC;QACnC,OAAO,QAAQ,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,QAAQ,GAAG,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IACxE,CAAC;IACD,IAAI,KAAK,IAAI,KAAK,EAAE,CAAC;QACpB,MAAM,SAAS,GAAG,KAAK,GAAG,KAAK,CAAC;QAChC,OAAO,SAAS,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,SAAS,GAAG,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IAC3E,CAAC;IACD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;AACtB,CAAC;AAED,MAAM,UAAU,6BAA6B,CAAC,MAAqC;IAClF,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,MAAM,UAAU,GAAa,EAAE,CAAC;IAChC,KAAK,MAAM,KAAK,IAAI,MAAM,IAAI,EAAE,EAAE,CAAC;QAClC,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC;YAAE,SAAS;QAC3D,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAChB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACxB,CAAC;IACD,OAAO,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;AACzC,CAAC;AAED,MAAM,UAAU,4BAA4B,CAAC,KAAiB;IAC7D,OAAO,iBAAiB,CAAC,KAAK,CAAC,oBAAoB,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,oBAAqB,CAAC,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC;AAC/G,CAAC;AAED,MAAM,UAAU,0BAA0B,CAAC,KAAiB;IAC3D,OAAO,6BAA6B,CAAC,CAAC,4BAA4B,CAAC,KAAK,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,oBAAoB,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;AACpH,CAAC;AAED,MAAM,UAAU,wBAAwB,CACvC,KAAkB,EAClB,oBAAuC;IAEvC,OAAO;QACN,GAAG,KAAK;QACR,oBAAoB,EAAE,4BAA4B,CAAC,KAAmB,CAAC;QACvE,oBAAoB,EAAE,6BAA6B,CAAC,oBAAoB,CAAC;KACzE,CAAC;AACH,CAAC;AAED,SAAS,8BAA8B,CACtC,KAAiB,EACjB,sBAA8B,EAC9B,SAA4B,EAC5B,OAAsC;IAEtC,IAAI,SAAS,CAAC,QAAQ,CAAC,sBAAsB,CAAC,EAAE,CAAC;QAChD,OAAO,sBAAsB,CAAC;IAC/B,CAAC;IAED,IAAI,OAAO,CAAC,+BAA+B,KAAK,IAAI,IAAI,KAAK,CAAC,QAAQ,KAAK,gBAAgB,EAAE,CAAC;QAC7F,OAAO,SAAS,CAAC;IAClB,CAAC;IAED,MAAM,oBAAoB,GAAG,4BAA4B,CAAC,KAAK,CAAC,CAAC;IACjE,MAAM,UAAU,GAAG,SAAS,CAAC,MAAM,CAClC,CAAC,aAAa,EAAE,EAAE,CAAC,aAAa,IAAI,sBAAsB,IAAI,aAAa,GAAG,oBAAoB,CAClG,CAAC;IACF,OAAO,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AACpE,CAAC;AAED,MAAM,UAAU,mBAAmB,CAClC,KAAkB,EAClB,aAAqB,EACrB,OAAO,GAAkC,EAAE;IAE3C,MAAM,eAAe,GAAG,0BAA0B,CAAC,aAAa,CAAC,CAAC;IAClE,IAAI,eAAe,EAAE,CAAC;QACrB,OAAO,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC;IACnC,CAAC;IAED,MAAM,QAAQ,GAAG,KAAmB,CAAC;IACrC,MAAM,SAAS,GAAG,0BAA0B,CAAC,QAAQ,CAAC,CAAC;IACvD,MAAM,qBAAqB,GAAG,8BAA8B,CAAC,QAAQ,EAAE,aAAa,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;IAC1G,IAAI,qBAAqB,KAAK,SAAS,EAAE,CAAC;QACzC,OAAO;YACN,KAAK,EAAE,kBAAkB,mBAAmB,CAAC,aAAa,CAAC,wBAAwB,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,EAAE,uBAAuB,SAAS,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG;SACpL,CAAC;IACH,CAAC;IAED,OAAO;QACN,KAAK,EAAE;YACN,GAAG,KAAK;YACR,oBAAoB,EAAE,4BAA4B,CAAC,QAAQ,CAAC;YAC5D,aAAa,EAAE,qBAAqB;YACpC,oBAAoB,EAAE,SAAS;SAC/B;QACD,aAAa,EAAE,qBAAqB;KACpC,CAAC;AACH,CAAC","sourcesContent":["import type { Api, Model } from \"@earendil-works/pi-ai\";\n\ndeclare module \"@earendil-works/pi-ai\" {\n\tinterface Model<TApi extends Api> {\n\t\t/** Selectable context-window sizes for this model. The scalar contextWindow remains the default/effective value. */\n\t\tcontextWindowOptions?: readonly number[];\n\t\t/** Original/default scalar context window, preserved when contextWindow is overridden for a session. */\n\t\tdefaultContextWindow?: number;\n\t}\n}\n\nexport interface ContextWindowParseResult {\n\tvalue?: number;\n\terror?: string;\n}\n\nexport interface ContextWindowSelection<TApi extends Api = Api> {\n\tmodel: Model<TApi>;\n\tcontextWindow: number;\n}\n\nexport interface ContextWindowSelectionError {\n\terror: string;\n}\n\nexport interface ContextWindowSelectionOptions {\n\t/**\n\t * GitHub Copilot advertises some long-context tiers below their branded 1M size\n\t * (for example 936k or 922k input tokens). When enabled, a request above an\n\t * advertised long tier selects the largest supported Copilot window not\n\t * exceeding the request, but never silently falls back to the model's base\n\t * window.\n\t */\n\tallowCopilotLongContextFallback?: boolean;\n}\n\nconst CONTEXT_WINDOW_UNITS: Record<string, number> = {\n\tk: 1_000,\n\tm: 1_000_000,\n};\n\nfunction isPositiveInteger(value: number): boolean {\n\treturn Number.isFinite(value) && Number.isInteger(value) && value > 0;\n}\n\nexport function validateContextWindowValue(value: number): string | undefined {\n\treturn isPositiveInteger(value) ? undefined : \"Context window must be a positive integer token count\";\n}\n\nexport function parseContextWindowValue(input: string): ContextWindowParseResult {\n\tconst trimmed = input.trim();\n\tif (!trimmed) {\n\t\treturn { error: \"Context window requires a value\" };\n\t}\n\n\tconst match = /^(\\d+(?:\\.\\d+)?)([kKmM])?$/.exec(trimmed);\n\tif (!match) {\n\t\treturn { error: `Invalid context window \"${input}\". Use a positive number, or a compact value like 400k or 1m.` };\n\t}\n\n\tconst numericValue = Number(match[1]);\n\tconst unit = match[2]?.toLowerCase();\n\tconst multiplier = unit ? CONTEXT_WINDOW_UNITS[unit] : 1;\n\tconst tokens = numericValue * multiplier;\n\tconst validationError = validateContextWindowValue(tokens);\n\tif (validationError) {\n\t\treturn { error: `Invalid context window \"${input}\". ${validationError}.` };\n\t}\n\n\treturn { value: tokens };\n}\n\nexport function formatContextWindow(value: number): string {\n\tif (value >= 1_000_000) {\n\t\tconst millions = value / 1_000_000;\n\t\treturn millions % 1 === 0 ? `${millions}m` : `${millions.toFixed(1)}m`;\n\t}\n\tif (value >= 1_000) {\n\t\tconst thousands = value / 1_000;\n\t\treturn thousands % 1 === 0 ? `${thousands}k` : `${thousands.toFixed(1)}k`;\n\t}\n\treturn String(value);\n}\n\nexport function normalizeContextWindowOptions(values: readonly number[] | undefined): number[] {\n\tconst seen = new Set<number>();\n\tconst normalized: number[] = [];\n\tfor (const value of values ?? []) {\n\t\tif (!isPositiveInteger(value) || seen.has(value)) continue;\n\t\tseen.add(value);\n\t\tnormalized.push(value);\n\t}\n\treturn normalized.sort((a, b) => a - b);\n}\n\nexport function getModelDefaultContextWindow(model: Model<Api>): number {\n\treturn isPositiveInteger(model.defaultContextWindow ?? 0) ? model.defaultContextWindow! : model.contextWindow;\n}\n\nexport function getSupportedContextWindows(model: Model<Api>): number[] {\n\treturn normalizeContextWindowOptions([getModelDefaultContextWindow(model), ...(model.contextWindowOptions ?? [])]);\n}\n\nexport function withContextWindowOptions<TApi extends Api>(\n\tmodel: Model<TApi>,\n\tcontextWindowOptions: readonly number[],\n): Model<TApi> {\n\treturn {\n\t\t...model,\n\t\tdefaultContextWindow: getModelDefaultContextWindow(model as Model<Api>),\n\t\tcontextWindowOptions: normalizeContextWindowOptions(contextWindowOptions),\n\t};\n}\n\nfunction resolveSelectableContextWindow(\n\tmodel: Model<Api>,\n\trequestedContextWindow: number,\n\tsupported: readonly number[],\n\toptions: ContextWindowSelectionOptions,\n): number | undefined {\n\tif (supported.includes(requestedContextWindow)) {\n\t\treturn requestedContextWindow;\n\t}\n\n\tif (options.allowCopilotLongContextFallback !== true || model.provider !== \"github-copilot\") {\n\t\treturn undefined;\n\t}\n\n\tconst defaultContextWindow = getModelDefaultContextWindow(model);\n\tconst candidates = supported.filter(\n\t\t(contextWindow) => contextWindow <= requestedContextWindow && contextWindow > defaultContextWindow,\n\t);\n\treturn candidates.length > 0 ? Math.max(...candidates) : undefined;\n}\n\nexport function selectContextWindow<TApi extends Api>(\n\tmodel: Model<TApi>,\n\tcontextWindow: number,\n\toptions: ContextWindowSelectionOptions = {},\n): ContextWindowSelection<TApi> | ContextWindowSelectionError {\n\tconst validationError = validateContextWindowValue(contextWindow);\n\tif (validationError) {\n\t\treturn { error: validationError };\n\t}\n\n\tconst apiModel = model as Model<Api>;\n\tconst supported = getSupportedContextWindows(apiModel);\n\tconst selectedContextWindow = resolveSelectableContextWindow(apiModel, contextWindow, supported, options);\n\tif (selectedContextWindow === undefined) {\n\t\treturn {\n\t\t\terror: `Context window ${formatContextWindow(contextWindow)} is not supported by ${model.provider}/${model.id}. Supported values: ${supported.map(formatContextWindow).join(\", \")}.`,\n\t\t};\n\t}\n\n\treturn {\n\t\tmodel: {\n\t\t\t...model,\n\t\t\tdefaultContextWindow: getModelDefaultContextWindow(apiModel),\n\t\t\tcontextWindow: selectedContextWindow,\n\t\t\tcontextWindowOptions: supported,\n\t\t},\n\t\tcontextWindow: selectedContextWindow,\n\t};\n}\n"]}
+{"version":3,"file":"context-window.js","sourceRoot":"","sources":["../../src/core/context-window.ts"],"names":[],"mappings":"AA2CA,MAAM,oBAAoB,GAA2B;IACpD,CAAC,EAAE,KAAK;IACR,CAAC,EAAE,SAAS;CACZ,CAAC;AAEF,SAAS,iBAAiB,CAAC,KAAa;IACvC,OAAO,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,CAAC;AACvE,CAAC;AAED,MAAM,UAAU,0BAA0B,CAAC,KAAa;IACvD,OAAO,iBAAiB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,uDAAuD,CAAC;AACvG,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,KAAa;IACpD,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;IAC7B,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,OAAO,EAAE,KAAK,EAAE,iCAAiC,EAAE,CAAC;IACrD,CAAC;IAED,MAAM,KAAK,GAAG,4BAA4B,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACzD,IAAI,CAAC,KAAK,EAAE,CAAC;QACZ,OAAO,EAAE,KAAK,EAAE,2BAA2B,KAAK,+DAA+D,EAAE,CAAC;IACnH,CAAC;IAED,MAAM,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IACtC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC;IACrC,MAAM,UAAU,GAAG,IAAI,CAAC,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACzD,MAAM,MAAM,GAAG,YAAY,GAAG,UAAU,CAAC;IACzC,MAAM,eAAe,GAAG,0BAA0B,CAAC,MAAM,CAAC,CAAC;IAC3D,IAAI,eAAe,EAAE,CAAC;QACrB,OAAO,EAAE,KAAK,EAAE,2BAA2B,KAAK,MAAM,eAAe,GAAG,EAAE,CAAC;IAC5E,CAAC;IAED,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,KAAa;IAChD,IAAI,KAAK,IAAI,SAAS,EAAE,CAAC;QACxB,MAAM,QAAQ,GAAG,KAAK,GAAG,SAAS,CAAC;QACnC,OAAO,QAAQ,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,QAAQ,GAAG,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IACxE,CAAC;IACD,IAAI,KAAK,IAAI,KAAK,EAAE,CAAC;QACpB,MAAM,SAAS,GAAG,KAAK,GAAG,KAAK,CAAC;QAChC,OAAO,SAAS,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,SAAS,GAAG,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IAC3E,CAAC;IACD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;AACtB,CAAC;AAED,MAAM,UAAU,6BAA6B,CAAC,MAAqC;IAClF,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,MAAM,UAAU,GAAa,EAAE,CAAC;IAChC,KAAK,MAAM,KAAK,IAAI,MAAM,IAAI,EAAE,EAAE,CAAC;QAClC,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC;YAAE,SAAS;QAC3D,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAChB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACxB,CAAC;IACD,OAAO,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;AACzC,CAAC;AAED,MAAM,UAAU,4BAA4B,CAAC,KAAiB;IAC7D,OAAO,iBAAiB,CAAC,KAAK,CAAC,oBAAoB,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,oBAAqB,CAAC,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC;AAC/G,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,uBAAuB,CAAC,KAAiB;IACxD,MAAM,GAAG,GAAG,KAAK,CAAC,cAAc,CAAC;IACjC,OAAO,iBAAiB,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,EAAE,GAAa,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC;AACzG,CAAC;AAED,MAAM,UAAU,0BAA0B,CAAC,KAAiB;IAC3D,OAAO,6BAA6B,CAAC,CAAC,4BAA4B,CAAC,KAAK,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,oBAAoB,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;AACpH,CAAC;AAED,MAAM,UAAU,wBAAwB,CACvC,KAAkB,EAClB,oBAAuC;IAEvC,OAAO;QACN,GAAG,KAAK;QACR,oBAAoB,EAAE,4BAA4B,CAAC,KAAmB,CAAC;QACvE,oBAAoB,EAAE,6BAA6B,CAAC,oBAAoB,CAAC;KACzE,CAAC;AACH,CAAC;AAED,SAAS,8BAA8B,CACtC,KAAiB,EACjB,sBAA8B,EAC9B,SAA4B,EAC5B,OAAsC;IAEtC,IAAI,SAAS,CAAC,QAAQ,CAAC,sBAAsB,CAAC,EAAE,CAAC;QAChD,OAAO,sBAAsB,CAAC;IAC/B,CAAC;IAED,IAAI,OAAO,CAAC,+BAA+B,KAAK,IAAI,IAAI,KAAK,CAAC,QAAQ,KAAK,gBAAgB,EAAE,CAAC;QAC7F,OAAO,SAAS,CAAC;IAClB,CAAC;IAED,MAAM,oBAAoB,GAAG,4BAA4B,CAAC,KAAK,CAAC,CAAC;IACjE,MAAM,UAAU,GAAG,SAAS,CAAC,MAAM,CAClC,CAAC,aAAa,EAAE,EAAE,CAAC,aAAa,IAAI,sBAAsB,IAAI,aAAa,GAAG,oBAAoB,CAClG,CAAC;IACF,OAAO,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AACpE,CAAC;AAED,MAAM,UAAU,mBAAmB,CAClC,KAAkB,EAClB,aAAqB,EACrB,OAAO,GAAkC,EAAE;IAE3C,MAAM,eAAe,GAAG,0BAA0B,CAAC,aAAa,CAAC,CAAC;IAClE,IAAI,eAAe,EAAE,CAAC;QACrB,OAAO,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC;IACnC,CAAC;IAED,MAAM,QAAQ,GAAG,KAAmB,CAAC;IACrC,MAAM,SAAS,GAAG,0BAA0B,CAAC,QAAQ,CAAC,CAAC;IACvD,MAAM,qBAAqB,GAAG,8BAA8B,CAAC,QAAQ,EAAE,aAAa,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;IAC1G,IAAI,qBAAqB,KAAK,SAAS,EAAE,CAAC;QACzC,OAAO;YACN,KAAK,EAAE,kBAAkB,mBAAmB,CAAC,aAAa,CAAC,wBAAwB,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,EAAE,uBAAuB,SAAS,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG;SACpL,CAAC;IACH,CAAC;IAED,OAAO;QACN,KAAK,EAAE;YACN,GAAG,KAAK;YACR,oBAAoB,EAAE,4BAA4B,CAAC,QAAQ,CAAC;YAC5D,aAAa,EAAE,qBAAqB;YACpC,oBAAoB,EAAE,SAAS;SAC/B;QACD,aAAa,EAAE,qBAAqB;KACpC,CAAC;AACH,CAAC","sourcesContent":["import type { Api, Model } from \"@earendil-works/pi-ai\";\n\ndeclare module \"@earendil-works/pi-ai\" {\n\tinterface Model<TApi extends Api> {\n\t\t/** Selectable context-window sizes for this model. The scalar contextWindow remains the default/effective value. */\n\t\tcontextWindowOptions?: readonly number[];\n\t\t/** Original/default scalar context window, preserved when contextWindow is overridden for a session. */\n\t\tdefaultContextWindow?: number;\n\t\t/**\n\t\t * Hard prompt/input cap for providers (e.g. GitHub Copilot `max_prompt_tokens`) that enforce an\n\t\t * input budget below the displayed context window. When set and below `contextWindow`, it is the\n\t\t * effective input budget for compaction thresholds and overflow recovery; `contextWindow` remains\n\t\t * the displayed/branded window.\n\t\t */\n\t\tmaxInputTokens?: number;\n\t}\n}\n\nexport interface ContextWindowParseResult {\n\tvalue?: number;\n\terror?: string;\n}\n\nexport interface ContextWindowSelection<TApi extends Api = Api> {\n\tmodel: Model<TApi>;\n\tcontextWindow: number;\n}\n\nexport interface ContextWindowSelectionError {\n\terror: string;\n}\n\nexport interface ContextWindowSelectionOptions {\n\t/**\n\t * GitHub Copilot advertises some long-context tiers below their branded 1M size\n\t * (for example 936k or 922k input tokens). When enabled, a request above an\n\t * advertised long tier selects the largest supported Copilot window not\n\t * exceeding the request, but never silently falls back to the model's base\n\t * window.\n\t */\n\tallowCopilotLongContextFallback?: boolean;\n}\n\nconst CONTEXT_WINDOW_UNITS: Record<string, number> = {\n\tk: 1_000,\n\tm: 1_000_000,\n};\n\nfunction isPositiveInteger(value: number): boolean {\n\treturn Number.isFinite(value) && Number.isInteger(value) && value > 0;\n}\n\nexport function validateContextWindowValue(value: number): string | undefined {\n\treturn isPositiveInteger(value) ? undefined : \"Context window must be a positive integer token count\";\n}\n\nexport function parseContextWindowValue(input: string): ContextWindowParseResult {\n\tconst trimmed = input.trim();\n\tif (!trimmed) {\n\t\treturn { error: \"Context window requires a value\" };\n\t}\n\n\tconst match = /^(\\d+(?:\\.\\d+)?)([kKmM])?$/.exec(trimmed);\n\tif (!match) {\n\t\treturn { error: `Invalid context window \"${input}\". Use a positive number, or a compact value like 400k or 1m.` };\n\t}\n\n\tconst numericValue = Number(match[1]);\n\tconst unit = match[2]?.toLowerCase();\n\tconst multiplier = unit ? CONTEXT_WINDOW_UNITS[unit] : 1;\n\tconst tokens = numericValue * multiplier;\n\tconst validationError = validateContextWindowValue(tokens);\n\tif (validationError) {\n\t\treturn { error: `Invalid context window \"${input}\". ${validationError}.` };\n\t}\n\n\treturn { value: tokens };\n}\n\nexport function formatContextWindow(value: number): string {\n\tif (value >= 1_000_000) {\n\t\tconst millions = value / 1_000_000;\n\t\treturn millions % 1 === 0 ? `${millions}m` : `${millions.toFixed(1)}m`;\n\t}\n\tif (value >= 1_000) {\n\t\tconst thousands = value / 1_000;\n\t\treturn thousands % 1 === 0 ? `${thousands}k` : `${thousands.toFixed(1)}k`;\n\t}\n\treturn String(value);\n}\n\nexport function normalizeContextWindowOptions(values: readonly number[] | undefined): number[] {\n\tconst seen = new Set<number>();\n\tconst normalized: number[] = [];\n\tfor (const value of values ?? []) {\n\t\tif (!isPositiveInteger(value) || seen.has(value)) continue;\n\t\tseen.add(value);\n\t\tnormalized.push(value);\n\t}\n\treturn normalized.sort((a, b) => a - b);\n}\n\nexport function getModelDefaultContextWindow(model: Model<Api>): number {\n\treturn isPositiveInteger(model.defaultContextWindow ?? 0) ? model.defaultContextWindow! : model.contextWindow;\n}\n\n/**\n * Effective input-token budget for compaction/overflow decisions. Equals the displayed\n * `contextWindow` unless a smaller hard input cap (`maxInputTokens`, e.g. GitHub Copilot's\n * `max_prompt_tokens`) is advertised, in which case the lower of the two is used. This lets a model\n * display its full/branded window while compaction and overflow recovery respect the real,\n * server-enforced input limit.\n */\nexport function getEffectiveInputBudget(model: Model<Api>): number {\n\tconst cap = model.maxInputTokens;\n\treturn isPositiveInteger(cap ?? 0) ? Math.min(model.contextWindow, cap as number) : model.contextWindow;\n}\n\nexport function getSupportedContextWindows(model: Model<Api>): number[] {\n\treturn normalizeContextWindowOptions([getModelDefaultContextWindow(model), ...(model.contextWindowOptions ?? [])]);\n}\n\nexport function withContextWindowOptions<TApi extends Api>(\n\tmodel: Model<TApi>,\n\tcontextWindowOptions: readonly number[],\n): Model<TApi> {\n\treturn {\n\t\t...model,\n\t\tdefaultContextWindow: getModelDefaultContextWindow(model as Model<Api>),\n\t\tcontextWindowOptions: normalizeContextWindowOptions(contextWindowOptions),\n\t};\n}\n\nfunction resolveSelectableContextWindow(\n\tmodel: Model<Api>,\n\trequestedContextWindow: number,\n\tsupported: readonly number[],\n\toptions: ContextWindowSelectionOptions,\n): number | undefined {\n\tif (supported.includes(requestedContextWindow)) {\n\t\treturn requestedContextWindow;\n\t}\n\n\tif (options.allowCopilotLongContextFallback !== true || model.provider !== \"github-copilot\") {\n\t\treturn undefined;\n\t}\n\n\tconst defaultContextWindow = getModelDefaultContextWindow(model);\n\tconst candidates = supported.filter(\n\t\t(contextWindow) => contextWindow <= requestedContextWindow && contextWindow > defaultContextWindow,\n\t);\n\treturn candidates.length > 0 ? Math.max(...candidates) : undefined;\n}\n\nexport function selectContextWindow<TApi extends Api>(\n\tmodel: Model<TApi>,\n\tcontextWindow: number,\n\toptions: ContextWindowSelectionOptions = {},\n): ContextWindowSelection<TApi> | ContextWindowSelectionError {\n\tconst validationError = validateContextWindowValue(contextWindow);\n\tif (validationError) {\n\t\treturn { error: validationError };\n\t}\n\n\tconst apiModel = model as Model<Api>;\n\tconst supported = getSupportedContextWindows(apiModel);\n\tconst selectedContextWindow = resolveSelectableContextWindow(apiModel, contextWindow, supported, options);\n\tif (selectedContextWindow === undefined) {\n\t\treturn {\n\t\t\terror: `Context window ${formatContextWindow(contextWindow)} is not supported by ${model.provider}/${model.id}. Supported values: ${supported.map(formatContextWindow).join(\", \")}.`,\n\t\t};\n\t}\n\n\treturn {\n\t\tmodel: {\n\t\t\t...model,\n\t\t\tdefaultContextWindow: getModelDefaultContextWindow(apiModel),\n\t\t\tcontextWindow: selectedContextWindow,\n\t\t\tcontextWindowOptions: supported,\n\t\t},\n\t\tcontextWindow: selectedContextWindow,\n\t};\n}\n"]}

package/dist/core/copilot-gemini-payload-sanitizer.d.ts

@@ -0,0 +1,72 @@
+import type { Api, Model } from "@earendil-works/pi-ai";
+/**
+ * Sanitizes outbound OpenAI-compatible request payloads for GitHub Copilot
+ * Gemini models so their tool/function JSON Schemas survive translation to
+ * Google's GenAI `FunctionDeclaration` schema.
+ *
+ * Why this exists
+ * ---------------
+ * `github-copilot` Gemini models (e.g. `gemini-3.1-pro-preview`) are served
+ * through Copilot's CAPI gateway at `api.*.githubcopilot.com` using the
+ * `openai-completions` API. CAPI receives the OpenAI chat-completions request
+ * and translates it into a Google GenAI `GenerateContent` request. During that
+ * translation CAPI forwards JSON Schema `anyOf`/`oneOf` verbatim into the Gemini
+ * `FunctionDeclaration` schema. Gemini's function-declaration schema rejects an
+ * `anyOf`/`oneOf` whose branch is a complex *object* schema, so Google returns
+ * HTTP 400 and CAPI relabels it `{"error":{"code":"invalid_request_body"}}`.
+ *
+ * Atomic's bundled tools (notably the `workflow` tool) use the TypeBox
+ * `Type.Union([Type.Object(...), Type.String()])` pattern for fields like
+ * `task`, `chain`, and `parallel`, which emit exactly that construct. Because
+ * those tools are present in normal chat turns, every Gemini request fails with
+ * `400 invalid request body` until the schema is sanitized.
+ *
+ * What it does
+ * ------------
+ * Recursively rewrites each tool's `parameters` JSON Schema into the reduced
+ * subset that CAPI/Gemini actually honors (`type`, `description`, `enum`,
+ * `properties`, `required`, `items`, `nullable`, and scalar-only `anyOf`):
+ *  - Resolves `anyOf`/`oneOf` that contain an object/array branch to the most
+ *    expressive object/array branch (the core fix), preserving `description`
+ *    and collapsing a `"null"` branch into `nullable: true`.
+ *  - Converts `const` (and `anyOf` of `const`/`enum` scalars) into `enum`,
+ *    which Gemini supports, so `Type.Literal(...)` unions keep their constraint.
+ *  - Drops JSON Schema keywords CAPI strips or Gemini rejects (`$schema`,
+ *    `$ref`, `$defs`, `patternProperties`, `additionalProperties`, `allOf`,
+ *    `not`, `format`, `pattern`, numeric/length bounds, `default`, etc.).
+ *  - Filters `required` down to keys still present under `properties`.
+ *
+ * The transform is gated to GitHub Copilot Gemini models only, so it never
+ * changes payloads for any currently-working provider/model.
+ */
+type JsonObject = {
+    [key: string]: JsonValue;
+};
+type JsonValue = string | number | boolean | null | JsonValue[] | JsonObject;
+/**
+ * Keywords that are dropped because CAPI strips them before Google and/or
+ * Gemini's function schema rejects them. Listing them explicitly keeps the
+ * intent auditable; any keyword not in {@link KEPT_SCHEMA_KEYWORDS} and not a
+ * union keyword is dropped regardless. Exported so the documented drop-list has
+ * a real consumer (and stays linted) without a per-iteration `void` reference.
+ */
+export declare const DROPPED_SCHEMA_KEYWORDS: Set<string>;
+/**
+ * Whether this model is a GitHub Copilot Gemini model routed through the
+ * OpenAI-completions CAPI path that needs schema sanitization.
+ */
+export declare function isCopilotGeminiModel(model: Pick<Model<Api>, "provider" | "api" | "id">): boolean;
+/**
+ * Recursively rewrite a JSON Schema node into the Gemini-compatible subset.
+ */
+export declare function sanitizeGeminiSchema(schema: JsonValue): JsonValue;
+/**
+ * Sanitize an outbound provider payload for GitHub Copilot Gemini models.
+ *
+ * Returns the payload unchanged for any other provider/model, or when the
+ * payload has no sanitizable `tools`. Mirrors the `onPayload`
+ * `(payload: unknown) => unknown` contract used elsewhere in the SDK wiring.
+ */
+export declare function sanitizeCopilotGeminiPayload(payload: unknown, model: Pick<Model<Api>, "provider" | "api" | "id">): unknown;
+export {};
+//# sourceMappingURL=copilot-gemini-payload-sanitizer.d.ts.map

package/dist/core/copilot-gemini-payload-sanitizer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"copilot-gemini-payload-sanitizer.d.ts","sourceRoot":"","sources":["../../src/core/copilot-gemini-payload-sanitizer.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC;AAExD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAuCG;AAEH,KAAK,UAAU,GAAG;IAAE,CAAC,GAAG,EAAE,MAAM,GAAG,SAAS,CAAA;CAAE,CAAC;AAC/C,KAAK,SAAS,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,IAAI,GAAG,SAAS,EAAE,GAAG,UAAU,CAAC;AAa7E;;;;;;GAMG;AACH,eAAO,MAAM,uBAAuB,aAwClC,CAAC;AAMH;;;GAGG;AACH,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,UAAU,GAAG,KAAK,GAAG,IAAI,CAAC,GAAG,OAAO,CAMhG;AAqCD;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,SAAS,GAAG,SAAS,CAgDjE;AA2GD;;;;;;GAMG;AACH,wBAAgB,4BAA4B,CAC1C,OAAO,EAAE,OAAO,EAChB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,UAAU,GAAG,KAAK,GAAG,IAAI,CAAC,GACjD,OAAO,CA+BT","sourcesContent":["import type { Api, Model } from \"@earendil-works/pi-ai\";\n\n/**\n * Sanitizes outbound OpenAI-compatible request payloads for GitHub Copilot\n * Gemini models so their tool/function JSON Schemas survive translation to\n * Google's GenAI `FunctionDeclaration` schema.\n *\n * Why this exists\n * ---------------\n * `github-copilot` Gemini models (e.g. `gemini-3.1-pro-preview`) are served\n * through Copilot's CAPI gateway at `api.*.githubcopilot.com` using the\n * `openai-completions` API. CAPI receives the OpenAI chat-completions request\n * and translates it into a Google GenAI `GenerateContent` request. During that\n * translation CAPI forwards JSON Schema `anyOf`/`oneOf` verbatim into the Gemini\n * `FunctionDeclaration` schema. Gemini's function-declaration schema rejects an\n * `anyOf`/`oneOf` whose branch is a complex *object* schema, so Google returns\n * HTTP 400 and CAPI relabels it `{\"error\":{\"code\":\"invalid_request_body\"}}`.\n *\n * Atomic's bundled tools (notably the `workflow` tool) use the TypeBox\n * `Type.Union([Type.Object(...), Type.String()])` pattern for fields like\n * `task`, `chain`, and `parallel`, which emit exactly that construct. Because\n * those tools are present in normal chat turns, every Gemini request fails with\n * `400 invalid request body` until the schema is sanitized.\n *\n * What it does\n * ------------\n * Recursively rewrites each tool's `parameters` JSON Schema into the reduced\n * subset that CAPI/Gemini actually honors (`type`, `description`, `enum`,\n * `properties`, `required`, `items`, `nullable`, and scalar-only `anyOf`):\n *  - Resolves `anyOf`/`oneOf` that contain an object/array branch to the most\n *    expressive object/array branch (the core fix), preserving `description`\n *    and collapsing a `\"null\"` branch into `nullable: true`.\n *  - Converts `const` (and `anyOf` of `const`/`enum` scalars) into `enum`,\n *    which Gemini supports, so `Type.Literal(...)` unions keep their constraint.\n *  - Drops JSON Schema keywords CAPI strips or Gemini rejects (`$schema`,\n *    `$ref`, `$defs`, `patternProperties`, `additionalProperties`, `allOf`,\n *    `not`, `format`, `pattern`, numeric/length bounds, `default`, etc.).\n *  - Filters `required` down to keys still present under `properties`.\n *\n * The transform is gated to GitHub Copilot Gemini models only, so it never\n * changes payloads for any currently-working provider/model.\n */\n\ntype JsonObject = { [key: string]: JsonValue };\ntype JsonValue = string | number | boolean | null | JsonValue[] | JsonObject;\n\n/** JSON Schema keywords Gemini's function-declaration schema honors. */\nconst KEPT_SCHEMA_KEYWORDS = new Set<string>([\n  \"type\",\n  \"description\",\n  \"enum\",\n  \"properties\",\n  \"required\",\n  \"items\",\n  \"nullable\",\n]);\n\n/**\n * Keywords that are dropped because CAPI strips them before Google and/or\n * Gemini's function schema rejects them. Listing them explicitly keeps the\n * intent auditable; any keyword not in {@link KEPT_SCHEMA_KEYWORDS} and not a\n * union keyword is dropped regardless. Exported so the documented drop-list has\n * a real consumer (and stays linted) without a per-iteration `void` reference.\n */\nexport const DROPPED_SCHEMA_KEYWORDS = new Set<string>([\n  \"$schema\",\n  \"$id\",\n  \"$ref\",\n  \"$defs\",\n  \"$comment\",\n  \"definitions\",\n  \"patternProperties\",\n  \"propertyNames\",\n  \"unevaluatedProperties\",\n  \"additionalProperties\",\n  \"additionalItems\",\n  \"unevaluatedItems\",\n  \"allOf\",\n  \"not\",\n  \"if\",\n  \"then\",\n  \"else\",\n  \"format\",\n  \"pattern\",\n  \"minLength\",\n  \"maxLength\",\n  \"minItems\",\n  \"maxItems\",\n  \"uniqueItems\",\n  \"minimum\",\n  \"maximum\",\n  \"exclusiveMinimum\",\n  \"exclusiveMaximum\",\n  \"multipleOf\",\n  \"minProperties\",\n  \"maxProperties\",\n  \"default\",\n  \"examples\",\n  \"title\",\n  \"readOnly\",\n  \"writeOnly\",\n  \"deprecated\",\n  \"contentEncoding\",\n  \"contentMediaType\",\n]);\n\nfunction isPlainObject(value: JsonValue | undefined): value is JsonObject {\n  return typeof value === \"object\" && value !== null && !Array.isArray(value);\n}\n\n/**\n * Whether this model is a GitHub Copilot Gemini model routed through the\n * OpenAI-completions CAPI path that needs schema sanitization.\n */\nexport function isCopilotGeminiModel(model: Pick<Model<Api>, \"provider\" | \"api\" | \"id\">): boolean {\n  return (\n    model.provider === \"github-copilot\" &&\n    model.api === \"openai-completions\" &&\n    /(^|[/-])gemini/i.test(model.id)\n  );\n}\n\nfunction jsonScalarType(value: JsonValue): string | undefined {\n  switch (typeof value) {\n    case \"string\":\n      return \"string\";\n    case \"boolean\":\n      return \"boolean\";\n    case \"number\":\n      return Number.isInteger(value) ? \"integer\" : \"number\";\n    default:\n      return value === null ? \"null\" : undefined;\n  }\n}\n\nfunction isObjectOrArraySchema(schema: JsonValue): boolean {\n  if (!isPlainObject(schema)) return false;\n  return (\n    schema.type === \"object\" ||\n    schema.type === \"array\" ||\n    \"properties\" in schema ||\n    \"items\" in schema\n  );\n}\n\nfunction isNullSchema(schema: JsonValue): boolean {\n  return isPlainObject(schema) && schema.type === \"null\";\n}\n\n/** Collect literal values from a scalar branch expressed as `const` or `enum`. */\nfunction literalValues(schema: JsonValue): JsonValue[] | undefined {\n  if (!isPlainObject(schema)) return undefined;\n  if (\"const\" in schema) return [schema.const as JsonValue];\n  if (Array.isArray(schema.enum)) return schema.enum as JsonValue[];\n  return undefined;\n}\n\n/**\n * Recursively rewrite a JSON Schema node into the Gemini-compatible subset.\n */\nexport function sanitizeGeminiSchema(schema: JsonValue): JsonValue {\n  if (Array.isArray(schema)) {\n    return schema.map((entry) => sanitizeGeminiSchema(entry));\n  }\n  if (!isPlainObject(schema)) {\n    return schema;\n  }\n\n  const union = (schema.anyOf ?? schema.oneOf) as JsonValue | undefined;\n  if (Array.isArray(union)) {\n    return sanitizeUnion(schema, union);\n  }\n\n  const result: JsonObject = {};\n\n  // const -> enum (Gemini supports enum, not const).\n  if (\"const\" in schema && !(\"enum\" in schema)) {\n    const constValue = schema.const as JsonValue;\n    const inferred = jsonScalarType(constValue);\n    if (inferred && inferred !== \"null\") result.type = inferred;\n    result.enum = [constValue];\n  }\n\n  for (const [key, value] of Object.entries(schema)) {\n    if (key === \"const\") continue;\n    if (key === \"properties\" && isPlainObject(value)) {\n      const props: JsonObject = {};\n      for (const [propName, propSchema] of Object.entries(value)) {\n        props[propName] = sanitizeGeminiSchema(propSchema);\n      }\n      result.properties = props;\n      continue;\n    }\n    if (key === \"items\") {\n      result.items = sanitizeItems(value);\n      continue;\n    }\n    if (KEPT_SCHEMA_KEYWORDS.has(key)) {\n      result[key] = value;\n      continue;\n    }\n    // Everything else (the documented DROPPED_SCHEMA_KEYWORDS plus any unknown\n    // keyword) is omitted: the rule is simply \"keep only KEPT_SCHEMA_KEYWORDS\".\n  }\n\n  inferContainerType(result);\n  pruneRequired(result);\n  return result;\n}\n\n/**\n * Resolve an `items` schema. Gemini's function-declaration schema expects a\n * single `items` schema, so a tuple-form `items` (array of schemas) is collapsed\n * to its most expressive (object/array) entry, falling back to the first entry.\n */\nfunction sanitizeItems(items: JsonValue): JsonValue {\n  if (!Array.isArray(items)) return sanitizeGeminiSchema(items);\n  const sanitized = items.map((entry) => sanitizeGeminiSchema(entry));\n  const objectOrArray = sanitized.find((entry) => isObjectOrArraySchema(entry));\n  return objectOrArray ?? sanitized[0] ?? { type: \"string\" };\n}\n\n/**\n * Gemini resolves function arguments more reliably when container nodes carry an\n * explicit `type`. Infer it when omitted: a node with `properties`/`required` is\n * an object, and a node with `items` is an array.\n */\nfunction inferContainerType(schema: JsonObject): void {\n  if (schema.type !== undefined) return;\n  if (isPlainObject(schema.properties) || Array.isArray(schema.required)) {\n    schema.type = \"object\";\n  } else if (schema.items !== undefined) {\n    schema.type = \"array\";\n  }\n}\n\n/** Resolve an `anyOf`/`oneOf` union node into the Gemini-compatible subset. */\nfunction sanitizeUnion(parent: JsonObject, branches: JsonValue[]): JsonValue {\n  const sanitizedBranches = branches.map((branch) => sanitizeGeminiSchema(branch));\n  const nullable =\n    branches.some((branch) => isNullSchema(branch)) || parent.nullable === true;\n  const nonNull = sanitizedBranches.filter((branch) => !isNullSchema(branch));\n\n  const carryDescription = (target: JsonValue): JsonValue => {\n    if (isPlainObject(target) && typeof parent.description === \"string\" && !(\"description\" in target)) {\n      target.description = parent.description;\n    }\n    if (nullable && isPlainObject(target)) target.nullable = true;\n    return target;\n  };\n\n  // Core fix: if any branch is an object/array schema, collapse to the first\n  // such branch (Gemini rejects unions whose branch is a complex object). For\n  // the TypeBox `Type.Union([Type.Object(...), Type.String()])` pattern this is\n  // the object branch; a union of two distinct object shapes keeps only the\n  // first, so the others' properties are intentionally dropped.\n  const objectOrArray = nonNull.find((branch) => isObjectOrArraySchema(branch));\n  if (objectOrArray) {\n    return carryDescription(objectOrArray);\n  }\n\n  if (nonNull.length === 0) {\n    return carryDescription({ type: \"string\" });\n  }\n  if (nonNull.length === 1) {\n    return carryDescription(nonNull[0]);\n  }\n\n  // All-scalar union: prefer an `enum` when every branch is a literal/enum of\n  // one underlying type; otherwise keep a scalar `anyOf` (Gemini accepts it).\n  const literals: JsonValue[] = [];\n  const scalarTypes = new Set<string>();\n  let allLiteral = true;\n  for (const branch of nonNull) {\n    const values = literalValues(branch);\n    if (!values) {\n      allLiteral = false;\n      break;\n    }\n    for (const value of values) {\n      literals.push(value);\n      const inferred = jsonScalarType(value);\n      if (inferred) scalarTypes.add(inferred);\n    }\n  }\n  if (allLiteral && literals.length > 0) {\n    const collapsed: JsonObject = { enum: literals };\n    if (scalarTypes.size === 1) collapsed.type = [...scalarTypes][0];\n    return carryDescription(collapsed);\n  }\n\n  const node: JsonObject = { anyOf: nonNull };\n  return carryDescription(node);\n}\n\n/** Drop `required` entries that are not present under `properties`. */\nfunction pruneRequired(schema: JsonObject): void {\n  if (!Array.isArray(schema.required)) return;\n  const properties = isPlainObject(schema.properties) ? schema.properties : undefined;\n  if (!properties) {\n    delete schema.required;\n    return;\n  }\n  const filtered = schema.required.filter(\n    (key): key is string => typeof key === \"string\" && key in properties,\n  );\n  if (filtered.length > 0) schema.required = filtered;\n  else delete schema.required;\n}\n\n/** Sanitize a single tool's `parameters` schema in place-safe (returns new value). */\nfunction sanitizeToolParameters(parameters: JsonValue): JsonValue {\n  return sanitizeGeminiSchema(parameters);\n}\n\n/**\n * Sanitize an outbound provider payload for GitHub Copilot Gemini models.\n *\n * Returns the payload unchanged for any other provider/model, or when the\n * payload has no sanitizable `tools`. Mirrors the `onPayload`\n * `(payload: unknown) => unknown` contract used elsewhere in the SDK wiring.\n */\nexport function sanitizeCopilotGeminiPayload(\n  payload: unknown,\n  model: Pick<Model<Api>, \"provider\" | \"api\" | \"id\">,\n): unknown {\n  if (!isCopilotGeminiModel(model)) return payload;\n  if (!isPlainObject(payload as JsonValue)) return payload;\n  const payloadObject = payload as JsonObject;\n  const tools = payloadObject.tools;\n  if (!Array.isArray(tools) || tools.length === 0) return payload;\n\n  let mutated = false;\n  const sanitizedTools = tools.map((tool) => {\n    if (!isPlainObject(tool)) return tool;\n    // OpenAI chat-completions tool shape: { type: \"function\", function: { parameters } }\n    if (isPlainObject(tool.function) && tool.function.parameters !== undefined) {\n      mutated = true;\n      return {\n        ...tool,\n        function: {\n          ...tool.function,\n          parameters: sanitizeToolParameters(tool.function.parameters),\n        },\n      };\n    }\n    // Defensive: flat tool shape { name, parameters }.\n    if (tool.parameters !== undefined) {\n      mutated = true;\n      return { ...tool, parameters: sanitizeToolParameters(tool.parameters) };\n    }\n    return tool;\n  });\n\n  if (!mutated) return payload;\n  return { ...payloadObject, tools: sanitizedTools };\n}\n"]}

package/dist/core/copilot-gemini-payload-sanitizer.js

@@ -0,0 +1,296 @@
+/** JSON Schema keywords Gemini's function-declaration schema honors. */
+const KEPT_SCHEMA_KEYWORDS = new Set([
+    "type",
+    "description",
+    "enum",
+    "properties",
+    "required",
+    "items",
+    "nullable",
+]);
+/**
+ * Keywords that are dropped because CAPI strips them before Google and/or
+ * Gemini's function schema rejects them. Listing them explicitly keeps the
+ * intent auditable; any keyword not in {@link KEPT_SCHEMA_KEYWORDS} and not a
+ * union keyword is dropped regardless. Exported so the documented drop-list has
+ * a real consumer (and stays linted) without a per-iteration `void` reference.
+ */
+export const DROPPED_SCHEMA_KEYWORDS = new Set([
+    "$schema",
+    "$id",
+    "$ref",
+    "$defs",
+    "$comment",
+    "definitions",
+    "patternProperties",
+    "propertyNames",
+    "unevaluatedProperties",
+    "additionalProperties",
+    "additionalItems",
+    "unevaluatedItems",
+    "allOf",
+    "not",
+    "if",
+    "then",
+    "else",
+    "format",
+    "pattern",
+    "minLength",
+    "maxLength",
+    "minItems",
+    "maxItems",
+    "uniqueItems",
+    "minimum",
+    "maximum",
+    "exclusiveMinimum",
+    "exclusiveMaximum",
+    "multipleOf",
+    "minProperties",
+    "maxProperties",
+    "default",
+    "examples",
+    "title",
+    "readOnly",
+    "writeOnly",
+    "deprecated",
+    "contentEncoding",
+    "contentMediaType",
+]);
+function isPlainObject(value) {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+/**
+ * Whether this model is a GitHub Copilot Gemini model routed through the
+ * OpenAI-completions CAPI path that needs schema sanitization.
+ */
+export function isCopilotGeminiModel(model) {
+    return (model.provider === "github-copilot" &&
+        model.api === "openai-completions" &&
+        /(^|[/-])gemini/i.test(model.id));
+}
+function jsonScalarType(value) {
+    switch (typeof value) {
+        case "string":
+            return "string";
+        case "boolean":
+            return "boolean";
+        case "number":
+            return Number.isInteger(value) ? "integer" : "number";
+        default:
+            return value === null ? "null" : undefined;
+    }
+}
+function isObjectOrArraySchema(schema) {
+    if (!isPlainObject(schema))
+        return false;
+    return (schema.type === "object" ||
+        schema.type === "array" ||
+        "properties" in schema ||
+        "items" in schema);
+}
+function isNullSchema(schema) {
+    return isPlainObject(schema) && schema.type === "null";
+}
+/** Collect literal values from a scalar branch expressed as `const` or `enum`. */
+function literalValues(schema) {
+    if (!isPlainObject(schema))
+        return undefined;
+    if ("const" in schema)
+        return [schema.const];
+    if (Array.isArray(schema.enum))
+        return schema.enum;
+    return undefined;
+}
+/**
+ * Recursively rewrite a JSON Schema node into the Gemini-compatible subset.
+ */
+export function sanitizeGeminiSchema(schema) {
+    if (Array.isArray(schema)) {
+        return schema.map((entry) => sanitizeGeminiSchema(entry));
+    }
+    if (!isPlainObject(schema)) {
+        return schema;
+    }
+    const union = (schema.anyOf ?? schema.oneOf);
+    if (Array.isArray(union)) {
+        return sanitizeUnion(schema, union);
+    }
+    const result = {};
+    // const -> enum (Gemini supports enum, not const).
+    if ("const" in schema && !("enum" in schema)) {
+        const constValue = schema.const;
+        const inferred = jsonScalarType(constValue);
+        if (inferred && inferred !== "null")
+            result.type = inferred;
+        result.enum = [constValue];
+    }
+    for (const [key, value] of Object.entries(schema)) {
+        if (key === "const")
+            continue;
+        if (key === "properties" && isPlainObject(value)) {
+            const props = {};
+            for (const [propName, propSchema] of Object.entries(value)) {
+                props[propName] = sanitizeGeminiSchema(propSchema);
+            }
+            result.properties = props;
+            continue;
+        }
+        if (key === "items") {
+            result.items = sanitizeItems(value);
+            continue;
+        }
+        if (KEPT_SCHEMA_KEYWORDS.has(key)) {
+            result[key] = value;
+            continue;
+        }
+        // Everything else (the documented DROPPED_SCHEMA_KEYWORDS plus any unknown
+        // keyword) is omitted: the rule is simply "keep only KEPT_SCHEMA_KEYWORDS".
+    }
+    inferContainerType(result);
+    pruneRequired(result);
+    return result;
+}
+/**
+ * Resolve an `items` schema. Gemini's function-declaration schema expects a
+ * single `items` schema, so a tuple-form `items` (array of schemas) is collapsed
+ * to its most expressive (object/array) entry, falling back to the first entry.
+ */
+function sanitizeItems(items) {
+    if (!Array.isArray(items))
+        return sanitizeGeminiSchema(items);
+    const sanitized = items.map((entry) => sanitizeGeminiSchema(entry));
+    const objectOrArray = sanitized.find((entry) => isObjectOrArraySchema(entry));
+    return objectOrArray ?? sanitized[0] ?? { type: "string" };
+}
+/**
+ * Gemini resolves function arguments more reliably when container nodes carry an
+ * explicit `type`. Infer it when omitted: a node with `properties`/`required` is
+ * an object, and a node with `items` is an array.
+ */
+function inferContainerType(schema) {
+    if (schema.type !== undefined)
+        return;
+    if (isPlainObject(schema.properties) || Array.isArray(schema.required)) {
+        schema.type = "object";
+    }
+    else if (schema.items !== undefined) {
+        schema.type = "array";
+    }
+}
+/** Resolve an `anyOf`/`oneOf` union node into the Gemini-compatible subset. */
+function sanitizeUnion(parent, branches) {
+    const sanitizedBranches = branches.map((branch) => sanitizeGeminiSchema(branch));
+    const nullable = branches.some((branch) => isNullSchema(branch)) || parent.nullable === true;
+    const nonNull = sanitizedBranches.filter((branch) => !isNullSchema(branch));
+    const carryDescription = (target) => {
+        if (isPlainObject(target) && typeof parent.description === "string" && !("description" in target)) {
+            target.description = parent.description;
+        }
+        if (nullable && isPlainObject(target))
+            target.nullable = true;
+        return target;
+    };
+    // Core fix: if any branch is an object/array schema, collapse to the first
+    // such branch (Gemini rejects unions whose branch is a complex object). For
+    // the TypeBox `Type.Union([Type.Object(...), Type.String()])` pattern this is
+    // the object branch; a union of two distinct object shapes keeps only the
+    // first, so the others' properties are intentionally dropped.
+    const objectOrArray = nonNull.find((branch) => isObjectOrArraySchema(branch));
+    if (objectOrArray) {
+        return carryDescription(objectOrArray);
+    }
+    if (nonNull.length === 0) {
+        return carryDescription({ type: "string" });
+    }
+    if (nonNull.length === 1) {
+        return carryDescription(nonNull[0]);
+    }
+    // All-scalar union: prefer an `enum` when every branch is a literal/enum of
+    // one underlying type; otherwise keep a scalar `anyOf` (Gemini accepts it).
+    const literals = [];
+    const scalarTypes = new Set();
+    let allLiteral = true;
+    for (const branch of nonNull) {
+        const values = literalValues(branch);
+        if (!values) {
+            allLiteral = false;
+            break;
+        }
+        for (const value of values) {
+            literals.push(value);
+            const inferred = jsonScalarType(value);
+            if (inferred)
+                scalarTypes.add(inferred);
+        }
+    }
+    if (allLiteral && literals.length > 0) {
+        const collapsed = { enum: literals };
+        if (scalarTypes.size === 1)
+            collapsed.type = [...scalarTypes][0];
+        return carryDescription(collapsed);
+    }
+    const node = { anyOf: nonNull };
+    return carryDescription(node);
+}
+/** Drop `required` entries that are not present under `properties`. */
+function pruneRequired(schema) {
+    if (!Array.isArray(schema.required))
+        return;
+    const properties = isPlainObject(schema.properties) ? schema.properties : undefined;
+    if (!properties) {
+        delete schema.required;
+        return;
+    }
+    const filtered = schema.required.filter((key) => typeof key === "string" && key in properties);
+    if (filtered.length > 0)
+        schema.required = filtered;
+    else
+        delete schema.required;
+}
+/** Sanitize a single tool's `parameters` schema in place-safe (returns new value). */
+function sanitizeToolParameters(parameters) {
+    return sanitizeGeminiSchema(parameters);
+}
+/**
+ * Sanitize an outbound provider payload for GitHub Copilot Gemini models.
+ *
+ * Returns the payload unchanged for any other provider/model, or when the
+ * payload has no sanitizable `tools`. Mirrors the `onPayload`
+ * `(payload: unknown) => unknown` contract used elsewhere in the SDK wiring.
+ */
+export function sanitizeCopilotGeminiPayload(payload, model) {
+    if (!isCopilotGeminiModel(model))
+        return payload;
+    if (!isPlainObject(payload))
+        return payload;
+    const payloadObject = payload;
+    const tools = payloadObject.tools;
+    if (!Array.isArray(tools) || tools.length === 0)
+        return payload;
+    let mutated = false;
+    const sanitizedTools = tools.map((tool) => {
+        if (!isPlainObject(tool))
+            return tool;
+        // OpenAI chat-completions tool shape: { type: "function", function: { parameters } }
+        if (isPlainObject(tool.function) && tool.function.parameters !== undefined) {
+            mutated = true;
+            return {
+                ...tool,
+                function: {
+                    ...tool.function,
+                    parameters: sanitizeToolParameters(tool.function.parameters),
+                },
+            };
+        }
+        // Defensive: flat tool shape { name, parameters }.
+        if (tool.parameters !== undefined) {
+            mutated = true;
+            return { ...tool, parameters: sanitizeToolParameters(tool.parameters) };
+        }
+        return tool;
+    });
+    if (!mutated)
+        return payload;
+    return { ...payloadObject, tools: sanitizedTools };
+}
+//# sourceMappingURL=copilot-gemini-payload-sanitizer.js.map

package/dist/core/copilot-gemini-payload-sanitizer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"copilot-gemini-payload-sanitizer.js","sourceRoot":"","sources":["../../src/core/copilot-gemini-payload-sanitizer.ts"],"names":[],"mappings":"AA8CA,wEAAwE;AACxE,MAAM,oBAAoB,GAAG,IAAI,GAAG,CAAS;IAC3C,MAAM;IACN,aAAa;IACb,MAAM;IACN,YAAY;IACZ,UAAU;IACV,OAAO;IACP,UAAU;CACX,CAAC,CAAC;AAEH;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG,IAAI,GAAG,CAAS;IACrD,SAAS;IACT,KAAK;IACL,MAAM;IACN,OAAO;IACP,UAAU;IACV,aAAa;IACb,mBAAmB;IACnB,eAAe;IACf,uBAAuB;IACvB,sBAAsB;IACtB,iBAAiB;IACjB,kBAAkB;IAClB,OAAO;IACP,KAAK;IACL,IAAI;IACJ,MAAM;IACN,MAAM;IACN,QAAQ;IACR,SAAS;IACT,WAAW;IACX,WAAW;IACX,UAAU;IACV,UAAU;IACV,aAAa;IACb,SAAS;IACT,SAAS;IACT,kBAAkB;IAClB,kBAAkB;IAClB,YAAY;IACZ,eAAe;IACf,eAAe;IACf,SAAS;IACT,UAAU;IACV,OAAO;IACP,UAAU;IACV,WAAW;IACX,YAAY;IACZ,iBAAiB;IACjB,kBAAkB;CACnB,CAAC,CAAC;AAEH,SAAS,aAAa,CAAC,KAA4B;IACjD,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,oBAAoB,CAAC,KAAkD;IACrF,OAAO,CACL,KAAK,CAAC,QAAQ,KAAK,gBAAgB;QACnC,KAAK,CAAC,GAAG,KAAK,oBAAoB;QAClC,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CACjC,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,KAAgB;IACtC,QAAQ,OAAO,KAAK,EAAE,CAAC;QACrB,KAAK,QAAQ;YACX,OAAO,QAAQ,CAAC;QAClB,KAAK,SAAS;YACZ,OAAO,SAAS,CAAC;QACnB,KAAK,QAAQ;YACX,OAAO,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC;QACxD;YACE,OAAO,KAAK,KAAK,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;IAC/C,CAAC;AACH,CAAC;AAED,SAAS,qBAAqB,CAAC,MAAiB;IAC9C,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC;QAAE,OAAO,KAAK,CAAC;IACzC,OAAO,CACL,MAAM,CAAC,IAAI,KAAK,QAAQ;QACxB,MAAM,CAAC,IAAI,KAAK,OAAO;QACvB,YAAY,IAAI,MAAM;QACtB,OAAO,IAAI,MAAM,CAClB,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,MAAiB;IACrC,OAAO,aAAa,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC;AACzD,CAAC;AAED,kFAAkF;AAClF,SAAS,aAAa,CAAC,MAAiB;IACtC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC;QAAE,OAAO,SAAS,CAAC;IAC7C,IAAI,OAAO,IAAI,MAAM;QAAE,OAAO,CAAC,MAAM,CAAC,KAAkB,CAAC,CAAC;IAC1D,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC;QAAE,OAAO,MAAM,CAAC,IAAmB,CAAC;IAClE,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAAC,MAAiB;IACpD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC;IAC5D,CAAC;IACD,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,EAAE,CAAC;QAC3B,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,MAAM,KAAK,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAA0B,CAAC;IACtE,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,aAAa,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IACtC,CAAC;IAED,MAAM,MAAM,GAAe,EAAE,CAAC;IAE9B,mDAAmD;IACnD,IAAI,OAAO,IAAI,MAAM,IAAI,CAAC,CAAC,MAAM,IAAI,MAAM,CAAC,EAAE,CAAC;QAC7C,MAAM,UAAU,GAAG,MAAM,CAAC,KAAkB,CAAC;QAC7C,MAAM,QAAQ,GAAG,cAAc,CAAC,UAAU,CAAC,CAAC;QAC5C,IAAI,QAAQ,IAAI,QAAQ,KAAK,MAAM;YAAE,MAAM,CAAC,IAAI,GAAG,QAAQ,CAAC;QAC5D,MAAM,CAAC,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC;IAC7B,CAAC;IAED,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAClD,IAAI,GAAG,KAAK,OAAO;YAAE,SAAS;QAC9B,IAAI,GAAG,KAAK,YAAY,IAAI,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;YACjD,MAAM,KAAK,GAAe,EAAE,CAAC;YAC7B,KAAK,MAAM,CAAC,QAAQ,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC3D,KAAK,CAAC,QAAQ,CAAC,GAAG,oBAAoB,CAAC,UAAU,CAAC,CAAC;YACrD,CAAC;YACD,MAAM,CAAC,UAAU,GAAG,KAAK,CAAC;YAC1B,SAAS;QACX,CAAC;QACD,IAAI,GAAG,KAAK,OAAO,EAAE,CAAC;YACpB,MAAM,CAAC,KAAK,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;YACpC,SAAS;QACX,CAAC;QACD,IAAI,oBAAoB,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YAClC,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;YACpB,SAAS;QACX,CAAC;QACD,2EAA2E;QAC3E,4EAA4E;IAC9E,CAAC;IAED,kBAAkB,CAAC,MAAM,CAAC,CAAC;IAC3B,aAAa,CAAC,MAAM,CAAC,CAAC;IACtB,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;GAIG;AACH,SAAS,aAAa,CAAC,KAAgB;IACrC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO,oBAAoB,CAAC,KAAK,CAAC,CAAC;IAC9D,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC;IACpE,MAAM,aAAa,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC,CAAC;IAC9E,OAAO,aAAa,IAAI,SAAS,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;AAC7D,CAAC;AAED;;;;GAIG;AACH,SAAS,kBAAkB,CAAC,MAAkB;IAC5C,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS;QAAE,OAAO;IACtC,IAAI,aAAa,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;QACvE,MAAM,CAAC,IAAI,GAAG,QAAQ,CAAC;IACzB,CAAC;SAAM,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QACtC,MAAM,CAAC,IAAI,GAAG,OAAO,CAAC;IACxB,CAAC;AACH,CAAC;AAED,+EAA+E;AAC/E,SAAS,aAAa,CAAC,MAAkB,EAAE,QAAqB;IAC9D,MAAM,iBAAiB,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC,CAAC;IACjF,MAAM,QAAQ,GACZ,QAAQ,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,IAAI,MAAM,CAAC,QAAQ,KAAK,IAAI,CAAC;IAC9E,MAAM,OAAO,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC;IAE5E,MAAM,gBAAgB,GAAG,CAAC,MAAiB,EAAa,EAAE;QACxD,IAAI,aAAa,CAAC,MAAM,CAAC,IAAI,OAAO,MAAM,CAAC,WAAW,KAAK,QAAQ,IAAI,CAAC,CAAC,aAAa,IAAI,MAAM,CAAC,EAAE,CAAC;YAClG,MAAM,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QAC1C,CAAC;QACD,IAAI,QAAQ,IAAI,aAAa,CAAC,MAAM,CAAC;YAAE,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC;QAC9D,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;IAEF,2EAA2E;IAC3E,4EAA4E;IAC5E,8EAA8E;IAC9E,0EAA0E;IAC1E,8DAA8D;IAC9D,MAAM,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC,CAAC;IAC9E,IAAI,aAAa,EAAE,CAAC;QAClB,OAAO,gBAAgB,CAAC,aAAa,CAAC,CAAC;IACzC,CAAC;IAED,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,gBAAgB,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC9C,CAAC;IACD,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IACtC,CAAC;IAED,4EAA4E;IAC5E,4EAA4E;IAC5E,MAAM,QAAQ,GAAgB,EAAE,CAAC;IACjC,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;IACtC,IAAI,UAAU,GAAG,IAAI,CAAC;IACtB,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;QACrC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,UAAU,GAAG,KAAK,CAAC;YACnB,MAAM;QACR,CAAC;QACD,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,MAAM,QAAQ,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;YACvC,IAAI,QAAQ;gBAAE,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IACD,IAAI,UAAU,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtC,MAAM,SAAS,GAAe,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;QACjD,IAAI,WAAW,CAAC,IAAI,KAAK,CAAC;YAAE,SAAS,CAAC,IAAI,GAAG,CAAC,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;QACjE,OAAO,gBAAgB,CAAC,SAAS,CAAC,CAAC;IACrC,CAAC;IAED,MAAM,IAAI,GAAe,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;IAC5C,OAAO,gBAAgB,CAAC,IAAI,CAAC,CAAC;AAChC,CAAC;AAED,uEAAuE;AACvE,SAAS,aAAa,CAAC,MAAkB;IACvC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC;QAAE,OAAO;IAC5C,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;IACpF,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,MAAM,CAAC,QAAQ,CAAC;QACvB,OAAO;IACT,CAAC;IACD,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CACrC,CAAC,GAAG,EAAiB,EAAE,CAAC,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,IAAI,UAAU,CACrE,CAAC;IACF,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC;QAAE,MAAM,CAAC,QAAQ,GAAG,QAAQ,CAAC;;QAC/C,OAAO,MAAM,CAAC,QAAQ,CAAC;AAC9B,CAAC;AAED,sFAAsF;AACtF,SAAS,sBAAsB,CAAC,UAAqB;IACnD,OAAO,oBAAoB,CAAC,UAAU,CAAC,CAAC;AAC1C,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,4BAA4B,CAC1C,OAAgB,EAChB,KAAkD;IAElD,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC;QAAE,OAAO,OAAO,CAAC;IACjD,IAAI,CAAC,aAAa,CAAC,OAAoB,CAAC;QAAE,OAAO,OAAO,CAAC;IACzD,MAAM,aAAa,GAAG,OAAqB,CAAC;IAC5C,MAAM,KAAK,GAAG,aAAa,CAAC,KAAK,CAAC;IAClC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,OAAO,CAAC;IAEhE,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,MAAM,cAAc,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACxC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;YAAE,OAAO,IAAI,CAAC;QACtC,qFAAqF;QACrF,IAAI,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;YAC3E,OAAO,GAAG,IAAI,CAAC;YACf,OAAO;gBACL,GAAG,IAAI;gBACP,QAAQ,EAAE;oBACR,GAAG,IAAI,CAAC,QAAQ;oBAChB,UAAU,EAAE,sBAAsB,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC;iBAC7D;aACF,CAAC;QACJ,CAAC;QACD,mDAAmD;QACnD,IAAI,IAAI,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;YAClC,OAAO,GAAG,IAAI,CAAC;YACf,OAAO,EAAE,GAAG,IAAI,EAAE,UAAU,EAAE,sBAAsB,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;QAC1E,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,OAAO;QAAE,OAAO,OAAO,CAAC;IAC7B,OAAO,EAAE,GAAG,aAAa,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC;AACrD,CAAC","sourcesContent":["import type { Api, Model } from \"@earendil-works/pi-ai\";\n\n/**\n * Sanitizes outbound OpenAI-compatible request payloads for GitHub Copilot\n * Gemini models so their tool/function JSON Schemas survive translation to\n * Google's GenAI `FunctionDeclaration` schema.\n *\n * Why this exists\n * ---------------\n * `github-copilot` Gemini models (e.g. `gemini-3.1-pro-preview`) are served\n * through Copilot's CAPI gateway at `api.*.githubcopilot.com` using the\n * `openai-completions` API. CAPI receives the OpenAI chat-completions request\n * and translates it into a Google GenAI `GenerateContent` request. During that\n * translation CAPI forwards JSON Schema `anyOf`/`oneOf` verbatim into the Gemini\n * `FunctionDeclaration` schema. Gemini's function-declaration schema rejects an\n * `anyOf`/`oneOf` whose branch is a complex *object* schema, so Google returns\n * HTTP 400 and CAPI relabels it `{\"error\":{\"code\":\"invalid_request_body\"}}`.\n *\n * Atomic's bundled tools (notably the `workflow` tool) use the TypeBox\n * `Type.Union([Type.Object(...), Type.String()])` pattern for fields like\n * `task`, `chain`, and `parallel`, which emit exactly that construct. Because\n * those tools are present in normal chat turns, every Gemini request fails with\n * `400 invalid request body` until the schema is sanitized.\n *\n * What it does\n * ------------\n * Recursively rewrites each tool's `parameters` JSON Schema into the reduced\n * subset that CAPI/Gemini actually honors (`type`, `description`, `enum`,\n * `properties`, `required`, `items`, `nullable`, and scalar-only `anyOf`):\n *  - Resolves `anyOf`/`oneOf` that contain an object/array branch to the most\n *    expressive object/array branch (the core fix), preserving `description`\n *    and collapsing a `\"null\"` branch into `nullable: true`.\n *  - Converts `const` (and `anyOf` of `const`/`enum` scalars) into `enum`,\n *    which Gemini supports, so `Type.Literal(...)` unions keep their constraint.\n *  - Drops JSON Schema keywords CAPI strips or Gemini rejects (`$schema`,\n *    `$ref`, `$defs`, `patternProperties`, `additionalProperties`, `allOf`,\n *    `not`, `format`, `pattern`, numeric/length bounds, `default`, etc.).\n *  - Filters `required` down to keys still present under `properties`.\n *\n * The transform is gated to GitHub Copilot Gemini models only, so it never\n * changes payloads for any currently-working provider/model.\n */\n\ntype JsonObject = { [key: string]: JsonValue };\ntype JsonValue = string | number | boolean | null | JsonValue[] | JsonObject;\n\n/** JSON Schema keywords Gemini's function-declaration schema honors. */\nconst KEPT_SCHEMA_KEYWORDS = new Set<string>([\n  \"type\",\n  \"description\",\n  \"enum\",\n  \"properties\",\n  \"required\",\n  \"items\",\n  \"nullable\",\n]);\n\n/**\n * Keywords that are dropped because CAPI strips them before Google and/or\n * Gemini's function schema rejects them. Listing them explicitly keeps the\n * intent auditable; any keyword not in {@link KEPT_SCHEMA_KEYWORDS} and not a\n * union keyword is dropped regardless. Exported so the documented drop-list has\n * a real consumer (and stays linted) without a per-iteration `void` reference.\n */\nexport const DROPPED_SCHEMA_KEYWORDS = new Set<string>([\n  \"$schema\",\n  \"$id\",\n  \"$ref\",\n  \"$defs\",\n  \"$comment\",\n  \"definitions\",\n  \"patternProperties\",\n  \"propertyNames\",\n  \"unevaluatedProperties\",\n  \"additionalProperties\",\n  \"additionalItems\",\n  \"unevaluatedItems\",\n  \"allOf\",\n  \"not\",\n  \"if\",\n  \"then\",\n  \"else\",\n  \"format\",\n  \"pattern\",\n  \"minLength\",\n  \"maxLength\",\n  \"minItems\",\n  \"maxItems\",\n  \"uniqueItems\",\n  \"minimum\",\n  \"maximum\",\n  \"exclusiveMinimum\",\n  \"exclusiveMaximum\",\n  \"multipleOf\",\n  \"minProperties\",\n  \"maxProperties\",\n  \"default\",\n  \"examples\",\n  \"title\",\n  \"readOnly\",\n  \"writeOnly\",\n  \"deprecated\",\n  \"contentEncoding\",\n  \"contentMediaType\",\n]);\n\nfunction isPlainObject(value: JsonValue | undefined): value is JsonObject {\n  return typeof value === \"object\" && value !== null && !Array.isArray(value);\n}\n\n/**\n * Whether this model is a GitHub Copilot Gemini model routed through the\n * OpenAI-completions CAPI path that needs schema sanitization.\n */\nexport function isCopilotGeminiModel(model: Pick<Model<Api>, \"provider\" | \"api\" | \"id\">): boolean {\n  return (\n    model.provider === \"github-copilot\" &&\n    model.api === \"openai-completions\" &&\n    /(^|[/-])gemini/i.test(model.id)\n  );\n}\n\nfunction jsonScalarType(value: JsonValue): string | undefined {\n  switch (typeof value) {\n    case \"string\":\n      return \"string\";\n    case \"boolean\":\n      return \"boolean\";\n    case \"number\":\n      return Number.isInteger(value) ? \"integer\" : \"number\";\n    default:\n      return value === null ? \"null\" : undefined;\n  }\n}\n\nfunction isObjectOrArraySchema(schema: JsonValue): boolean {\n  if (!isPlainObject(schema)) return false;\n  return (\n    schema.type === \"object\" ||\n    schema.type === \"array\" ||\n    \"properties\" in schema ||\n    \"items\" in schema\n  );\n}\n\nfunction isNullSchema(schema: JsonValue): boolean {\n  return isPlainObject(schema) && schema.type === \"null\";\n}\n\n/** Collect literal values from a scalar branch expressed as `const` or `enum`. */\nfunction literalValues(schema: JsonValue): JsonValue[] | undefined {\n  if (!isPlainObject(schema)) return undefined;\n  if (\"const\" in schema) return [schema.const as JsonValue];\n  if (Array.isArray(schema.enum)) return schema.enum as JsonValue[];\n  return undefined;\n}\n\n/**\n * Recursively rewrite a JSON Schema node into the Gemini-compatible subset.\n */\nexport function sanitizeGeminiSchema(schema: JsonValue): JsonValue {\n  if (Array.isArray(schema)) {\n    return schema.map((entry) => sanitizeGeminiSchema(entry));\n  }\n  if (!isPlainObject(schema)) {\n    return schema;\n  }\n\n  const union = (schema.anyOf ?? schema.oneOf) as JsonValue | undefined;\n  if (Array.isArray(union)) {\n    return sanitizeUnion(schema, union);\n  }\n\n  const result: JsonObject = {};\n\n  // const -> enum (Gemini supports enum, not const).\n  if (\"const\" in schema && !(\"enum\" in schema)) {\n    const constValue = schema.const as JsonValue;\n    const inferred = jsonScalarType(constValue);\n    if (inferred && inferred !== \"null\") result.type = inferred;\n    result.enum = [constValue];\n  }\n\n  for (const [key, value] of Object.entries(schema)) {\n    if (key === \"const\") continue;\n    if (key === \"properties\" && isPlainObject(value)) {\n      const props: JsonObject = {};\n      for (const [propName, propSchema] of Object.entries(value)) {\n        props[propName] = sanitizeGeminiSchema(propSchema);\n      }\n      result.properties = props;\n      continue;\n    }\n    if (key === \"items\") {\n      result.items = sanitizeItems(value);\n      continue;\n    }\n    if (KEPT_SCHEMA_KEYWORDS.has(key)) {\n      result[key] = value;\n      continue;\n    }\n    // Everything else (the documented DROPPED_SCHEMA_KEYWORDS plus any unknown\n    // keyword) is omitted: the rule is simply \"keep only KEPT_SCHEMA_KEYWORDS\".\n  }\n\n  inferContainerType(result);\n  pruneRequired(result);\n  return result;\n}\n\n/**\n * Resolve an `items` schema. Gemini's function-declaration schema expects a\n * single `items` schema, so a tuple-form `items` (array of schemas) is collapsed\n * to its most expressive (object/array) entry, falling back to the first entry.\n */\nfunction sanitizeItems(items: JsonValue): JsonValue {\n  if (!Array.isArray(items)) return sanitizeGeminiSchema(items);\n  const sanitized = items.map((entry) => sanitizeGeminiSchema(entry));\n  const objectOrArray = sanitized.find((entry) => isObjectOrArraySchema(entry));\n  return objectOrArray ?? sanitized[0] ?? { type: \"string\" };\n}\n\n/**\n * Gemini resolves function arguments more reliably when container nodes carry an\n * explicit `type`. Infer it when omitted: a node with `properties`/`required` is\n * an object, and a node with `items` is an array.\n */\nfunction inferContainerType(schema: JsonObject): void {\n  if (schema.type !== undefined) return;\n  if (isPlainObject(schema.properties) || Array.isArray(schema.required)) {\n    schema.type = \"object\";\n  } else if (schema.items !== undefined) {\n    schema.type = \"array\";\n  }\n}\n\n/** Resolve an `anyOf`/`oneOf` union node into the Gemini-compatible subset. */\nfunction sanitizeUnion(parent: JsonObject, branches: JsonValue[]): JsonValue {\n  const sanitizedBranches = branches.map((branch) => sanitizeGeminiSchema(branch));\n  const nullable =\n    branches.some((branch) => isNullSchema(branch)) || parent.nullable === true;\n  const nonNull = sanitizedBranches.filter((branch) => !isNullSchema(branch));\n\n  const carryDescription = (target: JsonValue): JsonValue => {\n    if (isPlainObject(target) && typeof parent.description === \"string\" && !(\"description\" in target)) {\n      target.description = parent.description;\n    }\n    if (nullable && isPlainObject(target)) target.nullable = true;\n    return target;\n  };\n\n  // Core fix: if any branch is an object/array schema, collapse to the first\n  // such branch (Gemini rejects unions whose branch is a complex object). For\n  // the TypeBox `Type.Union([Type.Object(...), Type.String()])` pattern this is\n  // the object branch; a union of two distinct object shapes keeps only the\n  // first, so the others' properties are intentionally dropped.\n  const objectOrArray = nonNull.find((branch) => isObjectOrArraySchema(branch));\n  if (objectOrArray) {\n    return carryDescription(objectOrArray);\n  }\n\n  if (nonNull.length === 0) {\n    return carryDescription({ type: \"string\" });\n  }\n  if (nonNull.length === 1) {\n    return carryDescription(nonNull[0]);\n  }\n\n  // All-scalar union: prefer an `enum` when every branch is a literal/enum of\n  // one underlying type; otherwise keep a scalar `anyOf` (Gemini accepts it).\n  const literals: JsonValue[] = [];\n  const scalarTypes = new Set<string>();\n  let allLiteral = true;\n  for (const branch of nonNull) {\n    const values = literalValues(branch);\n    if (!values) {\n      allLiteral = false;\n      break;\n    }\n    for (const value of values) {\n      literals.push(value);\n      const inferred = jsonScalarType(value);\n      if (inferred) scalarTypes.add(inferred);\n    }\n  }\n  if (allLiteral && literals.length > 0) {\n    const collapsed: JsonObject = { enum: literals };\n    if (scalarTypes.size === 1) collapsed.type = [...scalarTypes][0];\n    return carryDescription(collapsed);\n  }\n\n  const node: JsonObject = { anyOf: nonNull };\n  return carryDescription(node);\n}\n\n/** Drop `required` entries that are not present under `properties`. */\nfunction pruneRequired(schema: JsonObject): void {\n  if (!Array.isArray(schema.required)) return;\n  const properties = isPlainObject(schema.properties) ? schema.properties : undefined;\n  if (!properties) {\n    delete schema.required;\n    return;\n  }\n  const filtered = schema.required.filter(\n    (key): key is string => typeof key === \"string\" && key in properties,\n  );\n  if (filtered.length > 0) schema.required = filtered;\n  else delete schema.required;\n}\n\n/** Sanitize a single tool's `parameters` schema in place-safe (returns new value). */\nfunction sanitizeToolParameters(parameters: JsonValue): JsonValue {\n  return sanitizeGeminiSchema(parameters);\n}\n\n/**\n * Sanitize an outbound provider payload for GitHub Copilot Gemini models.\n *\n * Returns the payload unchanged for any other provider/model, or when the\n * payload has no sanitizable `tools`. Mirrors the `onPayload`\n * `(payload: unknown) => unknown` contract used elsewhere in the SDK wiring.\n */\nexport function sanitizeCopilotGeminiPayload(\n  payload: unknown,\n  model: Pick<Model<Api>, \"provider\" | \"api\" | \"id\">,\n): unknown {\n  if (!isCopilotGeminiModel(model)) return payload;\n  if (!isPlainObject(payload as JsonValue)) return payload;\n  const payloadObject = payload as JsonObject;\n  const tools = payloadObject.tools;\n  if (!Array.isArray(tools) || tools.length === 0) return payload;\n\n  let mutated = false;\n  const sanitizedTools = tools.map((tool) => {\n    if (!isPlainObject(tool)) return tool;\n    // OpenAI chat-completions tool shape: { type: \"function\", function: { parameters } }\n    if (isPlainObject(tool.function) && tool.function.parameters !== undefined) {\n      mutated = true;\n      return {\n        ...tool,\n        function: {\n          ...tool.function,\n          parameters: sanitizeToolParameters(tool.function.parameters),\n        },\n      };\n    }\n    // Defensive: flat tool shape { name, parameters }.\n    if (tool.parameters !== undefined) {\n      mutated = true;\n      return { ...tool, parameters: sanitizeToolParameters(tool.parameters) };\n    }\n    return tool;\n  });\n\n  if (!mutated) return payload;\n  return { ...payloadObject, tools: sanitizedTools };\n}\n"]}