antri_cli 1.26.0

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 (171) hide show
  1. package/README.md +92 -0
  2. package/bin/antri.js +2 -0
  3. package/dist/cli/banner.d.ts +22 -0
  4. package/dist/cli/banner.d.ts.map +1 -0
  5. package/dist/cli/banner.js +133 -0
  6. package/dist/cli/banner.js.map +1 -0
  7. package/dist/cli/dialogs/filePicker.d.ts +23 -0
  8. package/dist/cli/dialogs/filePicker.d.ts.map +1 -0
  9. package/dist/cli/dialogs/filePicker.js +106 -0
  10. package/dist/cli/dialogs/filePicker.js.map +1 -0
  11. package/dist/cli/dialogs/modelPicker.d.ts +10 -0
  12. package/dist/cli/dialogs/modelPicker.d.ts.map +1 -0
  13. package/dist/cli/dialogs/modelPicker.js +205 -0
  14. package/dist/cli/dialogs/modelPicker.js.map +1 -0
  15. package/dist/cli/dialogs/profilePicker.d.ts +11 -0
  16. package/dist/cli/dialogs/profilePicker.d.ts.map +1 -0
  17. package/dist/cli/dialogs/profilePicker.js +185 -0
  18. package/dist/cli/dialogs/profilePicker.js.map +1 -0
  19. package/dist/cli/dialogs/providerPicker.d.ts +19 -0
  20. package/dist/cli/dialogs/providerPicker.d.ts.map +1 -0
  21. package/dist/cli/dialogs/providerPicker.js +205 -0
  22. package/dist/cli/dialogs/providerPicker.js.map +1 -0
  23. package/dist/cli/prompt.d.ts +3 -0
  24. package/dist/cli/prompt.d.ts.map +1 -0
  25. package/dist/cli/prompt.js +50 -0
  26. package/dist/cli/prompt.js.map +1 -0
  27. package/dist/cli/promptToolkit.d.ts +14 -0
  28. package/dist/cli/promptToolkit.d.ts.map +1 -0
  29. package/dist/cli/promptToolkit.js +389 -0
  30. package/dist/cli/promptToolkit.js.map +1 -0
  31. package/dist/cli/renderer.d.ts +21 -0
  32. package/dist/cli/renderer.d.ts.map +1 -0
  33. package/dist/cli/renderer.js +97 -0
  34. package/dist/cli/renderer.js.map +1 -0
  35. package/dist/cli/shortcuts.d.ts +13 -0
  36. package/dist/cli/shortcuts.d.ts.map +1 -0
  37. package/dist/cli/shortcuts.js +414 -0
  38. package/dist/cli/shortcuts.js.map +1 -0
  39. package/dist/core/agent.d.ts +15 -0
  40. package/dist/core/agent.d.ts.map +1 -0
  41. package/dist/core/agent.js +232 -0
  42. package/dist/core/agent.js.map +1 -0
  43. package/dist/core/citations.d.ts +19 -0
  44. package/dist/core/citations.d.ts.map +1 -0
  45. package/dist/core/citations.js +40 -0
  46. package/dist/core/citations.js.map +1 -0
  47. package/dist/core/config.d.ts +16 -0
  48. package/dist/core/config.d.ts.map +1 -0
  49. package/dist/core/config.js +172 -0
  50. package/dist/core/config.js.map +1 -0
  51. package/dist/core/crawler.d.ts +18 -0
  52. package/dist/core/crawler.d.ts.map +1 -0
  53. package/dist/core/crawler.js +83 -0
  54. package/dist/core/crawler.js.map +1 -0
  55. package/dist/core/debugger.d.ts +20 -0
  56. package/dist/core/debugger.d.ts.map +1 -0
  57. package/dist/core/debugger.js +105 -0
  58. package/dist/core/debugger.js.map +1 -0
  59. package/dist/core/dialectic.d.ts +23 -0
  60. package/dist/core/dialectic.d.ts.map +1 -0
  61. package/dist/core/dialectic.js +300 -0
  62. package/dist/core/dialectic.js.map +1 -0
  63. package/dist/core/goalLoop.d.ts +14 -0
  64. package/dist/core/goalLoop.d.ts.map +1 -0
  65. package/dist/core/goalLoop.js +104 -0
  66. package/dist/core/goalLoop.js.map +1 -0
  67. package/dist/core/history.d.ts +13 -0
  68. package/dist/core/history.d.ts.map +1 -0
  69. package/dist/core/history.js +53 -0
  70. package/dist/core/history.js.map +1 -0
  71. package/dist/core/metaOptimizer.d.ts +27 -0
  72. package/dist/core/metaOptimizer.d.ts.map +1 -0
  73. package/dist/core/metaOptimizer.js +98 -0
  74. package/dist/core/metaOptimizer.js.map +1 -0
  75. package/dist/core/sandbox.d.ts +29 -0
  76. package/dist/core/sandbox.d.ts.map +1 -0
  77. package/dist/core/sandbox.js +141 -0
  78. package/dist/core/sandbox.js.map +1 -0
  79. package/dist/core/scraper.d.ts +22 -0
  80. package/dist/core/scraper.d.ts.map +1 -0
  81. package/dist/core/scraper.js +180 -0
  82. package/dist/core/scraper.js.map +1 -0
  83. package/dist/core/search.d.ts +27 -0
  84. package/dist/core/search.d.ts.map +1 -0
  85. package/dist/core/search.js +222 -0
  86. package/dist/core/search.js.map +1 -0
  87. package/dist/core/skillSynthesizer.d.ts +43 -0
  88. package/dist/core/skillSynthesizer.d.ts.map +1 -0
  89. package/dist/core/skillSynthesizer.js +147 -0
  90. package/dist/core/skillSynthesizer.js.map +1 -0
  91. package/dist/core/tools.d.ts +12 -0
  92. package/dist/core/tools.d.ts.map +1 -0
  93. package/dist/core/tools.js +467 -0
  94. package/dist/core/tools.js.map +1 -0
  95. package/dist/core/updater.d.ts +9 -0
  96. package/dist/core/updater.d.ts.map +1 -0
  97. package/dist/core/updater.js +63 -0
  98. package/dist/core/updater.js.map +1 -0
  99. package/dist/index.d.ts +3 -0
  100. package/dist/index.d.ts.map +1 -0
  101. package/dist/index.js +92 -0
  102. package/dist/index.js.map +1 -0
  103. package/dist/memory/consolidation.d.ts +17 -0
  104. package/dist/memory/consolidation.d.ts.map +1 -0
  105. package/dist/memory/consolidation.js +42 -0
  106. package/dist/memory/consolidation.js.map +1 -0
  107. package/dist/memory/episodic.d.ts +15 -0
  108. package/dist/memory/episodic.d.ts.map +1 -0
  109. package/dist/memory/episodic.js +97 -0
  110. package/dist/memory/episodic.js.map +1 -0
  111. package/dist/memory/manager.d.ts +24 -0
  112. package/dist/memory/manager.d.ts.map +1 -0
  113. package/dist/memory/manager.js +97 -0
  114. package/dist/memory/manager.js.map +1 -0
  115. package/dist/memory/profile.d.ts +21 -0
  116. package/dist/memory/profile.d.ts.map +1 -0
  117. package/dist/memory/profile.js +111 -0
  118. package/dist/memory/profile.js.map +1 -0
  119. package/dist/memory/semantic.d.ts +24 -0
  120. package/dist/memory/semantic.d.ts.map +1 -0
  121. package/dist/memory/semantic.js +102 -0
  122. package/dist/memory/semantic.js.map +1 -0
  123. package/dist/memory/types.d.ts +45 -0
  124. package/dist/memory/types.d.ts.map +1 -0
  125. package/dist/memory/types.js +2 -0
  126. package/dist/memory/types.js.map +1 -0
  127. package/dist/memory/vectorStore.d.ts +18 -0
  128. package/dist/memory/vectorStore.d.ts.map +1 -0
  129. package/dist/memory/vectorStore.js +103 -0
  130. package/dist/memory/vectorStore.js.map +1 -0
  131. package/dist/profiles/profileManager.d.ts +25 -0
  132. package/dist/profiles/profileManager.d.ts.map +1 -0
  133. package/dist/profiles/profileManager.js +205 -0
  134. package/dist/profiles/profileManager.js.map +1 -0
  135. package/dist/providers/anthropic.d.ts +14 -0
  136. package/dist/providers/anthropic.d.ts.map +1 -0
  137. package/dist/providers/anthropic.js +80 -0
  138. package/dist/providers/anthropic.js.map +1 -0
  139. package/dist/providers/base.d.ts +7 -0
  140. package/dist/providers/base.d.ts.map +1 -0
  141. package/dist/providers/base.js +2 -0
  142. package/dist/providers/base.js.map +1 -0
  143. package/dist/providers/gemini.d.ts +14 -0
  144. package/dist/providers/gemini.d.ts.map +1 -0
  145. package/dist/providers/gemini.js +77 -0
  146. package/dist/providers/gemini.js.map +1 -0
  147. package/dist/providers/index.d.ts +4 -0
  148. package/dist/providers/index.d.ts.map +1 -0
  149. package/dist/providers/index.js +65 -0
  150. package/dist/providers/index.js.map +1 -0
  151. package/dist/providers/mock.d.ts +8 -0
  152. package/dist/providers/mock.d.ts.map +1 -0
  153. package/dist/providers/mock.js +62 -0
  154. package/dist/providers/mock.js.map +1 -0
  155. package/dist/providers/models.d.ts +8 -0
  156. package/dist/providers/models.d.ts.map +1 -0
  157. package/dist/providers/models.js +403 -0
  158. package/dist/providers/models.js.map +1 -0
  159. package/dist/providers/openai.d.ts +18 -0
  160. package/dist/providers/openai.d.ts.map +1 -0
  161. package/dist/providers/openai.js +162 -0
  162. package/dist/providers/openai.js.map +1 -0
  163. package/dist/types.d.ts +116 -0
  164. package/dist/types.d.ts.map +1 -0
  165. package/dist/types.js +2 -0
  166. package/dist/types.js.map +1 -0
  167. package/dist/utils/logger.d.ts +23 -0
  168. package/dist/utils/logger.d.ts.map +1 -0
  169. package/dist/utils/logger.js +26 -0
  170. package/dist/utils/logger.js.map +1 -0
  171. package/package.json +57 -0
