atom-lite 0.0.0-stage → 0.4.1

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 (250) hide show
  1. package/CHANGELOG.md +72 -0
  2. package/LICENSE +21 -0
  3. package/README.md +313 -2
  4. package/dist/agent/loop.d.ts +32 -0
  5. package/dist/agent/loop.js +204 -0
  6. package/dist/agent/loop.js.map +1 -0
  7. package/dist/agent/types.d.ts +85 -0
  8. package/dist/agent/types.js +2 -0
  9. package/dist/agent/types.js.map +1 -0
  10. package/dist/cli/activity.d.ts +45 -0
  11. package/dist/cli/activity.js +141 -0
  12. package/dist/cli/activity.js.map +1 -0
  13. package/dist/cli/app.d.ts +1 -0
  14. package/dist/cli/app.js +1576 -0
  15. package/dist/cli/app.js.map +1 -0
  16. package/dist/cli/approval-prompt.d.ts +24 -0
  17. package/dist/cli/approval-prompt.js +82 -0
  18. package/dist/cli/approval-prompt.js.map +1 -0
  19. package/dist/cli/approvals.d.ts +1 -0
  20. package/dist/cli/approvals.js +2 -0
  21. package/dist/cli/approvals.js.map +1 -0
  22. package/dist/cli/banner.d.ts +12 -0
  23. package/dist/cli/banner.js +35 -0
  24. package/dist/cli/banner.js.map +1 -0
  25. package/dist/cli/commands.d.ts +94 -0
  26. package/dist/cli/commands.js +324 -0
  27. package/dist/cli/commands.js.map +1 -0
  28. package/dist/cli/diff.d.ts +1 -0
  29. package/dist/cli/diff.js +28 -0
  30. package/dist/cli/diff.js.map +1 -0
  31. package/dist/cli/input.d.ts +2 -0
  32. package/dist/cli/input.js +10 -0
  33. package/dist/cli/input.js.map +1 -0
  34. package/dist/cli/main.d.ts +2 -0
  35. package/dist/cli/main.js +22 -0
  36. package/dist/cli/main.js.map +1 -0
  37. package/dist/cli/mouse.d.ts +54 -0
  38. package/dist/cli/mouse.js +83 -0
  39. package/dist/cli/mouse.js.map +1 -0
  40. package/dist/cli/palette.d.ts +41 -0
  41. package/dist/cli/palette.js +212 -0
  42. package/dist/cli/palette.js.map +1 -0
  43. package/dist/cli/panel.d.ts +29 -0
  44. package/dist/cli/panel.js +48 -0
  45. package/dist/cli/panel.js.map +1 -0
  46. package/dist/cli/picker.d.ts +34 -0
  47. package/dist/cli/picker.js +173 -0
  48. package/dist/cli/picker.js.map +1 -0
  49. package/dist/cli/renderer.d.ts +1 -0
  50. package/dist/cli/renderer.js +5 -0
  51. package/dist/cli/renderer.js.map +1 -0
  52. package/dist/cli/secret-input.d.ts +21 -0
  53. package/dist/cli/secret-input.js +101 -0
  54. package/dist/cli/secret-input.js.map +1 -0
  55. package/dist/cli/status-line.d.ts +24 -0
  56. package/dist/cli/status-line.js +49 -0
  57. package/dist/cli/status-line.js.map +1 -0
  58. package/dist/cli/stream-text.d.ts +18 -0
  59. package/dist/cli/stream-text.js +63 -0
  60. package/dist/cli/stream-text.js.map +1 -0
  61. package/dist/cli/theme.d.ts +48 -0
  62. package/dist/cli/theme.js +158 -0
  63. package/dist/cli/theme.js.map +1 -0
  64. package/dist/cli/tool-box.d.ts +41 -0
  65. package/dist/cli/tool-box.js +177 -0
  66. package/dist/cli/tool-box.js.map +1 -0
  67. package/dist/cli/transcript.d.ts +26 -0
  68. package/dist/cli/transcript.js +64 -0
  69. package/dist/cli/transcript.js.map +1 -0
  70. package/dist/config/credentials.d.ts +26 -0
  71. package/dist/config/credentials.js +79 -0
  72. package/dist/config/credentials.js.map +1 -0
  73. package/dist/config/defaults.d.ts +34 -0
  74. package/dist/config/defaults.js +35 -0
  75. package/dist/config/defaults.js.map +1 -0
  76. package/dist/config/load.d.ts +9 -0
  77. package/dist/config/load.js +137 -0
  78. package/dist/config/load.js.map +1 -0
  79. package/dist/config/paths.d.ts +4 -0
  80. package/dist/config/paths.js +14 -0
  81. package/dist/config/paths.js.map +1 -0
  82. package/dist/config/schema.d.ts +132 -0
  83. package/dist/config/schema.js +83 -0
  84. package/dist/config/schema.js.map +1 -0
  85. package/dist/config/write.d.ts +20 -0
  86. package/dist/config/write.js +70 -0
  87. package/dist/config/write.js.map +1 -0
  88. package/dist/context/budget.d.ts +2 -0
  89. package/dist/context/budget.js +13 -0
  90. package/dist/context/budget.js.map +1 -0
  91. package/dist/context/builder.d.ts +7 -0
  92. package/dist/context/builder.js +11 -0
  93. package/dist/context/builder.js.map +1 -0
  94. package/dist/context/compaction.d.ts +1 -0
  95. package/dist/context/compaction.js +5 -0
  96. package/dist/context/compaction.js.map +1 -0
  97. package/dist/context/instructions.d.ts +3 -0
  98. package/dist/context/instructions.js +30 -0
  99. package/dist/context/instructions.js.map +1 -0
  100. package/dist/context/manager.d.ts +29 -0
  101. package/dist/context/manager.js +122 -0
  102. package/dist/context/manager.js.map +1 -0
  103. package/dist/extensions/api.d.ts +51 -0
  104. package/dist/extensions/api.js +101 -0
  105. package/dist/extensions/api.js.map +1 -0
  106. package/dist/extensions/discovery.d.ts +26 -0
  107. package/dist/extensions/discovery.js +243 -0
  108. package/dist/extensions/discovery.js.map +1 -0
  109. package/dist/extensions/loader.d.ts +27 -0
  110. package/dist/extensions/loader.js +113 -0
  111. package/dist/extensions/loader.js.map +1 -0
  112. package/dist/extensions/trust.d.ts +4 -0
  113. package/dist/extensions/trust.js +40 -0
  114. package/dist/extensions/trust.js.map +1 -0
  115. package/dist/extensions/types.d.ts +52 -0
  116. package/dist/extensions/types.js +2 -0
  117. package/dist/extensions/types.js.map +1 -0
  118. package/dist/index.d.ts +17 -0
  119. package/dist/index.js +17 -0
  120. package/dist/index.js.map +1 -0
  121. package/dist/mcp/client.d.ts +73 -0
  122. package/dist/mcp/client.js +363 -0
  123. package/dist/mcp/client.js.map +1 -0
  124. package/dist/mcp/format.d.ts +3 -0
  125. package/dist/mcp/format.js +37 -0
  126. package/dist/mcp/format.js.map +1 -0
  127. package/dist/mcp/index.d.ts +6 -0
  128. package/dist/mcp/index.js +7 -0
  129. package/dist/mcp/index.js.map +1 -0
  130. package/dist/mcp/manager.d.ts +53 -0
  131. package/dist/mcp/manager.js +349 -0
  132. package/dist/mcp/manager.js.map +1 -0
  133. package/dist/mcp/stdio-transport.d.ts +26 -0
  134. package/dist/mcp/stdio-transport.js +191 -0
  135. package/dist/mcp/stdio-transport.js.map +1 -0
  136. package/dist/mcp/tools.d.ts +22 -0
  137. package/dist/mcp/tools.js +162 -0
  138. package/dist/mcp/tools.js.map +1 -0
  139. package/dist/mcp/types.d.ts +92 -0
  140. package/dist/mcp/types.js +2 -0
  141. package/dist/mcp/types.js.map +1 -0
  142. package/dist/providers/catalog.d.ts +60 -0
  143. package/dist/providers/catalog.js +289 -0
  144. package/dist/providers/catalog.js.map +1 -0
  145. package/dist/providers/context-window.d.ts +11 -0
  146. package/dist/providers/context-window.js +87 -0
  147. package/dist/providers/context-window.js.map +1 -0
  148. package/dist/providers/factory.d.ts +2 -0
  149. package/dist/providers/factory.js +55 -0
  150. package/dist/providers/factory.js.map +1 -0
  151. package/dist/providers/fake.d.ts +14 -0
  152. package/dist/providers/fake.js +23 -0
  153. package/dist/providers/fake.js.map +1 -0
  154. package/dist/providers/gemini.d.ts +24 -0
  155. package/dist/providers/gemini.js +228 -0
  156. package/dist/providers/gemini.js.map +1 -0
  157. package/dist/providers/index.d.ts +5 -0
  158. package/dist/providers/index.js +4 -0
  159. package/dist/providers/index.js.map +1 -0
  160. package/dist/providers/ollama.d.ts +64 -0
  161. package/dist/providers/ollama.js +433 -0
  162. package/dist/providers/ollama.js.map +1 -0
  163. package/dist/providers/openai-compatible.d.ts +27 -0
  164. package/dist/providers/openai-compatible.js +282 -0
  165. package/dist/providers/openai-compatible.js.map +1 -0
  166. package/dist/providers/registry.d.ts +7 -0
  167. package/dist/providers/registry.js +13 -0
  168. package/dist/providers/registry.js.map +1 -0
  169. package/dist/providers/tool-calls.d.ts +18 -0
  170. package/dist/providers/tool-calls.js +42 -0
  171. package/dist/providers/tool-calls.js.map +1 -0
  172. package/dist/providers/types.d.ts +16 -0
  173. package/dist/providers/types.js +2 -0
  174. package/dist/providers/types.js.map +1 -0
  175. package/dist/runtime/approvals.d.ts +20 -0
  176. package/dist/runtime/approvals.js +20 -0
  177. package/dist/runtime/approvals.js.map +1 -0
  178. package/dist/runtime/executor.d.ts +2 -0
  179. package/dist/runtime/executor.js +4 -0
  180. package/dist/runtime/executor.js.map +1 -0
  181. package/dist/runtime/locks.d.ts +13 -0
  182. package/dist/runtime/locks.js +180 -0
  183. package/dist/runtime/locks.js.map +1 -0
  184. package/dist/runtime/scheduler.d.ts +7 -0
  185. package/dist/runtime/scheduler.js +110 -0
  186. package/dist/runtime/scheduler.js.map +1 -0
  187. package/dist/security/paths.d.ts +1 -0
  188. package/dist/security/paths.js +35 -0
  189. package/dist/security/paths.js.map +1 -0
  190. package/dist/security/process.d.ts +1 -0
  191. package/dist/security/process.js +8 -0
  192. package/dist/security/process.js.map +1 -0
  193. package/dist/security/redaction.d.ts +1 -0
  194. package/dist/security/redaction.js +4 -0
  195. package/dist/security/redaction.js.map +1 -0
  196. package/dist/sessions/events.d.ts +30 -0
  197. package/dist/sessions/events.js +92 -0
  198. package/dist/sessions/events.js.map +1 -0
  199. package/dist/sessions/history.d.ts +18 -0
  200. package/dist/sessions/history.js +76 -0
  201. package/dist/sessions/history.js.map +1 -0
  202. package/dist/sessions/index.d.ts +5 -0
  203. package/dist/sessions/index.js +6 -0
  204. package/dist/sessions/index.js.map +1 -0
  205. package/dist/sessions/recovery.d.ts +6 -0
  206. package/dist/sessions/recovery.js +35 -0
  207. package/dist/sessions/recovery.js.map +1 -0
  208. package/dist/sessions/search.d.ts +16 -0
  209. package/dist/sessions/search.js +136 -0
  210. package/dist/sessions/search.js.map +1 -0
  211. package/dist/sessions/store.d.ts +91 -0
  212. package/dist/sessions/store.js +360 -0
  213. package/dist/sessions/store.js.map +1 -0
  214. package/dist/skills/registry.d.ts +23 -0
  215. package/dist/skills/registry.js +129 -0
  216. package/dist/skills/registry.js.map +1 -0
  217. package/dist/tools/bash.d.ts +2 -0
  218. package/dist/tools/bash.js +159 -0
  219. package/dist/tools/bash.js.map +1 -0
  220. package/dist/tools/diff.d.ts +7 -0
  221. package/dist/tools/diff.js +40 -0
  222. package/dist/tools/diff.js.map +1 -0
  223. package/dist/tools/edit.d.ts +2 -0
  224. package/dist/tools/edit.js +118 -0
  225. package/dist/tools/edit.js.map +1 -0
  226. package/dist/tools/index.d.ts +8 -0
  227. package/dist/tools/index.js +8 -0
  228. package/dist/tools/index.js.map +1 -0
  229. package/dist/tools/read.d.ts +2 -0
  230. package/dist/tools/read.js +138 -0
  231. package/dist/tools/read.js.map +1 -0
  232. package/dist/tools/registry.d.ts +7 -0
  233. package/dist/tools/registry.js +13 -0
  234. package/dist/tools/registry.js.map +1 -0
  235. package/dist/tools/types.d.ts +27 -0
  236. package/dist/tools/types.js +2 -0
  237. package/dist/tools/types.js.map +1 -0
  238. package/dist/tools/webfetch.d.ts +2 -0
  239. package/dist/tools/webfetch.js +97 -0
  240. package/dist/tools/webfetch.js.map +1 -0
  241. package/dist/tools/websearch.d.ts +2 -0
  242. package/dist/tools/websearch.js +88 -0
  243. package/dist/tools/websearch.js.map +1 -0
  244. package/dist/tools/write.d.ts +2 -0
  245. package/dist/tools/write.js +91 -0
  246. package/dist/tools/write.js.map +1 -0
  247. package/dist/types.d.ts +84 -0
  248. package/dist/types.js +10 -0
  249. package/dist/types.js.map +1 -0
  250. package/package.json +74 -4
