galaxy-code 0.1.5 → 0.1.7
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 +149 -600
- package/dist/cli.bundle.js +588 -0
- package/package.json +14 -12
- package/dist/app.d.ts +0 -7
- package/dist/app.js +0 -597
- package/dist/auto-updater.d.ts +0 -21
- package/dist/auto-updater.js +0 -144
- package/dist/cli.d.ts +0 -2
- package/dist/cli.js +0 -159
- package/dist/connections/claude.d.ts +0 -71
- package/dist/connections/claude.js +0 -303
- package/dist/connections/gemini.d.ts +0 -40
- package/dist/connections/gemini.js +0 -232
- package/dist/connections/index.d.ts +0 -11
- package/dist/connections/index.js +0 -11
- package/dist/connections/ollama.d.ts +0 -37
- package/dist/connections/ollama.js +0 -73
- package/dist/connections/types.d.ts +0 -22
- package/dist/connections/types.js +0 -7
- package/dist/env.d.ts +0 -1
- package/dist/env.js +0 -29
- package/dist/node_modules/@workspace/agent-core/connections/claude.d.ts +0 -72
- package/dist/node_modules/@workspace/agent-core/connections/claude.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/connections/claude.js +0 -270
- package/dist/node_modules/@workspace/agent-core/connections/claude.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/connections/gemini.d.ts +0 -41
- package/dist/node_modules/@workspace/agent-core/connections/gemini.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/connections/gemini.js +0 -203
- package/dist/node_modules/@workspace/agent-core/connections/gemini.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/connections/index.d.ts +0 -12
- package/dist/node_modules/@workspace/agent-core/connections/index.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/connections/index.js +0 -12
- package/dist/node_modules/@workspace/agent-core/connections/index.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/connections/ollama.d.ts +0 -38
- package/dist/node_modules/@workspace/agent-core/connections/ollama.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/connections/ollama.js +0 -64
- package/dist/node_modules/@workspace/agent-core/connections/ollama.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/connections/types.d.ts +0 -23
- package/dist/node_modules/@workspace/agent-core/connections/types.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/connections/types.js +0 -8
- package/dist/node_modules/@workspace/agent-core/connections/types.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/index.d.ts +0 -13
- package/dist/node_modules/@workspace/agent-core/index.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/index.js +0 -22
- package/dist/node_modules/@workspace/agent-core/index.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/package.json +0 -53
- package/dist/node_modules/@workspace/agent-core/prompts/ba-it-analyzer.d.ts +0 -2
- package/dist/node_modules/@workspace/agent-core/prompts/ba-it-analyzer.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/prompts/ba-it-analyzer.js +0 -144
- package/dist/node_modules/@workspace/agent-core/prompts/ba-it-analyzer.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/prompts/index.d.ts +0 -12
- package/dist/node_modules/@workspace/agent-core/prompts/index.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/prompts/index.js +0 -12
- package/dist/node_modules/@workspace/agent-core/prompts/index.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/prompts/orchestrator.d.ts +0 -9
- package/dist/node_modules/@workspace/agent-core/prompts/orchestrator.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/prompts/orchestrator.js +0 -89
- package/dist/node_modules/@workspace/agent-core/prompts/orchestrator.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/prompts/planning-agent.d.ts +0 -9
- package/dist/node_modules/@workspace/agent-core/prompts/planning-agent.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/prompts/planning-agent.js +0 -196
- package/dist/node_modules/@workspace/agent-core/prompts/planning-agent.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/prompts/universal-agent.d.ts +0 -8
- package/dist/node_modules/@workspace/agent-core/prompts/universal-agent.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/prompts/universal-agent.js +0 -112
- package/dist/node_modules/@workspace/agent-core/prompts/universal-agent.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/providers/agent-selector.d.ts +0 -30
- package/dist/node_modules/@workspace/agent-core/providers/agent-selector.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/providers/agent-selector.js +0 -76
- package/dist/node_modules/@workspace/agent-core/providers/agent-selector.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/providers/claude-agent.d.ts +0 -30
- package/dist/node_modules/@workspace/agent-core/providers/claude-agent.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/providers/claude-agent.js +0 -107
- package/dist/node_modules/@workspace/agent-core/providers/claude-agent.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/providers/gemini-agent.d.ts +0 -37
- package/dist/node_modules/@workspace/agent-core/providers/gemini-agent.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/providers/gemini-agent.js +0 -144
- package/dist/node_modules/@workspace/agent-core/providers/gemini-agent.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/providers/index.d.ts +0 -13
- package/dist/node_modules/@workspace/agent-core/providers/index.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/providers/index.js +0 -13
- package/dist/node_modules/@workspace/agent-core/providers/index.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/providers/ollama-agent.d.ts +0 -54
- package/dist/node_modules/@workspace/agent-core/providers/ollama-agent.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/providers/ollama-agent.js +0 -247
- package/dist/node_modules/@workspace/agent-core/providers/ollama-agent.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/providers/orchestrator.d.ts +0 -17
- package/dist/node_modules/@workspace/agent-core/providers/orchestrator.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/providers/orchestrator.js +0 -62
- package/dist/node_modules/@workspace/agent-core/providers/orchestrator.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/ba-it-analyzer.d.ts +0 -67
- package/dist/node_modules/@workspace/agent-core/tools/ba-it-analyzer.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/ba-it-analyzer.js +0 -81
- package/dist/node_modules/@workspace/agent-core/tools/ba-it-analyzer.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/code-generate-agent.d.ts +0 -52
- package/dist/node_modules/@workspace/agent-core/tools/code-generate-agent.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/code-generate-agent.js +0 -150
- package/dist/node_modules/@workspace/agent-core/tools/code-generate-agent.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/command-runner.d.ts +0 -15
- package/dist/node_modules/@workspace/agent-core/tools/command-runner.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/command-runner.js +0 -121
- package/dist/node_modules/@workspace/agent-core/tools/command-runner.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/document-parser.d.ts +0 -12
- package/dist/node_modules/@workspace/agent-core/tools/document-parser.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/document-parser.js +0 -84
- package/dist/node_modules/@workspace/agent-core/tools/document-parser.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/file-operations.d.ts +0 -18
- package/dist/node_modules/@workspace/agent-core/tools/file-operations.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/file-operations.js +0 -126
- package/dist/node_modules/@workspace/agent-core/tools/file-operations.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/galaxy-ui-integration.d.ts +0 -95
- package/dist/node_modules/@workspace/agent-core/tools/galaxy-ui-integration.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/galaxy-ui-integration.js +0 -245
- package/dist/node_modules/@workspace/agent-core/tools/galaxy-ui-integration.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/git-operations.d.ts +0 -12
- package/dist/node_modules/@workspace/agent-core/tools/git-operations.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/git-operations.js +0 -115
- package/dist/node_modules/@workspace/agent-core/tools/git-operations.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/index.d.ts +0 -11
- package/dist/node_modules/@workspace/agent-core/tools/index.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/index.js +0 -11
- package/dist/node_modules/@workspace/agent-core/tools/index.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/planning-agent.d.ts +0 -30
- package/dist/node_modules/@workspace/agent-core/tools/planning-agent.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/planning-agent.js +0 -125
- package/dist/node_modules/@workspace/agent-core/tools/planning-agent.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/progress-reporter.d.ts +0 -20
- package/dist/node_modules/@workspace/agent-core/tools/progress-reporter.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/progress-reporter.js +0 -46
- package/dist/node_modules/@workspace/agent-core/tools/progress-reporter.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/registry.d.ts +0 -22
- package/dist/node_modules/@workspace/agent-core/tools/registry.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/registry.js +0 -651
- package/dist/node_modules/@workspace/agent-core/tools/registry.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/tool-event-emitter.d.ts +0 -25
- package/dist/node_modules/@workspace/agent-core/tools/tool-event-emitter.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/tool-event-emitter.js +0 -26
- package/dist/node_modules/@workspace/agent-core/tools/tool-event-emitter.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/types.d.ts +0 -32
- package/dist/node_modules/@workspace/agent-core/tools/types.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/tools/types.js +0 -2
- package/dist/node_modules/@workspace/agent-core/tools/types.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/types.d.ts +0 -40
- package/dist/node_modules/@workspace/agent-core/types.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/types.js +0 -9
- package/dist/node_modules/@workspace/agent-core/types.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/utils/config-manager.d.ts +0 -103
- package/dist/node_modules/@workspace/agent-core/utils/config-manager.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/utils/config-manager.js +0 -307
- package/dist/node_modules/@workspace/agent-core/utils/config-manager.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/utils/devtools.d.ts +0 -22
- package/dist/node_modules/@workspace/agent-core/utils/devtools.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/utils/devtools.js +0 -62
- package/dist/node_modules/@workspace/agent-core/utils/devtools.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/utils/message-formatters.d.ts +0 -33
- package/dist/node_modules/@workspace/agent-core/utils/message-formatters.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/utils/message-formatters.js +0 -591
- package/dist/node_modules/@workspace/agent-core/utils/message-formatters.js.map +0 -1
- package/dist/node_modules/@workspace/agent-core/utils/progress-tracker.d.ts +0 -60
- package/dist/node_modules/@workspace/agent-core/utils/progress-tracker.d.ts.map +0 -1
- package/dist/node_modules/@workspace/agent-core/utils/progress-tracker.js +0 -204
- package/dist/node_modules/@workspace/agent-core/utils/progress-tracker.js.map +0 -1
- package/dist/prompts/ba-it-analyzer.d.ts +0 -1
- package/dist/prompts/ba-it-analyzer.js +0 -143
- package/dist/prompts/index.d.ts +0 -11
- package/dist/prompts/index.js +0 -11
- package/dist/prompts/orchestrator.d.ts +0 -8
- package/dist/prompts/orchestrator.js +0 -88
- package/dist/prompts/planning-agent.d.ts +0 -8
- package/dist/prompts/planning-agent.js +0 -195
- package/dist/prompts/universal-agent.d.ts +0 -7
- package/dist/prompts/universal-agent.js +0 -111
- package/dist/providers/agent-selector.d.ts +0 -29
- package/dist/providers/agent-selector.js +0 -84
- package/dist/providers/claude-agent.d.ts +0 -29
- package/dist/providers/claude-agent.js +0 -121
- package/dist/providers/gemini-agent.d.ts +0 -36
- package/dist/providers/gemini-agent.js +0 -168
- package/dist/providers/index.d.ts +0 -12
- package/dist/providers/index.js +0 -12
- package/dist/providers/ollama-agent.d.ts +0 -53
- package/dist/providers/ollama-agent.js +0 -276
- package/dist/providers/orchestrator.d.ts +0 -16
- package/dist/providers/orchestrator.js +0 -76
- package/dist/tools/ba-it-analyzer.d.ts +0 -66
- package/dist/tools/ba-it-analyzer.js +0 -90
- package/dist/tools/code-generate-agent.d.ts +0 -51
- package/dist/tools/code-generate-agent.js +0 -159
- package/dist/tools/command-runner.d.ts +0 -14
- package/dist/tools/command-runner.js +0 -120
- package/dist/tools/document-parser.d.ts +0 -11
- package/dist/tools/document-parser.js +0 -83
- package/dist/tools/file-operations.d.ts +0 -17
- package/dist/tools/file-operations.js +0 -127
- package/dist/tools/galaxy-ui-integration.d.ts +0 -94
- package/dist/tools/galaxy-ui-integration.js +0 -244
- package/dist/tools/git-operations.d.ts +0 -11
- package/dist/tools/git-operations.js +0 -114
- package/dist/tools/index.d.ts +0 -10
- package/dist/tools/index.js +0 -10
- package/dist/tools/planning-agent.d.ts +0 -29
- package/dist/tools/planning-agent.js +0 -134
- package/dist/tools/progress-reporter.d.ts +0 -19
- package/dist/tools/progress-reporter.js +0 -52
- package/dist/tools/registry.d.ts +0 -21
- package/dist/tools/registry.js +0 -695
- package/dist/tools/tool-event-emitter.d.ts +0 -24
- package/dist/tools/tool-event-emitter.js +0 -25
- package/dist/tools/types.d.ts +0 -31
- package/dist/tools/types.js +0 -1
- package/dist/types.d.ts +0 -39
- package/dist/types.js +0 -8
- package/dist/update-checker.d.ts +0 -22
- package/dist/update-checker.js +0 -85
- package/dist/utils/config-manager.d.ts +0 -102
- package/dist/utils/config-manager.js +0 -326
- package/dist/utils/devtools.d.ts +0 -21
- package/dist/utils/devtools.js +0 -61
- package/dist/utils/message-formatters.d.ts +0 -32
- package/dist/utils/message-formatters.js +0 -590
- package/dist/utils/progress-tracker.d.ts +0 -59
- package/dist/utils/progress-tracker.js +0 -213
|
@@ -1,270 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* @author Bùi Trọng Hiếu
|
|
3
|
-
* @email kevinbui210191@gmail.com
|
|
4
|
-
* @create 2024-10-13
|
|
5
|
-
* @modify 2024-10-13
|
|
6
|
-
* @desc Claude API connection using Anthropic SDK
|
|
7
|
-
*/
|
|
8
|
-
import Anthropic from '@anthropic-ai/sdk';
|
|
9
|
-
import { ToolRegistry } from '../tools/index.js';
|
|
10
|
-
export class ClaudeConnection {
|
|
11
|
-
client;
|
|
12
|
-
model;
|
|
13
|
-
conversationHistory = [];
|
|
14
|
-
conversationHistoryWithTools = [];
|
|
15
|
-
systemPrompt;
|
|
16
|
-
toolsRegistry = new ToolRegistry();
|
|
17
|
-
constructor(config, systemPrompt) {
|
|
18
|
-
// Only get apiKey from config - no fallback to env
|
|
19
|
-
const apiKey = config?.apiKey;
|
|
20
|
-
// if (!apiKey) {
|
|
21
|
-
// throw new Error(
|
|
22
|
-
// 'Hiện tại bạn chưa thêm apikey cho model claude. Vui lòng thêm để dùng phần mềm'
|
|
23
|
-
// );
|
|
24
|
-
// }
|
|
25
|
-
this.client = new Anthropic({
|
|
26
|
-
apiKey,
|
|
27
|
-
});
|
|
28
|
-
this.model = config?.model || 'claude-sonnet-4.5-20250929';
|
|
29
|
-
this.systemPrompt = systemPrompt;
|
|
30
|
-
}
|
|
31
|
-
/**
|
|
32
|
-
* Send a message and get response
|
|
33
|
-
*/
|
|
34
|
-
async chat(userMessage) {
|
|
35
|
-
// Add user message to history
|
|
36
|
-
this.conversationHistory.push({
|
|
37
|
-
role: 'user',
|
|
38
|
-
content: userMessage,
|
|
39
|
-
});
|
|
40
|
-
try {
|
|
41
|
-
const response = await this.client.messages.create({
|
|
42
|
-
model: this.model,
|
|
43
|
-
max_tokens: 8192,
|
|
44
|
-
system: this.systemPrompt,
|
|
45
|
-
messages: this.conversationHistory,
|
|
46
|
-
});
|
|
47
|
-
// Extract assistant response
|
|
48
|
-
const assistantMessage = response.content[0]?.type === 'text' ? response.content[0].text : '';
|
|
49
|
-
// Add to history
|
|
50
|
-
this.conversationHistory.push({
|
|
51
|
-
role: 'assistant',
|
|
52
|
-
content: assistantMessage,
|
|
53
|
-
});
|
|
54
|
-
return assistantMessage;
|
|
55
|
-
}
|
|
56
|
-
catch (error) {
|
|
57
|
-
console.error('Claude API Error:', error);
|
|
58
|
-
throw error;
|
|
59
|
-
}
|
|
60
|
-
}
|
|
61
|
-
/**
|
|
62
|
-
* Stream response (for future implementation)
|
|
63
|
-
*/
|
|
64
|
-
async *chatStream(userMessage) {
|
|
65
|
-
this.conversationHistory.push({
|
|
66
|
-
role: 'user',
|
|
67
|
-
content: userMessage,
|
|
68
|
-
});
|
|
69
|
-
try {
|
|
70
|
-
const stream = await this.client.messages.create({
|
|
71
|
-
model: this.model,
|
|
72
|
-
max_tokens: 8192,
|
|
73
|
-
system: this.systemPrompt,
|
|
74
|
-
messages: this.conversationHistory,
|
|
75
|
-
stream: true,
|
|
76
|
-
});
|
|
77
|
-
let fullResponse = '';
|
|
78
|
-
for await (const chunk of stream) {
|
|
79
|
-
if (chunk.type === 'content_block_delta' && chunk.delta.type === 'text_delta') {
|
|
80
|
-
const text = chunk.delta.text || '';
|
|
81
|
-
fullResponse += text;
|
|
82
|
-
yield text;
|
|
83
|
-
}
|
|
84
|
-
}
|
|
85
|
-
// Add complete response to history
|
|
86
|
-
this.conversationHistory.push({
|
|
87
|
-
role: 'assistant',
|
|
88
|
-
content: fullResponse,
|
|
89
|
-
});
|
|
90
|
-
}
|
|
91
|
-
catch (error) {
|
|
92
|
-
console.error('Claude Stream Error:', error);
|
|
93
|
-
throw error;
|
|
94
|
-
}
|
|
95
|
-
}
|
|
96
|
-
/**
|
|
97
|
-
* Generate single response without conversation history
|
|
98
|
-
*/
|
|
99
|
-
async generate(prompt, systemPrompt) {
|
|
100
|
-
try {
|
|
101
|
-
const response = await this.client.messages.create({
|
|
102
|
-
model: this.model,
|
|
103
|
-
max_tokens: 8192,
|
|
104
|
-
system: systemPrompt || this.systemPrompt,
|
|
105
|
-
messages: [
|
|
106
|
-
{
|
|
107
|
-
role: 'user',
|
|
108
|
-
content: prompt,
|
|
109
|
-
},
|
|
110
|
-
],
|
|
111
|
-
});
|
|
112
|
-
return response.content[0]?.type === 'text' ? response.content[0].text : '';
|
|
113
|
-
}
|
|
114
|
-
catch (error) {
|
|
115
|
-
console.error('Claude Generate Error:', error);
|
|
116
|
-
throw error;
|
|
117
|
-
}
|
|
118
|
-
}
|
|
119
|
-
/**
|
|
120
|
-
* Clear conversation history
|
|
121
|
-
*/
|
|
122
|
-
clearHistory() {
|
|
123
|
-
this.conversationHistory = [];
|
|
124
|
-
}
|
|
125
|
-
/**
|
|
126
|
-
* Get conversation history
|
|
127
|
-
*/
|
|
128
|
-
getHistory() {
|
|
129
|
-
return [...this.conversationHistory];
|
|
130
|
-
}
|
|
131
|
-
/**
|
|
132
|
-
* Set system prompt
|
|
133
|
-
*/
|
|
134
|
-
setSystemPrompt(prompt) {
|
|
135
|
-
this.systemPrompt = prompt;
|
|
136
|
-
}
|
|
137
|
-
/**
|
|
138
|
-
* Get current model
|
|
139
|
-
*/
|
|
140
|
-
getModel() {
|
|
141
|
-
return this.model;
|
|
142
|
-
}
|
|
143
|
-
/**
|
|
144
|
-
* Change model
|
|
145
|
-
*/
|
|
146
|
-
setModel(model) {
|
|
147
|
-
this.model = model;
|
|
148
|
-
}
|
|
149
|
-
/**
|
|
150
|
-
* Chat with tools support - handles full tool loop
|
|
151
|
-
*/
|
|
152
|
-
async chatWithTools(userMessage, tools, onMessage) {
|
|
153
|
-
// Add user message to history
|
|
154
|
-
this.conversationHistoryWithTools.push({
|
|
155
|
-
role: 'user',
|
|
156
|
-
content: userMessage,
|
|
157
|
-
});
|
|
158
|
-
let iteration = 0;
|
|
159
|
-
const MAX_ITERATIONS = 100; // Safety limit
|
|
160
|
-
while (iteration < MAX_ITERATIONS) {
|
|
161
|
-
iteration++;
|
|
162
|
-
// Call Claude API
|
|
163
|
-
const response = await this.client.messages.create({
|
|
164
|
-
model: this.model,
|
|
165
|
-
max_tokens: 8192,
|
|
166
|
-
system: this.systemPrompt,
|
|
167
|
-
messages: this.conversationHistoryWithTools,
|
|
168
|
-
tools: tools,
|
|
169
|
-
});
|
|
170
|
-
// Check stop reason
|
|
171
|
-
if (response.stop_reason !== 'tool_use') {
|
|
172
|
-
// No more tools to execute - add final response to history
|
|
173
|
-
this.conversationHistoryWithTools.push({
|
|
174
|
-
role: 'assistant',
|
|
175
|
-
content: response.content,
|
|
176
|
-
});
|
|
177
|
-
// Extract text from response
|
|
178
|
-
const textBlock = response.content.find((block) => block.type === 'text');
|
|
179
|
-
return textBlock?.text || '';
|
|
180
|
-
}
|
|
181
|
-
// Add assistant response with tool_use to history
|
|
182
|
-
this.conversationHistoryWithTools.push({
|
|
183
|
-
role: 'assistant',
|
|
184
|
-
content: response.content,
|
|
185
|
-
});
|
|
186
|
-
// Execute tools and collect results
|
|
187
|
-
const toolResults = [];
|
|
188
|
-
for (const block of response.content) {
|
|
189
|
-
if (block.type === 'tool_use') {
|
|
190
|
-
const toolUseBlock = block;
|
|
191
|
-
const toolName = toolUseBlock.name;
|
|
192
|
-
const toolInput = toolUseBlock.input;
|
|
193
|
-
// Special handling for plan_task and progress_reporter
|
|
194
|
-
if (toolName === 'plan_task' || toolName === 'progress_reporter') {
|
|
195
|
-
try {
|
|
196
|
-
const result = await this.toolsRegistry.executeTool(toolName, toolInput);
|
|
197
|
-
const resultStr = typeof result === 'string' ? result : JSON.stringify(result);
|
|
198
|
-
toolResults.push({
|
|
199
|
-
type: 'tool_result',
|
|
200
|
-
tool_use_id: toolUseBlock.id,
|
|
201
|
-
content: resultStr,
|
|
202
|
-
});
|
|
203
|
-
}
|
|
204
|
-
catch (error) {
|
|
205
|
-
const errorMsg = error instanceof Error ? error.message : String(error);
|
|
206
|
-
toolResults.push({
|
|
207
|
-
type: 'tool_result',
|
|
208
|
-
tool_use_id: toolUseBlock.id,
|
|
209
|
-
content: `Error: ${errorMsg}`,
|
|
210
|
-
is_error: true,
|
|
211
|
-
});
|
|
212
|
-
}
|
|
213
|
-
}
|
|
214
|
-
else {
|
|
215
|
-
// Regular tool handling with notification
|
|
216
|
-
try {
|
|
217
|
-
const result = await this.toolsRegistry.executeTool(toolName, toolInput);
|
|
218
|
-
const resultStr = typeof result === 'string' ? result : JSON.stringify(result);
|
|
219
|
-
// Notify tool execution
|
|
220
|
-
if (onMessage) {
|
|
221
|
-
const success = !resultStr.startsWith('❌');
|
|
222
|
-
onMessage('tool', resultStr, toolName, {
|
|
223
|
-
success,
|
|
224
|
-
output: resultStr,
|
|
225
|
-
});
|
|
226
|
-
}
|
|
227
|
-
toolResults.push({
|
|
228
|
-
type: 'tool_result',
|
|
229
|
-
tool_use_id: toolUseBlock.id,
|
|
230
|
-
content: resultStr,
|
|
231
|
-
});
|
|
232
|
-
}
|
|
233
|
-
catch (error) {
|
|
234
|
-
const errorMsg = error instanceof Error ? error.message : String(error);
|
|
235
|
-
// Notify error
|
|
236
|
-
if (onMessage) {
|
|
237
|
-
onMessage('tool', `❌ Error: ${errorMsg}`, toolName, {
|
|
238
|
-
success: false,
|
|
239
|
-
error: errorMsg,
|
|
240
|
-
});
|
|
241
|
-
}
|
|
242
|
-
toolResults.push({
|
|
243
|
-
type: 'tool_result',
|
|
244
|
-
tool_use_id: toolUseBlock.id,
|
|
245
|
-
content: `Error: ${errorMsg}`,
|
|
246
|
-
is_error: true,
|
|
247
|
-
});
|
|
248
|
-
}
|
|
249
|
-
}
|
|
250
|
-
}
|
|
251
|
-
}
|
|
252
|
-
// Add tool results to history
|
|
253
|
-
this.conversationHistoryWithTools.push({
|
|
254
|
-
role: 'user',
|
|
255
|
-
content: toolResults,
|
|
256
|
-
});
|
|
257
|
-
// Loop continues to call API again with tool results
|
|
258
|
-
}
|
|
259
|
-
// Safety limit reached
|
|
260
|
-
console.warn(`⚠️ Safety limit (${MAX_ITERATIONS}) reached in Claude tool loop.`);
|
|
261
|
-
return 'Maximum iteration limit reached. Please try again.';
|
|
262
|
-
}
|
|
263
|
-
/**
|
|
264
|
-
* Clear tool conversation history
|
|
265
|
-
*/
|
|
266
|
-
clearToolHistory() {
|
|
267
|
-
this.conversationHistoryWithTools = [];
|
|
268
|
-
}
|
|
269
|
-
}
|
|
270
|
-
//# sourceMappingURL=claude.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"claude.js","sourceRoot":"","sources":["../../src/connections/claude.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,SAAS,MAAM,mBAAmB,CAAA;AAOzC,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAA;AAuBhD,MAAM,OAAO,gBAAgB;IACnB,MAAM,CAAW;IACjB,KAAK,CAAe;IACpB,mBAAmB,GAAoB,EAAE,CAAA;IACzC,4BAA4B,GAAmB,EAAE,CAAA;IACjD,YAAY,CAAS;IACrB,aAAa,GAAG,IAAI,YAAY,EAAE,CAAA;IAE1C,YAAY,MAAoB,EAAE,YAAqB;QACrD,mDAAmD;QACnD,MAAM,MAAM,GAAG,MAAM,EAAE,MAAM,CAAA;QAE7B,iBAAiB;QACjB,oBAAoB;QACpB,qFAAqF;QACrF,MAAM;QACN,IAAI;QAEJ,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC;YAC1B,MAAM;SACP,CAAC,CAAA;QAEF,IAAI,CAAC,KAAK,GAAI,MAAM,EAAE,KAAuB,IAAI,4BAA4B,CAAA;QAC7E,IAAI,CAAC,YAAY,GAAG,YAAY,CAAA;IAClC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAI,CAAC,WAAmB;QAC5B,8BAA8B;QAC9B,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC;YAC5B,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,WAAW;SACrB,CAAC,CAAA;QAEF,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;gBACjD,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,UAAU,EAAE,IAAI;gBAChB,MAAM,EAAE,IAAI,CAAC,YAAY;gBACzB,QAAQ,EAAE,IAAI,CAAC,mBAAmB;aACnC,CAAC,CAAA;YAEF,6BAA6B;YAC7B,MAAM,gBAAgB,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAA;YAE7F,iBAAiB;YACjB,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC;gBAC5B,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,gBAAgB;aAC1B,CAAC,CAAA;YAEF,OAAO,gBAAgB,CAAA;QACzB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,mBAAmB,EAAE,KAAK,CAAC,CAAA;YACzC,MAAM,KAAK,CAAA;QACb,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,CAAC,UAAU,CAAC,WAAmB;QACnC,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC;YAC5B,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,WAAW;SACrB,CAAC,CAAA;QAEF,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAC/C,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,UAAU,EAAE,IAAI;gBAChB,MAAM,EAAE,IAAI,CAAC,YAAY;gBACzB,QAAQ,EAAE,IAAI,CAAC,mBAAmB;gBAClC,MAAM,EAAE,IAAI;aACb,CAAC,CAAA;YAEF,IAAI,YAAY,GAAG,EAAE,CAAA;YAErB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBACjC,IAAI,KAAK,CAAC,IAAI,KAAK,qBAAqB,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;oBAC9E,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAA;oBACnC,YAAY,IAAI,IAAI,CAAA;oBACpB,MAAM,IAAI,CAAA;gBACZ,CAAC;YACH,CAAC;YAED,mCAAmC;YACnC,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC;gBAC5B,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,YAAY;aACtB,CAAC,CAAA;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAAA;YAC5C,MAAM,KAAK,CAAA;QACb,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CAAC,MAAc,EAAE,YAAqB;QAClD,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;gBACjD,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,UAAU,EAAE,IAAI;gBAChB,MAAM,EAAE,YAAY,IAAI,IAAI,CAAC,YAAY;gBACzC,QAAQ,EAAE;oBACR;wBACE,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,MAAM;qBAChB;iBACF;aACF,CAAC,CAAA;YAEF,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAA;QAC7E,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAA;YAC9C,MAAM,KAAK,CAAA;QACb,CAAC;IACH,CAAC;IAED;;OAEG;IACH,YAAY;QACV,IAAI,CAAC,mBAAmB,GAAG,EAAE,CAAA;IAC/B,CAAC;IAED;;OAEG;IACH,UAAU;QACR,OAAO,CAAC,GAAG,IAAI,CAAC,mBAAmB,CAAC,CAAA;IACtC,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,MAAc;QAC5B,IAAI,CAAC,YAAY,GAAG,MAAM,CAAA;IAC5B,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,OAAO,IAAI,CAAC,KAAK,CAAA;IACnB,CAAC;IAED;;OAEG;IACH,QAAQ,CAAC,KAAoB;QAC3B,IAAI,CAAC,KAAK,GAAG,KAAK,CAAA;IACpB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,aAAa,CACjB,WAAmB,EACnB,KAAuB,EACvB,SAA2B;QAE3B,8BAA8B;QAC9B,IAAI,CAAC,4BAA4B,CAAC,IAAI,CAAC;YACrC,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,WAAW;SACrB,CAAC,CAAA;QAEF,IAAI,SAAS,GAAG,CAAC,CAAA;QACjB,MAAM,cAAc,GAAG,GAAG,CAAA,CAAC,eAAe;QAE1C,OAAO,SAAS,GAAG,cAAc,EAAE,CAAC;YAClC,SAAS,EAAE,CAAA;YAEX,kBAAkB;YAClB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;gBACjD,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,UAAU,EAAE,IAAI;gBAChB,MAAM,EAAE,IAAI,CAAC,YAAY;gBACzB,QAAQ,EAAE,IAAI,CAAC,4BAA4B;gBAC3C,KAAK,EAAE,KAAK;aACb,CAAC,CAAA;YAEF,oBAAoB;YACpB,IAAI,QAAQ,CAAC,WAAW,KAAK,UAAU,EAAE,CAAC;gBACxC,2DAA2D;gBAC3D,IAAI,CAAC,4BAA4B,CAAC,IAAI,CAAC;oBACrC,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,QAAQ,CAAC,OAAO;iBAC1B,CAAC,CAAA;gBAEF,6BAA6B;gBAC7B,MAAM,SAAS,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CACrC,CAAC,KAAK,EAAsB,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CACrD,CAAA;gBACD,OAAO,SAAS,EAAE,IAAI,IAAI,EAAE,CAAA;YAC9B,CAAC;YAED,kDAAkD;YAClD,IAAI,CAAC,4BAA4B,CAAC,IAAI,CAAC;gBACrC,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,QAAQ,CAAC,OAAO;aAC1B,CAAC,CAAA;YAEF,oCAAoC;YACpC,MAAM,WAAW,GAAU,EAAE,CAAA;YAE7B,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;gBACrC,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAC9B,MAAM,YAAY,GAAG,KAAqB,CAAA;oBAC1C,MAAM,QAAQ,GAAG,YAAY,CAAC,IAAI,CAAA;oBAClC,MAAM,SAAS,GAAG,YAAY,CAAC,KAAK,CAAA;oBAEpC,uDAAuD;oBACvD,IAAI,QAAQ,KAAK,WAAW,IAAI,QAAQ,KAAK,mBAAmB,EAAE,CAAC;wBACjE,IAAI,CAAC;4BACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAA;4BACxE,MAAM,SAAS,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAA;4BAE9E,WAAW,CAAC,IAAI,CAAC;gCACf,IAAI,EAAE,aAAa;gCACnB,WAAW,EAAE,YAAY,CAAC,EAAE;gCAC5B,OAAO,EAAE,SAAS;6BACnB,CAAC,CAAA;wBACJ,CAAC;wBAAC,OAAO,KAAK,EAAE,CAAC;4BACf,MAAM,QAAQ,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;4BACvE,WAAW,CAAC,IAAI,CAAC;gCACf,IAAI,EAAE,aAAa;gCACnB,WAAW,EAAE,YAAY,CAAC,EAAE;gCAC5B,OAAO,EAAE,UAAU,QAAQ,EAAE;gCAC7B,QAAQ,EAAE,IAAI;6BACf,CAAC,CAAA;wBACJ,CAAC;oBACH,CAAC;yBAAM,CAAC;wBACN,0CAA0C;wBAC1C,IAAI,CAAC;4BACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAA;4BACxE,MAAM,SAAS,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAA;4BAE9E,wBAAwB;4BACxB,IAAI,SAAS,EAAE,CAAC;gCACd,MAAM,OAAO,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,CAAA;gCAC1C,SAAS,CAAC,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE;oCACrC,OAAO;oCACP,MAAM,EAAE,SAAS;iCAClB,CAAC,CAAA;4BACJ,CAAC;4BAED,WAAW,CAAC,IAAI,CAAC;gCACf,IAAI,EAAE,aAAa;gCACnB,WAAW,EAAE,YAAY,CAAC,EAAE;gCAC5B,OAAO,EAAE,SAAS;6BACnB,CAAC,CAAA;wBACJ,CAAC;wBAAC,OAAO,KAAK,EAAE,CAAC;4BACf,MAAM,QAAQ,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;4BAEvE,eAAe;4BACf,IAAI,SAAS,EAAE,CAAC;gCACd,SAAS,CAAC,MAAM,EAAE,YAAY,QAAQ,EAAE,EAAE,QAAQ,EAAE;oCAClD,OAAO,EAAE,KAAK;oCACd,KAAK,EAAE,QAAQ;iCAChB,CAAC,CAAA;4BACJ,CAAC;4BAED,WAAW,CAAC,IAAI,CAAC;gCACf,IAAI,EAAE,aAAa;gCACnB,WAAW,EAAE,YAAY,CAAC,EAAE;gCAC5B,OAAO,EAAE,UAAU,QAAQ,EAAE;gCAC7B,QAAQ,EAAE,IAAI;6BACf,CAAC,CAAA;wBACJ,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAED,8BAA8B;YAC9B,IAAI,CAAC,4BAA4B,CAAC,IAAI,CAAC;gBACrC,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,WAAW;aACrB,CAAC,CAAA;YAEF,qDAAqD;QACvD,CAAC;QAED,uBAAuB;QACvB,OAAO,CAAC,IAAI,CAAC,oBAAoB,cAAc,gCAAgC,CAAC,CAAA;QAChF,OAAO,oDAAoD,CAAA;IAC7D,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,IAAI,CAAC,4BAA4B,GAAG,EAAE,CAAA;IACxC,CAAC;CACF"}
|
|
@@ -1,41 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* @author Bùi Trọng Hiếu
|
|
3
|
-
* @email kevinbui210191@gmail.com
|
|
4
|
-
* @create 2024-10-08
|
|
5
|
-
* @modify 2024-10-13
|
|
6
|
-
* @desc Gemini connection helper using @google/genai with improved function call handling.
|
|
7
|
-
*/
|
|
8
|
-
import { FunctionDeclaration, type GenerateContentConfig } from '@google/genai';
|
|
9
|
-
import type { AgentConfig } from '../utils/config-manager.js';
|
|
10
|
-
import type { ToolExecutionInfo } from './types.js';
|
|
11
|
-
export type GeminiGenerateParameters = {
|
|
12
|
-
readonly prompt: string;
|
|
13
|
-
readonly config?: GenerateContentConfig;
|
|
14
|
-
};
|
|
15
|
-
export type GeminiChatParameters = {
|
|
16
|
-
readonly prompt: string;
|
|
17
|
-
readonly tools?: FunctionDeclaration[];
|
|
18
|
-
readonly message: string;
|
|
19
|
-
};
|
|
20
|
-
export type GeminiGenerateResult<T = unknown> = {
|
|
21
|
-
readonly text: string;
|
|
22
|
-
readonly raw: T;
|
|
23
|
-
};
|
|
24
|
-
export declare class GeminiConnection {
|
|
25
|
-
private client;
|
|
26
|
-
private readonly apiKey;
|
|
27
|
-
private chatSession;
|
|
28
|
-
private readonly systemInstruction?;
|
|
29
|
-
private readonly model;
|
|
30
|
-
private toolsRegistry;
|
|
31
|
-
constructor(config?: AgentConfig, systemPrompt?: string);
|
|
32
|
-
initChat(): void;
|
|
33
|
-
generate(parameters: GeminiGenerateParameters): Promise<GeminiGenerateResult>;
|
|
34
|
-
chat(message: string, onMessage?: (author: 'tool' | 'assistant', content: string, toolName?: string, toolInfo?: ToolExecutionInfo) => void): Promise<string>;
|
|
35
|
-
private getToolDeclarations;
|
|
36
|
-
private convertToFunctionDeclaration;
|
|
37
|
-
private convertSchemaToGeminiSchema;
|
|
38
|
-
private convertPropertyToGeminiSchema;
|
|
39
|
-
private mapTypeToGemini;
|
|
40
|
-
}
|
|
41
|
-
//# sourceMappingURL=gemini.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../src/connections/gemini.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAEL,mBAAmB,EAKnB,KAAK,qBAAqB,EAC3B,MAAM,eAAe,CAAA;AAEtB,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAA;AAC7D,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAA;AAInD,MAAM,MAAM,wBAAwB,GAAG;IACrC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IACvB,QAAQ,CAAC,MAAM,CAAC,EAAE,qBAAqB,CAAA;CACxC,CAAA;AACD,MAAM,MAAM,oBAAoB,GAAG;IACjC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IACvB,QAAQ,CAAC,KAAK,CAAC,EAAE,mBAAmB,EAAE,CAAA;IACtC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAA;CACzB,CAAA;AACD,MAAM,MAAM,oBAAoB,CAAC,CAAC,GAAG,OAAO,IAAI;IAC9C,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAA;IACrB,QAAQ,CAAC,GAAG,EAAE,CAAC,CAAA;CAChB,CAAA;AAED,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,MAAM,CAAa;IAC3B,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAoB;IAC3C,OAAO,CAAC,WAAW,CAAkB;IACrC,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAQ;IAC3C,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAQ;IAC9B,OAAO,CAAC,aAAa,CAAqB;gBAE9B,MAAM,CAAC,EAAE,WAAW,EAAE,YAAY,CAAC,EAAE,MAAM;IAUvD,QAAQ,IAAI,IAAI;IAcV,QAAQ,CAAC,UAAU,EAAE,wBAAwB,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAU7E,IAAI,CACR,OAAO,EAAE,MAAM,EACf,SAAS,CAAC,EAAE,CACV,MAAM,EAAE,MAAM,GAAG,WAAW,EAC5B,OAAO,EAAE,MAAM,EACf,QAAQ,CAAC,EAAE,MAAM,EACjB,QAAQ,CAAC,EAAE,iBAAiB,KACzB,IAAI,GACR,OAAO,CAAC,MAAM,CAAC;IAkGlB,OAAO,CAAC,mBAAmB;IAK3B,OAAO,CAAC,4BAA4B;IAYpC,OAAO,CAAC,2BAA2B;IA4BnC,OAAO,CAAC,6BAA6B;IA+BrC,OAAO,CAAC,eAAe;CAWxB"}
|
|
@@ -1,203 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* @author Bùi Trọng Hiếu
|
|
3
|
-
* @email kevinbui210191@gmail.com
|
|
4
|
-
* @create 2024-10-08
|
|
5
|
-
* @modify 2024-10-13
|
|
6
|
-
* @desc Gemini connection helper using @google/genai with improved function call handling.
|
|
7
|
-
*/
|
|
8
|
-
import { GoogleGenAI, Type, } from '@google/genai';
|
|
9
|
-
import { globalToolRegistry } from '../tools/registry.js';
|
|
10
|
-
const geminiModel = 'gemini-2.5-flash';
|
|
11
|
-
export class GeminiConnection {
|
|
12
|
-
client;
|
|
13
|
-
apiKey;
|
|
14
|
-
chatSession;
|
|
15
|
-
systemInstruction;
|
|
16
|
-
model;
|
|
17
|
-
toolsRegistry = globalToolRegistry;
|
|
18
|
-
constructor(config, systemPrompt) {
|
|
19
|
-
// Only get apiKey from config - no fallback to env
|
|
20
|
-
this.apiKey = config?.apiKey;
|
|
21
|
-
this.model = config?.model ?? 'gemini-2.5-flash';
|
|
22
|
-
this.client = new GoogleGenAI({ apiKey: this.apiKey });
|
|
23
|
-
this.systemInstruction = systemPrompt;
|
|
24
|
-
this.initChat();
|
|
25
|
-
}
|
|
26
|
-
initChat() {
|
|
27
|
-
this.chatSession = this.client.chats.create({
|
|
28
|
-
model: this.model,
|
|
29
|
-
config: {
|
|
30
|
-
systemInstruction: this.systemInstruction,
|
|
31
|
-
tools: [{ functionDeclarations: this.getToolDeclarations() }],
|
|
32
|
-
temperature: 0.7,
|
|
33
|
-
thinkingConfig: {
|
|
34
|
-
thinkingBudget: 0,
|
|
35
|
-
},
|
|
36
|
-
},
|
|
37
|
-
});
|
|
38
|
-
}
|
|
39
|
-
async generate(parameters) {
|
|
40
|
-
const response = await this.client.models.generateContent({
|
|
41
|
-
model: geminiModel,
|
|
42
|
-
contents: parameters.prompt,
|
|
43
|
-
});
|
|
44
|
-
const text = response.text ?? '';
|
|
45
|
-
return { text, raw: response };
|
|
46
|
-
}
|
|
47
|
-
async chat(message, onMessage) {
|
|
48
|
-
if (!this.chatSession) {
|
|
49
|
-
throw new Error('Chat session is not initialized.');
|
|
50
|
-
}
|
|
51
|
-
let currentMessage = message;
|
|
52
|
-
let response;
|
|
53
|
-
let iteration = 0;
|
|
54
|
-
const MAX_ITERATIONS = 15; // Tăng lên vì xử lý từng tool một
|
|
55
|
-
while (currentMessage && iteration < MAX_ITERATIONS) {
|
|
56
|
-
iteration++;
|
|
57
|
-
response = await this.chatSession.sendMessage({
|
|
58
|
-
message: currentMessage,
|
|
59
|
-
});
|
|
60
|
-
if (response.functionCalls && response.functionCalls.length > 0) {
|
|
61
|
-
const functionResponses = [];
|
|
62
|
-
// **IMPROVED: Sequential execution với early feedback**
|
|
63
|
-
for (const functionCall of response.functionCalls) {
|
|
64
|
-
try {
|
|
65
|
-
const toolName = functionCall.name || 'unknown';
|
|
66
|
-
const args = functionCall.args || {};
|
|
67
|
-
const toolResult = await this.toolsRegistry.executeTool(toolName, args);
|
|
68
|
-
const resultStr = typeof toolResult === 'string' ? toolResult : JSON.stringify(toolResult);
|
|
69
|
-
if (onMessage) {
|
|
70
|
-
// Pass tool execution info for better formatting
|
|
71
|
-
const toolInfo = {
|
|
72
|
-
toolName,
|
|
73
|
-
args,
|
|
74
|
-
result: toolResult,
|
|
75
|
-
};
|
|
76
|
-
onMessage('tool', resultStr, toolName, toolInfo);
|
|
77
|
-
}
|
|
78
|
-
functionResponses.push({
|
|
79
|
-
functionResponse: {
|
|
80
|
-
name: functionCall.name,
|
|
81
|
-
response: { result: resultStr },
|
|
82
|
-
},
|
|
83
|
-
});
|
|
84
|
-
// **NEW: Cho phép xử lý từng tool một nếu cần**
|
|
85
|
-
// Nếu có nhiều function calls, có thể break ở đây để feedback sớm
|
|
86
|
-
// Nhưng hiện tại vẫn xử lý hết để tương thích
|
|
87
|
-
}
|
|
88
|
-
catch (error) {
|
|
89
|
-
const errorMsg = error instanceof Error ? error.message : String(error);
|
|
90
|
-
// Log error
|
|
91
|
-
if (onMessage) {
|
|
92
|
-
const toolInfo = {
|
|
93
|
-
toolName: functionCall.name || 'unknown',
|
|
94
|
-
args: functionCall.args || {},
|
|
95
|
-
result: `❌ Error: ${errorMsg}`,
|
|
96
|
-
};
|
|
97
|
-
onMessage('tool', `❌ Error: ${errorMsg}`, functionCall.name, toolInfo);
|
|
98
|
-
}
|
|
99
|
-
functionResponses.push({
|
|
100
|
-
functionResponse: {
|
|
101
|
-
name: functionCall.name,
|
|
102
|
-
response: {
|
|
103
|
-
error: errorMsg,
|
|
104
|
-
},
|
|
105
|
-
},
|
|
106
|
-
});
|
|
107
|
-
// **NEW: Stop on critical errors**
|
|
108
|
-
// Nếu là lỗi nghiêm trọng, có thể break để không lãng phí
|
|
109
|
-
// nhưng hiện tại vẫn tiếp tục để AI có đầy đủ context
|
|
110
|
-
}
|
|
111
|
-
}
|
|
112
|
-
// Gửi tất cả function responses
|
|
113
|
-
currentMessage = functionResponses;
|
|
114
|
-
}
|
|
115
|
-
else {
|
|
116
|
-
// Không có function calls → trả về text response
|
|
117
|
-
return response.text ?? '';
|
|
118
|
-
}
|
|
119
|
-
}
|
|
120
|
-
if (iteration >= MAX_ITERATIONS) {
|
|
121
|
-
console.warn(`⚠️ Max iterations (${MAX_ITERATIONS}) reached. This may indicate:`);
|
|
122
|
-
console.warn(' - Too many sequential tool calls');
|
|
123
|
-
console.warn(' - AI stuck in a reasoning loop');
|
|
124
|
-
console.warn(' - Consider reviewing system prompt or tool definitions');
|
|
125
|
-
throw new Error(`Max iterations (${MAX_ITERATIONS}) reached`);
|
|
126
|
-
}
|
|
127
|
-
return '';
|
|
128
|
-
}
|
|
129
|
-
getToolDeclarations() {
|
|
130
|
-
const tools = this.toolsRegistry.list();
|
|
131
|
-
return tools.map(tool => this.convertToFunctionDeclaration(tool));
|
|
132
|
-
}
|
|
133
|
-
convertToFunctionDeclaration(tool) {
|
|
134
|
-
return {
|
|
135
|
-
name: tool.name,
|
|
136
|
-
description: tool.description,
|
|
137
|
-
parameters: this.convertSchemaToGeminiSchema(tool.inputSchema),
|
|
138
|
-
};
|
|
139
|
-
}
|
|
140
|
-
convertSchemaToGeminiSchema(schema) {
|
|
141
|
-
if (!schema) {
|
|
142
|
-
return {
|
|
143
|
-
type: Type.OBJECT,
|
|
144
|
-
properties: {},
|
|
145
|
-
};
|
|
146
|
-
}
|
|
147
|
-
const converted = {
|
|
148
|
-
type: Type.OBJECT,
|
|
149
|
-
properties: {},
|
|
150
|
-
};
|
|
151
|
-
if (schema['properties']) {
|
|
152
|
-
converted.properties = {};
|
|
153
|
-
for (const [key, value] of Object.entries(schema['properties'])) {
|
|
154
|
-
const prop = value;
|
|
155
|
-
converted.properties[key] = this.convertPropertyToGeminiSchema(prop);
|
|
156
|
-
}
|
|
157
|
-
}
|
|
158
|
-
if (schema['required']) {
|
|
159
|
-
converted.required = schema['required'];
|
|
160
|
-
}
|
|
161
|
-
return converted;
|
|
162
|
-
}
|
|
163
|
-
convertPropertyToGeminiSchema(prop) {
|
|
164
|
-
const propSchema = {
|
|
165
|
-
type: this.mapTypeToGemini(prop.type),
|
|
166
|
-
description: prop.description || '',
|
|
167
|
-
};
|
|
168
|
-
if (prop.type === 'object' && prop.properties) {
|
|
169
|
-
propSchema.properties = {};
|
|
170
|
-
for (const [key, value] of Object.entries(prop.properties)) {
|
|
171
|
-
propSchema.properties[key] = this.convertPropertyToGeminiSchema(value);
|
|
172
|
-
}
|
|
173
|
-
if (prop.required) {
|
|
174
|
-
propSchema.required = prop.required;
|
|
175
|
-
}
|
|
176
|
-
}
|
|
177
|
-
if (prop.type === 'array') {
|
|
178
|
-
if (prop.items && typeof prop.items === 'object') {
|
|
179
|
-
propSchema.items = this.convertPropertyToGeminiSchema(prop.items);
|
|
180
|
-
}
|
|
181
|
-
else {
|
|
182
|
-
// Gemini requires items field for arrays
|
|
183
|
-
propSchema.items = {
|
|
184
|
-
type: Type.STRING,
|
|
185
|
-
description: '',
|
|
186
|
-
};
|
|
187
|
-
}
|
|
188
|
-
}
|
|
189
|
-
return propSchema;
|
|
190
|
-
}
|
|
191
|
-
mapTypeToGemini(type) {
|
|
192
|
-
const typeMap = {
|
|
193
|
-
string: Type.STRING,
|
|
194
|
-
number: Type.NUMBER,
|
|
195
|
-
integer: Type.INTEGER,
|
|
196
|
-
boolean: Type.BOOLEAN,
|
|
197
|
-
object: Type.OBJECT,
|
|
198
|
-
array: Type.ARRAY,
|
|
199
|
-
};
|
|
200
|
-
return typeMap[type.toLowerCase()] || Type.STRING;
|
|
201
|
-
}
|
|
202
|
-
}
|
|
203
|
-
//# sourceMappingURL=gemini.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../src/connections/gemini.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAIL,WAAW,EAEX,IAAI,GAEL,MAAM,eAAe,CAAA;AACtB,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAA;AAIzD,MAAM,WAAW,GAAG,kBAAkB,CAAA;AAgBtC,MAAM,OAAO,gBAAgB;IACnB,MAAM,CAAa;IACV,MAAM,CAAoB;IACnC,WAAW,CAAkB;IACpB,iBAAiB,CAAS;IAC1B,KAAK,CAAQ;IACtB,aAAa,GAAG,kBAAkB,CAAA;IAE1C,YAAY,MAAoB,EAAE,YAAqB;QACrD,mDAAmD;QACnD,IAAI,CAAC,MAAM,GAAG,MAAM,EAAE,MAAM,CAAA;QAE5B,IAAI,CAAC,KAAK,GAAG,MAAM,EAAE,KAAK,IAAI,kBAAkB,CAAA;QAChD,IAAI,CAAC,MAAM,GAAG,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAA;QACtD,IAAI,CAAC,iBAAiB,GAAG,YAAY,CAAA;QACrC,IAAI,CAAC,QAAQ,EAAE,CAAA;IACjB,CAAC;IAED,QAAQ;QACN,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,MAAM,EAAE;gBACN,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;gBACzC,KAAK,EAAE,CAAC,EAAE,oBAAoB,EAAE,IAAI,CAAC,mBAAmB,EAAE,EAAE,CAAC;gBAC7D,WAAW,EAAE,GAAG;gBAChB,cAAc,EAAE;oBACd,cAAc,EAAE,CAAC;iBAClB;aACF;SACF,CAAC,CAAA;IACJ,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,UAAoC;QACjD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC;YACxD,KAAK,EAAE,WAAW;YAClB,QAAQ,EAAE,UAAU,CAAC,MAAM;SAC5B,CAAC,CAAA;QACF,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAA;QAEhC,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAA;IAChC,CAAC;IAED,KAAK,CAAC,IAAI,CACR,OAAe,EACf,SAKS;QAET,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAA;QACrD,CAAC;QAED,IAAI,cAAc,GAAmB,OAAO,CAAA;QAC5C,IAAI,QAAiC,CAAA;QACrC,IAAI,SAAS,GAAG,CAAC,CAAA;QACjB,MAAM,cAAc,GAAG,EAAE,CAAA,CAAC,kCAAkC;QAE5D,OAAO,cAAc,IAAI,SAAS,GAAG,cAAc,EAAE,CAAC;YACpD,SAAS,EAAE,CAAA;YAEX,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC;gBAC5C,OAAO,EAAE,cAAqB;aAC/B,CAAC,CAAA;YAEF,IAAI,QAAQ,CAAC,aAAa,IAAI,QAAQ,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAChE,MAAM,iBAAiB,GAAU,EAAE,CAAA;gBAEnC,wDAAwD;gBACxD,KAAK,MAAM,YAAY,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;oBAClD,IAAI,CAAC;wBACH,MAAM,QAAQ,GAAG,YAAY,CAAC,IAAI,IAAI,SAAS,CAAA;wBAC/C,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,IAAI,EAAE,CAAA;wBAEpC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAA;wBAEvE,MAAM,SAAS,GACb,OAAO,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAA;wBAE1E,IAAI,SAAS,EAAE,CAAC;4BACd,iDAAiD;4BACjD,MAAM,QAAQ,GAAsB;gCAClC,QAAQ;gCACR,IAAI;gCACJ,MAAM,EAAE,UAAU;6BACnB,CAAA;4BACD,SAAS,CAAC,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAA;wBAClD,CAAC;wBAED,iBAAiB,CAAC,IAAI,CAAC;4BACrB,gBAAgB,EAAE;gCAChB,IAAI,EAAE,YAAY,CAAC,IAAI;gCACvB,QAAQ,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE;6BAChC;yBACF,CAAC,CAAA;wBAEF,gDAAgD;wBAChD,kEAAkE;wBAClE,8CAA8C;oBAChD,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,MAAM,QAAQ,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;wBAEvE,YAAY;wBACZ,IAAI,SAAS,EAAE,CAAC;4BACd,MAAM,QAAQ,GAAsB;gCAClC,QAAQ,EAAE,YAAY,CAAC,IAAI,IAAI,SAAS;gCACxC,IAAI,EAAE,YAAY,CAAC,IAAI,IAAI,EAAE;gCAC7B,MAAM,EAAE,YAAY,QAAQ,EAAE;6BAC/B,CAAA;4BACD,SAAS,CAAC,MAAM,EAAE,YAAY,QAAQ,EAAE,EAAE,YAAY,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAA;wBACxE,CAAC;wBAED,iBAAiB,CAAC,IAAI,CAAC;4BACrB,gBAAgB,EAAE;gCAChB,IAAI,EAAE,YAAY,CAAC,IAAI;gCACvB,QAAQ,EAAE;oCACR,KAAK,EAAE,QAAQ;iCAChB;6BACF;yBACF,CAAC,CAAA;wBAEF,mCAAmC;wBACnC,0DAA0D;wBAC1D,sDAAsD;oBACxD,CAAC;gBACH,CAAC;gBAED,gCAAgC;gBAChC,cAAc,GAAG,iBAAiB,CAAA;YACpC,CAAC;iBAAM,CAAC;gBACN,iDAAiD;gBACjD,OAAO,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAA;YAC5B,CAAC;QACH,CAAC;QAED,IAAI,SAAS,IAAI,cAAc,EAAE,CAAC;YAChC,OAAO,CAAC,IAAI,CAAC,sBAAsB,cAAc,+BAA+B,CAAC,CAAA;YACjF,OAAO,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAA;YAClD,OAAO,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAA;YAChD,OAAO,CAAC,IAAI,CAAC,0DAA0D,CAAC,CAAA;YACxE,MAAM,IAAI,KAAK,CAAC,mBAAmB,cAAc,WAAW,CAAC,CAAA;QAC/D,CAAC;QAED,OAAO,EAAE,CAAA;IACX,CAAC;IAEO,mBAAmB;QACzB,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAA;QACvC,OAAO,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,4BAA4B,CAAC,IAAI,CAAC,CAAC,CAAA;IACnE,CAAC;IAEO,4BAA4B,CAAC,IAIpC;QACC,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,WAAW,CAAC;SAC/D,CAAA;IACH,CAAC;IAEO,2BAA2B,CAAC,MAA2C;QAC7E,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;gBACL,IAAI,EAAE,IAAI,CAAC,MAAM;gBACjB,UAAU,EAAE,EAAE;aACf,CAAA;QACH,CAAC;QAED,MAAM,SAAS,GAAW;YACxB,IAAI,EAAE,IAAI,CAAC,MAAM;YACjB,UAAU,EAAE,EAAE;SACf,CAAA;QAED,IAAI,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC;YACzB,SAAS,CAAC,UAAU,GAAG,EAAE,CAAA;YACzB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC;gBAChE,MAAM,IAAI,GAAG,KAAY,CAAA;gBACzB,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,6BAA6B,CAAC,IAAI,CAAC,CAAA;YACtE,CAAC;QACH,CAAC;QAED,IAAI,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC;YACvB,SAAS,CAAC,QAAQ,GAAG,MAAM,CAAC,UAAU,CAAa,CAAA;QACrD,CAAC;QAED,OAAO,SAAS,CAAA;IAClB,CAAC;IAEO,6BAA6B,CAAC,IAAS;QAC7C,MAAM,UAAU,GAAW;YACzB,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC;YACrC,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;SACpC,CAAA;QAED,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YAC9C,UAAU,CAAC,UAAU,GAAG,EAAE,CAAA;YAC1B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC3D,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,6BAA6B,CAAC,KAAY,CAAC,CAAA;YAC/E,CAAC;YACD,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClB,UAAU,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAA;YACrC,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAC1B,IAAI,IAAI,CAAC,KAAK,IAAI,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;gBACjD,UAAU,CAAC,KAAK,GAAG,IAAI,CAAC,6BAA6B,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YACnE,CAAC;iBAAM,CAAC;gBACN,yCAAyC;gBACzC,UAAU,CAAC,KAAK,GAAG;oBACjB,IAAI,EAAE,IAAI,CAAC,MAAM;oBACjB,WAAW,EAAE,EAAE;iBAChB,CAAA;YACH,CAAC;QACH,CAAC;QAED,OAAO,UAAU,CAAA;IACnB,CAAC;IAEO,eAAe,CAAC,IAAY;QAClC,MAAM,OAAO,GAAyB;YACpC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAA;QACD,OAAO,OAAO,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,CAAA;IACnD,CAAC;CACF"}
|
|
@@ -1,12 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* @author Bùi Trọng Hiếu
|
|
3
|
-
* @email kevinbui210191@gmail.com
|
|
4
|
-
* @create 2024-10-08
|
|
5
|
-
* @modify 2024-10-13
|
|
6
|
-
* @desc Export entry for connection modules.
|
|
7
|
-
*/
|
|
8
|
-
export * from './types.js';
|
|
9
|
-
export * from './ollama.js';
|
|
10
|
-
export * from './gemini.js';
|
|
11
|
-
export * from './claude.js';
|
|
12
|
-
//# sourceMappingURL=index.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/connections/index.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,cAAc,YAAY,CAAC;AAC3B,cAAc,aAAa,CAAC;AAC5B,cAAc,aAAa,CAAC;AAC5B,cAAc,aAAa,CAAC"}
|
|
@@ -1,12 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* @author Bùi Trọng Hiếu
|
|
3
|
-
* @email kevinbui210191@gmail.com
|
|
4
|
-
* @create 2024-10-08
|
|
5
|
-
* @modify 2024-10-13
|
|
6
|
-
* @desc Export entry for connection modules.
|
|
7
|
-
*/
|
|
8
|
-
export * from './types.js';
|
|
9
|
-
export * from './ollama.js';
|
|
10
|
-
export * from './gemini.js';
|
|
11
|
-
export * from './claude.js';
|
|
12
|
-
//# sourceMappingURL=index.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/connections/index.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,cAAc,YAAY,CAAC;AAC3B,cAAc,aAAa,CAAC;AAC5B,cAAc,aAAa,CAAC;AAC5B,cAAc,aAAa,CAAC"}
|
|
@@ -1,38 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* @author Bùi Trọng Hiếu
|
|
3
|
-
* @email kevinbui210191@gmail.com
|
|
4
|
-
* @create 2024-10-08
|
|
5
|
-
* @modify 2024-10-08
|
|
6
|
-
* @desc Ollama connection helper wrapping chat completion endpoint.
|
|
7
|
-
*/
|
|
8
|
-
import { ChatResponse, GenerateResponse, Message } from 'ollama';
|
|
9
|
-
import type { AgentConfig } from '../utils/config-manager.js';
|
|
10
|
-
export declare const ollamaModels: readonly ["deepseek-v3.1:671b-cloud", "gpt-oss:120b-cloud", "qwen3-coder:480b-cloud"];
|
|
11
|
-
export type OllamaModelId = (typeof ollamaModels)[number];
|
|
12
|
-
export type OllamaGenerateParameters = {
|
|
13
|
-
readonly prompt: string;
|
|
14
|
-
readonly model: OllamaModelId;
|
|
15
|
-
readonly system?: string;
|
|
16
|
-
};
|
|
17
|
-
export type OllamaGenerateResult<T = unknown> = {
|
|
18
|
-
readonly text: string;
|
|
19
|
-
readonly raw: T;
|
|
20
|
-
};
|
|
21
|
-
export declare class OllamaConnection {
|
|
22
|
-
private ollama;
|
|
23
|
-
private model;
|
|
24
|
-
constructor(config?: AgentConfig | {
|
|
25
|
-
model?: OllamaModelId;
|
|
26
|
-
host?: string;
|
|
27
|
-
apiKey?: string;
|
|
28
|
-
});
|
|
29
|
-
generate(parameters: OllamaGenerateParameters): Promise<GenerateResponse>;
|
|
30
|
-
chat(userMessage: string): Promise<string>;
|
|
31
|
-
chatWithTools(options: {
|
|
32
|
-
model: string;
|
|
33
|
-
messages: Message[];
|
|
34
|
-
tools?: any[];
|
|
35
|
-
stream?: false;
|
|
36
|
-
}): Promise<ChatResponse>;
|
|
37
|
-
}
|
|
38
|
-
//# sourceMappingURL=ollama.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"ollama.d.ts","sourceRoot":"","sources":["../../src/connections/ollama.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAAE,YAAY,EAAE,gBAAgB,EAAE,OAAO,EAAU,MAAM,QAAQ,CAAA;AACxE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAA;AAE7D,eAAO,MAAM,YAAY,uFAIf,CAAA;AAEV,MAAM,MAAM,aAAa,GAAG,CAAC,OAAO,YAAY,CAAC,CAAC,MAAM,CAAC,CAAA;AAEzD,MAAM,MAAM,wBAAwB,GAAG;IACrC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IACvB,QAAQ,CAAC,KAAK,EAAE,aAAa,CAAA;IAC7B,QAAQ,CAAC,MAAM,CAAC,EAAE,MAAM,CAAA;CACzB,CAAA;AACD,MAAM,MAAM,oBAAoB,CAAC,CAAC,GAAG,OAAO,IAAI;IAC9C,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAA;IACrB,QAAQ,CAAC,GAAG,EAAE,CAAC,CAAA;CAChB,CAAA;AAED,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,MAAM,CAAQ;IACtB,OAAO,CAAC,KAAK,CAA4C;gBAC7C,MAAM,CAAC,EAAE,WAAW,GAAG;QAAE,KAAK,CAAC,EAAE,aAAa,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE;IAwBtF,QAAQ,CAAC,UAAU,EAAE,wBAAwB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAWzE,IAAI,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAU1C,aAAa,CAAC,OAAO,EAAE;QAC3B,KAAK,EAAE,MAAM,CAAA;QACb,QAAQ,EAAE,OAAO,EAAE,CAAA;QACnB,KAAK,CAAC,EAAE,GAAG,EAAE,CAAA;QACb,MAAM,CAAC,EAAE,KAAK,CAAA;KACf,GAAG,OAAO,CAAC,YAAY,CAAC;CAQ1B"}
|