@bastani/atomic 0.8.30 → 0.8.31-alpha.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (205) hide show
  1. package/CHANGELOG.md +29 -0
  2. package/README.md +12 -10
  3. package/dist/builtin/cursor/CHANGELOG.md +4 -0
  4. package/dist/builtin/cursor/package.json +2 -2
  5. package/dist/builtin/intercom/CHANGELOG.md +4 -0
  6. package/dist/builtin/intercom/package.json +2 -2
  7. package/dist/builtin/mcp/CHANGELOG.md +4 -0
  8. package/dist/builtin/mcp/package.json +3 -3
  9. package/dist/builtin/subagents/CHANGELOG.md +13 -0
  10. package/dist/builtin/subagents/agents/codebase-online-researcher.md +8 -8
  11. package/dist/builtin/subagents/agents/debugger.md +6 -6
  12. package/dist/builtin/subagents/package.json +4 -4
  13. package/dist/builtin/subagents/skills/effective-liteparse/SKILL.md +118 -0
  14. package/dist/builtin/subagents/skills/effective-liteparse/scripts/search.py +128 -0
  15. package/dist/builtin/subagents/skills/playwright-cli/SKILL.md +404 -0
  16. package/dist/builtin/subagents/skills/playwright-cli/references/element-attributes.md +23 -0
  17. package/dist/builtin/subagents/skills/playwright-cli/references/playwright-tests.md +39 -0
  18. package/dist/builtin/subagents/skills/playwright-cli/references/request-mocking.md +87 -0
  19. package/dist/builtin/subagents/skills/playwright-cli/references/running-code.md +241 -0
  20. package/dist/builtin/subagents/skills/playwright-cli/references/session-management.md +225 -0
  21. package/dist/builtin/subagents/skills/playwright-cli/references/spec-driven-testing.md +305 -0
  22. package/dist/builtin/subagents/skills/playwright-cli/references/storage-state.md +275 -0
  23. package/dist/builtin/subagents/skills/playwright-cli/references/test-generation.md +134 -0
  24. package/dist/builtin/subagents/skills/playwright-cli/references/tracing.md +139 -0
  25. package/dist/builtin/subagents/skills/playwright-cli/references/video-recording.md +143 -0
  26. package/dist/builtin/web-access/CHANGELOG.md +4 -0
  27. package/dist/builtin/web-access/package.json +2 -2
  28. package/dist/builtin/workflows/CHANGELOG.md +16 -0
  29. package/dist/builtin/workflows/README.md +4 -4
  30. package/dist/builtin/workflows/builtin/deep-research-codebase.ts +1 -1
  31. package/dist/builtin/workflows/builtin/goal.ts +2 -2
  32. package/dist/builtin/workflows/builtin/open-claude-design.ts +60 -57
  33. package/dist/builtin/workflows/builtin/ralph.ts +117 -14
  34. package/dist/builtin/workflows/builtin/shared-prompts.ts +1 -1
  35. package/dist/builtin/workflows/package.json +2 -2
  36. package/dist/builtin/workflows/skills/research-codebase/SKILL.md +1 -1
  37. package/dist/builtin/workflows/src/extension/workflow-schema.ts +3 -1
  38. package/dist/builtin/workflows/src/runs/foreground/stage-runner.ts +5 -0
  39. package/dist/builtin/workflows/src/runs/shared/model-fallback.ts +95 -8
  40. package/dist/builtin/workflows/src/shared/authoring-contract.d.ts +11 -0
  41. package/dist/cli/args.d.ts +1 -0
  42. package/dist/cli/args.d.ts.map +1 -1
  43. package/dist/cli/args.js +21 -1
  44. package/dist/cli/args.js.map +1 -1
  45. package/dist/cli/list-models.d.ts.map +1 -1
  46. package/dist/cli/list-models.js +2 -1
  47. package/dist/cli/list-models.js.map +1 -1
  48. package/dist/core/agent-session-services.d.ts +2 -0
  49. package/dist/core/agent-session-services.d.ts.map +1 -1
  50. package/dist/core/agent-session-services.js +2 -0
  51. package/dist/core/agent-session-services.js.map +1 -1
  52. package/dist/core/agent-session.d.ts +18 -0
  53. package/dist/core/agent-session.d.ts.map +1 -1
  54. package/dist/core/agent-session.js +182 -19
  55. package/dist/core/agent-session.js.map +1 -1
  56. package/dist/core/compaction/branch-summarization.d.ts.map +1 -1
  57. package/dist/core/compaction/branch-summarization.js +20 -5
  58. package/dist/core/compaction/branch-summarization.js.map +1 -1
  59. package/dist/core/compaction/context-compaction.d.ts.map +1 -1
  60. package/dist/core/compaction/context-compaction.js +14 -3
  61. package/dist/core/compaction/context-compaction.js.map +1 -1
  62. package/dist/core/context-window.d.ts +39 -0
  63. package/dist/core/context-window.d.ts.map +1 -0
  64. package/dist/core/context-window.js +99 -0
  65. package/dist/core/context-window.js.map +1 -0
  66. package/dist/core/copilot-errors.d.ts +9 -0
  67. package/dist/core/copilot-errors.d.ts.map +1 -0
  68. package/dist/core/copilot-errors.js +32 -0
  69. package/dist/core/copilot-errors.js.map +1 -0
  70. package/dist/core/copilot-model-catalog.d.ts +135 -0
  71. package/dist/core/copilot-model-catalog.d.ts.map +1 -0
  72. package/dist/core/copilot-model-catalog.js +257 -0
  73. package/dist/core/copilot-model-catalog.js.map +1 -0
  74. package/dist/core/export-html/template.js +10 -1
  75. package/dist/core/extensions/types.d.ts +3 -1
  76. package/dist/core/extensions/types.d.ts.map +1 -1
  77. package/dist/core/extensions/types.js.map +1 -1
  78. package/dist/core/model-registry.d.ts +10 -0
  79. package/dist/core/model-registry.d.ts.map +1 -1
  80. package/dist/core/model-registry.js +107 -4
  81. package/dist/core/model-registry.js.map +1 -1
  82. package/dist/core/model-resolver.d.ts.map +1 -1
  83. package/dist/core/model-resolver.js +4 -0
  84. package/dist/core/model-resolver.js.map +1 -1
  85. package/dist/core/project-trust.d.ts.map +1 -1
  86. package/dist/core/project-trust.js +2 -1
  87. package/dist/core/project-trust.js.map +1 -1
  88. package/dist/core/provider-attribution.d.ts.map +1 -1
  89. package/dist/core/provider-attribution.js +17 -7
  90. package/dist/core/provider-attribution.js.map +1 -1
  91. package/dist/core/sdk.d.ts +8 -0
  92. package/dist/core/sdk.d.ts.map +1 -1
  93. package/dist/core/sdk.js +58 -0
  94. package/dist/core/sdk.js.map +1 -1
  95. package/dist/core/session-manager.d.ts +8 -1
  96. package/dist/core/session-manager.d.ts.map +1 -1
  97. package/dist/core/session-manager.js +19 -3
  98. package/dist/core/session-manager.js.map +1 -1
  99. package/dist/core/settings-manager.d.ts +15 -0
  100. package/dist/core/settings-manager.d.ts.map +1 -1
  101. package/dist/core/settings-manager.js +124 -1
  102. package/dist/core/settings-manager.js.map +1 -1
  103. package/dist/core/system-prompt.d.ts.map +1 -1
  104. package/dist/core/system-prompt.js +1 -0
  105. package/dist/core/system-prompt.js.map +1 -1
  106. package/dist/core/tools/edit-diff.d.ts +1 -2
  107. package/dist/core/tools/edit-diff.d.ts.map +1 -1
  108. package/dist/core/tools/edit-diff.js +1 -2
  109. package/dist/core/tools/edit-diff.js.map +1 -1
  110. package/dist/index.d.ts +3 -1
  111. package/dist/index.d.ts.map +1 -1
  112. package/dist/index.js +2 -0
  113. package/dist/index.js.map +1 -1
  114. package/dist/main.d.ts.map +1 -1
  115. package/dist/main.js +24 -1
  116. package/dist/main.js.map +1 -1
  117. package/dist/modes/index.d.ts +1 -1
  118. package/dist/modes/index.d.ts.map +1 -1
  119. package/dist/modes/index.js.map +1 -1
  120. package/dist/modes/interactive/components/config-selector.d.ts.map +1 -1
  121. package/dist/modes/interactive/components/config-selector.js +5 -7
  122. package/dist/modes/interactive/components/config-selector.js.map +1 -1
  123. package/dist/modes/interactive/components/context-window-selector.d.ts +53 -0
  124. package/dist/modes/interactive/components/context-window-selector.d.ts.map +1 -0
  125. package/dist/modes/interactive/components/context-window-selector.js +136 -0
  126. package/dist/modes/interactive/components/context-window-selector.js.map +1 -0
  127. package/dist/modes/interactive/components/model-selector.d.ts.map +1 -1
  128. package/dist/modes/interactive/components/model-selector.js +2 -1
  129. package/dist/modes/interactive/components/model-selector.js.map +1 -1
  130. package/dist/modes/interactive/components/scoped-models-selector.d.ts.map +1 -1
  131. package/dist/modes/interactive/components/scoped-models-selector.js +4 -1
  132. package/dist/modes/interactive/components/scoped-models-selector.js.map +1 -1
  133. package/dist/modes/interactive/components/settings-selector.d.ts +2 -0
  134. package/dist/modes/interactive/components/settings-selector.d.ts.map +1 -1
  135. package/dist/modes/interactive/components/settings-selector.js +165 -15
  136. package/dist/modes/interactive/components/settings-selector.js.map +1 -1
  137. package/dist/modes/interactive/components/tree-selector.d.ts.map +1 -1
  138. package/dist/modes/interactive/components/tree-selector.js +51 -4
  139. package/dist/modes/interactive/components/tree-selector.js.map +1 -1
  140. package/dist/modes/interactive/interactive-mode.d.ts +6 -1
  141. package/dist/modes/interactive/interactive-mode.d.ts.map +1 -1
  142. package/dist/modes/interactive/interactive-mode.js +115 -55
  143. package/dist/modes/interactive/interactive-mode.js.map +1 -1
  144. package/dist/modes/interactive/model-search.d.ts +7 -0
  145. package/dist/modes/interactive/model-search.d.ts.map +1 -0
  146. package/dist/modes/interactive/model-search.js +6 -0
  147. package/dist/modes/interactive/model-search.js.map +1 -0
  148. package/dist/modes/interactive/theme/theme-controller.d.ts +30 -0
  149. package/dist/modes/interactive/theme/theme-controller.d.ts.map +1 -0
  150. package/dist/modes/interactive/theme/theme-controller.js +108 -0
  151. package/dist/modes/interactive/theme/theme-controller.js.map +1 -0
  152. package/dist/modes/interactive/theme/theme-schema.json +2 -1
  153. package/dist/modes/interactive/theme/theme.d.ts +5 -0
  154. package/dist/modes/interactive/theme/theme.d.ts.map +1 -1
  155. package/dist/modes/interactive/theme/theme.js +70 -29
  156. package/dist/modes/interactive/theme/theme.js.map +1 -1
  157. package/dist/modes/rpc/rpc-client.d.ts +14 -2
  158. package/dist/modes/rpc/rpc-client.d.ts.map +1 -1
  159. package/dist/modes/rpc/rpc-client.js +23 -3
  160. package/dist/modes/rpc/rpc-client.js.map +1 -1
  161. package/dist/modes/rpc/rpc-mode.d.ts +1 -1
  162. package/dist/modes/rpc/rpc-mode.d.ts.map +1 -1
  163. package/dist/modes/rpc/rpc-mode.js +31 -2
  164. package/dist/modes/rpc/rpc-mode.js.map +1 -1
  165. package/dist/modes/rpc/rpc-types.d.ts +23 -0
  166. package/dist/modes/rpc/rpc-types.d.ts.map +1 -1
  167. package/dist/modes/rpc/rpc-types.js.map +1 -1
  168. package/dist/package-manager-cli.d.ts.map +1 -1
  169. package/dist/package-manager-cli.js +39 -9
  170. package/dist/package-manager-cli.js.map +1 -1
  171. package/docs/custom-provider.md +4 -1
  172. package/docs/extensions.md +21 -0
  173. package/docs/json.md +3 -1
  174. package/docs/models.md +78 -2
  175. package/docs/packages.md +13 -9
  176. package/docs/providers.md +3 -0
  177. package/docs/quickstart.md +14 -0
  178. package/docs/rpc.md +80 -1
  179. package/docs/sdk.md +35 -11
  180. package/docs/session-format.md +15 -1
  181. package/docs/sessions.md +1 -1
  182. package/docs/settings.md +12 -2
  183. package/docs/themes.md +3 -1
  184. package/docs/tui.md +1 -1
  185. package/docs/usage.md +12 -9
  186. package/docs/workflows.md +34 -10
  187. package/examples/extensions/custom-provider-anthropic/package-lock.json +2 -2
  188. package/examples/extensions/custom-provider-anthropic/package.json +1 -1
  189. package/examples/extensions/custom-provider-gitlab-duo/package.json +1 -1
  190. package/examples/extensions/gondolin/package-lock.json +2 -2
  191. package/examples/extensions/gondolin/package.json +1 -1
  192. package/examples/extensions/preset.ts +10 -4
  193. package/examples/extensions/provider-payload.ts +5 -5
  194. package/examples/extensions/sandbox/index.ts +2 -2
  195. package/examples/extensions/sandbox/package-lock.json +3 -3
  196. package/examples/extensions/sandbox/package.json +2 -2
  197. package/examples/extensions/subagent/agents.ts +2 -2
  198. package/examples/extensions/subagent/index.ts +4 -2
  199. package/examples/extensions/with-deps/package-lock.json +2 -2
  200. package/examples/extensions/with-deps/package.json +1 -1
  201. package/package.json +5 -5
  202. package/dist/builtin/subagents/skills/browser/EXAMPLES.md +0 -151
  203. package/dist/builtin/subagents/skills/browser/LICENSE.txt +0 -21
  204. package/dist/builtin/subagents/skills/browser/REFERENCE.md +0 -451
  205. package/dist/builtin/subagents/skills/browser/SKILL.md +0 -170