@@ -0,0 +1,87 @@
1
+ // Context-window metadata helpers. A model's window is resolved from (in
2
+ // order): an explicit provider config override, live provider metadata, or the
3
+ // known-model table below.
4
+ /** Extract a context window from an OpenAI-compatible /models entry. */
5
+ export function contextWindowFromModelEntry(entry) {
6
+ if (!entry || typeof entry !== "object")
7
+ return undefined;
8
+ const e = entry;
9
+ const candidates = [
10
+ e.context_length,
11
+ e.context_window,
12
+ e.contextWindow,
13
+ e.max_context_length,
14
+ e.max_context_window,
15
+ e.limits?.context,
16
+ e.top_provider?.context_length,
17
+ e.top_provider?.context_window,
18
+ ];
19
+ for (const value of candidates) {
20
+ if (typeof value === "number" && Number.isFinite(value) && value > 0) {
21
+ return Math.floor(value);
22
+ }
23
+ }
24
+ return undefined;
25
+ }
26
+ /**
27
+ * Best-effort context windows for models whose provider does not expose one.
28
+ * Most specific patterns come first. These are display/compaction defaults and
29
+ * are always overridden by live provider metadata when available.
30
+ */
31
+ const KNOWN_WINDOWS = [
32
+ [/gpt-4\.1/, 1_047_576],
33
+ [/gpt-4o|gpt-4-turbo|gpt-4-0125|gpt-4-1106/, 128_000],
34
+ [/o3|o4-mini/, 200_000],
35
+ [/gpt-3\.5/, 16_385],
36
+ [/claude/, 200_000],
37
+ [/gemini-1\.5-pro/, 2_097_152],
38
+ [/gemini-1\.5|gemini-2/, 1_048_576],
39
+ [/grok/, 131_072],
40
+ [/codestral/, 256_000],
41
+ [/mistral-large/, 128_000],
42
+ [/llama-3|llama3|llama-4|mixtral|deepseek|qwen/, 131_072],
43
+ [/kilo-auto\/free/, 256_000],
44
+ [/kilo-auto\/efficient/, 1_000_000],
45
+ ];
46
+ /** Compact token count used by the status line: 999, 1.2k, 256k, 1M. */
47
+ export function formatTokenCount(count) {
48
+ if (!Number.isFinite(count) || count < 0)
49
+ return "0";
50
+ const n = Math.floor(count);
51
+ if (n < 1000)
52
+ return String(n);
53
+ if (n < 1_000_000) {
54
+ if (n < 10_000) {
55
+ const k = (n / 1000).toFixed(1).replace(/\.0$/, "");
56
+ return `${k}k`;
57
+ }
58
+ const rounded = Math.round(n / 1000);
59
+ return rounded >= 1000 ? "1M" : `${rounded}k`;
60
+ }
61
+ if (n < 10_000_000) {
62
+ return `${(n / 1_000_000).toFixed(1).replace(/\.0$/, "")}M`;
63
+ }
64
+ return `${Math.round(n / 1_000_000)}M`;
65
+ }
66
+ /**
67
+ * Render `used/window`, matching the compact peer style (`1.2k/256k`).
68
+ * Unknown values are shown as `?` rather than a fake number.
69
+ */
70
+ export function formatContextUsage(used, window) {
71
+ const usedText = used !== undefined && used > 0 ? formatTokenCount(used) : used === 0 ? "0" : "?";
72
+ if (window === undefined || window <= 0)
73
+ return usedText;
74
+ return `${usedText}/${formatTokenCount(window)}`;
75
+ }
76
+ /** Guess a context window from the model id when no metadata is available. */
77
+ export function knownContextWindow(model) {
78
+ if (!model)
79
+ return undefined;
80
+ const id = model.toLowerCase();
81
+ for (const [pattern, window] of KNOWN_WINDOWS) {
82
+ if (pattern.test(id))
83
+ return window;
84
+ }
85
+ return undefined;
86
+ }
87
+ //# sourceMappingURL=context-window.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"context-window.js","sourceRoot":"","sources":["../../src/providers/context-window.ts"],"names":[],"mappings":"AAAA,yEAAyE;AACzE,+EAA+E;AAC/E,2BAA2B;AAE3B,wEAAwE;AACxE,MAAM,UAAU,2BAA2B,CAAC,KAAc;IACxD,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC;IAC1D,MAAM,CAAC,GAAG,KAA4B,CAAC;IACvC,MAAM,UAAU,GAAG;QACjB,CAAC,CAAC,cAAc;QAChB,CAAC,CAAC,cAAc;QAChB,CAAC,CAAC,aAAa;QACf,CAAC,CAAC,kBAAkB;QACpB,CAAC,CAAC,kBAAkB;QACpB,CAAC,CAAC,MAAM,EAAE,OAAO;QACjB,CAAC,CAAC,YAAY,EAAE,cAAc;QAC9B,CAAC,CAAC,YAAY,EAAE,cAAc;KAC/B,CAAC;IACF,KAAK,MAAM,KAAK,IAAI,UAAU,EAAE,CAAC;QAC/B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;YACrE,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;;;GAIG;AACH,MAAM,aAAa,GAA4B;IAC7C,CAAC,UAAU,EAAE,SAAS,CAAC;IACvB,CAAC,0CAA0C,EAAE,OAAO,CAAC;IACrD,CAAC,YAAY,EAAE,OAAO,CAAC;IACvB,CAAC,UAAU,EAAE,MAAM,CAAC;IACpB,CAAC,QAAQ,EAAE,OAAO,CAAC;IACnB,CAAC,iBAAiB,EAAE,SAAS,CAAC;IAC9B,CAAC,sBAAsB,EAAE,SAAS,CAAC;IACnC,CAAC,MAAM,EAAE,OAAO,CAAC;IACjB,CAAC,WAAW,EAAE,OAAO,CAAC;IACtB,CAAC,eAAe,EAAE,OAAO,CAAC;IAC1B,CAAC,8CAA8C,EAAE,OAAO,CAAC;IACzD,CAAC,iBAAiB,EAAE,OAAO,CAAC;IAC5B,CAAC,sBAAsB,EAAE,SAAS,CAAC;CACpC,CAAC;AAEF,wEAAwE;AACxE,MAAM,UAAU,gBAAgB,CAAC,KAAa;IAC5C,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC;QAAE,OAAO,GAAG,CAAC;IACrD,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAC5B,IAAI,CAAC,GAAG,IAAI;QAAE,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC;IAC/B,IAAI,CAAC,GAAG,SAAS,EAAE,CAAC;QAClB,IAAI,CAAC,GAAG,MAAM,EAAE,CAAC;YACf,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YACpD,OAAO,GAAG,CAAC,GAAG,CAAC;QACjB,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;QACrC,OAAO,OAAO,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,OAAO,GAAG,CAAC;IAChD,CAAC;IACD,IAAI,CAAC,GAAG,UAAU,EAAE,CAAC;QACnB,OAAO,GAAG,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,GAAG,CAAC;IAC9D,CAAC;IACD,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,SAAS,CAAC,GAAG,CAAC;AACzC,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,kBAAkB,CAChC,IAAwB,EACxB,MAA0B;IAE1B,MAAM,QAAQ,GAAG,IAAI,KAAK,SAAS,IAAI,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC;IAClG,IAAI,MAAM,KAAK,SAAS,IAAI,MAAM,IAAI,CAAC;QAAE,OAAO,QAAQ,CAAC;IACzD,OAAO,GAAG,QAAQ,IAAI,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;AACnD,CAAC;AAED,8EAA8E;AAC9E,MAAM,UAAU,kBAAkB,CAAC,KAAyB;IAC1D,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAC7B,MAAM,EAAE,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IAC/B,KAAK,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,aAAa,EAAE,CAAC;QAC9C,IAAI,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YAAE,OAAO,MAAM,CAAC;IACtC,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC"}
@@ -0,0 +1,2 @@
1
+ import type { Config } from "../config/schema.js";
2
+ export declare function createProvider(config: Config, profileName?: string, modelOverride?: string): any;
@@ -0,0 +1,55 @@
1
+ import { resolveApiKey } from "../config/credentials.js";
2
+ import { OpenAICompatibleAdapter } from "./openai-compatible.js";
3
+ import { OllamaAdapter } from "./ollama.js";
4
+ import { GeminiAdapter } from "./gemini.js";
5
+ import { FakeProvider } from "./fake.js";
6
+ export function createProvider(config, profileName, modelOverride) {
7
+ const name = profileName ?? config.defaultProvider;
8
+ const pcfg = config.providers[name];
9
+ if (!pcfg)
10
+ return new FakeProvider([{ text: "No provider configured; using fake." }]);
11
+ if (pcfg.type === "openai-compatible") {
12
+ const apiKey = resolveApiKey(name, {
13
+ apiKeyEnv: pcfg.apiKeyEnv,
14
+ inline: pcfg.apiKey,
15
+ });
16
+ return new OpenAICompatibleAdapter({
17
+ id: name,
18
+ baseUrl: pcfg.baseUrl ?? "http://127.0.0.1:1234/v1",
19
+ model: modelOverride ?? pcfg.model,
20
+ apiKey,
21
+ headers: pcfg.headers,
22
+ contextWindow: pcfg.contextWindow,
23
+ });
24
+ }
25
+ if (pcfg.type === "ollama") {
26
+ return new OllamaAdapter({
27
+ id: name,
28
+ baseUrl: pcfg.baseUrl ?? "http://127.0.0.1:11434",
29
+ model: modelOverride ?? pcfg.model,
30
+ headers: pcfg.headers,
31
+ keepAlive: pcfg.keepAlive,
32
+ think: pcfg.think,
33
+ options: pcfg.options,
34
+ contextWindow: pcfg.contextWindow,
35
+ });
36
+ }
37
+ if (pcfg.type === "gemini") {
38
+ const apiKey = resolveApiKey(name, {
39
+ apiKeyEnv: pcfg.apiKeyEnv,
40
+ inline: pcfg.apiKey,
41
+ });
42
+ return new GeminiAdapter({
43
+ id: name,
44
+ baseUrl: pcfg.baseUrl,
45
+ model: modelOverride ?? pcfg.model,
46
+ apiKey,
47
+ headers: pcfg.headers,
48
+ contextWindow: pcfg.contextWindow,
49
+ });
50
+ }
51
+ return new FakeProvider([
52
+ { text: `Provider type ${pcfg.type} not implemented yet.` },
53
+ ]);
54
+ }
55
+ //# sourceMappingURL=factory.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"factory.js","sourceRoot":"","sources":["../../src/providers/factory.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AACzD,OAAO,EAAE,uBAAuB,EAAE,MAAM,wBAAwB,CAAC;AACjE,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAEzC,MAAM,UAAU,cAAc,CAC5B,MAAc,EACd,WAAoB,EACpB,aAAsB;IAEtB,MAAM,IAAI,GAAG,WAAW,IAAI,MAAM,CAAC,eAAe,CAAC;IACnD,MAAM,IAAI,GAAI,MAAM,CAAC,SAAiC,CAAC,IAAI,CAAC,CAAC;IAC7D,IAAI,CAAC,IAAI;QACP,OAAO,IAAI,YAAY,CAAC,CAAC,EAAE,IAAI,EAAE,qCAAqC,EAAE,CAAC,CAAC,CAAC;IAC7E,IAAI,IAAI,CAAC,IAAI,KAAK,mBAAmB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,EAAE;YACjC,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC,CAAC;QACH,OAAO,IAAI,uBAAuB,CAAC;YACjC,EAAE,EAAE,IAAI;YACR,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,0BAA0B;YACnD,KAAK,EAAE,aAAa,IAAI,IAAI,CAAC,KAAK;YAClC,MAAM;YACN,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,aAAa,EAAE,IAAI,CAAC,aAAa;SAClC,CAAC,CAAC;IACL,CAAC;IACD,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC3B,OAAO,IAAI,aAAa,CAAC;YACvB,EAAE,EAAE,IAAI;YACR,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,wBAAwB;YACjD,KAAK,EAAE,aAAa,IAAI,IAAI,CAAC,KAAK;YAClC,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,aAAa,EAAE,IAAI,CAAC,aAAa;SAClC,CAAC,CAAC;IACL,CAAC;IACD,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC3B,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,EAAE;YACjC,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC,CAAC;QACH,OAAO,IAAI,aAAa,CAAC;YACvB,EAAE,EAAE,IAAI;YACR,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,KAAK,EAAE,aAAa,IAAI,IAAI,CAAC,KAAK;YAClC,MAAM;YACN,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,aAAa,EAAE,IAAI,CAAC,aAAa;SAClC,CAAC,CAAC;IACL,CAAC;IACD,OAAO,IAAI,YAAY,CAAC;QACtB,EAAE,IAAI,EAAE,iBAAiB,IAAI,CAAC,IAAI,uBAAuB,EAAE;KAC5D,CAAC,CAAC;AACL,CAAC"}
@@ -0,0 +1,14 @@
1
+ import type { ModelRequest, ModelResponse, TokenUsage, ToolCall } from "../types.js";
2
+ import type { ProviderAdapter } from "./types.js";
3
+ export interface FakeResponse {
4
+ text?: string;
5
+ toolCalls?: ToolCall[];
6
+ usage?: TokenUsage;
7
+ }
8
+ export declare class FakeProvider implements ProviderAdapter {
9
+ readonly id = "fake";
10
+ private responses;
11
+ private index;
12
+ constructor(responses: FakeResponse[]);
13
+ generate(_request: ModelRequest): Promise<ModelResponse>;
14
+ }
@@ -0,0 +1,23 @@
1
+ export class FakeProvider {
2
+ id = "fake";
3
+ responses;
4
+ index = 0;
5
+ constructor(responses) {
6
+ this.responses = responses;
7
+ }
8
+ async generate(_request) {
9
+ const r = this.responses[this.index++] ?? { text: "" };
10
+ const toolCalls = r.toolCalls ?? [];
11
+ return {
12
+ message: {
13
+ role: "assistant",
14
+ content: r.text ? [{ type: "text", text: r.text }] : undefined,
15
+ toolCalls: toolCalls.length ? toolCalls : undefined,
16
+ },
17
+ toolCalls,
18
+ finishReason: toolCalls.length ? "tool_calls" : "stop",
19
+ usage: r.usage,
20
+ };
21
+ }
22
+ }
23
+ //# sourceMappingURL=fake.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"fake.js","sourceRoot":"","sources":["../../src/providers/fake.ts"],"names":[],"mappings":"AASA,MAAM,OAAO,YAAY;IACd,EAAE,GAAG,MAAM,CAAC;IACb,SAAS,CAAiB;IAC1B,KAAK,GAAG,CAAC,CAAC;IAElB,YAAY,SAAyB;QACnC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,QAAsB;QACnC,MAAM,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;QACvD,MAAM,SAAS,GAAG,CAAC,CAAC,SAAS,IAAI,EAAE,CAAC;QACpC,OAAO;YACL,OAAO,EAAE;gBACP,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS;gBAC9D,SAAS,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;aACpD;YACD,SAAS;YACT,YAAY,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM;YACtD,KAAK,EAAE,CAAC,CAAC,KAAK;SACf,CAAC;IACJ,CAAC;CACF"}
@@ -0,0 +1,24 @@
1
+ import type { ModelRequest, ModelResponse, ModelStreamEvent } from "../types.js";
2
+ import type { ProviderAdapter } from "./types.js";
3
+ export interface GeminiConfig {
4
+ id: string;
5
+ model: string;
6
+ apiKey?: string | null;
7
+ baseUrl?: string;
8
+ headers?: Record<string, string>;
9
+ contextWindow?: number;
10
+ }
11
+ export declare class GeminiAdapter implements ProviderAdapter {
12
+ readonly id: string;
13
+ model: string;
14
+ private apiKey;
15
+ private baseUrl;
16
+ private contextWindowOverride;
17
+ private modelList?;
18
+ constructor(config: GeminiConfig);
19
+ private fetchModelList;
20
+ listModels(): Promise<string[]>;
21
+ getModelInfo(): Promise<import("../types.js").ModelInfo | undefined>;
22
+ generate(request: ModelRequest): Promise<ModelResponse>;
23
+ stream(request: ModelRequest): AsyncIterable<ModelStreamEvent>;
24
+ }
@@ -0,0 +1,228 @@
1
+ import { parseArgs } from "./tool-calls.js";
2
+ function toGeminiContents(messages) {
3
+ const systemParts = [];
4
+ const contents = [];
5
+ for (const m of messages) {
6
+ if (m.role === "system") {
7
+ systemParts.push({ text: m.content });
8
+ continue;
9
+ }
10
+ if (m.role === "user") {
11
+ contents.push({
12
+ role: "user",
13
+ parts: m.content.map((p) => p.type === "text" ? { text: p.text } : p),
14
+ });
15
+ continue;
16
+ }
17
+ if (m.role === "assistant") {
18
+ const parts = [];
19
+ const text = m.content
20
+ ?.map((p) => (p.type === "text" ? p.text : ""))
21
+ .join("") ?? "";
22
+ if (text)
23
+ parts.push({ text });
24
+ if (m.toolCalls?.length) {
25
+ for (const tc of m.toolCalls) {
26
+ parts.push({
27
+ functionCall: { name: tc.name, args: tc.arguments ?? {} },
28
+ });
29
+ }
30
+ }
31
+ // preserve provider state if present
32
+ const state = m.providerState;
33
+ contents.push({
34
+ role: "model",
35
+ parts,
36
+ ...(state ? { _state: state } : {}),
37
+ });
38
+ continue;
39
+ }
40
+ if (m.role === "tool") {
41
+ contents.push({
42
+ role: "function",
43
+ parts: [
44
+ {
45
+ functionResponse: {
46
+ name: m.name,
47
+ response: { result: m.content },
48
+ },
49
+ },
50
+ ],
51
+ });
52
+ continue;
53
+ }
54
+ }
55
+ const systemInstruction = systemParts.length
56
+ ? { parts: systemParts }
57
+ : undefined;
58
+ return { systemInstruction, contents };
59
+ }
60
+ export class GeminiAdapter {
61
+ id;
62
+ model;
63
+ apiKey;
64
+ baseUrl;
65
+ contextWindowOverride;
66
+ modelList;
67
+ constructor(config) {
68
+ this.id = config.id;
69
+ this.model = config.model;
70
+ this.apiKey = config.apiKey ?? null;
71
+ this.baseUrl = (config.baseUrl ?? "https://generativelanguage.googleapis.com").replace(/\/$/, "");
72
+ this.contextWindowOverride = config.contextWindow;
73
+ }
74
+ fetchModelList() {
75
+ if (!this.apiKey)
76
+ return Promise.resolve([]);
77
+ if (!this.modelList) {
78
+ this.modelList = (async () => {
79
+ const res = await fetch(`${this.baseUrl}/v1beta/models?key=${encodeURIComponent(this.apiKey)}`);
80
+ if (!res.ok)
81
+ return [];
82
+ const json = (await res.json());
83
+ return json.models ?? [];
84
+ })().catch(() => []);
85
+ }
86
+ return this.modelList;
87
+ }
88
+ async listModels() {
89
+ const models = await this.fetchModelList();
90
+ if (!models.length)
91
+ this.modelList = undefined;
92
+ const ids = models
93
+ .map((m) => (m.name ?? "").replace(/^models\//, ""))
94
+ .filter(Boolean);
95
+ return ids.length ? ids : [this.model];
96
+ }
97
+ async getModelInfo() {
98
+ const models = await this.fetchModelList();
99
+ if (!models.length)
100
+ this.modelList = undefined;
101
+ const entry = models.find((m) => (m.name ?? "").replace(/^models\//, "") === this.model);
102
+ const limit = entry?.inputTokenLimit;
103
+ return {
104
+ id: this.model,
105
+ contextWindow: typeof limit === "number" && limit > 0
106
+ ? Math.floor(limit)
107
+ : this.contextWindowOverride,
108
+ supportsTools: true,
109
+ supportsStreaming: true,
110
+ };
111
+ }
112
+ async generate(request) {
113
+ if (!this.apiKey)
114
+ throw new Error("PROVIDER_AUTH_FAILED: missing API key");
115
+ const { systemInstruction, contents } = toGeminiContents(request.messages);
116
+ const url = `${this.baseUrl}/v1beta/models/${encodeURIComponent(this.model)}:generateContent?key=${encodeURIComponent(this.apiKey)}`;
117
+ const body = { contents };
118
+ if (systemInstruction)
119
+ body.systemInstruction = systemInstruction;
120
+ const effort = request.options?.effort;
121
+ if (effort) {
122
+ const budget = effort === "low" ? 1024 : effort === "high" ? 16384 : 4096;
123
+ body.generationConfig = { ...(body.generationConfig ?? {}), thinkingConfig: { thinkingBudget: budget } };
124
+ }
125
+ if (request.tools && request.tools.length) {
126
+ body.tools = [
127
+ {
128
+ functionDeclarations: request.tools.map((t) => ({
129
+ name: t.name,
130
+ description: t.description,
131
+ parameters: t.parameters,
132
+ })),
133
+ },
134
+ ];
135
+ }
136
+ let res;
137
+ let lastErr;
138
+ for (let attempt = 0; attempt < 3; attempt++) {
139
+ try {
140
+ res = await fetch(url, {
141
+ method: "POST",
142
+ headers: { "Content-Type": "application/json" },
143
+ body: JSON.stringify(body),
144
+ signal: request.signal,
145
+ });
146
+ }
147
+ catch (err) {
148
+ lastErr = new Error(`PROVIDER_CONNECTION_FAILED: ${err instanceof Error ? err.message : String(err)}`);
149
+ if (attempt < 2) {
150
+ await new Promise((r) => setTimeout(r, 200 * (attempt + 1)));
151
+ continue;
152
+ }
153
+ throw lastErr;
154
+ }
155
+ if (res.ok)
156
+ break;
157
+ const text = await res.text().catch(() => "");
158
+ if (res.status === 401 || res.status === 403)
159
+ throw new Error(`PROVIDER_AUTH_FAILED: ${text}`);
160
+ if (res.status === 429) {
161
+ lastErr = new Error(`PROVIDER_RATE_LIMITED: ${text}`);
162
+ if (attempt < 2) {
163
+ await new Promise((r) => setTimeout(r, 500 * (attempt + 1)));
164
+ continue;
165
+ }
166
+ throw lastErr;
167
+ }
168
+ throw new Error(`PROVIDER_INVALID_RESPONSE: ${res.status} ${text}`);
169
+ }
170
+ if (!res || !res.ok)
171
+ throw lastErr ?? new Error("PROVIDER_INVALID_RESPONSE");
172
+ const json = (await res.json());
173
+ const candidate = json.candidates?.[0];
174
+ const parts = candidate?.content?.parts ?? [];
175
+ const thinkingParts = parts
176
+ .filter((p) => p.thought && typeof p.text === "string")
177
+ .map((p) => p.text);
178
+ const textParts = parts
179
+ .filter((p) => !p.thought && typeof p.text === "string")
180
+ .map((p) => p.text);
181
+ const text = textParts.join("");
182
+ const thinking = thinkingParts.join("");
183
+ const toolCalls = parts
184
+ .filter((p) => p.functionCall)
185
+ .map((p, i) => ({
186
+ id: `gemini_${i}`,
187
+ name: p.functionCall.name,
188
+ arguments: parseArgs(JSON.stringify(p.functionCall.args ?? {})),
189
+ }));
190
+ const usage = json.usageMetadata
191
+ ? {
192
+ input: json.usageMetadata.promptTokenCount,
193
+ // Thinking tokens count toward the next request's context, so fold
194
+ // them into generated output.
195
+ output: (json.usageMetadata.candidatesTokenCount ?? 0) +
196
+ (json.usageMetadata.thoughtsTokenCount ?? 0),
197
+ total: json.usageMetadata.totalTokenCount,
198
+ }
199
+ : undefined;
200
+ // safety block handling
201
+ if (candidate?.finishReason === "SAFETY") {
202
+ throw new Error("PROVIDER_INVALID_RESPONSE: safety block");
203
+ }
204
+ return {
205
+ message: {
206
+ role: "assistant",
207
+ content: text ? [{ type: "text", text }] : undefined,
208
+ toolCalls: toolCalls.length ? toolCalls : undefined,
209
+ providerState: candidate?.content,
210
+ },
211
+ toolCalls,
212
+ finishReason: toolCalls.length ? "tool_calls" : "stop",
213
+ usage,
214
+ metadata: thinking ? { thinking } : undefined,
215
+ };
216
+ }
217
+ async *stream(request) {
218
+ const resp = await this.generate(request);
219
+ const thinking = resp.metadata?.thinking;
220
+ if (thinking)
221
+ yield { type: "thinking", text: thinking };
222
+ if (resp.message.content?.[0]?.type === "text") {
223
+ yield { type: "text", text: resp.message.content[0].text };
224
+ }
225
+ yield { type: "done", response: resp };
226
+ }
227
+ }
228
+ //# sourceMappingURL=gemini.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../src/providers/gemini.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAW5C,SAAS,gBAAgB,CAAC,QAAmB;IAI3C,MAAM,WAAW,GAAU,EAAE,CAAC;IAC9B,MAAM,QAAQ,GAAU,EAAE,CAAC;IAC3B,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;QACzB,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACxB,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;YACtC,SAAS;QACX,CAAC;QACD,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACtB,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAC9B,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CACzC;aACF,CAAC,CAAC;YACH,SAAS;QACX,CAAC;QACD,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC3B,MAAM,KAAK,GAAU,EAAE,CAAC;YACxB,MAAM,IAAI,GACR,CAAC,CAAC,OAAO;gBACP,EAAE,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;iBACnD,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YACpB,IAAI,IAAI;gBAAE,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;YAC/B,IAAI,CAAC,CAAC,SAAS,EAAE,MAAM,EAAE,CAAC;gBACxB,KAAK,MAAM,EAAE,IAAI,CAAC,CAAC,SAAS,EAAE,CAAC;oBAC7B,KAAK,CAAC,IAAI,CAAC;wBACT,YAAY,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC,SAAS,IAAI,EAAE,EAAE;qBAC1D,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YACD,qCAAqC;YACrC,MAAM,KAAK,GAAI,CAAS,CAAC,aAAa,CAAC;YACvC,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,OAAO;gBACb,KAAK;gBACL,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACpC,CAAC,CAAC;YACH,SAAS;QACX,CAAC;QACD,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACtB,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,UAAU;gBAChB,KAAK,EAAE;oBACL;wBACE,gBAAgB,EAAE;4BAChB,IAAI,EAAE,CAAC,CAAC,IAAI;4BACZ,QAAQ,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE;yBAChC;qBACF;iBACF;aACF,CAAC,CAAC;YACH,SAAS;QACX,CAAC;IACH,CAAC;IACD,MAAM,iBAAiB,GAAG,WAAW,CAAC,MAAM;QAC1C,CAAC,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE;QACxB,CAAC,CAAC,SAAS,CAAC;IACd,OAAO,EAAE,iBAAiB,EAAE,QAAQ,EAAE,CAAC;AACzC,CAAC;AAED,MAAM,OAAO,aAAa;IACf,EAAE,CAAS;IACb,KAAK,CAAS;IACb,MAAM,CAAgB;IACtB,OAAO,CAAS;IAChB,qBAAqB,CAAqB;IAC1C,SAAS,CAAkB;IAEnC,YAAY,MAAoB;QAC9B,IAAI,CAAC,EAAE,GAAG,MAAM,CAAC,EAAE,CAAC;QACpB,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC1B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC;QACpC,IAAI,CAAC,OAAO,GAAG,CACb,MAAM,CAAC,OAAO,IAAI,2CAA2C,CAC9D,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACrB,IAAI,CAAC,qBAAqB,GAAG,MAAM,CAAC,aAAa,CAAC;IACpD,CAAC;IAEO,cAAc;QACpB,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC7C,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,GAAG,CAAC,KAAK,IAAI,EAAE;gBAC3B,MAAM,GAAG,GAAG,MAAM,KAAK,CACrB,GAAG,IAAI,CAAC,OAAO,sBAAsB,kBAAkB,CAAC,IAAI,CAAC,MAAO,CAAC,EAAE,CACxE,CAAC;gBACF,IAAI,CAAC,GAAG,CAAC,EAAE;oBAAE,OAAO,EAAE,CAAC;gBACvB,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAQ,CAAC;gBACvC,OAAO,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC;YAC3B,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;QACvB,CAAC;QACD,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC3C,IAAI,CAAC,MAAM,CAAC,MAAM;YAAE,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC/C,MAAM,GAAG,GAAG,MAAM;aACf,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;aACxD,MAAM,CAAC,OAAO,CAAC,CAAC;QACnB,OAAO,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC3C,IAAI,CAAC,MAAM,CAAC,MAAM;YAAE,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC/C,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CACvB,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,KAAK,IAAI,CAAC,KAAK,CACnE,CAAC;QACF,MAAM,KAAK,GAAG,KAAK,EAAE,eAAe,CAAC;QACrC,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,KAAK;YACd,aAAa,EACX,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,GAAG,CAAC;gBACpC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC;gBACnB,CAAC,CAAC,IAAI,CAAC,qBAAqB;YAChC,aAAa,EAAE,IAAI;YACnB,iBAAiB,EAAE,IAAI;SACxB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAAqB;QAClC,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;QAC3E,MAAM,EAAE,iBAAiB,EAAE,QAAQ,EAAE,GAAG,gBAAgB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC3E,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,kBAAkB,kBAAkB,CAAC,IAAI,CAAC,KAAK,CAAC,wBAAwB,kBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;QACrI,MAAM,IAAI,GAAQ,EAAE,QAAQ,EAAE,CAAC;QAC/B,IAAI,iBAAiB;YAAE,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAClE,MAAM,MAAM,GAAI,OAAO,CAAC,OAAe,EAAE,MAAM,CAAC;QAChD,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,MAAM,GAAG,MAAM,KAAK,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;YAC1E,IAAI,CAAC,gBAAgB,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,gBAAgB,IAAI,EAAE,CAAC,EAAE,cAAc,EAAE,EAAE,cAAc,EAAE,MAAM,EAAE,EAAE,CAAC;QAC3G,CAAC;QACD,IAAI,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;YAC1C,IAAI,CAAC,KAAK,GAAG;gBACX;oBACE,oBAAoB,EAAE,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBAC9C,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;wBAC1B,UAAU,EAAE,CAAC,CAAC,UAAU;qBACzB,CAAC,CAAC;iBACJ;aACF,CAAC;QACJ,CAAC;QACD,IAAI,GAAyB,CAAC;QAC9B,IAAI,OAA0B,CAAC;QAC/B,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC;YAC7C,IAAI,CAAC;gBACH,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;oBACrB,MAAM,EAAE,MAAM;oBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;oBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;oBAC1B,MAAM,EAAE,OAAO,CAAC,MAAM;iBACvB,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,GAAG,IAAI,KAAK,CACjB,+BAA+B,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAClF,CAAC;gBACF,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;oBAChB,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,GAAG,GAAG,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC7D,SAAS;gBACX,CAAC;gBACD,MAAM,OAAO,CAAC;YAChB,CAAC;YACD,IAAI,GAAG,CAAC,EAAE;gBAAE,MAAM;YAClB,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;YAC9C,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG;gBAC1C,MAAM,IAAI,KAAK,CAAC,yBAAyB,IAAI,EAAE,CAAC,CAAC;YACnD,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACvB,OAAO,GAAG,IAAI,KAAK,CAAC,0BAA0B,IAAI,EAAE,CAAC,CAAC;gBACtD,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;oBAChB,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,GAAG,GAAG,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC7D,SAAS;gBACX,CAAC;gBACD,MAAM,OAAO,CAAC;YAChB,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,8BAA8B,GAAG,CAAC,MAAM,IAAI,IAAI,EAAE,CAAC,CAAC;QACtE,CAAC;QACD,IAAI,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE;YACjB,MAAM,OAAO,IAAI,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC1D,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAQ,CAAC;QACvC,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,KAAK,GAAG,SAAS,EAAE,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC;QAC9C,MAAM,aAAa,GAAG,KAAK;aACxB,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;aAC3D,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC3B,MAAM,SAAS,GAAG,KAAK;aACpB,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,IAAI,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;aAC5D,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC3B,MAAM,IAAI,GAAG,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAChC,MAAM,QAAQ,GAAG,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACxC,MAAM,SAAS,GAAe,KAAK;aAChC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC;aAClC,GAAG,CAAC,CAAC,CAAM,EAAE,CAAS,EAAE,EAAE,CAAC,CAAC;YAC3B,EAAE,EAAE,UAAU,CAAC,EAAE;YACjB,IAAI,EAAE,CAAC,CAAC,YAAY,CAAC,IAAI;YACzB,SAAS,EAAE,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;SAChE,CAAC,CAAC,CAAC;QACN,MAAM,KAAK,GAA2B,IAAI,CAAC,aAAa;YACtD,CAAC,CAAC;gBACE,KAAK,EAAE,IAAI,CAAC,aAAa,CAAC,gBAAgB;gBAC1C,mEAAmE;gBACnE,8BAA8B;gBAC9B,MAAM,EACJ,CAAC,IAAI,CAAC,aAAa,CAAC,oBAAoB,IAAI,CAAC,CAAC;oBAC9C,CAAC,IAAI,CAAC,aAAa,CAAC,kBAAkB,IAAI,CAAC,CAAC;gBAC9C,KAAK,EAAE,IAAI,CAAC,aAAa,CAAC,eAAe;aAC1C;YACH,CAAC,CAAC,SAAS,CAAC;QACd,wBAAwB;QACxB,IAAI,SAAS,EAAE,YAAY,KAAK,QAAQ,EAAE,CAAC;YACzC,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;QAC7D,CAAC;QACD,OAAO;YACL,OAAO,EAAE;gBACP,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS;gBACpD,SAAS,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;gBACnD,aAAa,EAAE,SAAS,EAAE,OAAO;aAClC;YACD,SAAS;YACT,YAAY,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM;YACtD,KAAK;YACL,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,SAAS;SAC9C,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,OAAqB;QACjC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAC1C,MAAM,QAAQ,GAAI,IAAI,CAAC,QAAgB,EAAE,QAAQ,CAAC;QAClD,IAAI,QAAQ;YAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;QACzD,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;YAC/C,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAS,CAAC,IAAI,EAAE,CAAC;QACtE,CAAC;QACD,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;IACzC,CAAC;CACF"}
@@ -0,0 +1,5 @@
1
+ export { FakeProvider } from "./fake.js";
2
+ export { InMemoryProviderRegistry } from "./registry.js";
3
+ export { OllamaAdapter, OllamaProvider } from "./ollama.js";
4
+ export type { OllamaConfig } from "./ollama.js";
5
+ export type { ProviderAdapter, ProviderRegistry } from "./types.js";
@@ -0,0 +1,4 @@
1
+ export { FakeProvider } from "./fake.js";
2
+ export { InMemoryProviderRegistry } from "./registry.js";
3
+ export { OllamaAdapter, OllamaProvider } from "./ollama.js";
4
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,wBAAwB,EAAE,MAAM,eAAe,CAAC;AACzD,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC"}
@@ -0,0 +1,64 @@
1
+ import type { ModelRequest, ModelResponse, ModelStreamEvent, ModelInfo } from "../types.js";
2
+ import type { ProviderAdapter } from "./types.js";
3
+ export interface OllamaConfig {
4
+ id?: string;
5
+ baseUrl?: string;
6
+ model: string;
7
+ apiKey?: string | null;
8
+ headers?: Record<string, string>;
9
+ keepAlive?: string | number;
10
+ think?: boolean | string;
11
+ options?: Record<string, unknown>;
12
+ contextWindow?: number;
13
+ supportsTools?: boolean;
14
+ supportsStreaming?: boolean;
15
+ }
16
+ export declare class OllamaAdapter implements ProviderAdapter {
17
+ readonly id: string;
18
+ private baseUrl;
19
+ model: string;
20
+ private headers;
21
+ private keepAlive;
22
+ private think;
23
+ private options;
24
+ private contextWindow;
25
+ private supportsToolsFlag;
26
+ private supportsStreamingFlag;
27
+ constructor(config: OllamaConfig);
28
+ get capabilities(): {
29
+ toolCalling: boolean;
30
+ streaming: boolean;
31
+ contextWindow: number | undefined;
32
+ };
33
+ get supportsTools(): boolean;
34
+ get supportsStreaming(): boolean;
35
+ private buildHeaders;
36
+ private resolveModel;
37
+ private toOllamaMessages;
38
+ private buildBody;
39
+ private parseToolCalls;
40
+ listModels(): Promise<string[]>;
41
+ generate(request: ModelRequest): Promise<ModelResponse>;
42
+ stream(request: ModelRequest): AsyncIterable<ModelStreamEvent>;
43
+ getModelInfo(): Promise<ModelInfo | undefined>;
44
+ doctor(): Promise<{
45
+ ok: boolean;
46
+ connectivity: boolean;
47
+ model: string;
48
+ supportsTools: boolean;
49
+ contextWindow: number | undefined;
50
+ error?: string;
51
+ }>;
52
+ healthCheck(): Promise<{
53
+ ok: boolean;
54
+ connectivity: boolean;
55
+ model: string;
56
+ supportsTools: boolean;
57
+ contextWindow: number | undefined;
58
+ error?: string;
59
+ }>;
60
+ checkConnectivity(): Promise<boolean>;
61
+ verifyToolCalling(): Promise<boolean>;
62
+ }
63
+ export declare const OllamaProvider: typeof OllamaAdapter;
64
+ export type OllamaProvider = OllamaAdapter;