agent-sory 1.0.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/LICENSE +21 -0
- package/README.md +265 -0
- package/dist/commands/base.d.ts +26 -0
- package/dist/commands/base.d.ts.map +1 -0
- package/dist/commands/base.js +3 -0
- package/dist/commands/base.js.map +1 -0
- package/dist/commands/definitions/clear.d.ts +3 -0
- package/dist/commands/definitions/clear.d.ts.map +1 -0
- package/dist/commands/definitions/clear.js +12 -0
- package/dist/commands/definitions/clear.js.map +1 -0
- package/dist/commands/definitions/help.d.ts +3 -0
- package/dist/commands/definitions/help.d.ts.map +1 -0
- package/dist/commands/definitions/help.js +28 -0
- package/dist/commands/definitions/help.js.map +1 -0
- package/dist/commands/definitions/init.d.ts +3 -0
- package/dist/commands/definitions/init.d.ts.map +1 -0
- package/dist/commands/definitions/init.js +23 -0
- package/dist/commands/definitions/init.js.map +1 -0
- package/dist/commands/definitions/login.d.ts +3 -0
- package/dist/commands/definitions/login.d.ts.map +1 -0
- package/dist/commands/definitions/login.js +8 -0
- package/dist/commands/definitions/login.js.map +1 -0
- package/dist/commands/definitions/model.d.ts +3 -0
- package/dist/commands/definitions/model.d.ts.map +1 -0
- package/dist/commands/definitions/model.js +10 -0
- package/dist/commands/definitions/model.js.map +1 -0
- package/dist/commands/definitions/reasoning.d.ts +3 -0
- package/dist/commands/definitions/reasoning.d.ts.map +1 -0
- package/dist/commands/definitions/reasoning.js +21 -0
- package/dist/commands/definitions/reasoning.js.map +1 -0
- package/dist/commands/definitions/stats.d.ts +3 -0
- package/dist/commands/definitions/stats.d.ts.map +1 -0
- package/dist/commands/definitions/stats.js +22 -0
- package/dist/commands/definitions/stats.js.map +1 -0
- package/dist/commands/index.d.ts +6 -0
- package/dist/commands/index.d.ts.map +1 -0
- package/dist/commands/index.js +38 -0
- package/dist/commands/index.js.map +1 -0
- package/dist/core/agent.d.ts +41 -0
- package/dist/core/agent.d.ts.map +1 -0
- package/dist/core/agent.js +266 -0
- package/dist/core/agent.js.map +1 -0
- package/dist/core/cli.d.ts +3 -0
- package/dist/core/cli.d.ts.map +1 -0
- package/dist/core/cli.js +48 -0
- package/dist/core/cli.js.map +1 -0
- package/dist/core/llm/factory.d.ts +12 -0
- package/dist/core/llm/factory.d.ts.map +1 -0
- package/dist/core/llm/factory.js +44 -0
- package/dist/core/llm/factory.js.map +1 -0
- package/dist/core/llm/providers/gemini.d.ts +22 -0
- package/dist/core/llm/providers/gemini.d.ts.map +1 -0
- package/dist/core/llm/providers/gemini.js +167 -0
- package/dist/core/llm/providers/gemini.js.map +1 -0
- package/dist/core/llm/providers/groq.d.ts +23 -0
- package/dist/core/llm/providers/groq.d.ts.map +1 -0
- package/dist/core/llm/providers/groq.js +96 -0
- package/dist/core/llm/providers/groq.js.map +1 -0
- package/dist/core/llm/providers/openai-compatible.d.ts +24 -0
- package/dist/core/llm/providers/openai-compatible.d.ts.map +1 -0
- package/dist/core/llm/providers/openai-compatible.js +86 -0
- package/dist/core/llm/providers/openai-compatible.js.map +1 -0
- package/dist/core/llm/types.d.ts +37 -0
- package/dist/core/llm/types.d.ts.map +1 -0
- package/dist/core/llm/types.js +2 -0
- package/dist/core/llm/types.js.map +1 -0
- package/dist/core/prompts/system.d.ts +6 -0
- package/dist/core/prompts/system.d.ts.map +1 -0
- package/dist/core/prompts/system.js +36 -0
- package/dist/core/prompts/system.js.map +1 -0
- package/dist/tests/proxy-config.test.d.ts +2 -0
- package/dist/tests/proxy-config.test.d.ts.map +1 -0
- package/dist/tests/proxy-config.test.js +169 -0
- package/dist/tests/proxy-config.test.js.map +1 -0
- package/dist/tools/tool-schemas.d.ts +36 -0
- package/dist/tools/tool-schemas.d.ts.map +1 -0
- package/dist/tools/tool-schemas.js +482 -0
- package/dist/tools/tool-schemas.js.map +1 -0
- package/dist/tools/tools.d.ts +114 -0
- package/dist/tools/tools.d.ts.map +1 -0
- package/dist/tools/tools.js +1015 -0
- package/dist/tools/tools.js.map +1 -0
- package/dist/tools/utils/edit-logic.d.ts +9 -0
- package/dist/tools/utils/edit-logic.d.ts.map +1 -0
- package/dist/tools/utils/edit-logic.js +149 -0
- package/dist/tools/utils/edit-logic.js.map +1 -0
- package/dist/tools/utils/ripgrep-runner.d.ts +15 -0
- package/dist/tools/utils/ripgrep-runner.d.ts.map +1 -0
- package/dist/tools/utils/ripgrep-runner.js +90 -0
- package/dist/tools/utils/ripgrep-runner.js.map +1 -0
- package/dist/tools/validators.d.ts +4 -0
- package/dist/tools/validators.d.ts.map +1 -0
- package/dist/tools/validators.js +18 -0
- package/dist/tools/validators.js.map +1 -0
- package/dist/ui/App.d.ts +7 -0
- package/dist/ui/App.d.ts.map +1 -0
- package/dist/ui/App.js +12 -0
- package/dist/ui/App.js.map +1 -0
- package/dist/ui/components/core/Chat.d.ts +7 -0
- package/dist/ui/components/core/Chat.d.ts.map +1 -0
- package/dist/ui/components/core/Chat.js +60 -0
- package/dist/ui/components/core/Chat.js.map +1 -0
- package/dist/ui/components/core/MessageHistory.d.ts +7 -0
- package/dist/ui/components/core/MessageHistory.d.ts.map +1 -0
- package/dist/ui/components/core/MessageHistory.js +39 -0
- package/dist/ui/components/core/MessageHistory.js.map +1 -0
- package/dist/ui/components/core/MessageInput.d.ts +10 -0
- package/dist/ui/components/core/MessageInput.d.ts.map +1 -0
- package/dist/ui/components/core/MessageInput.js +130 -0
- package/dist/ui/components/core/MessageInput.js.map +1 -0
- package/dist/ui/components/display/DiffPreview.d.ts +15 -0
- package/dist/ui/components/display/DiffPreview.d.ts.map +1 -0
- package/dist/ui/components/display/DiffPreview.js +298 -0
- package/dist/ui/components/display/DiffPreview.js.map +1 -0
- package/dist/ui/components/display/Stats.d.ts +19 -0
- package/dist/ui/components/display/Stats.d.ts.map +1 -0
- package/dist/ui/components/display/Stats.js +31 -0
- package/dist/ui/components/display/Stats.js.map +1 -0
- package/dist/ui/components/display/TokenMetrics.d.ts +11 -0
- package/dist/ui/components/display/TokenMetrics.d.ts.map +1 -0
- package/dist/ui/components/display/TokenMetrics.js +16 -0
- package/dist/ui/components/display/TokenMetrics.js.map +1 -0
- package/dist/ui/components/display/ToolHistoryItem.d.ts +7 -0
- package/dist/ui/components/display/ToolHistoryItem.d.ts.map +1 -0
- package/dist/ui/components/display/ToolHistoryItem.js +102 -0
- package/dist/ui/components/display/ToolHistoryItem.js.map +1 -0
- package/dist/ui/components/input-overlays/ErrorRetry.d.ts +8 -0
- package/dist/ui/components/input-overlays/ErrorRetry.d.ts.map +1 -0
- package/dist/ui/components/input-overlays/ErrorRetry.js +6 -0
- package/dist/ui/components/input-overlays/ErrorRetry.js.map +1 -0
- package/dist/ui/components/input-overlays/Login.d.ts +7 -0
- package/dist/ui/components/input-overlays/Login.d.ts.map +1 -0
- package/dist/ui/components/input-overlays/Login.js +32 -0
- package/dist/ui/components/input-overlays/Login.js.map +1 -0
- package/dist/ui/components/input-overlays/MaxIterationsContinue.d.ts +8 -0
- package/dist/ui/components/input-overlays/MaxIterationsContinue.d.ts.map +1 -0
- package/dist/ui/components/input-overlays/MaxIterationsContinue.js +32 -0
- package/dist/ui/components/input-overlays/MaxIterationsContinue.js.map +1 -0
- package/dist/ui/components/input-overlays/ModelSelector.d.ts +12 -0
- package/dist/ui/components/input-overlays/ModelSelector.d.ts.map +1 -0
- package/dist/ui/components/input-overlays/ModelSelector.js +114 -0
- package/dist/ui/components/input-overlays/ModelSelector.js.map +1 -0
- package/dist/ui/components/input-overlays/PendingToolApproval.d.ts +10 -0
- package/dist/ui/components/input-overlays/PendingToolApproval.d.ts.map +1 -0
- package/dist/ui/components/input-overlays/PendingToolApproval.js +51 -0
- package/dist/ui/components/input-overlays/PendingToolApproval.js.map +1 -0
- package/dist/ui/components/input-overlays/SlashCommandSuggestions.d.ts +8 -0
- package/dist/ui/components/input-overlays/SlashCommandSuggestions.d.ts.map +1 -0
- package/dist/ui/components/input-overlays/SlashCommandSuggestions.js +13 -0
- package/dist/ui/components/input-overlays/SlashCommandSuggestions.js.map +1 -0
- package/dist/ui/hooks/useAgent.d.ts +41 -0
- package/dist/ui/hooks/useAgent.d.ts.map +1 -0
- package/dist/ui/hooks/useAgent.js +205 -0
- package/dist/ui/hooks/useAgent.js.map +1 -0
- package/dist/ui/hooks/useMouseScroll.d.ts +3 -0
- package/dist/ui/hooks/useMouseScroll.d.ts.map +1 -0
- package/dist/ui/hooks/useMouseScroll.js +32 -0
- package/dist/ui/hooks/useMouseScroll.js.map +1 -0
- package/dist/ui/hooks/useSessionStats.d.ts +20 -0
- package/dist/ui/hooks/useSessionStats.d.ts.map +1 -0
- package/dist/ui/hooks/useSessionStats.js +36 -0
- package/dist/ui/hooks/useSessionStats.js.map +1 -0
- package/dist/ui/hooks/useTokenMetrics.d.ts +21 -0
- package/dist/ui/hooks/useTokenMetrics.d.ts.map +1 -0
- package/dist/ui/hooks/useTokenMetrics.js +102 -0
- package/dist/ui/hooks/useTokenMetrics.js.map +1 -0
- package/dist/ui/utils/CodeColorizer.d.ts +5 -0
- package/dist/ui/utils/CodeColorizer.d.ts.map +1 -0
- package/dist/ui/utils/CodeColorizer.js +47 -0
- package/dist/ui/utils/CodeColorizer.js.map +1 -0
- package/dist/utils/constants.d.ts +8 -0
- package/dist/utils/constants.d.ts.map +1 -0
- package/dist/utils/constants.js +104 -0
- package/dist/utils/constants.js.map +1 -0
- package/dist/utils/context.d.ts +36 -0
- package/dist/utils/context.d.ts.map +1 -0
- package/dist/utils/context.js +239 -0
- package/dist/utils/context.js.map +1 -0
- package/dist/utils/file-ops.d.ts +21 -0
- package/dist/utils/file-ops.d.ts.map +1 -0
- package/dist/utils/file-ops.js +125 -0
- package/dist/utils/file-ops.js.map +1 -0
- package/dist/utils/local-settings.d.ts +18 -0
- package/dist/utils/local-settings.d.ts.map +1 -0
- package/dist/utils/local-settings.js +176 -0
- package/dist/utils/local-settings.js.map +1 -0
- package/dist/utils/markdown.d.ts +13 -0
- package/dist/utils/markdown.d.ts.map +1 -0
- package/dist/utils/markdown.js +122 -0
- package/dist/utils/markdown.js.map +1 -0
- package/dist/utils/proxy-config.d.ts +25 -0
- package/dist/utils/proxy-config.d.ts.map +1 -0
- package/dist/utils/proxy-config.js +145 -0
- package/dist/utils/proxy-config.js.map +1 -0
- package/package.json +76 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../../src/core/agent.ts"],"names":[],"mappings":"AAWA,qBAAa,KAAK;IAChB,OAAO,CAAC,QAAQ,CAA4B;IAC5C,OAAO,CAAC,QAAQ,CAAiB;IACjC,OAAO,CAAC,MAAM,CAAuB;IACrC,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,kBAAkB,CAAkB;IAC5C,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,aAAa,CAAC,CAAS;IAExB,WAAW,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,IAAI,CAAC;IAChE,SAAS,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,KAAK,IAAI,CAAC;IAChD,cAAc,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,OAAO,CAAC;QAAE,QAAQ,EAAE,OAAO,CAAC;QAAC,kBAAkB,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC;IACnI,cAAc,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,KAAK,IAAI,CAAC;IAC/D,cAAc,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,KAAK,IAAI,CAAC;IAC/D,eAAe,CAAC,EAAE,CAAC,aAAa,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;IAC9D,UAAU,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,IAAI,CAAC;IAClC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;IAErD,OAAO,CAAC,sBAAsB,CAAgC;IAC9D,OAAO,CAAC,aAAa,CAAkB;IAEvC,OAAO;WAoBM,MAAM,CACjB,KAAK,EAAE,MAAM,EACb,WAAW,EAAE,MAAM,EACnB,aAAa,EAAE,MAAM,GAAG,IAAI,EAC5B,KAAK,CAAC,EAAE,OAAO,EACf,aAAa,CAAC,EAAE,MAAM,GACrB,OAAO,CAAC,KAAK,CAAC;IAQjB,OAAO,CAAC,WAAW;IA4BnB,OAAO,CAAC,yBAAyB;IAI1B,gBAAgB,CAAC,SAAS,EAAE,GAAG;IAW/B,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAU/B,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAKhC,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAKpC,WAAW,IAAI,IAAI;IAMnB,YAAY,IAAI,IAAI;IAIpB,QAAQ,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAW7B,eAAe,IAAI,MAAM;IAIzB,qBAAqB,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI;IAI7C,SAAS,IAAI,IAAI;IAYlB,IAAI,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;YAmE9B,eAAe;CAmD9B"}
|
|
@@ -0,0 +1,266 @@
|
|
|
1
|
+
import { executeTool } from '../tools/tools.js';
|
|
2
|
+
import { validateReadBeforeEdit, getReadBeforeEditError } from '../tools/validators.js';
|
|
3
|
+
import { ALL_TOOL_SCHEMAS, DANGEROUS_TOOLS, APPROVAL_REQUIRED_TOOLS } from '../tools/tool-schemas.js';
|
|
4
|
+
import { ConfigManager } from '../utils/local-settings.js';
|
|
5
|
+
import { LLMFactory } from './llm/factory.js';
|
|
6
|
+
import { getSystemPrompt } from './prompts/system.js';
|
|
7
|
+
import fs from 'fs';
|
|
8
|
+
import path from 'path';
|
|
9
|
+
export class Agent {
|
|
10
|
+
provider = null;
|
|
11
|
+
messages = [];
|
|
12
|
+
apiKey = null;
|
|
13
|
+
model;
|
|
14
|
+
temperature;
|
|
15
|
+
sessionAutoApprove = false;
|
|
16
|
+
systemMessage;
|
|
17
|
+
configManager;
|
|
18
|
+
proxyOverride;
|
|
19
|
+
onToolStart;
|
|
20
|
+
onToolEnd;
|
|
21
|
+
onToolApproval;
|
|
22
|
+
onThinkingText;
|
|
23
|
+
onFinalMessage;
|
|
24
|
+
onMaxIterations;
|
|
25
|
+
onApiUsage;
|
|
26
|
+
onError;
|
|
27
|
+
currentAbortController = null;
|
|
28
|
+
isInterrupted = false;
|
|
29
|
+
constructor(model, temperature, systemMessage, debug, proxyOverride) {
|
|
30
|
+
this.model = model;
|
|
31
|
+
this.temperature = temperature;
|
|
32
|
+
this.configManager = new ConfigManager();
|
|
33
|
+
this.proxyOverride = proxyOverride;
|
|
34
|
+
// Set debug mode
|
|
35
|
+
debugEnabled = debug || false;
|
|
36
|
+
this.systemMessage = systemMessage || this.buildDefaultSystemMessage();
|
|
37
|
+
this.messages.push({ role: 'system', content: this.systemMessage });
|
|
38
|
+
this.loadContext();
|
|
39
|
+
}
|
|
40
|
+
static async create(model, temperature, systemMessage, debug, proxyOverride) {
|
|
41
|
+
const configManager = new ConfigManager();
|
|
42
|
+
const defaultModel = configManager.getDefaultModel();
|
|
43
|
+
const selectedModel = defaultModel || model;
|
|
44
|
+
return new Agent(selectedModel, temperature, systemMessage, debug, proxyOverride);
|
|
45
|
+
}
|
|
46
|
+
loadContext() {
|
|
47
|
+
try {
|
|
48
|
+
const baseDir = process.env.SORY_CONTEXT_DIR || process.cwd();
|
|
49
|
+
// 1. Charger le contexte projet (.sory/context.md)
|
|
50
|
+
const contextPath = path.join(baseDir, '.sory', 'context.md');
|
|
51
|
+
if (fs.existsSync(contextPath)) {
|
|
52
|
+
const ctx = fs.readFileSync(contextPath, 'utf-8');
|
|
53
|
+
this.messages.push({
|
|
54
|
+
role: 'system',
|
|
55
|
+
content: `Contexte du projet (.sory/context.md) :\n\n${ctx}`
|
|
56
|
+
});
|
|
57
|
+
}
|
|
58
|
+
// 2. Charger la mémoire à long terme (SORY.md)
|
|
59
|
+
const memoryPath = path.join(baseDir, 'SORY.md');
|
|
60
|
+
if (fs.existsSync(memoryPath)) {
|
|
61
|
+
const mem = fs.readFileSync(memoryPath, 'utf-8');
|
|
62
|
+
this.messages.push({
|
|
63
|
+
role: 'system',
|
|
64
|
+
content: `Mémoire à long terme (SORY.md) :\n\n${mem}`
|
|
65
|
+
});
|
|
66
|
+
}
|
|
67
|
+
}
|
|
68
|
+
catch (error) {
|
|
69
|
+
debugLog('Failed to load project context/memory:', error);
|
|
70
|
+
}
|
|
71
|
+
}
|
|
72
|
+
buildDefaultSystemMessage() {
|
|
73
|
+
return getSystemPrompt();
|
|
74
|
+
}
|
|
75
|
+
setToolCallbacks(callbacks) {
|
|
76
|
+
this.onToolStart = callbacks.onToolStart;
|
|
77
|
+
this.onToolEnd = callbacks.onToolEnd;
|
|
78
|
+
this.onToolApproval = callbacks.onToolApproval;
|
|
79
|
+
this.onThinkingText = callbacks.onThinkingText;
|
|
80
|
+
this.onFinalMessage = callbacks.onFinalMessage;
|
|
81
|
+
this.onMaxIterations = callbacks.onMaxIterations;
|
|
82
|
+
this.onApiUsage = callbacks.onApiUsage;
|
|
83
|
+
this.onError = callbacks.onError;
|
|
84
|
+
}
|
|
85
|
+
setApiKey(apiKey) {
|
|
86
|
+
debugLog('Setting API key...');
|
|
87
|
+
this.apiKey = apiKey;
|
|
88
|
+
const detected = LLMFactory.detectProvider(apiKey);
|
|
89
|
+
const providerName = this.configManager.getActiveProvider() || detected;
|
|
90
|
+
this.provider = LLMFactory.getProvider(providerName, this.proxyOverride);
|
|
91
|
+
this.provider.configure(apiKey);
|
|
92
|
+
}
|
|
93
|
+
saveApiKey(apiKey) {
|
|
94
|
+
this.configManager.setApiKey(apiKey);
|
|
95
|
+
this.setApiKey(apiKey);
|
|
96
|
+
}
|
|
97
|
+
hasApiKey(provider) {
|
|
98
|
+
const key = this.configManager.getApiKey(provider);
|
|
99
|
+
return !!key && key.length > 0;
|
|
100
|
+
}
|
|
101
|
+
clearApiKey() {
|
|
102
|
+
this.configManager.clearApiKey();
|
|
103
|
+
this.apiKey = null;
|
|
104
|
+
this.provider = null;
|
|
105
|
+
}
|
|
106
|
+
clearHistory() {
|
|
107
|
+
this.messages = this.messages.filter(msg => msg.role === 'system');
|
|
108
|
+
}
|
|
109
|
+
setModel(model) {
|
|
110
|
+
this.model = model;
|
|
111
|
+
this.configManager.setDefaultModel(model);
|
|
112
|
+
const newMsg = this.buildDefaultSystemMessage();
|
|
113
|
+
this.systemMessage = newMsg;
|
|
114
|
+
// Replace the first message (system message) with the new one
|
|
115
|
+
if (this.messages.length > 0 && this.messages[0].role === 'system') {
|
|
116
|
+
this.messages[0].content = newMsg;
|
|
117
|
+
}
|
|
118
|
+
}
|
|
119
|
+
getCurrentModel() {
|
|
120
|
+
return this.model;
|
|
121
|
+
}
|
|
122
|
+
setSessionAutoApprove(enabled) {
|
|
123
|
+
this.sessionAutoApprove = enabled;
|
|
124
|
+
}
|
|
125
|
+
interrupt() {
|
|
126
|
+
debugLog('Interrupting current request');
|
|
127
|
+
this.isInterrupted = true;
|
|
128
|
+
if (this.currentAbortController) {
|
|
129
|
+
this.currentAbortController.abort();
|
|
130
|
+
}
|
|
131
|
+
this.messages.push({
|
|
132
|
+
role: 'system',
|
|
133
|
+
content: 'User has interrupted the request.'
|
|
134
|
+
});
|
|
135
|
+
}
|
|
136
|
+
async chat(userInput) {
|
|
137
|
+
this.isInterrupted = false;
|
|
138
|
+
if (!this.provider) {
|
|
139
|
+
const apiKey = this.configManager.getApiKey();
|
|
140
|
+
if (!apiKey)
|
|
141
|
+
throw new Error('No API key available. Please use /login.');
|
|
142
|
+
this.setApiKey(apiKey);
|
|
143
|
+
}
|
|
144
|
+
this.messages.push({ role: 'user', content: userInput });
|
|
145
|
+
const maxIterations = 50;
|
|
146
|
+
let iteration = 0;
|
|
147
|
+
while (iteration < maxIterations && !this.isInterrupted) {
|
|
148
|
+
try {
|
|
149
|
+
if (!this.provider)
|
|
150
|
+
throw new Error('Provider not initialized');
|
|
151
|
+
let currentContent = '';
|
|
152
|
+
this.currentAbortController = new AbortController();
|
|
153
|
+
const res = await this.provider.streamChat(this.model, this.messages, (chunk) => {
|
|
154
|
+
currentContent += chunk;
|
|
155
|
+
if (this.onThinkingText)
|
|
156
|
+
this.onThinkingText(currentContent);
|
|
157
|
+
}, {
|
|
158
|
+
tools: ALL_TOOL_SCHEMAS,
|
|
159
|
+
temperature: this.temperature,
|
|
160
|
+
maxTokens: 8192, // Increased for long code files
|
|
161
|
+
abortSignal: this.currentAbortController.signal
|
|
162
|
+
});
|
|
163
|
+
const { content, tool_calls, usage, reasoning } = res;
|
|
164
|
+
if (usage && this.onApiUsage)
|
|
165
|
+
this.onApiUsage(usage);
|
|
166
|
+
if (tool_calls && tool_calls.length > 0) {
|
|
167
|
+
const assistantMsg = { role: 'assistant', content: content || '' };
|
|
168
|
+
assistantMsg.tool_calls = tool_calls;
|
|
169
|
+
this.messages.push(assistantMsg);
|
|
170
|
+
for (const toolCall of tool_calls) {
|
|
171
|
+
if (this.isInterrupted)
|
|
172
|
+
return;
|
|
173
|
+
const result = await this.executeToolCall(toolCall);
|
|
174
|
+
this.messages.push({
|
|
175
|
+
role: 'tool',
|
|
176
|
+
tool_call_id: toolCall.id,
|
|
177
|
+
name: toolCall.function.name,
|
|
178
|
+
content: JSON.stringify(result)
|
|
179
|
+
});
|
|
180
|
+
if (result.userRejected)
|
|
181
|
+
return;
|
|
182
|
+
}
|
|
183
|
+
iteration++;
|
|
184
|
+
continue;
|
|
185
|
+
}
|
|
186
|
+
if (this.onFinalMessage)
|
|
187
|
+
this.onFinalMessage(content || '', reasoning);
|
|
188
|
+
this.messages.push({ role: 'assistant', content: content || '' });
|
|
189
|
+
return;
|
|
190
|
+
}
|
|
191
|
+
catch (error) {
|
|
192
|
+
if (error.name === 'AbortError')
|
|
193
|
+
return;
|
|
194
|
+
debugLog('Error occurred during chat:', error);
|
|
195
|
+
throw error;
|
|
196
|
+
}
|
|
197
|
+
}
|
|
198
|
+
}
|
|
199
|
+
async executeToolCall(toolCall) {
|
|
200
|
+
try {
|
|
201
|
+
let toolName = toolCall.function.name;
|
|
202
|
+
if (toolName.startsWith('repo_browser.'))
|
|
203
|
+
toolName = toolName.substring(13);
|
|
204
|
+
let toolArgs;
|
|
205
|
+
try {
|
|
206
|
+
toolArgs = typeof toolCall.function.arguments === 'string'
|
|
207
|
+
? JSON.parse(toolCall.function.arguments)
|
|
208
|
+
: toolCall.function.arguments;
|
|
209
|
+
}
|
|
210
|
+
catch (error) {
|
|
211
|
+
return { error: `Invalid JSON arguments: ${error}`, success: false };
|
|
212
|
+
}
|
|
213
|
+
if (this.onToolStart)
|
|
214
|
+
this.onToolStart(toolName, toolArgs);
|
|
215
|
+
if (toolName === 'edit_file' && toolArgs.file_path) {
|
|
216
|
+
if (!validateReadBeforeEdit(toolArgs.file_path)) {
|
|
217
|
+
const errorMessage = getReadBeforeEditError(toolArgs.file_path);
|
|
218
|
+
const result = { error: errorMessage, success: false };
|
|
219
|
+
if (this.onToolEnd)
|
|
220
|
+
this.onToolEnd(toolName, result);
|
|
221
|
+
return result;
|
|
222
|
+
}
|
|
223
|
+
}
|
|
224
|
+
const isDangerous = DANGEROUS_TOOLS.includes(toolName);
|
|
225
|
+
const requiresApproval = APPROVAL_REQUIRED_TOOLS.includes(toolName);
|
|
226
|
+
const needsApproval = isDangerous || (requiresApproval && !this.sessionAutoApprove);
|
|
227
|
+
if (needsApproval) {
|
|
228
|
+
if (this.onToolApproval) {
|
|
229
|
+
const approvalResult = await this.onToolApproval(toolName, toolArgs);
|
|
230
|
+
if (approvalResult.autoApproveSession && !isDangerous) {
|
|
231
|
+
this.sessionAutoApprove = true;
|
|
232
|
+
}
|
|
233
|
+
if (!approvalResult.approved) {
|
|
234
|
+
const result = { error: 'Tool execution canceled by user', success: false, userRejected: true };
|
|
235
|
+
if (this.onToolEnd)
|
|
236
|
+
this.onToolEnd(toolName, result);
|
|
237
|
+
return result;
|
|
238
|
+
}
|
|
239
|
+
}
|
|
240
|
+
}
|
|
241
|
+
const result = await executeTool(toolName, toolArgs);
|
|
242
|
+
if (this.onToolEnd)
|
|
243
|
+
this.onToolEnd(toolName, result);
|
|
244
|
+
return result;
|
|
245
|
+
}
|
|
246
|
+
catch (error) {
|
|
247
|
+
return { error: `Tool execution error: ${error}`, success: false };
|
|
248
|
+
}
|
|
249
|
+
}
|
|
250
|
+
}
|
|
251
|
+
// Debug logging
|
|
252
|
+
const DEBUG_LOG_FILE = path.join(process.cwd(), 'debug-agent.log');
|
|
253
|
+
let debugLogCleared = false;
|
|
254
|
+
let debugEnabled = false;
|
|
255
|
+
function debugLog(message, data) {
|
|
256
|
+
if (!debugEnabled)
|
|
257
|
+
return;
|
|
258
|
+
if (!debugLogCleared) {
|
|
259
|
+
fs.writeFileSync(DEBUG_LOG_FILE, '');
|
|
260
|
+
debugLogCleared = true;
|
|
261
|
+
}
|
|
262
|
+
const timestamp = new Date().toISOString();
|
|
263
|
+
const logEntry = `[${timestamp}] ${message}${data ? '\n' + JSON.stringify(data, null, 2) : ''}\n`;
|
|
264
|
+
fs.appendFileSync(DEBUG_LOG_FILE, logEntry);
|
|
265
|
+
}
|
|
266
|
+
//# sourceMappingURL=agent.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../src/core/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,sBAAsB,EAAE,sBAAsB,EAAE,MAAM,wBAAwB,CAAC;AACxF,OAAO,EAAE,gBAAgB,EAAE,eAAe,EAAE,uBAAuB,EAAE,MAAM,0BAA0B,CAAC;AACtG,OAAO,EAAE,aAAa,EAAE,MAAM,4BAA4B,CAAC;AAG3D,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AAExB,MAAM,OAAO,KAAK;IACR,QAAQ,GAAuB,IAAI,CAAC;IACpC,QAAQ,GAAc,EAAE,CAAC;IACzB,MAAM,GAAkB,IAAI,CAAC;IAC7B,KAAK,CAAS;IACd,WAAW,CAAS;IACpB,kBAAkB,GAAY,KAAK,CAAC;IACpC,aAAa,CAAS;IACtB,aAAa,CAAgB;IAC7B,aAAa,CAAU;IAExB,WAAW,CAAqD;IAChE,SAAS,CAAuC;IAChD,cAAc,CAAqH;IACnI,cAAc,CAAiD;IAC/D,cAAc,CAAiD;IAC/D,eAAe,CAA+C;IAC9D,UAAU,CAAwB;IAClC,OAAO,CAAuC;IAE7C,sBAAsB,GAA2B,IAAI,CAAC;IACtD,aAAa,GAAY,KAAK,CAAC;IAEvC,YACE,KAAa,EACb,WAAmB,EACnB,aAA4B,EAC5B,KAAe,EACf,aAAsB;QAEtB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC;QACzC,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QAEnC,iBAAiB;QACjB,YAAY,GAAG,KAAK,IAAI,KAAK,CAAC;QAE9B,IAAI,CAAC,aAAa,GAAG,aAAa,IAAI,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACvE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC;QACpE,IAAI,CAAC,WAAW,EAAE,CAAC;IACrB,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,MAAM,CACjB,KAAa,EACb,WAAmB,EACnB,aAA4B,EAC5B,KAAe,EACf,aAAsB;QAEtB,MAAM,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC;QAC1C,MAAM,YAAY,GAAG,aAAa,CAAC,eAAe,EAAE,CAAC;QACrD,MAAM,aAAa,GAAG,YAAY,IAAI,KAAK,CAAC;QAE5C,OAAO,IAAI,KAAK,CAAC,aAAa,EAAE,WAAW,EAAE,aAAa,EAAE,KAAK,EAAE,aAAa,CAAC,CAAC;IACpF,CAAC;IAEO,WAAW;QACjB,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;YAE9D,mDAAmD;YACnD,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC;YAC9D,IAAI,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;gBAC/B,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;gBAClD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;oBACjB,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,8CAA8C,GAAG,EAAE;iBAC7D,CAAC,CAAC;YACL,CAAC;YAED,+CAA+C;YAC/C,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YACjD,IAAI,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC9B,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;gBACjD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;oBACjB,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,uCAAuC,GAAG,EAAE;iBACtD,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,QAAQ,CAAC,wCAAwC,EAAE,KAAK,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;IAEO,yBAAyB;QAC/B,OAAO,eAAe,EAAE,CAAC;IAC3B,CAAC;IAEM,gBAAgB,CAAC,SAAc;QACpC,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC,WAAW,CAAC;QACzC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC,SAAS,CAAC;QACrC,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC,cAAc,CAAC;QAC/C,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC,cAAc,CAAC;QAC/C,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC,cAAc,CAAC;QAC/C,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC,eAAe,CAAC;QACjD,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC,UAAU,CAAC;QACvC,IAAI,CAAC,OAAO,GAAG,SAAS,CAAC,OAAO,CAAC;IACnC,CAAC;IAEM,SAAS,CAAC,MAAc;QAC7B,QAAQ,CAAC,oBAAoB,CAAC,CAAC;QAC/B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,MAAM,QAAQ,GAAG,UAAU,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QACnD,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,iBAAiB,EAAE,IAAI,QAAQ,CAAC;QAExE,IAAI,CAAC,QAAQ,GAAG,UAAU,CAAC,WAAW,CAAC,YAAY,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QACzE,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IAEM,UAAU,CAAC,MAAc;QAC9B,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACrC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IACzB,CAAC;IAEM,SAAS,CAAC,QAAgB;QAC/B,MAAM,GAAG,GAAG,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACnD,OAAO,CAAC,CAAC,GAAG,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC;IACjC,CAAC;IAEM,WAAW;QAChB,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,CAAC;QACjC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;IACvB,CAAC;IAEM,YAAY;QACjB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;IACrE,CAAC;IAEM,QAAQ,CAAC,KAAa;QAC3B,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,yBAAyB,EAAE,CAAC;QAChD,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC;QAC5B,8DAA8D;QAC9D,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACnE,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,GAAG,MAAM,CAAC;QACpC,CAAC;IACH,CAAC;IAEM,eAAe;QACpB,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAEM,qBAAqB,CAAC,OAAgB;QAC3C,IAAI,CAAC,kBAAkB,GAAG,OAAO,CAAC;IACpC,CAAC;IAEM,SAAS;QACd,QAAQ,CAAC,8BAA8B,CAAC,CAAC;QACzC,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;QAC1B,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;YAChC,IAAI,CAAC,sBAAsB,CAAC,KAAK,EAAE,CAAC;QACtC,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,mCAAmC;SAC7C,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,SAAiB;QAC1B,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;QAE3B,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnB,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,CAAC;YAC9C,IAAI,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;YACzE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACzB,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;QAEzD,MAAM,aAAa,GAAG,EAAE,CAAC;QACzB,IAAI,SAAS,GAAG,CAAC,CAAC;QAElB,OAAO,SAAS,GAAG,aAAa,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;YACxD,IAAI,CAAC;gBACH,IAAI,CAAC,IAAI,CAAC,QAAQ;oBAAE,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;gBAEhE,IAAI,cAAc,GAAG,EAAE,CAAC;gBACxB,IAAI,CAAC,sBAAsB,GAAG,IAAI,eAAe,EAAE,CAAC;gBAEpD,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,QAAS,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE;oBAC/E,cAAc,IAAI,KAAK,CAAC;oBACxB,IAAI,IAAI,CAAC,cAAc;wBAAE,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;gBAC/D,CAAC,EAAE;oBACD,KAAK,EAAE,gBAAgB;oBACvB,WAAW,EAAE,IAAI,CAAC,WAAW;oBAC7B,SAAS,EAAE,IAAI,EAAE,gCAAgC;oBACjD,WAAW,EAAE,IAAI,CAAC,sBAAsB,CAAC,MAAM;iBAChD,CAAC,CAAC;gBAEH,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,GAAG,CAAC;gBAEtD,IAAI,KAAK,IAAI,IAAI,CAAC,UAAU;oBAAE,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;gBAErD,IAAI,UAAU,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACxC,MAAM,YAAY,GAAY,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,EAAE,CAAC;oBAC5E,YAAY,CAAC,UAAU,GAAG,UAAU,CAAC;oBACrC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;oBAEjC,KAAK,MAAM,QAAQ,IAAI,UAAU,EAAE,CAAC;wBAClC,IAAI,IAAI,CAAC,aAAa;4BAAE,OAAO;wBAC/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;wBACpD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;4BACjB,IAAI,EAAE,MAAM;4BACZ,YAAY,EAAE,QAAQ,CAAC,EAAE;4BACzB,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;4BAC5B,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;yBAChC,CAAC,CAAC;wBACH,IAAI,MAAM,CAAC,YAAY;4BAAE,OAAO;oBAClC,CAAC;oBACD,SAAS,EAAE,CAAC;oBACZ,SAAS;gBACX,CAAC;gBAED,IAAI,IAAI,CAAC,cAAc;oBAAE,IAAI,CAAC,cAAc,CAAC,OAAO,IAAI,EAAE,EAAE,SAAS,CAAC,CAAC;gBACvE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,EAAE,CAAC,CAAC;gBAClE,OAAO;YAET,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY;oBAAE,OAAO;gBACxC,QAAQ,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;gBAC/C,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,QAAa;QACzC,IAAI,CAAC;YACH,IAAI,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC;YACtC,IAAI,QAAQ,CAAC,UAAU,CAAC,eAAe,CAAC;gBAAE,QAAQ,GAAG,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;YAE5E,IAAI,QAAa,CAAC;YAClB,IAAI,CAAC;gBACH,QAAQ,GAAG,OAAO,QAAQ,CAAC,QAAQ,CAAC,SAAS,KAAK,QAAQ;oBACxD,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;oBACzC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;YAClC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,EAAE,KAAK,EAAE,2BAA2B,KAAK,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YACvE,CAAC;YAED,IAAI,IAAI,CAAC,WAAW;gBAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YAE3D,IAAI,QAAQ,KAAK,WAAW,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;gBACnD,IAAI,CAAC,sBAAsB,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;oBAChD,MAAM,YAAY,GAAG,sBAAsB,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;oBAChE,MAAM,MAAM,GAAG,EAAE,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;oBACvD,IAAI,IAAI,CAAC,SAAS;wBAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;oBACrD,OAAO,MAAM,CAAC;gBAChB,CAAC;YACH,CAAC;YAED,MAAM,WAAW,GAAG,eAAe,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACvD,MAAM,gBAAgB,GAAG,uBAAuB,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACpE,MAAM,aAAa,GAAG,WAAW,IAAI,CAAC,gBAAgB,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YAEpF,IAAI,aAAa,EAAE,CAAC;gBAClB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;oBACxB,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;oBACrE,IAAI,cAAc,CAAC,kBAAkB,IAAI,CAAC,WAAW,EAAE,CAAC;wBACtD,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC;oBACjC,CAAC;oBACD,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,CAAC;wBAC7B,MAAM,MAAM,GAAG,EAAE,KAAK,EAAE,iCAAiC,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;wBAChG,IAAI,IAAI,CAAC,SAAS;4BAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;wBACrD,OAAO,MAAM,CAAC;oBAChB,CAAC;gBACH,CAAC;YACH,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACrD,IAAI,IAAI,CAAC,SAAS;gBAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YACrD,OAAO,MAAM,CAAC;QAEhB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,EAAE,KAAK,EAAE,yBAAyB,KAAK,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;QACrE,CAAC;IACH,CAAC;CACF;AAED,gBAAgB;AAChB,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,iBAAiB,CAAC,CAAC;AACnE,IAAI,eAAe,GAAG,KAAK,CAAC;AAC5B,IAAI,YAAY,GAAG,KAAK,CAAC;AAEzB,SAAS,QAAQ,CAAC,OAAe,EAAE,IAAU;IAC3C,IAAI,CAAC,YAAY;QAAE,OAAO;IAC1B,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,EAAE,CAAC,aAAa,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;QACrC,eAAe,GAAG,IAAI,CAAC;IACzB,CAAC;IACD,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAC3C,MAAM,QAAQ,GAAG,IAAI,SAAS,KAAK,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;IAClG,EAAE,CAAC,cAAc,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;AAC9C,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cli.d.ts","sourceRoot":"","sources":["../../src/core/cli.ts"],"names":[],"mappings":""}
|
package/dist/core/cli.js
ADDED
|
@@ -0,0 +1,48 @@
|
|
|
1
|
+
#!/usr/bin/env node
|
|
2
|
+
import { Command } from 'commander';
|
|
3
|
+
import chalk from 'chalk';
|
|
4
|
+
import { render } from 'ink';
|
|
5
|
+
import React from 'react';
|
|
6
|
+
import { Agent } from './agent.js';
|
|
7
|
+
import App from '../ui/App.js';
|
|
8
|
+
const program = new Command();
|
|
9
|
+
async function startChat(temperature, system, debug, proxy) {
|
|
10
|
+
console.log(chalk.hex('#FF4500')(`
|
|
11
|
+
███████ ██████ ██████ ██ ██
|
|
12
|
+
██ ██ ██ ██ ██ ██ ██
|
|
13
|
+
███████ ██ ██ ██████ ████
|
|
14
|
+
██ ██ ██ ██ ██ ██
|
|
15
|
+
███████ ██████ ██ ██ ██
|
|
16
|
+
`));
|
|
17
|
+
let defaultModel = 'mistral-large-latest';
|
|
18
|
+
if (proxy) {
|
|
19
|
+
try {
|
|
20
|
+
new URL(proxy);
|
|
21
|
+
}
|
|
22
|
+
catch (error) {
|
|
23
|
+
console.log(chalk.red('Invalid proxy URL provided'));
|
|
24
|
+
process.exit(1);
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
try {
|
|
28
|
+
const agent = await Agent.create(defaultModel, temperature, system, debug, proxy);
|
|
29
|
+
render(React.createElement(App, { agent }));
|
|
30
|
+
}
|
|
31
|
+
catch (error) {
|
|
32
|
+
console.log(chalk.red(`Error initializing agent: ${error}`));
|
|
33
|
+
process.exit(1);
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
program
|
|
37
|
+
.name('sory')
|
|
38
|
+
.description('SORY AI CLI')
|
|
39
|
+
.version('1.0.0')
|
|
40
|
+
.option('-t, --temperature <temperature>', 'Temperature for generation', parseFloat, 1.0)
|
|
41
|
+
.option('-s, --system <message>', 'Custom system message')
|
|
42
|
+
.option('-d, --debug', 'Enable debug logging')
|
|
43
|
+
.option('-p, --proxy <url>', 'Proxy URL')
|
|
44
|
+
.action(async (options) => {
|
|
45
|
+
await startChat(options.temperature, options.system || null, options.debug, options.proxy);
|
|
46
|
+
});
|
|
47
|
+
program.parse();
|
|
48
|
+
//# sourceMappingURL=cli.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cli.js","sourceRoot":"","sources":["../../src/core/cli.ts"],"names":[],"mappings":";AACA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,MAAM,EAAE,MAAM,KAAK,CAAC;AAC7B,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,GAAG,MAAM,cAAc,CAAC;AAE/B,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;AAE9B,KAAK,UAAU,SAAS,CACtB,WAAmB,EACnB,MAAqB,EACrB,KAAe,EACf,KAAc;IAEd,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;;;;;;GAMhC,CAAC,CAAC,CAAC;IAEJ,IAAI,YAAY,GAAG,sBAAsB,CAAC;IAE1C,IAAI,KAAK,EAAE,CAAC;QACV,IAAI,CAAC;YACH,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC;QACjB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC,CAAC;YACrD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC;IAED,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,YAAY,EAAE,WAAW,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;QAClF,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;IAC9C,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,6BAA6B,KAAK,EAAE,CAAC,CAAC,CAAC;QAC7D,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC;AAED,OAAO;KACJ,IAAI,CAAC,MAAM,CAAC;KACZ,WAAW,CAAC,aAAa,CAAC;KAC1B,OAAO,CAAC,OAAO,CAAC;KAChB,MAAM,CAAC,iCAAiC,EAAE,4BAA4B,EAAE,UAAU,EAAE,GAAG,CAAC;KACxF,MAAM,CAAC,wBAAwB,EAAE,uBAAuB,CAAC;KACzD,MAAM,CAAC,aAAa,EAAE,sBAAsB,CAAC;KAC7C,MAAM,CAAC,mBAAmB,EAAE,WAAW,CAAC;KACxC,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;IACxB,MAAM,SAAS,CACb,OAAO,CAAC,WAAW,EACnB,OAAO,CAAC,MAAM,IAAI,IAAI,EACtB,OAAO,CAAC,KAAK,EACb,OAAO,CAAC,KAAK,CACd,CAAC;AACJ,CAAC,CAAC,CAAC;AAEL,OAAO,CAAC,KAAK,EAAE,CAAC"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import { LLMProvider } from './types.js';
|
|
2
|
+
export declare class LLMFactory {
|
|
3
|
+
/**
|
|
4
|
+
* Get a provider instance by name
|
|
5
|
+
*/
|
|
6
|
+
static getProvider(name: string, proxyUrl?: string): LLMProvider;
|
|
7
|
+
/**
|
|
8
|
+
* Auto-detect provider based on API key format
|
|
9
|
+
*/
|
|
10
|
+
static detectProvider(apiKey: string): string;
|
|
11
|
+
}
|
|
12
|
+
//# sourceMappingURL=factory.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"factory.d.ts","sourceRoot":"","sources":["../../../src/core/llm/factory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAKzC,qBAAa,UAAU;IACrB;;OAEG;IACH,MAAM,CAAC,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,WAAW;IA0ChE;;OAEG;IACH,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM;CAiB9C"}
|
|
@@ -0,0 +1,44 @@
|
|
|
1
|
+
import { GroqProvider } from './providers/groq.js';
|
|
2
|
+
import { OpenAICompatibleProvider } from './providers/openai-compatible.js';
|
|
3
|
+
import { GeminiProvider } from './providers/gemini.js';
|
|
4
|
+
export class LLMFactory {
|
|
5
|
+
/**
|
|
6
|
+
* Get a provider instance by name
|
|
7
|
+
*/
|
|
8
|
+
static getProvider(name, proxyUrl) {
|
|
9
|
+
switch (name.toLowerCase()) {
|
|
10
|
+
case 'groq':
|
|
11
|
+
return new GroqProvider(proxyUrl);
|
|
12
|
+
case 'gemini':
|
|
13
|
+
return new GeminiProvider(); // Gemini SDK uses generic fetch/node http, might need custom proxy setup later
|
|
14
|
+
case 'openrouter':
|
|
15
|
+
return new OpenAICompatibleProvider('openrouter', 'https://openrouter.ai/api/v1', proxyUrl);
|
|
16
|
+
case 'openai':
|
|
17
|
+
return new OpenAICompatibleProvider('openai', 'https://api.openai.com/v1', proxyUrl);
|
|
18
|
+
case 'deepseek':
|
|
19
|
+
return new OpenAICompatibleProvider('deepseek', 'https://api.deepseek.com', proxyUrl);
|
|
20
|
+
case 'mistral':
|
|
21
|
+
return new OpenAICompatibleProvider('mistral', 'https://api.mistral.ai/v1', proxyUrl);
|
|
22
|
+
default:
|
|
23
|
+
console.warn(`Unknown provider '${name}', defaulting to Groq`);
|
|
24
|
+
return new GroqProvider(proxyUrl);
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
/**
|
|
28
|
+
* Auto-detect provider based on API key format
|
|
29
|
+
*/
|
|
30
|
+
static detectProvider(apiKey) {
|
|
31
|
+
const key = apiKey.trim();
|
|
32
|
+
if (key.startsWith('gsk_')) {
|
|
33
|
+
return 'groq';
|
|
34
|
+
}
|
|
35
|
+
if (key.startsWith('AIza')) { // Google API Keys start with AIza
|
|
36
|
+
return 'gemini';
|
|
37
|
+
}
|
|
38
|
+
if (key.startsWith('sk-or-')) {
|
|
39
|
+
return 'openrouter';
|
|
40
|
+
}
|
|
41
|
+
return 'groq';
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
//# sourceMappingURL=factory.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"factory.js","sourceRoot":"","sources":["../../../src/core/llm/factory.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAEvD,MAAM,OAAO,UAAU;IACrB;;OAEG;IACH,MAAM,CAAC,WAAW,CAAC,IAAY,EAAE,QAAiB;QAChD,QAAQ,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;YAC3B,KAAK,MAAM;gBACT,OAAO,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;YAEpC,KAAK,QAAQ;gBACX,OAAO,IAAI,cAAc,EAAE,CAAC,CAAC,+EAA+E;YAE9G,KAAK,YAAY;gBACf,OAAO,IAAI,wBAAwB,CACjC,YAAY,EACZ,8BAA8B,EAC9B,QAAQ,CACT,CAAC;YAEJ,KAAK,QAAQ;gBACX,OAAO,IAAI,wBAAwB,CACjC,QAAQ,EACR,2BAA2B,EAC3B,QAAQ,CACT,CAAC;YAEJ,KAAK,UAAU;gBACb,OAAO,IAAI,wBAAwB,CACjC,UAAU,EACV,0BAA0B,EAC1B,QAAQ,CACT,CAAC;YAEJ,KAAK,SAAS;gBACZ,OAAO,IAAI,wBAAwB,CACjC,SAAS,EACT,2BAA2B,EAC3B,QAAQ,CACT,CAAC;YAEJ;gBACE,OAAO,CAAC,IAAI,CAAC,qBAAqB,IAAI,uBAAuB,CAAC,CAAC;gBAC/D,OAAO,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,cAAc,CAAC,MAAc;QAClC,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;QAE1B,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3B,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,kCAAkC;YAC9D,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,IAAI,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC7B,OAAO,YAAY,CAAC;QACtB,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;CACF"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import { LLMProvider, Message, CompletionResponse } from '../types.js';
|
|
2
|
+
export declare class GeminiProvider implements LLMProvider {
|
|
3
|
+
name: string;
|
|
4
|
+
private genAI;
|
|
5
|
+
private model;
|
|
6
|
+
configure(apiKey: string): void;
|
|
7
|
+
chat(modelName: string, messages: Message[], options?: {
|
|
8
|
+
temperature?: number;
|
|
9
|
+
maxTokens?: number;
|
|
10
|
+
tools?: any[];
|
|
11
|
+
toolChoice?: 'auto' | 'none' | 'required';
|
|
12
|
+
abortSignal?: AbortSignal;
|
|
13
|
+
}): Promise<CompletionResponse>;
|
|
14
|
+
streamChat(modelName: string, messages: Message[], onChunk: (chunk: string) => void, options?: {
|
|
15
|
+
temperature?: number;
|
|
16
|
+
maxTokens?: number;
|
|
17
|
+
tools?: any[];
|
|
18
|
+
toolChoice?: 'auto' | 'none' | 'required';
|
|
19
|
+
abortSignal?: AbortSignal;
|
|
20
|
+
}): Promise<CompletionResponse>;
|
|
21
|
+
}
|
|
22
|
+
//# sourceMappingURL=gemini.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../../../src/core/llm/providers/gemini.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AAEvE,qBAAa,cAAe,YAAW,WAAW;IAChD,IAAI,SAAY;IAChB,OAAO,CAAC,KAAK,CAAmC;IAChD,OAAO,CAAC,KAAK,CAAgC;IAE7C,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAIzB,IAAI,CACR,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,OAAO,EAAE,EACnB,OAAO,CAAC,EAAE;QACR,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,KAAK,CAAC,EAAE,GAAG,EAAE,CAAC;QACd,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,UAAU,CAAC;QAC1C,WAAW,CAAC,EAAE,WAAW,CAAC;KAC3B,GACA,OAAO,CAAC,kBAAkB,CAAC;IAyIxB,UAAU,CACd,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,OAAO,EAAE,EACnB,OAAO,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,EAChC,OAAO,CAAC,EAAE;QACR,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,KAAK,CAAC,EAAE,GAAG,EAAE,CAAC;QACd,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,UAAU,CAAC;QAC1C,WAAW,CAAC,EAAE,WAAW,CAAC;KAC3B,GACA,OAAO,CAAC,kBAAkB,CAAC;CA8C/B"}
|
|
@@ -0,0 +1,167 @@
|
|
|
1
|
+
import { GoogleGenerativeAI } from '@google/generative-ai';
|
|
2
|
+
export class GeminiProvider {
|
|
3
|
+
name = 'gemini';
|
|
4
|
+
genAI = null;
|
|
5
|
+
model = null;
|
|
6
|
+
configure(apiKey) {
|
|
7
|
+
this.genAI = new GoogleGenerativeAI(apiKey);
|
|
8
|
+
}
|
|
9
|
+
async chat(modelName, messages, options) {
|
|
10
|
+
if (!this.genAI) {
|
|
11
|
+
throw new Error('Gemini provider not configured. Call configure() first.');
|
|
12
|
+
}
|
|
13
|
+
// Convert OpenAI-style tools to Gemini tools format
|
|
14
|
+
const geminiTools = options?.tools ? [{
|
|
15
|
+
functionDeclarations: options.tools.map((t) => t.function)
|
|
16
|
+
}] : undefined;
|
|
17
|
+
this.model = this.genAI.getGenerativeModel({
|
|
18
|
+
model: modelName,
|
|
19
|
+
generationConfig: {
|
|
20
|
+
temperature: options?.temperature,
|
|
21
|
+
maxOutputTokens: options?.maxTokens,
|
|
22
|
+
},
|
|
23
|
+
tools: geminiTools
|
|
24
|
+
});
|
|
25
|
+
// Convert messages to Gemini history format
|
|
26
|
+
// Gemini separates "System Instructions" from the chat history in newer models,
|
|
27
|
+
// or treats the first part of history as context.
|
|
28
|
+
// However, the cleanest way in the SDK is usually startChat() with history.
|
|
29
|
+
// Extract system message
|
|
30
|
+
const systemMessage = messages.find(m => m.role === 'system')?.content;
|
|
31
|
+
const chatMessages = messages.filter(m => m.role !== 'system');
|
|
32
|
+
// Convert conversation history
|
|
33
|
+
const history = chatMessages.slice(0, -1).map(msg => {
|
|
34
|
+
let role = 'user';
|
|
35
|
+
if (msg.role === 'assistant')
|
|
36
|
+
role = 'model';
|
|
37
|
+
else if (msg.role === 'tool')
|
|
38
|
+
role = 'function'; // Gemini SDK handles function responses specifically
|
|
39
|
+
// Gemini is strict: 'user' or 'model'. Function outputs are handled differently in SDK usually,
|
|
40
|
+
// but here we are mapping "tool" role.
|
|
41
|
+
// Actually, standard Gemini chat history expects 'user' or 'model'.
|
|
42
|
+
// Tool outputs are sent as 'functionResponse' parts within a 'function' role (conceptually).
|
|
43
|
+
const parts = [];
|
|
44
|
+
if (msg.content) {
|
|
45
|
+
parts.push({ text: msg.content });
|
|
46
|
+
}
|
|
47
|
+
if (msg.tool_calls) {
|
|
48
|
+
msg.tool_calls.forEach(tc => {
|
|
49
|
+
parts.push({
|
|
50
|
+
functionCall: {
|
|
51
|
+
name: tc.function.name,
|
|
52
|
+
args: JSON.parse(tc.function.arguments)
|
|
53
|
+
}
|
|
54
|
+
});
|
|
55
|
+
});
|
|
56
|
+
}
|
|
57
|
+
if (msg.role === 'tool') {
|
|
58
|
+
role = 'function'; // Special internal handling, mapped to 'user' with functionResponse below?
|
|
59
|
+
// Wait, the SDK `history` array format is strictly { role: 'user' | 'model', parts: [...] }
|
|
60
|
+
// A function response from the tool (us) back to the model IS a 'user' turn (or 'function' role in REST API).
|
|
61
|
+
// In the Node SDK, it's confusing. Let's stick to the REST API structure mapping if possible.
|
|
62
|
+
// CORRECTION: In SDK `startChat`, history items must be { role: 'user' | 'model', parts: [...] }.
|
|
63
|
+
// Tool responses are 'user' role with `functionResponse` part.
|
|
64
|
+
return {
|
|
65
|
+
role: role === 'function' ? 'function' : role, // We'll fix 'function' role below
|
|
66
|
+
parts: msg.role === 'tool' ? [{
|
|
67
|
+
functionResponse: {
|
|
68
|
+
name: msg.name || '', // We need the tool name here! OpenAI format doesn't always have it in the tool_message
|
|
69
|
+
response: { result: JSON.parse(msg.content) }
|
|
70
|
+
}
|
|
71
|
+
}] : parts
|
|
72
|
+
};
|
|
73
|
+
}
|
|
74
|
+
return { role, parts };
|
|
75
|
+
});
|
|
76
|
+
// Fix roles for history: 'function' role isn't valid in startChat history, it must be 'user' or 'model'
|
|
77
|
+
// But wait, the SDK documentation says function responses come from 'function' role?
|
|
78
|
+
// Actually, for `sendMessage`, the history is implicit.
|
|
79
|
+
// Let's manually construct the prompt parts for the LAST message if we can't use startChat easily with tools history.
|
|
80
|
+
// STRATEGY CHANGE: Gemini multi-turn with tools is complex to map 1:1 with OpenAI history.
|
|
81
|
+
// We will use `startChat` and reconstruct the history carefully.
|
|
82
|
+
const validHistory = history.map(h => ({
|
|
83
|
+
role: h.role === 'function' ? 'user' : h.role, // Map function response to user turn
|
|
84
|
+
parts: h.parts
|
|
85
|
+
}));
|
|
86
|
+
const chatSession = this.model.startChat({
|
|
87
|
+
history: validHistory,
|
|
88
|
+
systemInstruction: systemMessage ? { role: 'system', parts: [{ text: systemMessage }] } : undefined
|
|
89
|
+
});
|
|
90
|
+
const lastMsg = chatMessages[chatMessages.length - 1];
|
|
91
|
+
let result;
|
|
92
|
+
if (lastMsg.role === 'tool') {
|
|
93
|
+
// This is a tool response turn.
|
|
94
|
+
const toolResponsePart = {
|
|
95
|
+
functionResponse: {
|
|
96
|
+
name: lastMsg.name || '',
|
|
97
|
+
response: { content: JSON.parse(lastMsg.content) }
|
|
98
|
+
}
|
|
99
|
+
};
|
|
100
|
+
result = await chatSession.sendMessage([toolResponsePart]);
|
|
101
|
+
}
|
|
102
|
+
else {
|
|
103
|
+
// Normal user message
|
|
104
|
+
result = await chatSession.sendMessage(lastMsg.content);
|
|
105
|
+
}
|
|
106
|
+
const response = await result.response;
|
|
107
|
+
const text = response.text();
|
|
108
|
+
const calls = response.functionCalls();
|
|
109
|
+
const tool_calls = calls ? calls.map(call => ({
|
|
110
|
+
id: 'call_' + Math.random().toString(36).substr(2, 9), // Gemini doesn't give IDs, generate one
|
|
111
|
+
type: 'function',
|
|
112
|
+
function: {
|
|
113
|
+
name: call.name,
|
|
114
|
+
arguments: JSON.stringify(call.args)
|
|
115
|
+
}
|
|
116
|
+
})) : undefined;
|
|
117
|
+
return {
|
|
118
|
+
content: text,
|
|
119
|
+
tool_calls: tool_calls,
|
|
120
|
+
usage: {
|
|
121
|
+
prompt_tokens: 0,
|
|
122
|
+
completion_tokens: 0,
|
|
123
|
+
total_tokens: 0
|
|
124
|
+
}
|
|
125
|
+
};
|
|
126
|
+
}
|
|
127
|
+
async streamChat(modelName, messages, onChunk, options) {
|
|
128
|
+
if (!this.genAI)
|
|
129
|
+
throw new Error('Gemini not configured');
|
|
130
|
+
const geminiTools = options?.tools ? [{
|
|
131
|
+
functionDeclarations: options.tools.map((t) => t.function)
|
|
132
|
+
}] : undefined;
|
|
133
|
+
const genModel = this.genAI.getGenerativeModel({
|
|
134
|
+
model: modelName,
|
|
135
|
+
tools: geminiTools
|
|
136
|
+
});
|
|
137
|
+
const systemMessage = messages.find(m => m.role === 'system')?.content;
|
|
138
|
+
const chatMessages = messages.filter(m => m.role !== 'system');
|
|
139
|
+
const history = chatMessages.slice(0, -1).map(m => ({
|
|
140
|
+
role: m.role === 'assistant' ? 'model' : 'user',
|
|
141
|
+
parts: [{ text: m.content || '' }]
|
|
142
|
+
}));
|
|
143
|
+
const chatSession = genModel.startChat({
|
|
144
|
+
history,
|
|
145
|
+
systemInstruction: systemMessage ? { role: 'system', parts: [{ text: systemMessage }] } : undefined
|
|
146
|
+
});
|
|
147
|
+
const lastMsg = chatMessages[chatMessages.length - 1];
|
|
148
|
+
const result = await chatSession.sendMessageStream(lastMsg.content || '');
|
|
149
|
+
let fullText = '';
|
|
150
|
+
for await (const chunk of result.stream) {
|
|
151
|
+
const chunkText = chunk.text();
|
|
152
|
+
fullText += chunkText;
|
|
153
|
+
onChunk(chunkText);
|
|
154
|
+
}
|
|
155
|
+
const response = await result.response;
|
|
156
|
+
const calls = response.functionCalls();
|
|
157
|
+
return {
|
|
158
|
+
content: fullText,
|
|
159
|
+
tool_calls: calls?.map(c => ({
|
|
160
|
+
id: 'call_' + Math.random().toString(36).substr(2, 9),
|
|
161
|
+
function: { name: c.name, arguments: JSON.stringify(c.args) }
|
|
162
|
+
})),
|
|
163
|
+
usage: { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 }
|
|
164
|
+
};
|
|
165
|
+
}
|
|
166
|
+
}
|
|
167
|
+
//# sourceMappingURL=gemini.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../../../src/core/llm/providers/gemini.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAyB,MAAM,uBAAuB,CAAC;AAGlF,MAAM,OAAO,cAAc;IACzB,IAAI,GAAG,QAAQ,CAAC;IACR,KAAK,GAA8B,IAAI,CAAC;IACxC,KAAK,GAA2B,IAAI,CAAC;IAE7C,SAAS,CAAC,MAAc;QACtB,IAAI,CAAC,KAAK,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,IAAI,CACR,SAAiB,EACjB,QAAmB,EACnB,OAMC;QAED,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;QAC7E,CAAC;QAED,oDAAoD;QACpD,MAAM,WAAW,GAAG,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;gBACpC,oBAAoB,EAAE,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC;aAChE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAEf,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC;YACzC,KAAK,EAAE,SAAS;YAChB,gBAAgB,EAAE;gBAChB,WAAW,EAAE,OAAO,EAAE,WAAW;gBACjC,eAAe,EAAE,OAAO,EAAE,SAAS;aACpC;YACD,KAAK,EAAE,WAAW;SACnB,CAAC,CAAC;QAEH,4CAA4C;QAC5C,gFAAgF;QAChF,kDAAkD;QAClD,4EAA4E;QAE5E,yBAAyB;QACzB,MAAM,aAAa,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,EAAE,OAAO,CAAC;QACvE,MAAM,YAAY,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QAE/D,+BAA+B;QAC/B,MAAM,OAAO,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;YAClD,IAAI,IAAI,GAAG,MAAM,CAAC;YAClB,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW;gBAAE,IAAI,GAAG,OAAO,CAAC;iBACxC,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM;gBAAE,IAAI,GAAG,UAAU,CAAC,CAAC,qDAAqD;YAEtG,gGAAgG;YAChG,uCAAuC;YACvC,oEAAoE;YACpE,6FAA6F;YAE7F,MAAM,KAAK,GAAW,EAAE,CAAC;YAEzB,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;gBAChB,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YACpC,CAAC;YAED,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;gBACnB,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE;oBAC1B,KAAK,CAAC,IAAI,CAAC;wBACT,YAAY,EAAE;4BACZ,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;4BACtB,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;yBACxC;qBACF,CAAC,CAAC;gBACL,CAAC,CAAC,CAAC;YACL,CAAC;YAED,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACxB,IAAI,GAAG,UAAU,CAAC,CAAC,2EAA2E;gBAC9F,4FAA4F;gBAC5F,8GAA8G;gBAC9G,8FAA8F;gBAE9F,kGAAkG;gBAClG,+DAA+D;gBAC/D,OAAO;oBACL,IAAI,EAAE,IAAI,KAAK,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,EAAE,kCAAkC;oBACjF,KAAK,EAAE,GAAG,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC;4BAC5B,gBAAgB,EAAE;gCAChB,IAAI,EAAE,GAAG,CAAC,IAAI,IAAI,EAAE,EAAE,uFAAuF;gCAC7G,QAAQ,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;6BAC9C;yBACF,CAAC,CAAC,CAAC,CAAC,KAAK;iBACX,CAAC;YACJ,CAAC;YAED,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,wGAAwG;QACxG,qFAAqF;QACrF,wDAAwD;QACxD,sHAAsH;QAEtH,2FAA2F;QAC3F,iEAAiE;QAEjE,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACrC,IAAI,EAAE,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,qCAAqC;YACpF,KAAK,EAAE,CAAC,CAAC,KAAK;SACf,CAAC,CAAC,CAAC;QAEJ,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC;YACvC,OAAO,EAAE,YAAY;YACrB,iBAAiB,EAAE,aAAa,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;SACpG,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,YAAY,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACtD,IAAI,MAAM,CAAC;QAEX,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC3B,gCAAgC;YAChC,MAAM,gBAAgB,GAAG;gBACvB,gBAAgB,EAAE;oBACf,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,EAAE;oBACxB,QAAQ,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;iBACpD;aACF,CAAC;YACF,MAAM,GAAG,MAAM,WAAW,CAAC,WAAW,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC;QAC9D,CAAC;aAAM,CAAC;YACL,sBAAsB;YACtB,MAAM,GAAG,MAAM,WAAW,CAAC,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC3D,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC;QACvC,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC7B,MAAM,KAAK,GAAG,QAAQ,CAAC,aAAa,EAAE,CAAC;QAEvC,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC5C,EAAE,EAAE,OAAO,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,wCAAwC;YAC/F,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE;gBACR,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;aACrC;SACF,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAEhB,OAAO;YACL,OAAO,EAAE,IAAI;YACb,UAAU,EAAE,UAAU;YACtB,KAAK,EAAE;gBACJ,aAAa,EAAE,CAAC;gBAChB,iBAAiB,EAAE,CAAC;gBACpB,YAAY,EAAE,CAAC;aACjB;SACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,UAAU,CACd,SAAiB,EACjB,QAAmB,EACnB,OAAgC,EAChC,OAMC;QAED,IAAI,CAAC,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAE1D,MAAM,WAAW,GAAG,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;gBACpC,oBAAoB,EAAE,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC;aAChE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAEf,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC;YAC7C,KAAK,EAAE,SAAS;YAChB,KAAK,EAAE,WAAW;SACnB,CAAC,CAAC;QAEH,MAAM,aAAa,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,EAAE,OAAO,CAAC;QACvE,MAAM,YAAY,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QAC/D,MAAM,OAAO,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAClD,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,IAAI,EAAE,EAAE,CAAC;SACnC,CAAC,CAAC,CAAC;QAEJ,MAAM,WAAW,GAAG,QAAQ,CAAC,SAAS,CAAC;YACrC,OAAO;YACP,iBAAiB,EAAE,aAAa,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;SACpG,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,YAAY,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACtD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,iBAAiB,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;QAE1E,IAAI,QAAQ,GAAG,EAAE,CAAC;QAClB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YACxC,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;YAC/B,QAAQ,IAAI,SAAS,CAAC;YACtB,OAAO,CAAC,SAAS,CAAC,CAAC;QACrB,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC;QACvC,MAAM,KAAK,GAAG,QAAQ,CAAC,aAAa,EAAE,CAAC;QAEvC,OAAO;YACL,OAAO,EAAE,QAAQ;YACjB,UAAU,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBAC3B,EAAE,EAAE,OAAO,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC;gBACrD,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE;aAC9D,CAAC,CAAC;YACH,KAAK,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE;SACnE,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
import { LLMProvider, Message, CompletionResponse } from '../types.js';
|
|
2
|
+
export declare class GroqProvider implements LLMProvider {
|
|
3
|
+
name: string;
|
|
4
|
+
private client;
|
|
5
|
+
private proxyUrl?;
|
|
6
|
+
constructor(proxyUrl?: string);
|
|
7
|
+
configure(apiKey: string): void;
|
|
8
|
+
chat(model: string, messages: Message[], options?: {
|
|
9
|
+
temperature?: number;
|
|
10
|
+
maxTokens?: number;
|
|
11
|
+
tools?: any[];
|
|
12
|
+
toolChoice?: 'auto' | 'none' | 'required';
|
|
13
|
+
abortSignal?: AbortSignal;
|
|
14
|
+
}): Promise<CompletionResponse>;
|
|
15
|
+
streamChat(model: string, messages: Message[], onChunk: (chunk: string) => void, options?: {
|
|
16
|
+
temperature?: number;
|
|
17
|
+
maxTokens?: number;
|
|
18
|
+
tools?: any[];
|
|
19
|
+
toolChoice?: 'auto' | 'none' | 'required';
|
|
20
|
+
abortSignal?: AbortSignal;
|
|
21
|
+
}): Promise<CompletionResponse>;
|
|
22
|
+
}
|
|
23
|
+
//# sourceMappingURL=groq.d.ts.map
|