@@ -0,0 +1,39 @@
1
+ import type { Api, Model } from "@earendil-works/pi-ai";
2
+ declare module "@earendil-works/pi-ai" {
3
+ interface Model<TApi extends Api> {
4
+ /** Selectable context-window sizes for this model. The scalar contextWindow remains the default/effective value. */
5
+ contextWindowOptions?: readonly number[];
6
+ /** Original/default scalar context window, preserved when contextWindow is overridden for a session. */
7
+ defaultContextWindow?: number;
8
+ }
9
+ }
10
+ export interface ContextWindowParseResult {
11
+ value?: number;
12
+ error?: string;
13
+ }
14
+ export interface ContextWindowSelection<TApi extends Api = Api> {
15
+ model: Model<TApi>;
16
+ contextWindow: number;
17
+ }
18
+ export interface ContextWindowSelectionError {
19
+ error: string;
20
+ }
21
+ export interface ContextWindowSelectionOptions {
22
+ /**
23
+ * GitHub Copilot advertises some long-context tiers below their branded 1M size
24
+ * (for example 936k or 922k input tokens). When enabled, a request above an
25
+ * advertised long tier selects the largest supported Copilot window not
26
+ * exceeding the request, but never silently falls back to the model's base
27
+ * window.
28
+ */
29
+ allowCopilotLongContextFallback?: boolean;
30
+ }
31
+ export declare function validateContextWindowValue(value: number): string | undefined;
32
+ export declare function parseContextWindowValue(input: string): ContextWindowParseResult;
33
+ export declare function formatContextWindow(value: number): string;
34
+ export declare function normalizeContextWindowOptions(values: readonly number[] | undefined): number[];
35
+ export declare function getModelDefaultContextWindow(model: Model<Api>): number;
36
+ export declare function getSupportedContextWindows(model: Model<Api>): number[];
37
+ export declare function withContextWindowOptions<TApi extends Api>(model: Model<TApi>, contextWindowOptions: readonly number[]): Model<TApi>;
38
+ export declare function selectContextWindow<TApi extends Api>(model: Model<TApi>, contextWindow: number, options?: ContextWindowSelectionOptions): ContextWindowSelection<TApi> | ContextWindowSelectionError;
39
+ //# sourceMappingURL=context-window.d.ts.map
@@ -0,0 +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"]}
@@ -0,0 +1,99 @@
1
+ const CONTEXT_WINDOW_UNITS = {
2
+ k: 1_000,
3
+ m: 1_000_000,
4
+ };
5
+ function isPositiveInteger(value) {
6
+ return Number.isFinite(value) && Number.isInteger(value) && value > 0;
7
+ }
8
+ export function validateContextWindowValue(value) {
9
+ return isPositiveInteger(value) ? undefined : "Context window must be a positive integer token count";
10
+ }
11
+ export function parseContextWindowValue(input) {
12
+ const trimmed = input.trim();
13
+ if (!trimmed) {
14
+ return { error: "Context window requires a value" };
15
+ }
16
+ const match = /^(\d+(?:\.\d+)?)([kKmM])?$/.exec(trimmed);
17
+ if (!match) {
18
+ return { error: `Invalid context window "${input}". Use a positive number, or a compact value like 400k or 1m.` };
19
+ }
20
+ const numericValue = Number(match[1]);
21
+ const unit = match[2]?.toLowerCase();
22
+ const multiplier = unit ? CONTEXT_WINDOW_UNITS[unit] : 1;
23
+ const tokens = numericValue * multiplier;
24
+ const validationError = validateContextWindowValue(tokens);
25
+ if (validationError) {
26
+ return { error: `Invalid context window "${input}". ${validationError}.` };
27
+ }
28
+ return { value: tokens };
29
+ }
30
+ export function formatContextWindow(value) {
31
+ if (value >= 1_000_000) {
32
+ const millions = value / 1_000_000;
33
+ return millions % 1 === 0 ? `${millions}m` : `${millions.toFixed(1)}m`;
34
+ }
35
+ if (value >= 1_000) {
36
+ const thousands = value / 1_000;
37
+ return thousands % 1 === 0 ? `${thousands}k` : `${thousands.toFixed(1)}k`;
38
+ }
39
+ return String(value);
40
+ }
41
+ export function normalizeContextWindowOptions(values) {
42
+ const seen = new Set();
43
+ const normalized = [];
44
+ for (const value of values ?? []) {
45
+ if (!isPositiveInteger(value) || seen.has(value))
46
+ continue;
47
+ seen.add(value);
48
+ normalized.push(value);
49
+ }
50
+ return normalized.sort((a, b) => a - b);
51
+ }
52
+ export function getModelDefaultContextWindow(model) {
53
+ return isPositiveInteger(model.defaultContextWindow ?? 0) ? model.defaultContextWindow : model.contextWindow;
54
+ }
55
+ export function getSupportedContextWindows(model) {
56
+ return normalizeContextWindowOptions([getModelDefaultContextWindow(model), ...(model.contextWindowOptions ?? [])]);
57
+ }
58
+ export function withContextWindowOptions(model, contextWindowOptions) {
59
+ return {
60
+ ...model,
61
+ defaultContextWindow: getModelDefaultContextWindow(model),
62
+ contextWindowOptions: normalizeContextWindowOptions(contextWindowOptions),
63
+ };
64
+ }
65
+ function resolveSelectableContextWindow(model, requestedContextWindow, supported, options) {
66
+ if (supported.includes(requestedContextWindow)) {
67
+ return requestedContextWindow;
68
+ }
69
+ if (options.allowCopilotLongContextFallback !== true || model.provider !== "github-copilot") {
70
+ return undefined;
71
+ }
72
+ const defaultContextWindow = getModelDefaultContextWindow(model);
73
+ const candidates = supported.filter((contextWindow) => contextWindow <= requestedContextWindow && contextWindow > defaultContextWindow);
74
+ return candidates.length > 0 ? Math.max(...candidates) : undefined;
75
+ }
76
+ export function selectContextWindow(model, contextWindow, options = {}) {
77
+ const validationError = validateContextWindowValue(contextWindow);
78
+ if (validationError) {
79
+ return { error: validationError };
80
+ }
81
+ const apiModel = model;
82
+ const supported = getSupportedContextWindows(apiModel);
83
+ const selectedContextWindow = resolveSelectableContextWindow(apiModel, contextWindow, supported, options);
84
+ if (selectedContextWindow === undefined) {
85
+ return {
86
+ error: `Context window ${formatContextWindow(contextWindow)} is not supported by ${model.provider}/${model.id}. Supported values: ${supported.map(formatContextWindow).join(", ")}.`,
87
+ };
88
+ }
89
+ return {
90
+ model: {
91
+ ...model,
92
+ defaultContextWindow: getModelDefaultContextWindow(apiModel),
93
+ contextWindow: selectedContextWindow,
94
+ contextWindowOptions: supported,
95
+ },
96
+ contextWindow: selectedContextWindow,
97
+ };
98
+ }
99
+ //# sourceMappingURL=context-window.js.map
@@ -0,0 +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"]}
@@ -0,0 +1,9 @@
1
+ export interface CopilotPromptLimitError {
2
+ promptTokens: number;
3
+ limitTokens: number;
4
+ }
5
+ export declare function parseCopilotPromptLimitError(errorMessage: string): CopilotPromptLimitError | undefined;
6
+ export declare function isCopilotPromptLimitError(errorMessage: string): boolean;
7
+ export declare function formatCopilotPromptLimitError(errorMessage: string): string;
8
+ export declare function formatCopilotProviderError(modelProvider: string, errorMessage: string): string;
9
+ //# sourceMappingURL=copilot-errors.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"copilot-errors.d.ts","sourceRoot":"","sources":["../../src/core/copilot-errors.ts"],"names":[],"mappings":"AAGA,MAAM,WAAW,uBAAuB;IACvC,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;CACpB;AAMD,wBAAgB,4BAA4B,CAAC,YAAY,EAAE,MAAM,GAAG,uBAAuB,GAAG,SAAS,CAUtG;AAED,wBAAgB,yBAAyB,CAAC,YAAY,EAAE,MAAM,GAAG,OAAO,CAEvE;AAED,wBAAgB,6BAA6B,CAAC,YAAY,EAAE,MAAM,GAAG,MAAM,CAM1E;AAED,wBAAgB,0BAA0B,CAAC,aAAa,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,GAAG,MAAM,CAE9F","sourcesContent":["const COPILOT_PROMPT_LIMIT_PATTERN = /\\bprompt token count of ([\\d,]+) exceeds the limit of ([\\d,]+)/i;\nconst COPILOT_LONG_CONTEXT_GUIDANCE_MARKER = \"Copilot long-context/usage-based billing\";\n\nexport interface CopilotPromptLimitError {\n\tpromptTokens: number;\n\tlimitTokens: number;\n}\n\nfunction parseTokenCount(value: string): number {\n\treturn Number(value.replace(/,/g, \"\"));\n}\n\nexport function parseCopilotPromptLimitError(errorMessage: string): CopilotPromptLimitError | undefined {\n\tconst match = COPILOT_PROMPT_LIMIT_PATTERN.exec(errorMessage);\n\tif (!match) return undefined;\n\tconst promptTokenText = match[1];\n\tconst limitTokenText = match[2];\n\tif (!promptTokenText || !limitTokenText) return undefined;\n\tconst promptTokens = parseTokenCount(promptTokenText);\n\tconst limitTokens = parseTokenCount(limitTokenText);\n\tif (!Number.isFinite(promptTokens) || !Number.isFinite(limitTokens)) return undefined;\n\treturn { promptTokens, limitTokens };\n}\n\nexport function isCopilotPromptLimitError(errorMessage: string): boolean {\n\treturn parseCopilotPromptLimitError(errorMessage) !== undefined;\n}\n\nexport function formatCopilotPromptLimitError(errorMessage: string): string {\n\tif (!isCopilotPromptLimitError(errorMessage) || errorMessage.includes(COPILOT_LONG_CONTEXT_GUIDANCE_MARKER)) {\n\t\treturn errorMessage;\n\t}\n\n\treturn `${errorMessage}\\n\\nGitHub Copilot rejected this prompt at the API/server context cap. Atomic raises the local token budget for selected larger context windows and sends X-GitHub-Api-Version: 2026-06-01 so Copilot can choose its long-context tier server-side by prompt token count, but GitHub still requires the account to have Copilot long-context/usage-based billing entitlement enabled. Long-context Copilot requests consume higher-cost AI credits. Reduce the session context, choose a smaller context window/model, or enable the required Copilot long-context/usage-based billing entitlement for this account.`;\n}\n\nexport function formatCopilotProviderError(modelProvider: string, errorMessage: string): string {\n\treturn modelProvider === \"github-copilot\" ? formatCopilotPromptLimitError(errorMessage) : errorMessage;\n}\n"]}
@@ -0,0 +1,32 @@
1
+ const COPILOT_PROMPT_LIMIT_PATTERN = /\bprompt token count of ([\d,]+) exceeds the limit of ([\d,]+)/i;
2
+ const COPILOT_LONG_CONTEXT_GUIDANCE_MARKER = "Copilot long-context/usage-based billing";
3
+ function parseTokenCount(value) {
4
+ return Number(value.replace(/,/g, ""));
5
+ }
6
+ export function parseCopilotPromptLimitError(errorMessage) {
7
+ const match = COPILOT_PROMPT_LIMIT_PATTERN.exec(errorMessage);
8
+ if (!match)
9
+ return undefined;
10
+ const promptTokenText = match[1];
11
+ const limitTokenText = match[2];
12
+ if (!promptTokenText || !limitTokenText)
13
+ return undefined;
14
+ const promptTokens = parseTokenCount(promptTokenText);
15
+ const limitTokens = parseTokenCount(limitTokenText);
16
+ if (!Number.isFinite(promptTokens) || !Number.isFinite(limitTokens))
17
+ return undefined;
18
+ return { promptTokens, limitTokens };
19
+ }
20
+ export function isCopilotPromptLimitError(errorMessage) {
21
+ return parseCopilotPromptLimitError(errorMessage) !== undefined;
22
+ }
23
+ export function formatCopilotPromptLimitError(errorMessage) {
24
+ if (!isCopilotPromptLimitError(errorMessage) || errorMessage.includes(COPILOT_LONG_CONTEXT_GUIDANCE_MARKER)) {
25
+ return errorMessage;
26
+ }
27
+ return `${errorMessage}\n\nGitHub Copilot rejected this prompt at the API/server context cap. Atomic raises the local token budget for selected larger context windows and sends X-GitHub-Api-Version: 2026-06-01 so Copilot can choose its long-context tier server-side by prompt token count, but GitHub still requires the account to have Copilot long-context/usage-based billing entitlement enabled. Long-context Copilot requests consume higher-cost AI credits. Reduce the session context, choose a smaller context window/model, or enable the required Copilot long-context/usage-based billing entitlement for this account.`;
28
+ }
29
+ export function formatCopilotProviderError(modelProvider, errorMessage) {
30
+ return modelProvider === "github-copilot" ? formatCopilotPromptLimitError(errorMessage) : errorMessage;
31
+ }
32
+ //# sourceMappingURL=copilot-errors.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"copilot-errors.js","sourceRoot":"","sources":["../../src/core/copilot-errors.ts"],"names":[],"mappings":"AAAA,MAAM,4BAA4B,GAAG,iEAAiE,CAAC;AACvG,MAAM,oCAAoC,GAAG,0CAA0C,CAAC;AAOxF,SAAS,eAAe,CAAC,KAAa;IACrC,OAAO,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;AACxC,CAAC;AAED,MAAM,UAAU,4BAA4B,CAAC,YAAoB;IAChE,MAAM,KAAK,GAAG,4BAA4B,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC9D,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAC7B,MAAM,eAAe,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IACjC,MAAM,cAAc,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IAChC,IAAI,CAAC,eAAe,IAAI,CAAC,cAAc;QAAE,OAAO,SAAS,CAAC;IAC1D,MAAM,YAAY,GAAG,eAAe,CAAC,eAAe,CAAC,CAAC;IACtD,MAAM,WAAW,GAAG,eAAe,CAAC,cAAc,CAAC,CAAC;IACpD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC;QAAE,OAAO,SAAS,CAAC;IACtF,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC;AACtC,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,YAAoB;IAC7D,OAAO,4BAA4B,CAAC,YAAY,CAAC,KAAK,SAAS,CAAC;AACjE,CAAC;AAED,MAAM,UAAU,6BAA6B,CAAC,YAAoB;IACjE,IAAI,CAAC,yBAAyB,CAAC,YAAY,CAAC,IAAI,YAAY,CAAC,QAAQ,CAAC,oCAAoC,CAAC,EAAE,CAAC;QAC7G,OAAO,YAAY,CAAC;IACrB,CAAC;IAED,OAAO,GAAG,YAAY,slBAAslB,CAAC;AAC9mB,CAAC;AAED,MAAM,UAAU,0BAA0B,CAAC,aAAqB,EAAE,YAAoB;IACrF,OAAO,aAAa,KAAK,gBAAgB,CAAC,CAAC,CAAC,6BAA6B,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC;AACxG,CAAC","sourcesContent":["const COPILOT_PROMPT_LIMIT_PATTERN = /\\bprompt token count of ([\\d,]+) exceeds the limit of ([\\d,]+)/i;\nconst COPILOT_LONG_CONTEXT_GUIDANCE_MARKER = \"Copilot long-context/usage-based billing\";\n\nexport interface CopilotPromptLimitError {\n\tpromptTokens: number;\n\tlimitTokens: number;\n}\n\nfunction parseTokenCount(value: string): number {\n\treturn Number(value.replace(/,/g, \"\"));\n}\n\nexport function parseCopilotPromptLimitError(errorMessage: string): CopilotPromptLimitError | undefined {\n\tconst match = COPILOT_PROMPT_LIMIT_PATTERN.exec(errorMessage);\n\tif (!match) return undefined;\n\tconst promptTokenText = match[1];\n\tconst limitTokenText = match[2];\n\tif (!promptTokenText || !limitTokenText) return undefined;\n\tconst promptTokens = parseTokenCount(promptTokenText);\n\tconst limitTokens = parseTokenCount(limitTokenText);\n\tif (!Number.isFinite(promptTokens) || !Number.isFinite(limitTokens)) return undefined;\n\treturn { promptTokens, limitTokens };\n}\n\nexport function isCopilotPromptLimitError(errorMessage: string): boolean {\n\treturn parseCopilotPromptLimitError(errorMessage) !== undefined;\n}\n\nexport function formatCopilotPromptLimitError(errorMessage: string): string {\n\tif (!isCopilotPromptLimitError(errorMessage) || errorMessage.includes(COPILOT_LONG_CONTEXT_GUIDANCE_MARKER)) {\n\t\treturn errorMessage;\n\t}\n\n\treturn `${errorMessage}\\n\\nGitHub Copilot rejected this prompt at the API/server context cap. Atomic raises the local token budget for selected larger context windows and sends X-GitHub-Api-Version: 2026-06-01 so Copilot can choose its long-context tier server-side by prompt token count, but GitHub still requires the account to have Copilot long-context/usage-based billing entitlement enabled. Long-context Copilot requests consume higher-cost AI credits. Reduce the session context, choose a smaller context window/model, or enable the required Copilot long-context/usage-based billing entitlement for this account.`;\n}\n\nexport function formatCopilotProviderError(modelProvider: string, errorMessage: string): string {\n\treturn modelProvider === \"github-copilot\" ? formatCopilotPromptLimitError(errorMessage) : errorMessage;\n}\n"]}
@@ -0,0 +1,135 @@
1
+ /**
2
+ * GitHub Copilot model catalog (CAPI) — dynamic prompt-token budgets.
3
+ *
4
+ * GitHub's Copilot API (CAPI) exposes distinct model limits via `GET {baseUrl}/models`:
5
+ *
6
+ * - `capabilities.limits.max_context_window_tokens` is the model's total context capacity
7
+ * (prompt + completion reserve).
8
+ * - `capabilities.limits.max_prompt_tokens` is the maximum prompt/input budget Atomic can safely
9
+ * fill before the provider must reserve output tokens.
10
+ * - `billing.token_prices.<tier>.context_max` is a prompt-token billing/selection threshold. The
11
+ * `default` tier is the short prompt budget (e.g. gpt-5.5 272k, Claude 200k); a
12
+ * `long_context` tier adds a selectable larger prompt budget (e.g. gpt-5.5 922k, Claude 936k).
13
+ *
14
+ * Atomic's current `contextWindow` drives local prompt collection, compaction thresholds, footer
15
+ * usage, and overflow avoidance, so Copilot selectable windows intentionally use prompt-token
16
+ * budgets rather than total context capacities such as 1_000_000/1_050_000. The total context field
17
+ * remains a compatibility fallback for older/sparse payloads that omit `max_prompt_tokens`.
18
+ *
19
+ * This data is intentionally NOT baked into a static map: GitHub adds/removes models and retiers
20
+ * windows over time (e.g. a model that disappears from the catalog), so a hardcoded snapshot goes
21
+ * stale. Instead the catalog is fetched live (gated on the user actually having the GitHub Copilot
22
+ * provider) and cached on disk for a short TTL, exactly like the Copilot CLI.
23
+ */
24
+ /** Resolved input-token context window(s) for a single Copilot model. */
25
+ export interface CopilotModelContext {
26
+ /**
27
+ * Base context window in INPUT tokens — shown in the footer and used for compaction. The
28
+ * default tier's `context_max`, or the model-level `max_prompt_tokens` fallback otherwise.
29
+ */
30
+ contextWindow: number;
31
+ /**
32
+ * Selectable input-token windows (`[default, long]`) when the model exposes a `long_context`
33
+ * tier larger than its default; absent for single-window models.
34
+ */
35
+ contextWindowOptions?: readonly number[];
36
+ }
37
+ /** Map of model id → resolved input-token context window(s). */
38
+ export type CopilotModelCatalog = ReadonlyMap<string, CopilotModelContext>;
39
+ /** Safety fallback when a model reports neither `max_prompt_tokens` nor `max_context_window_tokens`. */
40
+ export declare const COPILOT_CONTEXT_WINDOW_FALLBACK = 128000;
41
+ export declare const COPILOT_CATALOG_API_VERSION = "2026-06-01";
42
+ /**
43
+ * Headers GitHub's CAPI expects for catalog reads. Mirrors the editor headers pi-ai already sends
44
+ * for Copilot token refresh and model-policy calls, plus the dated API version.
45
+ */
46
+ export declare const COPILOT_CATALOG_HEADERS: Readonly<Record<string, string>>;
47
+ /** Default (non-enterprise) Copilot CAPI base URL when the token has no resolvable `proxy-ep`. */
48
+ export declare const DEFAULT_COPILOT_API_BASE_URL = "https://api.individual.githubcopilot.com";
49
+ /** Disk-cache freshness window, matching the Copilot CLI's list-models cache TTL. */
50
+ export declare const COPILOT_CATALOG_CACHE_TTL_MS: number;
51
+ /** Current on-disk cache schema version. */
52
+ export declare const COPILOT_CATALOG_CACHE_VERSION: 2;
53
+ /**
54
+ * Resolve the Copilot CAPI base URL.
55
+ *
56
+ * Copilot access tokens embed a `proxy-ep=proxy.<host>` segment; the API host is the same host with
57
+ * `proxy.` swapped for `api.`. Falls back to the enterprise host or the individual default. (pi-ai
58
+ * exposes an equivalent helper, but its published `dist` mangles the export name, so the small,
59
+ * stable parsing logic is reimplemented here.)
60
+ */
61
+ export declare function copilotApiBaseUrlFromToken(token: string | undefined, enterpriseDomain?: string): string;
62
+ /** Raw token limits parsed from a CAPI model entry. */
63
+ export interface CopilotModelLimits {
64
+ /** `capabilities.limits.max_prompt_tokens` — maximum prompt/input budget. */
65
+ maxPromptTokens?: number;
66
+ /** `capabilities.limits.max_context_window_tokens` — total context capacity, used only as a fallback. */
67
+ maxContextWindowTokens?: number;
68
+ /** `billing.token_prices.default.context_max` — default-tier prompt threshold. */
69
+ defaultContextMax?: number;
70
+ /** `billing.token_prices.long_context.context_max` — long-context prompt threshold. */
71
+ longContextMax?: number;
72
+ }
73
+ /**
74
+ * Resolve a model's input-token context window(s) from its CAPI limits.
75
+ *
76
+ * `contextWindow` is the model's base input budget — the default tier's `context_max` when tiered,
77
+ * otherwise `max_prompt_tokens ?? max_context_window_tokens ?? 128_000`. A `long_context` tier that
78
+ * is larger than the base adds a second selectable window. Returns `undefined` when the entry
79
+ * carries no usable limit signal at all.
80
+ */
81
+ export declare function resolveCopilotModelContext(limits: CopilotModelLimits): CopilotModelContext | undefined;
82
+ /**
83
+ * Parse a raw CAPI `/models` response body into an input-token context-window catalog.
84
+ */
85
+ export declare function parseCopilotModelCatalog(body: unknown): CopilotModelCatalog;
86
+ export interface FetchCopilotModelCatalogOptions {
87
+ /** Valid Copilot CAPI bearer token (e.g. from `modelRegistry.getApiKeyForProvider`). */
88
+ token: string;
89
+ /** Override the resolved base URL; defaults to one derived from the token. */
90
+ baseUrl?: string;
91
+ /** Enterprise domain, used for base-URL resolution when the token lacks a `proxy-ep`. */
92
+ enterpriseDomain?: string;
93
+ /** Extra/override request headers. */
94
+ headers?: Record<string, string>;
95
+ /** Injectable `fetch` for testing. */
96
+ fetchImpl?: typeof fetch;
97
+ /** Abort signal. */
98
+ signal?: AbortSignal;
99
+ }
100
+ /** Fetch and parse the live Copilot model catalog from CAPI `GET {baseUrl}/models`. */
101
+ export declare function fetchCopilotModelCatalog(options: FetchCopilotModelCatalogOptions): Promise<CopilotModelCatalog>;
102
+ /** Replace the active catalog the registry derives context windows from. */
103
+ export declare function setActiveCopilotModelCatalog(catalog: CopilotModelCatalog): void;
104
+ /** The active catalog (empty until a successful auth-gated fetch/cache load). */
105
+ export declare function getActiveCopilotModelCatalog(): CopilotModelCatalog;
106
+ /** Reset the active catalog (primarily for tests). */
107
+ export declare function clearActiveCopilotModelCatalog(): void;
108
+ export interface ReadCopilotCatalogCacheOptions {
109
+ /** Expected CAPI host; a cached file from a different host is ignored. */
110
+ host: string;
111
+ /** Current epoch ms (injectable for tests). */
112
+ now?: number;
113
+ /** Freshness window; defaults to {@link COPILOT_CATALOG_CACHE_TTL_MS}. */
114
+ ttlMs?: number;
115
+ }
116
+ /** Read a fresh, host-matching catalog from the cache file, or `undefined` if missing/stale/invalid. */
117
+ export declare function readCopilotCatalogCache(path: string, options: ReadCopilotCatalogCacheOptions): CopilotModelCatalog | undefined;
118
+ /** Write the catalog to the cache file (creating parent dirs). Best-effort; never throws. */
119
+ export declare function writeCopilotCatalogCache(path: string, baseUrl: string, catalog: CopilotModelCatalog, now?: number): void;
120
+ /** Host component of a base URL, for matching {@link readCopilotCatalogCache} `host`. */
121
+ export declare function copilotCatalogCacheHost(baseUrl: string): string;
122
+ /** Standard on-disk cache path for the Copilot model catalog under an agent directory. */
123
+ export declare function copilotCatalogCachePath(agentDir: string): string;
124
+ /**
125
+ * Seed the active catalog synchronously from the on-disk cache, gated on a Copilot access token.
126
+ *
127
+ * Called at model-registry construction so a returning user's previously selected long-context
128
+ * window is recognized before startup validation runs — otherwise the persisted choice would warn
129
+ * ("context window 936k is not supported…") and reset until the async refresh completes. The cache
130
+ * TTL is intentionally ignored here: stale-but-present windows are still valid for selection, and
131
+ * the async loader independently refetches on its own freshness window. Returns true when a catalog
132
+ * was applied. No-op (returns false) without a token or a host-matching cached catalog.
133
+ */
134
+ export declare function seedActiveCopilotModelCatalogFromCache(accessToken: string | undefined, cachePath: string, now?: number): boolean;
135
+ //# sourceMappingURL=copilot-model-catalog.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"copilot-model-catalog.d.ts","sourceRoot":"","sources":["../../src/core/copilot-model-catalog.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;GAsBG;AAKH,yEAAyE;AACzE,MAAM,WAAW,mBAAmB;IACnC;;;OAGG;IACH,aAAa,EAAE,MAAM,CAAC;IACtB;;;OAGG;IACH,oBAAoB,CAAC,EAAE,SAAS,MAAM,EAAE,CAAC;CACzC;AAED,gEAAgE;AAChE,MAAM,MAAM,mBAAmB,GAAG,WAAW,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAAC;AAE3E,wGAAwG;AACxG,eAAO,MAAM,+BAA+B,SAAU,CAAC;AAEvD,eAAO,MAAM,2BAA2B,eAAe,CAAC;AAExD;;;GAGG;AACH,eAAO,MAAM,uBAAuB,EAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAMpE,CAAC;AAEF,kGAAkG;AAClG,eAAO,MAAM,4BAA4B,6CAA6C,CAAC;AAEvF,qFAAqF;AACrF,eAAO,MAAM,4BAA4B,QAAiB,CAAC;AAE3D,4CAA4C;AAC5C,eAAO,MAAM,6BAA6B,EAAG,CAAU,CAAC;AAExD;;;;;;;GAOG;AACH,wBAAgB,0BAA0B,CAAC,KAAK,EAAE,MAAM,GAAG,SAAS,EAAE,gBAAgB,CAAC,EAAE,MAAM,GAAG,MAAM,CASvG;AAcD,uDAAuD;AACvD,MAAM,WAAW,kBAAkB;IAClC,6EAA6E;IAC7E,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,yGAAyG;IACzG,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,kFAAkF;IAClF,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,uFAAuF;IACvF,cAAc,CAAC,EAAE,MAAM,CAAC;CACxB;AAED;;;;;;;GAOG;AACH,wBAAgB,0BAA0B,CAAC,MAAM,EAAE,kBAAkB,GAAG,mBAAmB,GAAG,SAAS,CActG;AAED;;GAEG;AACH,wBAAgB,wBAAwB,CAAC,IAAI,EAAE,OAAO,GAAG,mBAAmB,CAuB3E;AAED,MAAM,WAAW,+BAA+B;IAC/C,wFAAwF;IACxF,KAAK,EAAE,MAAM,CAAC;IACd,8EAA8E;IAC9E,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,yFAAyF;IACzF,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,sCAAsC;IACtC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,sCAAsC;IACtC,SAAS,CAAC,EAAE,OAAO,KAAK,CAAC;IACzB,oBAAoB;IACpB,MAAM,CAAC,EAAE,WAAW,CAAC;CACrB;AAED,uFAAuF;AACvF,wBAAsB,wBAAwB,CAAC,OAAO,EAAE,+BAA+B,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAiBrH;AAWD,4EAA4E;AAC5E,wBAAgB,4BAA4B,CAAC,OAAO,EAAE,mBAAmB,GAAG,IAAI,CAE/E;AAED,iFAAiF;AACjF,wBAAgB,4BAA4B,IAAI,mBAAmB,CAElE;AAED,sDAAsD;AACtD,wBAAgB,8BAA8B,IAAI,IAAI,CAErD;AAuBD,MAAM,WAAW,8BAA8B;IAC9C,0EAA0E;IAC1E,IAAI,EAAE,MAAM,CAAC;IACb,+CAA+C;IAC/C,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,0EAA0E;IAC1E,KAAK,CAAC,EAAE,MAAM,CAAC;CACf;AAcD,wGAAwG;AACxG,wBAAgB,uBAAuB,CACtC,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,8BAA8B,GACrC,mBAAmB,GAAG,SAAS,CAsBjC;AAED,6FAA6F;AAC7F,wBAAgB,wBAAwB,CACvC,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,OAAO,EAAE,mBAAmB,EAC5B,GAAG,CAAC,EAAE,MAAM,GACV,IAAI,CAaN;AAED,yFAAyF;AACzF,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAE/D;AAED,0FAA0F;AAC1F,wBAAgB,uBAAuB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,CAEhE;AAED;;;;;;;;;GASG;AACH,wBAAgB,sCAAsC,CACrD,WAAW,EAAE,MAAM,GAAG,SAAS,EAC/B,SAAS,EAAE,MAAM,EACjB,GAAG,CAAC,EAAE,MAAM,GACV,OAAO,CAOT","sourcesContent":["/**\n * GitHub Copilot model catalog (CAPI) — dynamic prompt-token budgets.\n *\n * GitHub's Copilot API (CAPI) exposes distinct model limits via `GET {baseUrl}/models`:\n *\n * - `capabilities.limits.max_context_window_tokens` is the model's total context capacity\n * (prompt + completion reserve).\n * - `capabilities.limits.max_prompt_tokens` is the maximum prompt/input budget Atomic can safely\n * fill before the provider must reserve output tokens.\n * - `billing.token_prices.<tier>.context_max` is a prompt-token billing/selection threshold. The\n * `default` tier is the short prompt budget (e.g. gpt-5.5 272k, Claude 200k); a\n * `long_context` tier adds a selectable larger prompt budget (e.g. gpt-5.5 922k, Claude 936k).\n *\n * Atomic's current `contextWindow` drives local prompt collection, compaction thresholds, footer\n * usage, and overflow avoidance, so Copilot selectable windows intentionally use prompt-token\n * budgets rather than total context capacities such as 1_000_000/1_050_000. The total context field\n * remains a compatibility fallback for older/sparse payloads that omit `max_prompt_tokens`.\n *\n * This data is intentionally NOT baked into a static map: GitHub adds/removes models and retiers\n * windows over time (e.g. a model that disappears from the catalog), so a hardcoded snapshot goes\n * stale. Instead the catalog is fetched live (gated on the user actually having the GitHub Copilot\n * provider) and cached on disk for a short TTL, exactly like the Copilot CLI.\n */\n\nimport { existsSync, mkdirSync, readFileSync, writeFileSync } from \"node:fs\";\nimport { dirname, join } from \"node:path\";\n\n/** Resolved input-token context window(s) for a single Copilot model. */\nexport interface CopilotModelContext {\n\t/**\n\t * Base context window in INPUT tokens — shown in the footer and used for compaction. The\n\t * default tier's `context_max`, or the model-level `max_prompt_tokens` fallback otherwise.\n\t */\n\tcontextWindow: number;\n\t/**\n\t * Selectable input-token windows (`[default, long]`) when the model exposes a `long_context`\n\t * tier larger than its default; absent for single-window models.\n\t */\n\tcontextWindowOptions?: readonly number[];\n}\n\n/** Map of model id → resolved input-token context window(s). */\nexport type CopilotModelCatalog = ReadonlyMap<string, CopilotModelContext>;\n\n/** Safety fallback when a model reports neither `max_prompt_tokens` nor `max_context_window_tokens`. */\nexport const COPILOT_CONTEXT_WINDOW_FALLBACK = 128_000;\n\nexport const COPILOT_CATALOG_API_VERSION = \"2026-06-01\";\n\n/**\n * Headers GitHub's CAPI expects for catalog reads. Mirrors the editor headers pi-ai already sends\n * for Copilot token refresh and model-policy calls, plus the dated API version.\n */\nexport const COPILOT_CATALOG_HEADERS: Readonly<Record<string, string>> = {\n\t\"User-Agent\": \"GitHubCopilotChat/0.35.0\",\n\t\"Editor-Version\": \"vscode/1.107.0\",\n\t\"Editor-Plugin-Version\": \"copilot-chat/0.35.0\",\n\t\"Copilot-Integration-Id\": \"vscode-chat\",\n\t\"X-GitHub-Api-Version\": COPILOT_CATALOG_API_VERSION,\n};\n\n/** Default (non-enterprise) Copilot CAPI base URL when the token has no resolvable `proxy-ep`. */\nexport const DEFAULT_COPILOT_API_BASE_URL = \"https://api.individual.githubcopilot.com\";\n\n/** Disk-cache freshness window, matching the Copilot CLI's list-models cache TTL. */\nexport const COPILOT_CATALOG_CACHE_TTL_MS = 30 * 60 * 1000;\n\n/** Current on-disk cache schema version. */\nexport const COPILOT_CATALOG_CACHE_VERSION = 2 as const;\n\n/**\n * Resolve the Copilot CAPI base URL.\n *\n * Copilot access tokens embed a `proxy-ep=proxy.<host>` segment; the API host is the same host with\n * `proxy.` swapped for `api.`. Falls back to the enterprise host or the individual default. (pi-ai\n * exposes an equivalent helper, but its published `dist` mangles the export name, so the small,\n * stable parsing logic is reimplemented here.)\n */\nexport function copilotApiBaseUrlFromToken(token: string | undefined, enterpriseDomain?: string): string {\n\tif (token) {\n\t\tconst match = token.match(/proxy-ep=([^;]+)/);\n\t\tif (match) {\n\t\t\treturn `https://${match[1].replace(/^proxy\\./, \"api.\")}`;\n\t\t}\n\t}\n\tif (enterpriseDomain) return `https://copilot-api.${enterpriseDomain}`;\n\treturn DEFAULT_COPILOT_API_BASE_URL;\n}\n\nfunction trimTrailingSlash(url: string): string {\n\treturn url.replace(/\\/+$/, \"\");\n}\n\nfunction asRecord(value: unknown): Record<string, unknown> | undefined {\n\treturn value && typeof value === \"object\" ? (value as Record<string, unknown>) : undefined;\n}\n\nfunction toPositiveInt(value: unknown): number | undefined {\n\treturn typeof value === \"number\" && Number.isInteger(value) && value > 0 ? value : undefined;\n}\n\n/** Raw token limits parsed from a CAPI model entry. */\nexport interface CopilotModelLimits {\n\t/** `capabilities.limits.max_prompt_tokens` — maximum prompt/input budget. */\n\tmaxPromptTokens?: number;\n\t/** `capabilities.limits.max_context_window_tokens` — total context capacity, used only as a fallback. */\n\tmaxContextWindowTokens?: number;\n\t/** `billing.token_prices.default.context_max` — default-tier prompt threshold. */\n\tdefaultContextMax?: number;\n\t/** `billing.token_prices.long_context.context_max` — long-context prompt threshold. */\n\tlongContextMax?: number;\n}\n\n/**\n * Resolve a model's input-token context window(s) from its CAPI limits.\n *\n * `contextWindow` is the model's base input budget — the default tier's `context_max` when tiered,\n * otherwise `max_prompt_tokens ?? max_context_window_tokens ?? 128_000`. A `long_context` tier that\n * is larger than the base adds a second selectable window. Returns `undefined` when the entry\n * carries no usable limit signal at all.\n */\nexport function resolveCopilotModelContext(limits: CopilotModelLimits): CopilotModelContext | undefined {\n\tconst hasSignal =\n\t\tlimits.maxPromptTokens !== undefined ||\n\t\tlimits.maxContextWindowTokens !== undefined ||\n\t\tlimits.defaultContextMax !== undefined ||\n\t\tlimits.longContextMax !== undefined;\n\tif (!hasSignal) return undefined;\n\n\tconst maxInput = limits.maxPromptTokens ?? limits.maxContextWindowTokens ?? COPILOT_CONTEXT_WINDOW_FALLBACK;\n\tconst base = limits.defaultContextMax ?? maxInput;\n\tif (limits.longContextMax !== undefined && limits.longContextMax > base) {\n\t\treturn { contextWindow: base, contextWindowOptions: [base, limits.longContextMax] };\n\t}\n\treturn { contextWindow: base };\n}\n\n/**\n * Parse a raw CAPI `/models` response body into an input-token context-window catalog.\n */\nexport function parseCopilotModelCatalog(body: unknown): CopilotModelCatalog {\n\tconst catalog = new Map<string, CopilotModelContext>();\n\tconst data = asRecord(body)?.data;\n\tif (!Array.isArray(data)) return catalog;\n\n\tfor (const entry of data) {\n\t\tconst record = asRecord(entry);\n\t\tif (!record) continue;\n\t\tconst id = record.id;\n\t\tif (typeof id !== \"string\" || id.length === 0) continue;\n\n\t\tconst limits = asRecord(asRecord(record.capabilities)?.limits);\n\t\tconst prices = asRecord(asRecord(record.billing)?.token_prices);\n\t\tconst context = resolveCopilotModelContext({\n\t\t\tmaxPromptTokens: toPositiveInt(limits?.max_prompt_tokens),\n\t\t\tmaxContextWindowTokens: toPositiveInt(limits?.max_context_window_tokens),\n\t\t\tdefaultContextMax: toPositiveInt(asRecord(prices?.default)?.context_max),\n\t\t\tlongContextMax: toPositiveInt(asRecord(prices?.long_context)?.context_max),\n\t\t});\n\t\tif (context) catalog.set(id, context);\n\t}\n\n\treturn catalog;\n}\n\nexport interface FetchCopilotModelCatalogOptions {\n\t/** Valid Copilot CAPI bearer token (e.g. from `modelRegistry.getApiKeyForProvider`). */\n\ttoken: string;\n\t/** Override the resolved base URL; defaults to one derived from the token. */\n\tbaseUrl?: string;\n\t/** Enterprise domain, used for base-URL resolution when the token lacks a `proxy-ep`. */\n\tenterpriseDomain?: string;\n\t/** Extra/override request headers. */\n\theaders?: Record<string, string>;\n\t/** Injectable `fetch` for testing. */\n\tfetchImpl?: typeof fetch;\n\t/** Abort signal. */\n\tsignal?: AbortSignal;\n}\n\n/** Fetch and parse the live Copilot model catalog from CAPI `GET {baseUrl}/models`. */\nexport async function fetchCopilotModelCatalog(options: FetchCopilotModelCatalogOptions): Promise<CopilotModelCatalog> {\n\tconst fetchImpl = options.fetchImpl ?? fetch;\n\tconst baseUrl = options.baseUrl ?? copilotApiBaseUrlFromToken(options.token, options.enterpriseDomain);\n\tconst response = await fetchImpl(`${trimTrailingSlash(baseUrl)}/models`, {\n\t\tmethod: \"GET\",\n\t\theaders: {\n\t\t\tAccept: \"application/json\",\n\t\t\tAuthorization: `Bearer ${options.token}`,\n\t\t\t...COPILOT_CATALOG_HEADERS,\n\t\t\t...options.headers,\n\t\t},\n\t\t...(options.signal ? { signal: options.signal } : {}),\n\t});\n\tif (!response.ok) {\n\t\tthrow new Error(`GitHub Copilot /models request failed: ${response.status} ${response.statusText}`);\n\t}\n\treturn parseCopilotModelCatalog(await response.json());\n}\n\n// ----------------------------------------------------------------------------\n// Active in-memory catalog (consulted by the model registry).\n//\n// Empty by default, so with no Copilot auth / no successful fetch the registry leaves Copilot\n// model context windows untouched and the picker never appears.\n// ----------------------------------------------------------------------------\n\nlet activeCatalog: CopilotModelCatalog = new Map();\n\n/** Replace the active catalog the registry derives context windows from. */\nexport function setActiveCopilotModelCatalog(catalog: CopilotModelCatalog): void {\n\tactiveCatalog = catalog;\n}\n\n/** The active catalog (empty until a successful auth-gated fetch/cache load). */\nexport function getActiveCopilotModelCatalog(): CopilotModelCatalog {\n\treturn activeCatalog;\n}\n\n/** Reset the active catalog (primarily for tests). */\nexport function clearActiveCopilotModelCatalog(): void {\n\tactiveCatalog = new Map();\n}\n\n// ----------------------------------------------------------------------------\n// Disk cache.\n// ----------------------------------------------------------------------------\n\ninterface CopilotCatalogCacheFile {\n\tversion: typeof COPILOT_CATALOG_CACHE_VERSION;\n\t/** CAPI host the catalog was fetched from; cache misses on host change (e.g. enterprise switch). */\n\thost: string;\n\t/** Epoch ms the catalog was fetched. */\n\tfetchedAt: number;\n\tmodels: Record<string, CopilotModelContext>;\n}\n\nfunction hostFromBaseUrl(baseUrl: string): string {\n\ttry {\n\t\treturn new URL(baseUrl).host;\n\t} catch {\n\t\treturn baseUrl;\n\t}\n}\n\nexport interface ReadCopilotCatalogCacheOptions {\n\t/** Expected CAPI host; a cached file from a different host is ignored. */\n\thost: string;\n\t/** Current epoch ms (injectable for tests). */\n\tnow?: number;\n\t/** Freshness window; defaults to {@link COPILOT_CATALOG_CACHE_TTL_MS}. */\n\tttlMs?: number;\n}\n\nfunction sanitizeCachedContext(value: unknown): CopilotModelContext | undefined {\n\tconst record = asRecord(value);\n\tconst contextWindow = toPositiveInt(record?.contextWindow);\n\tif (contextWindow === undefined) return undefined;\n\tconst rawOptions = record?.contextWindowOptions;\n\tif (Array.isArray(rawOptions)) {\n\t\tconst options = rawOptions.map(toPositiveInt).filter((n): n is number => n !== undefined);\n\t\tif (options.length > 1) return { contextWindow, contextWindowOptions: options };\n\t}\n\treturn { contextWindow };\n}\n\n/** Read a fresh, host-matching catalog from the cache file, or `undefined` if missing/stale/invalid. */\nexport function readCopilotCatalogCache(\n\tpath: string,\n\toptions: ReadCopilotCatalogCacheOptions,\n): CopilotModelCatalog | undefined {\n\tlet parsed: CopilotCatalogCacheFile;\n\ttry {\n\t\tif (!existsSync(path)) return undefined;\n\t\tparsed = JSON.parse(readFileSync(path, \"utf8\")) as CopilotCatalogCacheFile;\n\t} catch {\n\t\treturn undefined;\n\t}\n\tif (!parsed || parsed.version !== COPILOT_CATALOG_CACHE_VERSION) return undefined;\n\tif (parsed.host !== options.host) return undefined;\n\tconst now = options.now ?? Date.now();\n\tconst ttlMs = options.ttlMs ?? COPILOT_CATALOG_CACHE_TTL_MS;\n\tif (typeof parsed.fetchedAt !== \"number\" || now - parsed.fetchedAt >= ttlMs) return undefined;\n\tconst models = asRecord(parsed.models);\n\tif (!models) return undefined;\n\n\tconst catalog = new Map<string, CopilotModelContext>();\n\tfor (const [id, value] of Object.entries(models)) {\n\t\tconst context = sanitizeCachedContext(value);\n\t\tif (context) catalog.set(id, context);\n\t}\n\treturn catalog;\n}\n\n/** Write the catalog to the cache file (creating parent dirs). Best-effort; never throws. */\nexport function writeCopilotCatalogCache(\n\tpath: string,\n\tbaseUrl: string,\n\tcatalog: CopilotModelCatalog,\n\tnow?: number,\n): void {\n\tconst payload: CopilotCatalogCacheFile = {\n\t\tversion: COPILOT_CATALOG_CACHE_VERSION,\n\t\thost: hostFromBaseUrl(baseUrl),\n\t\tfetchedAt: now ?? Date.now(),\n\t\tmodels: Object.fromEntries(catalog),\n\t};\n\ttry {\n\t\tmkdirSync(dirname(path), { recursive: true });\n\t\twriteFileSync(path, JSON.stringify(payload), \"utf8\");\n\t} catch {\n\t\t// best-effort cache; ignore write failures\n\t}\n}\n\n/** Host component of a base URL, for matching {@link readCopilotCatalogCache} `host`. */\nexport function copilotCatalogCacheHost(baseUrl: string): string {\n\treturn hostFromBaseUrl(baseUrl);\n}\n\n/** Standard on-disk cache path for the Copilot model catalog under an agent directory. */\nexport function copilotCatalogCachePath(agentDir: string): string {\n\treturn join(agentDir, \"cache\", \"copilot-models.json\");\n}\n\n/**\n * Seed the active catalog synchronously from the on-disk cache, gated on a Copilot access token.\n *\n * Called at model-registry construction so a returning user's previously selected long-context\n * window is recognized before startup validation runs — otherwise the persisted choice would warn\n * (\"context window 936k is not supported…\") and reset until the async refresh completes. The cache\n * TTL is intentionally ignored here: stale-but-present windows are still valid for selection, and\n * the async loader independently refetches on its own freshness window. Returns true when a catalog\n * was applied. No-op (returns false) without a token or a host-matching cached catalog.\n */\nexport function seedActiveCopilotModelCatalogFromCache(\n\taccessToken: string | undefined,\n\tcachePath: string,\n\tnow?: number,\n): boolean {\n\tif (typeof accessToken !== \"string\" || accessToken.length === 0) return false;\n\tconst host = copilotCatalogCacheHost(copilotApiBaseUrlFromToken(accessToken));\n\tconst cached = readCopilotCatalogCache(cachePath, { host, now, ttlMs: Number.POSITIVE_INFINITY });\n\tif (!cached) return false;\n\tsetActiveCopilotModelCatalog(cached);\n\treturn true;\n}\n"]}