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.
- package/README.md +92 -0
- package/bin/antri.js +2 -0
- package/dist/cli/banner.d.ts +22 -0
- package/dist/cli/banner.d.ts.map +1 -0
- package/dist/cli/banner.js +133 -0
- package/dist/cli/banner.js.map +1 -0
- package/dist/cli/dialogs/filePicker.d.ts +23 -0
- package/dist/cli/dialogs/filePicker.d.ts.map +1 -0
- package/dist/cli/dialogs/filePicker.js +106 -0
- package/dist/cli/dialogs/filePicker.js.map +1 -0
- package/dist/cli/dialogs/modelPicker.d.ts +10 -0
- package/dist/cli/dialogs/modelPicker.d.ts.map +1 -0
- package/dist/cli/dialogs/modelPicker.js +205 -0
- package/dist/cli/dialogs/modelPicker.js.map +1 -0
- package/dist/cli/dialogs/profilePicker.d.ts +11 -0
- package/dist/cli/dialogs/profilePicker.d.ts.map +1 -0
- package/dist/cli/dialogs/profilePicker.js +185 -0
- package/dist/cli/dialogs/profilePicker.js.map +1 -0
- package/dist/cli/dialogs/providerPicker.d.ts +19 -0
- package/dist/cli/dialogs/providerPicker.d.ts.map +1 -0
- package/dist/cli/dialogs/providerPicker.js +205 -0
- package/dist/cli/dialogs/providerPicker.js.map +1 -0
- package/dist/cli/prompt.d.ts +3 -0
- package/dist/cli/prompt.d.ts.map +1 -0
- package/dist/cli/prompt.js +50 -0
- package/dist/cli/prompt.js.map +1 -0
- package/dist/cli/promptToolkit.d.ts +14 -0
- package/dist/cli/promptToolkit.d.ts.map +1 -0
- package/dist/cli/promptToolkit.js +389 -0
- package/dist/cli/promptToolkit.js.map +1 -0
- package/dist/cli/renderer.d.ts +21 -0
- package/dist/cli/renderer.d.ts.map +1 -0
- package/dist/cli/renderer.js +97 -0
- package/dist/cli/renderer.js.map +1 -0
- package/dist/cli/shortcuts.d.ts +13 -0
- package/dist/cli/shortcuts.d.ts.map +1 -0
- package/dist/cli/shortcuts.js +414 -0
- package/dist/cli/shortcuts.js.map +1 -0
- package/dist/core/agent.d.ts +15 -0
- package/dist/core/agent.d.ts.map +1 -0
- package/dist/core/agent.js +232 -0
- package/dist/core/agent.js.map +1 -0
- package/dist/core/citations.d.ts +19 -0
- package/dist/core/citations.d.ts.map +1 -0
- package/dist/core/citations.js +40 -0
- package/dist/core/citations.js.map +1 -0
- package/dist/core/config.d.ts +16 -0
- package/dist/core/config.d.ts.map +1 -0
- package/dist/core/config.js +172 -0
- package/dist/core/config.js.map +1 -0
- package/dist/core/crawler.d.ts +18 -0
- package/dist/core/crawler.d.ts.map +1 -0
- package/dist/core/crawler.js +83 -0
- package/dist/core/crawler.js.map +1 -0
- package/dist/core/debugger.d.ts +20 -0
- package/dist/core/debugger.d.ts.map +1 -0
- package/dist/core/debugger.js +105 -0
- package/dist/core/debugger.js.map +1 -0
- package/dist/core/dialectic.d.ts +23 -0
- package/dist/core/dialectic.d.ts.map +1 -0
- package/dist/core/dialectic.js +300 -0
- package/dist/core/dialectic.js.map +1 -0
- package/dist/core/goalLoop.d.ts +14 -0
- package/dist/core/goalLoop.d.ts.map +1 -0
- package/dist/core/goalLoop.js +104 -0
- package/dist/core/goalLoop.js.map +1 -0
- package/dist/core/history.d.ts +13 -0
- package/dist/core/history.d.ts.map +1 -0
- package/dist/core/history.js +53 -0
- package/dist/core/history.js.map +1 -0
- package/dist/core/metaOptimizer.d.ts +27 -0
- package/dist/core/metaOptimizer.d.ts.map +1 -0
- package/dist/core/metaOptimizer.js +98 -0
- package/dist/core/metaOptimizer.js.map +1 -0
- package/dist/core/sandbox.d.ts +29 -0
- package/dist/core/sandbox.d.ts.map +1 -0
- package/dist/core/sandbox.js +141 -0
- package/dist/core/sandbox.js.map +1 -0
- package/dist/core/scraper.d.ts +22 -0
- package/dist/core/scraper.d.ts.map +1 -0
- package/dist/core/scraper.js +180 -0
- package/dist/core/scraper.js.map +1 -0
- package/dist/core/search.d.ts +27 -0
- package/dist/core/search.d.ts.map +1 -0
- package/dist/core/search.js +222 -0
- package/dist/core/search.js.map +1 -0
- package/dist/core/skillSynthesizer.d.ts +43 -0
- package/dist/core/skillSynthesizer.d.ts.map +1 -0
- package/dist/core/skillSynthesizer.js +147 -0
- package/dist/core/skillSynthesizer.js.map +1 -0
- package/dist/core/tools.d.ts +12 -0
- package/dist/core/tools.d.ts.map +1 -0
- package/dist/core/tools.js +467 -0
- package/dist/core/tools.js.map +1 -0
- package/dist/core/updater.d.ts +9 -0
- package/dist/core/updater.d.ts.map +1 -0
- package/dist/core/updater.js +63 -0
- package/dist/core/updater.js.map +1 -0
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +92 -0
- package/dist/index.js.map +1 -0
- package/dist/memory/consolidation.d.ts +17 -0
- package/dist/memory/consolidation.d.ts.map +1 -0
- package/dist/memory/consolidation.js +42 -0
- package/dist/memory/consolidation.js.map +1 -0
- package/dist/memory/episodic.d.ts +15 -0
- package/dist/memory/episodic.d.ts.map +1 -0
- package/dist/memory/episodic.js +97 -0
- package/dist/memory/episodic.js.map +1 -0
- package/dist/memory/manager.d.ts +24 -0
- package/dist/memory/manager.d.ts.map +1 -0
- package/dist/memory/manager.js +97 -0
- package/dist/memory/manager.js.map +1 -0
- package/dist/memory/profile.d.ts +21 -0
- package/dist/memory/profile.d.ts.map +1 -0
- package/dist/memory/profile.js +111 -0
- package/dist/memory/profile.js.map +1 -0
- package/dist/memory/semantic.d.ts +24 -0
- package/dist/memory/semantic.d.ts.map +1 -0
- package/dist/memory/semantic.js +102 -0
- package/dist/memory/semantic.js.map +1 -0
- package/dist/memory/types.d.ts +45 -0
- package/dist/memory/types.d.ts.map +1 -0
- package/dist/memory/types.js +2 -0
- package/dist/memory/types.js.map +1 -0
- package/dist/memory/vectorStore.d.ts +18 -0
- package/dist/memory/vectorStore.d.ts.map +1 -0
- package/dist/memory/vectorStore.js +103 -0
- package/dist/memory/vectorStore.js.map +1 -0
- package/dist/profiles/profileManager.d.ts +25 -0
- package/dist/profiles/profileManager.d.ts.map +1 -0
- package/dist/profiles/profileManager.js +205 -0
- package/dist/profiles/profileManager.js.map +1 -0
- package/dist/providers/anthropic.d.ts +14 -0
- package/dist/providers/anthropic.d.ts.map +1 -0
- package/dist/providers/anthropic.js +80 -0
- package/dist/providers/anthropic.js.map +1 -0
- package/dist/providers/base.d.ts +7 -0
- package/dist/providers/base.d.ts.map +1 -0
- package/dist/providers/base.js +2 -0
- package/dist/providers/base.js.map +1 -0
- package/dist/providers/gemini.d.ts +14 -0
- package/dist/providers/gemini.d.ts.map +1 -0
- package/dist/providers/gemini.js +77 -0
- package/dist/providers/gemini.js.map +1 -0
- package/dist/providers/index.d.ts +4 -0
- package/dist/providers/index.d.ts.map +1 -0
- package/dist/providers/index.js +65 -0
- package/dist/providers/index.js.map +1 -0
- package/dist/providers/mock.d.ts +8 -0
- package/dist/providers/mock.d.ts.map +1 -0
- package/dist/providers/mock.js +62 -0
- package/dist/providers/mock.js.map +1 -0
- package/dist/providers/models.d.ts +8 -0
- package/dist/providers/models.d.ts.map +1 -0
- package/dist/providers/models.js +403 -0
- package/dist/providers/models.js.map +1 -0
- package/dist/providers/openai.d.ts +18 -0
- package/dist/providers/openai.d.ts.map +1 -0
- package/dist/providers/openai.js +162 -0
- package/dist/providers/openai.js.map +1 -0
- package/dist/types.d.ts +116 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/dist/utils/logger.d.ts +23 -0
- package/dist/utils/logger.d.ts.map +1 -0
- package/dist/utils/logger.js +26 -0
- package/dist/utils/logger.js.map +1 -0
- 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 @@
|
|
|
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 @@
|
|
|
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"}
|