@@ -0,0 +1,7 @@
1
+ import { ChatMessage, ToolDefinition, StreamCallbacks } from '../types.js';
2
+ export interface LLMProvider {
3
+ name: string;
4
+ defaultModel: string;
5
+ sendMessageStream(messages: ChatMessage[], tools: ToolDefinition[], callbacks: StreamCallbacks): Promise<string>;
6
+ }
7
+ //# sourceMappingURL=base.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../src/providers/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAE3E,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,YAAY,EAAE,MAAM,CAAC;IACrB,iBAAiB,CACf,QAAQ,EAAE,WAAW,EAAE,EACvB,KAAK,EAAE,cAAc,EAAE,EACvB,SAAS,EAAE,eAAe,GACzB,OAAO,CAAC,MAAM,CAAC,CAAC;CACpB"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=base.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/providers/base.ts"],"names":[],"mappings":""}
@@ -0,0 +1,14 @@
1
+ import { LLMProvider } from './base.js';
2
+ import { ChatMessage, ToolDefinition, StreamCallbacks } from '../types.js';
3
+ export declare class GeminiProvider implements LLMProvider {
4
+ name: string;
5
+ defaultModel: string;
6
+ private apiKey;
7
+ private model;
8
+ constructor(options: {
9
+ apiKey?: string;
10
+ model?: string;
11
+ });
12
+ sendMessageStream(messages: ChatMessage[], tools: ToolDefinition[], callbacks: StreamCallbacks): Promise<string>;
13
+ }
14
+ //# sourceMappingURL=gemini.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../src/providers/gemini.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,WAAW,CAAC;AACxC,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAE3E,qBAAa,cAAe,YAAW,WAAW;IACzC,IAAI,EAAE,MAAM,CAAY;IACxB,YAAY,EAAE,MAAM,CAAC;IAC5B,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAS;gBAEV,OAAO,EAAE;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE;IAM3C,iBAAiB,CAC5B,QAAQ,EAAE,WAAW,EAAE,EACvB,KAAK,EAAE,cAAc,EAAE,EACvB,SAAS,EAAE,eAAe,GACzB,OAAO,CAAC,MAAM,CAAC;CA6EnB"}
@@ -0,0 +1,77 @@
1
+ export class GeminiProvider {
2
+ name = 'gemini';
3
+ defaultModel;
4
+ apiKey;
5
+ model;
6
+ constructor(options) {
7
+ this.apiKey = options.apiKey || '';
8
+ this.model = options.model || 'gemini-2.5-flash';
9
+ this.defaultModel = this.model;
10
+ }
11
+ async sendMessageStream(messages, tools, callbacks) {
12
+ if (!this.apiKey) {
13
+ throw new Error('GEMINI_API_KEY is not set. Use /key gemini <your-key> or set it in .env');
14
+ }
15
+ const contents = messages
16
+ .filter((m) => m.role !== 'system')
17
+ .map((m) => ({
18
+ role: m.role === 'assistant' ? 'model' : 'user',
19
+ parts: [{ text: m.content }],
20
+ }));
21
+ const systemMsg = messages.find((m) => m.role === 'system');
22
+ const payload = {
23
+ contents,
24
+ };
25
+ if (systemMsg) {
26
+ payload.systemInstruction = {
27
+ parts: [{ text: systemMsg.content }],
28
+ };
29
+ }
30
+ const url = `https://generativelanguage.googleapis.com/v1beta/models/${this.model}:streamGenerateContent?alt=sse&key=${this.apiKey}`;
31
+ const response = await fetch(url, {
32
+ method: 'POST',
33
+ headers: { 'Content-Type': 'application/json' },
34
+ body: JSON.stringify(payload),
35
+ });
36
+ if (!response.ok) {
37
+ const errText = await response.text();
38
+ throw new Error(`Gemini API error (${response.status}): ${errText}`);
39
+ }
40
+ if (!response.body) {
41
+ throw new Error('Gemini response body is null');
42
+ }
43
+ const reader = response.body.getReader();
44
+ const decoder = new TextDecoder('utf-8');
45
+ let fullContent = '';
46
+ let buffer = '';
47
+ while (true) {
48
+ const { done, value } = await reader.read();
49
+ if (done)
50
+ break;
51
+ buffer += decoder.decode(value, { stream: true });
52
+ const lines = buffer.split('\n');
53
+ buffer = lines.pop() || '';
54
+ for (const line of lines) {
55
+ const trimmed = line.trim();
56
+ if (trimmed.startsWith('data: ')) {
57
+ try {
58
+ const data = JSON.parse(trimmed.slice(6));
59
+ const text = data.candidates?.[0]?.content?.parts?.[0]?.text;
60
+ if (text) {
61
+ fullContent += text;
62
+ callbacks.onToken(text);
63
+ }
64
+ }
65
+ catch {
66
+ // Ignore parse errors on chunks
67
+ }
68
+ }
69
+ }
70
+ }
71
+ if (callbacks.onComplete) {
72
+ callbacks.onComplete(fullContent);
73
+ }
74
+ return fullContent;
75
+ }
76
+ }
77
+ //# sourceMappingURL=gemini.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../src/providers/gemini.ts"],"names":[],"mappings":"AAGA,MAAM,OAAO,cAAc;IAClB,IAAI,GAAW,QAAQ,CAAC;IACxB,YAAY,CAAS;IACpB,MAAM,CAAS;IACf,KAAK,CAAS;IAEtB,YAAY,OAA4C;QACtD,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC;QACnC,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,kBAAkB,CAAC;QACjD,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC;IACjC,CAAC;IAEM,KAAK,CAAC,iBAAiB,CAC5B,QAAuB,EACvB,KAAuB,EACvB,SAA0B;QAE1B,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,yEAAyE,CAAC,CAAC;QAC7F,CAAC;QAED,MAAM,QAAQ,GAAG,QAAQ;aACtB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;aAClC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACX,IAAI,EAAE,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;YAC/C,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;SAC7B,CAAC,CAAC,CAAC;QAEN,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QAE5D,MAAM,OAAO,GAAQ;YACnB,QAAQ;SACT,CAAC;QAEF,IAAI,SAAS,EAAE,CAAC;YACd,OAAO,CAAC,iBAAiB,GAAG;gBAC1B,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,CAAC,OAAO,EAAE,CAAC;aACrC,CAAC;QACJ,CAAC;QAED,MAAM,GAAG,GAAG,2DAA2D,IAAI,CAAC,KAAK,sCAAsC,IAAI,CAAC,MAAM,EAAE,CAAC;QAErI,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAChC,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;SAC9B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACtC,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,CAAC,MAAM,MAAM,OAAO,EAAE,CAAC,CAAC;QACvE,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAClD,CAAC;QAED,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;QACzC,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,OAAO,CAAC,CAAC;QACzC,IAAI,WAAW,GAAG,EAAE,CAAC;QACrB,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI;gBAAE,MAAM;YAEhB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAClD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;YAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;gBAC5B,IAAI,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACjC,IAAI,CAAC;wBACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;wBAC1C,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;wBAC7D,IAAI,IAAI,EAAE,CAAC;4BACT,WAAW,IAAI,IAAI,CAAC;4BACpB,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;wBAC1B,CAAC;oBACH,CAAC;oBAAC,MAAM,CAAC;wBACP,gCAAgC;oBAClC,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,SAAS,CAAC,UAAU,EAAE,CAAC;YACzB,SAAS,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QACpC,CAAC;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;CACF"}
@@ -0,0 +1,4 @@
1
+ import { LLMProvider } from './base.js';
2
+ import { AntriConfig } from '../types.js';
3
+ export declare function createProvider(config: AntriConfig): LLMProvider;
4
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,WAAW,CAAC;AAKxC,OAAO,EAAE,WAAW,EAAgB,MAAM,aAAa,CAAC;AAExD,wBAAgB,cAAc,CAAC,MAAM,EAAE,WAAW,GAAG,WAAW,CAoE/D"}
@@ -0,0 +1,65 @@
1
+ import { OpenAICompatibleProvider } from './openai.js';
2
+ import { GeminiProvider } from './gemini.js';
3
+ import { AnthropicProvider } from './anthropic.js';
4
+ import { MockProvider } from './mock.js';
5
+ export function createProvider(config) {
6
+ const providerType = config.provider;
7
+ switch (providerType) {
8
+ case 'ollama':
9
+ return new OpenAICompatibleProvider({
10
+ name: 'ollama',
11
+ apiKey: 'ollama',
12
+ baseUrl: (config.customBaseUrls?.ollama || 'http://localhost:11434') + '/v1',
13
+ model: config.model || 'llama3.3:70b',
14
+ });
15
+ case 'anthropic':
16
+ return new AnthropicProvider({
17
+ apiKey: config.apiKeys.anthropic,
18
+ model: config.model || 'claude-3-7-sonnet-20250219',
19
+ });
20
+ case 'nvidia-nim':
21
+ return new OpenAICompatibleProvider({
22
+ name: 'nvidia-nim',
23
+ apiKey: config.apiKeys.nvidia_nim,
24
+ baseUrl: config.customBaseUrls?.nvidia_nim || 'https://integrate.api.nvidia.com/v1',
25
+ model: config.model || 'meta/llama-3.3-70b-instruct',
26
+ });
27
+ case 'openai':
28
+ return new OpenAICompatibleProvider({
29
+ name: 'openai',
30
+ apiKey: config.apiKeys.openai,
31
+ baseUrl: config.customBaseUrls?.openai || 'https://api.openai.com/v1',
32
+ model: config.model || 'gpt-4o',
33
+ });
34
+ case 'gemini':
35
+ return new GeminiProvider({
36
+ apiKey: config.apiKeys.gemini,
37
+ model: config.model || 'gemini-2.5-flash',
38
+ });
39
+ case 'deepseek':
40
+ return new OpenAICompatibleProvider({
41
+ name: 'deepseek',
42
+ apiKey: config.apiKeys.deepseek,
43
+ baseUrl: config.customBaseUrls?.deepseek || 'https://api.deepseek.com/v1',
44
+ model: config.model || 'deepseek-v4-flash-(latest)',
45
+ });
46
+ case 'custom':
47
+ return new OpenAICompatibleProvider({
48
+ name: 'custom',
49
+ apiKey: config.apiKeys.custom || 'custom-key',
50
+ baseUrl: config.customBaseUrls?.custom || 'http://localhost:8000/v1',
51
+ model: config.model || 'custom-model',
52
+ });
53
+ case 'openrouter':
54
+ return new OpenAICompatibleProvider({
55
+ name: 'openrouter',
56
+ apiKey: config.apiKeys.openrouter,
57
+ baseUrl: 'https://openrouter.ai/api/v1',
58
+ model: config.model || 'deepseek/deepseek-r1',
59
+ });
60
+ case 'mock':
61
+ default:
62
+ return new MockProvider();
63
+ }
64
+ }
65
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAGzC,MAAM,UAAU,cAAc,CAAC,MAAmB;IAChD,MAAM,YAAY,GAAiB,MAAM,CAAC,QAAQ,CAAC;IAEnD,QAAQ,YAAY,EAAE,CAAC;QACrB,KAAK,QAAQ;YACX,OAAO,IAAI,wBAAwB,CAAC;gBAClC,IAAI,EAAE,QAAQ;gBACd,MAAM,EAAE,QAAQ;gBAChB,OAAO,EAAE,CAAC,MAAM,CAAC,cAAc,EAAE,MAAM,IAAI,wBAAwB,CAAC,GAAG,KAAK;gBAC5E,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,cAAc;aACtC,CAAC,CAAC;QAEL,KAAK,WAAW;YACd,OAAO,IAAI,iBAAiB,CAAC;gBAC3B,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,SAAS;gBAChC,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,4BAA4B;aACpD,CAAC,CAAC;QAEL,KAAK,YAAY;YACf,OAAO,IAAI,wBAAwB,CAAC;gBAClC,IAAI,EAAE,YAAY;gBAClB,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,UAAU;gBACjC,OAAO,EAAE,MAAM,CAAC,cAAc,EAAE,UAAU,IAAI,qCAAqC;gBACnF,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,6BAA6B;aACrD,CAAC,CAAC;QAEL,KAAK,QAAQ;YACX,OAAO,IAAI,wBAAwB,CAAC;gBAClC,IAAI,EAAE,QAAQ;gBACd,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM;gBAC7B,OAAO,EAAE,MAAM,CAAC,cAAc,EAAE,MAAM,IAAI,2BAA2B;gBACrE,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,QAAQ;aAChC,CAAC,CAAC;QAEL,KAAK,QAAQ;YACX,OAAO,IAAI,cAAc,CAAC;gBACxB,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM;gBAC7B,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,kBAAkB;aAC1C,CAAC,CAAC;QAEL,KAAK,UAAU;YACb,OAAO,IAAI,wBAAwB,CAAC;gBAClC,IAAI,EAAE,UAAU;gBAChB,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,QAAQ;gBAC/B,OAAO,EAAE,MAAM,CAAC,cAAc,EAAE,QAAQ,IAAI,6BAA6B;gBACzE,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,4BAA4B;aACpD,CAAC,CAAC;QAEL,KAAK,QAAQ;YACX,OAAO,IAAI,wBAAwB,CAAC;gBAClC,IAAI,EAAE,QAAQ;gBACd,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM,IAAI,YAAY;gBAC7C,OAAO,EAAE,MAAM,CAAC,cAAc,EAAE,MAAM,IAAI,0BAA0B;gBACpE,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,cAAc;aACtC,CAAC,CAAC;QAEL,KAAK,YAAY;YACf,OAAO,IAAI,wBAAwB,CAAC;gBAClC,IAAI,EAAE,YAAY;gBAClB,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,UAAU;gBACjC,OAAO,EAAE,8BAA8B;gBACvC,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,sBAAsB;aAC9C,CAAC,CAAC;QAEL,KAAK,MAAM,CAAC;QACZ;YACE,OAAO,IAAI,YAAY,EAAE,CAAC;IAC9B,CAAC;AACH,CAAC"}
@@ -0,0 +1,8 @@
1
+ import { LLMProvider } from './base.js';
2
+ import { ChatMessage, ToolDefinition, StreamCallbacks } from '../types.js';
3
+ export declare class MockProvider implements LLMProvider {
4
+ name: string;
5
+ defaultModel: string;
6
+ sendMessageStream(messages: ChatMessage[], tools: ToolDefinition[], callbacks: StreamCallbacks): Promise<string>;
7
+ }
8
+ //# sourceMappingURL=mock.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"mock.d.ts","sourceRoot":"","sources":["../../src/providers/mock.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,WAAW,CAAC;AACxC,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAE3E,qBAAa,YAAa,YAAW,WAAW;IACvC,IAAI,EAAE,MAAM,CAAU;IACtB,YAAY,EAAE,MAAM,CAAgC;IAE9C,iBAAiB,CAC5B,QAAQ,EAAE,WAAW,EAAE,EACvB,KAAK,EAAE,cAAc,EAAE,EACvB,SAAS,EAAE,eAAe,GACzB,OAAO,CAAC,MAAM,CAAC;CA4DnB"}
@@ -0,0 +1,62 @@
1
+ export class MockProvider {
2
+ name = 'mock';
3
+ defaultModel = 'deepseek-v4-flash-(latest)';
4
+ async sendMessageStream(messages, tools, callbacks) {
5
+ const lastMsg = messages[messages.length - 1]?.content.toLowerCase() || '';
6
+ let responseText = '';
7
+ if (lastMsg.includes('help') || lastMsg.includes('shortcut')) {
8
+ responseText = `### 🌟 Welcome to ANTRI Code CLI!
9
+
10
+ Here are quick shortcuts and commands:
11
+ - **\`?\`** or **\`/help\`**: Show this help manual
12
+ - **\`/model [name]\`**: Switch AI model (e.g., \`deepseek-v4-flash\`, \`gpt-4o\`, \`claude-3-5-sonnet\`)
13
+ - **\`/provider [name]\`**: Switch provider (\`deepseek\`, \`openai\`, \`gemini\`, \`anthropic\`, \`ollama\`)
14
+ - **\`/key <provider> <key>\`**: Set API key dynamically
15
+ - **\`/clear\`**: Clear chat session & screen
16
+ - **\`/exit\`**: Exit ANTRI CLI
17
+
18
+ > 💡 **To connect a real LLM model:**
19
+ > Set your API key in \`.env\` (e.g., \`DEEPSEEK_API_KEY=your_key\` or \`OPENAI_API_KEY=your_key\`), or type \`/key deepseek <your_key>\`.`;
20
+ }
21
+ else if (lastMsg.includes('file') || lastMsg.includes('read') || lastMsg.includes('list')) {
22
+ if (callbacks.onToolCall) {
23
+ callbacks.onToolCall({
24
+ id: `mock_call_${Date.now()}`,
25
+ type: 'function',
26
+ function: {
27
+ name: 'list_dir',
28
+ arguments: JSON.stringify({ dir_path: '.' }),
29
+ },
30
+ });
31
+ return '';
32
+ }
33
+ responseText = `I analyzed your project directory. All files are ready for editing!`;
34
+ }
35
+ else {
36
+ responseText = `Hello! I am **ANTRI**, your terminal-first AI coding companion. âš¡
37
+
38
+ You asked:
39
+ > *${messages[messages.length - 1]?.content}*
40
+
41
+ To connect your favorite LLM provider:
42
+ 1. **DeepSeek**: Set \`DEEPSEEK_API_KEY\` in your \`.env\` or run \`/key deepseek <key>\`
43
+ 2. **OpenAI**: Set \`OPENAI_API_KEY\` or run \`/key openai <key>\`
44
+ 3. **Gemini**: Set \`GEMINI_API_KEY\` or run \`/key gemini <key>\`
45
+ 4. **Ollama (Local)**: Run \`/provider ollama\` and ensure Ollama is running at \`localhost:11434\`
46
+
47
+ How can I assist you with your project today?`;
48
+ }
49
+ // Stream the tokens with realistic delay
50
+ const words = responseText.split(' ');
51
+ for (let i = 0; i < words.length; i++) {
52
+ const token = (i === 0 ? '' : ' ') + words[i];
53
+ callbacks.onToken(token);
54
+ await new Promise((r) => setTimeout(r, 18));
55
+ }
56
+ if (callbacks.onComplete) {
57
+ callbacks.onComplete(responseText);
58
+ }
59
+ return responseText;
60
+ }
61
+ }
62
+ //# sourceMappingURL=mock.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"mock.js","sourceRoot":"","sources":["../../src/providers/mock.ts"],"names":[],"mappings":"AAGA,MAAM,OAAO,YAAY;IAChB,IAAI,GAAW,MAAM,CAAC;IACtB,YAAY,GAAW,4BAA4B,CAAC;IAEpD,KAAK,CAAC,iBAAiB,CAC5B,QAAuB,EACvB,KAAuB,EACvB,SAA0B;QAE1B,MAAM,OAAO,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC;QAE3E,IAAI,YAAY,GAAG,EAAE,CAAC;QAEtB,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YAC7D,YAAY,GAAG;;;;;;;;;;;2IAWsH,CAAC;QACxI,CAAC;aAAM,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC5F,IAAI,SAAS,CAAC,UAAU,EAAE,CAAC;gBACzB,SAAS,CAAC,UAAU,CAAC;oBACnB,EAAE,EAAE,aAAa,IAAI,CAAC,GAAG,EAAE,EAAE;oBAC7B,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE;wBACR,IAAI,EAAE,UAAU;wBAChB,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC;qBAC7C;iBACF,CAAC,CAAC;gBACH,OAAO,EAAE,CAAC;YACZ,CAAC;YACD,YAAY,GAAG,qEAAqE,CAAC;QACvF,CAAC;aAAM,CAAC;YACN,YAAY,GAAG;;;KAGhB,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,OAAO;;;;;;;;8CAQG,CAAC;QAC3C,CAAC;QAED,yCAAyC;QACzC,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACtC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YAC9C,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACzB,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,SAAS,CAAC,UAAU,EAAE,CAAC;YACzB,SAAS,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;QACrC,CAAC;QAED,OAAO,YAAY,CAAC;IACtB,CAAC;CACF"}
@@ -0,0 +1,8 @@
1
+ import { ModelInfo, ProviderType, AntriConfig } from '../types.js';
2
+ export declare const VERIFIED_NVIDIA_MODELS: string[];
3
+ export declare const PROVIDER_CATALOGS: Record<ProviderType, ModelInfo[]>;
4
+ /**
5
+ * Returns available models scoped strictly to the active provider
6
+ */
7
+ export declare function getAvailableModels(config: AntriConfig): Promise<ModelInfo[]>;
8
+ //# sourceMappingURL=models.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../../src/providers/models.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAEnE,eAAO,MAAM,sBAAsB,EAAE,MAAM,EAc1C,CAAC;AAGF,eAAO,MAAM,iBAAiB,EAAE,MAAM,CAAC,YAAY,EAAE,SAAS,EAAE,CA8R/D,CAAC;AAEF;;GAEG;AACH,wBAAsB,kBAAkB,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC,CAyHlF"}