thatgfsj-code 0.3.0 → 0.4.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/cmd/index.js +56 -8
- package/dist/cmd/index.js.map +1 -1
- package/dist/config/index.d.ts +5 -1
- package/dist/config/index.d.ts.map +1 -1
- package/dist/config/index.js +22 -4
- package/dist/config/index.js.map +1 -1
- package/dist/config/providers.d.ts +5 -10
- package/dist/config/providers.d.ts.map +1 -1
- package/dist/config/providers.js +152 -78
- package/dist/config/providers.js.map +1 -1
- package/dist/config/types.d.ts +1 -1
- package/dist/config/types.d.ts.map +1 -1
- package/dist/llm/anthropic.d.ts +24 -6
- package/dist/llm/anthropic.d.ts.map +1 -1
- package/dist/llm/anthropic.js +136 -46
- package/dist/llm/anthropic.js.map +1 -1
- package/dist/llm/gemini.d.ts +4 -4
- package/dist/llm/gemini.d.ts.map +1 -1
- package/dist/llm/gemini.js +67 -30
- package/dist/llm/gemini.js.map +1 -1
- package/dist/llm/index.d.ts +10 -23
- package/dist/llm/index.d.ts.map +1 -1
- package/dist/llm/index.js +73 -78
- package/dist/llm/index.js.map +1 -1
- package/dist/llm/openai.d.ts +19 -6
- package/dist/llm/openai.d.ts.map +1 -1
- package/dist/llm/openai.js +86 -35
- package/dist/llm/openai.js.map +1 -1
- package/dist/llm/provider.d.ts +11 -16
- package/dist/llm/provider.d.ts.map +1 -1
- package/dist/tui/input.d.ts +0 -1
- package/dist/tui/input.d.ts.map +1 -1
- package/dist/tui/input.js +2 -5
- package/dist/tui/input.js.map +1 -1
- package/dist/tui/output.d.ts +27 -12
- package/dist/tui/output.d.ts.map +1 -1
- package/dist/tui/output.js +176 -40
- package/dist/tui/output.js.map +1 -1
- package/dist/tui/repl.d.ts +0 -10
- package/dist/tui/repl.d.ts.map +1 -1
- package/dist/tui/repl.js +76 -42
- package/dist/tui/repl.js.map +1 -1
- package/dist/tui/welcome.d.ts +2 -8
- package/dist/tui/welcome.d.ts.map +1 -1
- package/dist/tui/welcome.js +101 -51
- package/dist/tui/welcome.js.map +1 -1
- package/package.json +1 -1
- package/src/cmd/index.ts +49 -8
- package/src/config/index.ts +21 -4
- package/src/config/providers.ts +153 -79
- package/src/config/types.ts +11 -5
- package/src/llm/anthropic.ts +148 -51
- package/src/llm/gemini.ts +81 -39
- package/src/llm/index.ts +75 -83
- package/src/llm/openai.ts +96 -41
- package/src/llm/provider.ts +12 -16
- package/src/tui/input.ts +2 -5
- package/src/tui/output.ts +193 -40
- package/src/tui/repl.ts +82 -43
- package/src/tui/welcome.ts +110 -53
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../src/llm/anthropic.ts"],"names":[],"mappings":"AAAA
|
|
1
|
+
{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../src/llm/anthropic.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,WAAW,EAAY,MAAM,aAAa,CAAC;AACpF,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,mBAAmB,CAAC;AAC9C,OAAO,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAE9E,qBAAa,iBAAkB,YAAW,WAAW;IACnD,QAAQ,CAAC,IAAI,eAAe;IAC5B,SAAS,CAAC,MAAM,EAAE,cAAc,CAAC;gBAErB,MAAM,EAAE,cAAc;IAIlC,UAAU,CAAC,KAAK,EAAE,IAAI,EAAE,GAAG,GAAG,EAAE;IAc1B,IAAI,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,OAAO,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,YAAY,CAAC;IAsC1F,UAAU,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,OAAO,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,EAAE,IAAI,EAAE,GAAG,cAAc,CAAC,WAAW,EAAE,YAAY,CAAC;IAwF5H;;OAEG;IACH,SAAS,CAAC,YAAY,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,EAAE,IAAI,EAAE;IAwDtG;;OAEG;cACa,SAAS,CAAC,IAAI,EAAE,GAAG,GAAG,OAAO,CAAC,QAAQ,CAAC;CAYxD"}
|
package/dist/llm/anthropic.js
CHANGED
|
@@ -1,6 +1,17 @@
|
|
|
1
1
|
/**
|
|
2
2
|
* Anthropic Provider
|
|
3
|
-
* Works with: Anthropic Claude
|
|
3
|
+
* Works with: Anthropic Claude API
|
|
4
|
+
* Also works with any Anthropic-compatible relay station (中转站)
|
|
5
|
+
*
|
|
6
|
+
* API: POST {baseUrl}/messages
|
|
7
|
+
* Auth: x-api-key: {apiKey}, anthropic-version: 2023-06-01
|
|
8
|
+
* Streaming: SSE with "event: ..." and "data: {...}" lines
|
|
9
|
+
*
|
|
10
|
+
* Key differences from OpenAI:
|
|
11
|
+
* - System message is separate (top-level "system" field)
|
|
12
|
+
* - No "system" role in messages array
|
|
13
|
+
* - Tool use blocks have type "tool_use" with "input" (not "arguments")
|
|
14
|
+
* - Tool results use role "tool_result" (not "tool")
|
|
4
15
|
*/
|
|
5
16
|
export class AnthropicProvider {
|
|
6
17
|
name = 'anthropic';
|
|
@@ -19,54 +30,41 @@ export class AnthropicProvider {
|
|
|
19
30
|
},
|
|
20
31
|
}));
|
|
21
32
|
}
|
|
22
|
-
async chat(messages, options) {
|
|
23
|
-
const body = this.buildRequest(messages, false, options);
|
|
24
|
-
const response = await
|
|
25
|
-
method: 'POST',
|
|
26
|
-
headers: {
|
|
27
|
-
'Content-Type': 'application/json',
|
|
28
|
-
'x-api-key': this.config.apiKey,
|
|
29
|
-
'anthropic-version': '2023-06-01',
|
|
30
|
-
},
|
|
31
|
-
body: JSON.stringify(body),
|
|
32
|
-
});
|
|
33
|
+
async chat(messages, options, tools) {
|
|
34
|
+
const body = this.buildRequest(messages, false, options, tools);
|
|
35
|
+
const response = await this.doRequest(body);
|
|
33
36
|
if (!response.ok) {
|
|
34
37
|
const text = await response.text().catch(() => '');
|
|
35
38
|
throw new Error(`Anthropic API error ${response.status}: ${text}`);
|
|
36
39
|
}
|
|
37
40
|
const data = await response.json();
|
|
38
|
-
|
|
41
|
+
// Extract text content
|
|
42
|
+
const textBlocks = data.content?.filter((b) => b.type === 'text') || [];
|
|
43
|
+
const content = textBlocks.map((b) => b.text).join('');
|
|
44
|
+
// Extract tool calls
|
|
39
45
|
const toolUseBlocks = data.content?.filter((b) => b.type === 'tool_use') || [];
|
|
46
|
+
const toolCalls = toolUseBlocks.length > 0 ? toolUseBlocks.map((b) => ({
|
|
47
|
+
id: b.id,
|
|
48
|
+
type: 'function',
|
|
49
|
+
function: {
|
|
50
|
+
name: b.name,
|
|
51
|
+
arguments: JSON.stringify(b.input),
|
|
52
|
+
},
|
|
53
|
+
})) : undefined;
|
|
40
54
|
return {
|
|
41
|
-
content
|
|
55
|
+
content,
|
|
42
56
|
role: 'assistant',
|
|
43
57
|
usage: data.usage ? {
|
|
44
58
|
prompt_tokens: data.usage.input_tokens || 0,
|
|
45
59
|
completion_tokens: data.usage.output_tokens || 0,
|
|
46
60
|
total_tokens: (data.usage.input_tokens || 0) + (data.usage.output_tokens || 0),
|
|
47
61
|
} : undefined,
|
|
48
|
-
tool_calls:
|
|
49
|
-
id: b.id,
|
|
50
|
-
type: 'function',
|
|
51
|
-
function: {
|
|
52
|
-
name: b.name,
|
|
53
|
-
arguments: JSON.stringify(b.input),
|
|
54
|
-
},
|
|
55
|
-
})) : undefined,
|
|
62
|
+
tool_calls: toolCalls,
|
|
56
63
|
};
|
|
57
64
|
}
|
|
58
|
-
async *chatStream(messages, options) {
|
|
59
|
-
const body = this.buildRequest(messages, true, options);
|
|
60
|
-
const response = await
|
|
61
|
-
method: 'POST',
|
|
62
|
-
headers: {
|
|
63
|
-
'Content-Type': 'application/json',
|
|
64
|
-
'x-api-key': this.config.apiKey,
|
|
65
|
-
'anthropic-version': '2023-06-01',
|
|
66
|
-
'anthropic-dangerous-direct-browser-access': 'true',
|
|
67
|
-
},
|
|
68
|
-
body: JSON.stringify(body),
|
|
69
|
-
});
|
|
65
|
+
async *chatStream(messages, options, tools) {
|
|
66
|
+
const body = this.buildRequest(messages, true, options, tools);
|
|
67
|
+
const response = await this.doRequest(body);
|
|
70
68
|
if (!response.ok || !response.body) {
|
|
71
69
|
const text = await response.text().catch(() => '');
|
|
72
70
|
throw new Error(`Anthropic API error ${response.status}: ${text}`);
|
|
@@ -75,6 +73,10 @@ export class AnthropicProvider {
|
|
|
75
73
|
const decoder = new TextDecoder();
|
|
76
74
|
let fullContent = '';
|
|
77
75
|
let buffer = '';
|
|
76
|
+
// Track tool use blocks
|
|
77
|
+
const toolUseBlocks = new Map();
|
|
78
|
+
let currentBlockIndex = -1;
|
|
79
|
+
let currentBlockType = '';
|
|
78
80
|
try {
|
|
79
81
|
while (true) {
|
|
80
82
|
const { done, value } = await reader.read();
|
|
@@ -89,10 +91,31 @@ export class AnthropicProvider {
|
|
|
89
91
|
continue;
|
|
90
92
|
try {
|
|
91
93
|
const data = JSON.parse(trimmed.slice(6));
|
|
92
|
-
|
|
93
|
-
|
|
94
|
-
|
|
95
|
-
|
|
94
|
+
// content_block_start: track block types
|
|
95
|
+
if (data.type === 'content_block_start') {
|
|
96
|
+
currentBlockIndex = data.index ?? 0;
|
|
97
|
+
currentBlockType = data.content_block?.type || '';
|
|
98
|
+
if (currentBlockType === 'tool_use') {
|
|
99
|
+
toolUseBlocks.set(currentBlockIndex, {
|
|
100
|
+
id: data.content_block.id || '',
|
|
101
|
+
name: data.content_block.name || '',
|
|
102
|
+
input: '',
|
|
103
|
+
});
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
// content_block_delta: incremental text or tool input
|
|
107
|
+
if (data.type === 'content_block_delta') {
|
|
108
|
+
if (data.delta?.type === 'text_delta') {
|
|
109
|
+
const text = data.delta.text;
|
|
110
|
+
fullContent += text;
|
|
111
|
+
yield { type: 'text', content: text };
|
|
112
|
+
}
|
|
113
|
+
else if (data.delta?.type === 'input_json_delta') {
|
|
114
|
+
const buf = toolUseBlocks.get(currentBlockIndex);
|
|
115
|
+
if (buf) {
|
|
116
|
+
buf.input += data.delta.partial_json || '';
|
|
117
|
+
}
|
|
118
|
+
}
|
|
96
119
|
}
|
|
97
120
|
}
|
|
98
121
|
catch {
|
|
@@ -104,23 +127,90 @@ export class AnthropicProvider {
|
|
|
104
127
|
finally {
|
|
105
128
|
reader.releaseLock();
|
|
106
129
|
}
|
|
107
|
-
|
|
130
|
+
// Convert tool use blocks to ToolCall[]
|
|
131
|
+
const toolCalls = [];
|
|
132
|
+
for (const [, buf] of toolUseBlocks) {
|
|
133
|
+
if (buf.id && buf.name) {
|
|
134
|
+
toolCalls.push({
|
|
135
|
+
id: buf.id,
|
|
136
|
+
type: 'function',
|
|
137
|
+
function: { name: buf.name, arguments: buf.input },
|
|
138
|
+
});
|
|
139
|
+
}
|
|
140
|
+
}
|
|
141
|
+
if (toolCalls.length > 0) {
|
|
142
|
+
yield { type: 'tool_calls', toolCalls };
|
|
143
|
+
}
|
|
144
|
+
return { content: fullContent, role: 'assistant', tool_calls: toolCalls.length > 0 ? toolCalls : undefined };
|
|
108
145
|
}
|
|
109
|
-
|
|
146
|
+
/**
|
|
147
|
+
* Build request body for Anthropic API
|
|
148
|
+
*/
|
|
149
|
+
buildRequest(messages, stream, options, tools) {
|
|
110
150
|
// Anthropic uses separate system message
|
|
111
151
|
const systemMsg = messages.find(m => m.role === 'system');
|
|
112
152
|
const nonSystemMsgs = messages.filter(m => m.role !== 'system');
|
|
113
|
-
|
|
153
|
+
// Convert messages to Anthropic format
|
|
154
|
+
const anthropicMessages = nonSystemMsgs.map(m => {
|
|
155
|
+
if (m.role === 'tool') {
|
|
156
|
+
// Tool result message
|
|
157
|
+
return {
|
|
158
|
+
role: 'user',
|
|
159
|
+
content: [{
|
|
160
|
+
type: 'tool_result',
|
|
161
|
+
tool_use_id: m.tool_call_id,
|
|
162
|
+
content: m.content,
|
|
163
|
+
}],
|
|
164
|
+
};
|
|
165
|
+
}
|
|
166
|
+
if (m.tool_calls && m.tool_calls.length > 0) {
|
|
167
|
+
// Assistant message with tool calls
|
|
168
|
+
const blocks = [];
|
|
169
|
+
if (m.content) {
|
|
170
|
+
blocks.push({ type: 'text', text: m.content });
|
|
171
|
+
}
|
|
172
|
+
for (const tc of m.tool_calls) {
|
|
173
|
+
blocks.push({
|
|
174
|
+
type: 'tool_use',
|
|
175
|
+
id: tc.id,
|
|
176
|
+
name: tc.function.name,
|
|
177
|
+
input: JSON.parse(tc.function.arguments || '{}'),
|
|
178
|
+
});
|
|
179
|
+
}
|
|
180
|
+
return { role: 'assistant', content: blocks };
|
|
181
|
+
}
|
|
182
|
+
return {
|
|
183
|
+
role: m.role === 'assistant' ? 'assistant' : 'user',
|
|
184
|
+
content: m.content,
|
|
185
|
+
};
|
|
186
|
+
});
|
|
187
|
+
const body = {
|
|
114
188
|
model: this.config.model,
|
|
115
|
-
max_tokens: options?.maxTokens ?? this.config.maxTokens,
|
|
189
|
+
max_tokens: options?.maxTokens ?? this.config.maxTokens ?? 4096,
|
|
116
190
|
temperature: options?.temperature ?? this.config.temperature,
|
|
117
191
|
...(systemMsg && { system: systemMsg.content }),
|
|
118
|
-
messages:
|
|
119
|
-
role: m.role === 'assistant' ? 'assistant' : 'user',
|
|
120
|
-
content: m.content,
|
|
121
|
-
})),
|
|
192
|
+
messages: anthropicMessages,
|
|
122
193
|
stream,
|
|
123
194
|
};
|
|
195
|
+
if (tools && tools.length > 0) {
|
|
196
|
+
body.tools = this.buildTools(tools);
|
|
197
|
+
}
|
|
198
|
+
return body;
|
|
199
|
+
}
|
|
200
|
+
/**
|
|
201
|
+
* Execute the HTTP request
|
|
202
|
+
*/
|
|
203
|
+
async doRequest(body) {
|
|
204
|
+
const url = `${this.config.baseUrl}/messages`;
|
|
205
|
+
return fetch(url, {
|
|
206
|
+
method: 'POST',
|
|
207
|
+
headers: {
|
|
208
|
+
'Content-Type': 'application/json',
|
|
209
|
+
'x-api-key': this.config.apiKey,
|
|
210
|
+
'anthropic-version': '2023-06-01',
|
|
211
|
+
},
|
|
212
|
+
body: JSON.stringify(body),
|
|
213
|
+
});
|
|
124
214
|
}
|
|
125
215
|
}
|
|
126
216
|
//# sourceMappingURL=anthropic.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../src/llm/anthropic.ts"],"names":[],"mappings":"AAAA
|
|
1
|
+
{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../src/llm/anthropic.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAMH,MAAM,OAAO,iBAAiB;IACnB,IAAI,GAAG,WAAW,CAAC;IAClB,MAAM,CAAiB;IAEjC,YAAY,MAAsB;QAChC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,UAAU,CAAC,KAAa;QACtB,OAAO,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACxB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,YAAY,EAAE,IAAI,CAAC,WAAW,IAAI;gBAChC,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,MAAM,CAAC,WAAW,CAC5B,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CACjF;gBACD,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;aACnE;SACF,CAAC,CAAC,CAAC;IACN,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,QAAuB,EAAE,OAAqB,EAAE,KAAc;QACvE,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;QAChE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAE5C,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;YACnD,MAAM,IAAI,KAAK,CAAC,uBAAuB,QAAQ,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC,CAAC;QACrE,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QAEnC,uBAAuB;QACvB,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,IAAI,EAAE,CAAC;QAC7E,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAE5D,qBAAqB;QACrB,MAAM,aAAa,GAAG,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,IAAI,EAAE,CAAC;QACpF,MAAM,SAAS,GAAG,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;YAC1E,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,IAAI,EAAE,UAAmB;YACzB,QAAQ,EAAE;gBACR,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC;aACnC;SACF,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAEhB,OAAO;YACL,OAAO;YACP,IAAI,EAAE,WAAW;YACjB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;gBAClB,aAAa,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC;gBAC3C,iBAAiB,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC;gBAChD,YAAY,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;aAC/E,CAAC,CAAC,CAAC,SAAS;YACb,UAAU,EAAE,SAAS;SACtB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,CAAC,UAAU,CAAC,QAAuB,EAAE,OAAqB,EAAE,KAAc;QAC9E,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;QAC/D,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAE5C,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;YACnD,MAAM,IAAI,KAAK,CAAC,uBAAuB,QAAQ,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC,CAAC;QACrE,CAAC;QAED,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;QACzC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;QAClC,IAAI,WAAW,GAAG,EAAE,CAAC;QACrB,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,wBAAwB;QACxB,MAAM,aAAa,GAA6D,IAAI,GAAG,EAAE,CAAC;QAC1F,IAAI,iBAAiB,GAAG,CAAC,CAAC,CAAC;QAC3B,IAAI,gBAAgB,GAAG,EAAE,CAAC;QAE1B,IAAI,CAAC;YACH,OAAO,IAAI,EAAE,CAAC;gBACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5C,IAAI,IAAI;oBAAE,MAAM;gBAEhB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBAClD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;gBAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACzB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;oBAC5B,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC;wBAAE,SAAS;oBAExD,IAAI,CAAC;wBACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;wBAE1C,yCAAyC;wBACzC,IAAI,IAAI,CAAC,IAAI,KAAK,qBAAqB,EAAE,CAAC;4BACxC,iBAAiB,GAAG,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC;4BACpC,gBAAgB,GAAG,IAAI,CAAC,aAAa,EAAE,IAAI,IAAI,EAAE,CAAC;4BAClD,IAAI,gBAAgB,KAAK,UAAU,EAAE,CAAC;gCACpC,aAAa,CAAC,GAAG,CAAC,iBAAiB,EAAE;oCACnC,EAAE,EAAE,IAAI,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE;oCAC/B,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,IAAI,EAAE;oCACnC,KAAK,EAAE,EAAE;iCACV,CAAC,CAAC;4BACL,CAAC;wBACH,CAAC;wBAED,sDAAsD;wBACtD,IAAI,IAAI,CAAC,IAAI,KAAK,qBAAqB,EAAE,CAAC;4BACxC,IAAI,IAAI,CAAC,KAAK,EAAE,IAAI,KAAK,YAAY,EAAE,CAAC;gCACtC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;gCAC7B,WAAW,IAAI,IAAI,CAAC;gCACpB,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;4BACxC,CAAC;iCAAM,IAAI,IAAI,CAAC,KAAK,EAAE,IAAI,KAAK,kBAAkB,EAAE,CAAC;gCACnD,MAAM,GAAG,GAAG,aAAa,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;gCACjD,IAAI,GAAG,EAAE,CAAC;oCACR,GAAG,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,YAAY,IAAI,EAAE,CAAC;gCAC7C,CAAC;4BACH,CAAC;wBACH,CAAC;oBACH,CAAC;oBAAC,MAAM,CAAC;wBACP,oBAAoB;oBACtB,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,CAAC;QAED,wCAAwC;QACxC,MAAM,SAAS,GAAe,EAAE,CAAC;QACjC,KAAK,MAAM,CAAC,EAAE,GAAG,CAAC,IAAI,aAAa,EAAE,CAAC;YACpC,IAAI,GAAG,CAAC,EAAE,IAAI,GAAG,CAAC,IAAI,EAAE,CAAC;gBACvB,SAAS,CAAC,IAAI,CAAC;oBACb,EAAE,EAAE,GAAG,CAAC,EAAE;oBACV,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,SAAS,EAAE,GAAG,CAAC,KAAK,EAAE;iBACnD,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC;QAC1C,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,WAAW,EAAE,UAAU,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;IAC/G,CAAC;IAED;;OAEG;IACO,YAAY,CAAC,QAAuB,EAAE,MAAe,EAAE,OAAqB,EAAE,KAAc;QACpG,yCAAyC;QACzC,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QAC1D,MAAM,aAAa,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QAEhE,uCAAuC;QACvC,MAAM,iBAAiB,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;YAC9C,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACtB,sBAAsB;gBACtB,OAAO;oBACL,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,CAAC;4BACR,IAAI,EAAE,aAAa;4BACnB,WAAW,EAAE,CAAC,CAAC,YAAY;4BAC3B,OAAO,EAAE,CAAC,CAAC,OAAO;yBACnB,CAAC;iBACH,CAAC;YACJ,CAAC;YACD,IAAI,CAAC,CAAC,UAAU,IAAI,CAAC,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC5C,oCAAoC;gBACpC,MAAM,MAAM,GAAU,EAAE,CAAC;gBACzB,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC;oBACd,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;gBACjD,CAAC;gBACD,KAAK,MAAM,EAAE,IAAI,CAAC,CAAC,UAAU,EAAE,CAAC;oBAC9B,MAAM,CAAC,IAAI,CAAC;wBACV,IAAI,EAAE,UAAU;wBAChB,EAAE,EAAE,EAAE,CAAC,EAAE;wBACT,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;wBACtB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,IAAI,IAAI,CAAC;qBACjD,CAAC,CAAC;gBACL,CAAC;gBACD,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;YAChD,CAAC;YACD,OAAO;gBACL,IAAI,EAAE,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM;gBACnD,OAAO,EAAE,CAAC,CAAC,OAAO;aACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,MAAM,IAAI,GAAQ;YAChB,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;YACxB,UAAU,EAAE,OAAO,EAAE,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,IAAI;YAC/D,WAAW,EAAE,OAAO,EAAE,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW;YAC5D,GAAG,CAAC,SAAS,IAAI,EAAE,MAAM,EAAE,SAAS,CAAC,OAAO,EAAE,CAAC;YAC/C,QAAQ,EAAE,iBAAiB;YAC3B,MAAM;SACP,CAAC;QAEF,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACtC,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACO,KAAK,CAAC,SAAS,CAAC,IAAS;QACjC,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,WAAW,CAAC;QAC9C,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM;gBAC/B,mBAAmB,EAAE,YAAY;aAClC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CAAC,CAAC;IACL,CAAC;CACF"}
|
package/dist/llm/gemini.d.ts
CHANGED
|
@@ -3,14 +3,14 @@
|
|
|
3
3
|
*/
|
|
4
4
|
import type { ChatMessage, ChatResponse, ChatOptions } from '../types.js';
|
|
5
5
|
import type { Tool } from '../tools/types.js';
|
|
6
|
-
import type { LLMProvider, ProviderConfig } from './provider.js';
|
|
6
|
+
import type { LLMProvider, StreamChunk, ProviderConfig } from './provider.js';
|
|
7
7
|
export declare class GeminiProvider implements LLMProvider {
|
|
8
8
|
readonly name = "gemini";
|
|
9
|
-
|
|
9
|
+
protected config: ProviderConfig;
|
|
10
10
|
constructor(config: ProviderConfig);
|
|
11
11
|
buildTools(tools: Tool[]): any[];
|
|
12
|
-
chat(messages: ChatMessage[], options?: ChatOptions): Promise<ChatResponse>;
|
|
13
|
-
chatStream(messages: ChatMessage[], options?: ChatOptions): AsyncGenerator<
|
|
12
|
+
chat(messages: ChatMessage[], options?: ChatOptions, tools?: Tool[]): Promise<ChatResponse>;
|
|
13
|
+
chatStream(messages: ChatMessage[], options?: ChatOptions, tools?: Tool[]): AsyncGenerator<StreamChunk, ChatResponse>;
|
|
14
14
|
private buildRequest;
|
|
15
15
|
}
|
|
16
16
|
//# sourceMappingURL=gemini.d.ts.map
|
package/dist/llm/gemini.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../src/llm/gemini.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,WAAW,
|
|
1
|
+
{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../src/llm/gemini.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,WAAW,EAAY,MAAM,aAAa,CAAC;AACpF,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,mBAAmB,CAAC;AAC9C,OAAO,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAE9E,qBAAa,cAAe,YAAW,WAAW;IAChD,QAAQ,CAAC,IAAI,YAAY;IACzB,SAAS,CAAC,MAAM,EAAE,cAAc,CAAC;gBAErB,MAAM,EAAE,cAAc;IAIlC,UAAU,CAAC,KAAK,EAAE,IAAI,EAAE,GAAG,GAAG,EAAE;IAgB1B,IAAI,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,OAAO,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,YAAY,CAAC;IAyC1F,UAAU,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,OAAO,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,EAAE,IAAI,EAAE,GAAG,cAAc,CAAC,WAAW,EAAE,YAAY,CAAC;IAqE5H,OAAO,CAAC,YAAY;CA0BrB"}
|
package/dist/llm/gemini.js
CHANGED
|
@@ -8,18 +8,20 @@ export class GeminiProvider {
|
|
|
8
8
|
this.config = config;
|
|
9
9
|
}
|
|
10
10
|
buildTools(tools) {
|
|
11
|
-
return
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
11
|
+
return [{
|
|
12
|
+
functionDeclarations: tools.map(tool => ({
|
|
13
|
+
name: tool.name,
|
|
14
|
+
description: tool.description,
|
|
15
|
+
parameters: tool.inputSchema || {
|
|
16
|
+
type: 'object',
|
|
17
|
+
properties: Object.fromEntries(tool.parameters.map(p => [p.name, { type: p.type, description: p.description }])),
|
|
18
|
+
required: tool.parameters.filter(p => p.required).map(p => p.name),
|
|
19
|
+
},
|
|
20
|
+
})),
|
|
21
|
+
}];
|
|
20
22
|
}
|
|
21
|
-
async chat(messages, options) {
|
|
22
|
-
const body = this.buildRequest(messages, options);
|
|
23
|
+
async chat(messages, options, tools) {
|
|
24
|
+
const body = this.buildRequest(messages, options, tools);
|
|
23
25
|
const url = `${this.config.baseUrl}/models/${this.config.model}:generateContent?key=${this.config.apiKey}`;
|
|
24
26
|
const response = await fetch(url, {
|
|
25
27
|
method: 'POST',
|
|
@@ -32,7 +34,16 @@ export class GeminiProvider {
|
|
|
32
34
|
}
|
|
33
35
|
const data = await response.json();
|
|
34
36
|
const candidate = data.candidates?.[0];
|
|
35
|
-
const
|
|
37
|
+
const parts = candidate?.content?.parts || [];
|
|
38
|
+
const text = parts.map((p) => p.text).filter(Boolean).join('');
|
|
39
|
+
const functionCalls = parts.filter((p) => p.functionCall).map((p) => ({
|
|
40
|
+
id: `call_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`,
|
|
41
|
+
type: 'function',
|
|
42
|
+
function: {
|
|
43
|
+
name: p.functionCall.name,
|
|
44
|
+
arguments: JSON.stringify(p.functionCall.args || {}),
|
|
45
|
+
},
|
|
46
|
+
}));
|
|
36
47
|
return {
|
|
37
48
|
content: text,
|
|
38
49
|
role: 'assistant',
|
|
@@ -41,11 +52,12 @@ export class GeminiProvider {
|
|
|
41
52
|
completion_tokens: data.usageMetadata.candidatesTokenCount || 0,
|
|
42
53
|
total_tokens: data.usageMetadata.totalTokenCount || 0,
|
|
43
54
|
} : undefined,
|
|
55
|
+
tool_calls: functionCalls.length > 0 ? functionCalls : undefined,
|
|
44
56
|
};
|
|
45
57
|
}
|
|
46
|
-
async *chatStream(messages, options) {
|
|
47
|
-
const body = this.buildRequest(messages, options);
|
|
48
|
-
const url = `${this.config.baseUrl}/models/${this.config.model}:streamGenerateContent?key=${this.config.apiKey}`;
|
|
58
|
+
async *chatStream(messages, options, tools) {
|
|
59
|
+
const body = this.buildRequest(messages, options, tools);
|
|
60
|
+
const url = `${this.config.baseUrl}/models/${this.config.model}:streamGenerateContent?alt=sse&key=${this.config.apiKey}`;
|
|
49
61
|
const response = await fetch(url, {
|
|
50
62
|
method: 'POST',
|
|
51
63
|
headers: { 'Content-Type': 'application/json' },
|
|
@@ -59,49 +71,74 @@ export class GeminiProvider {
|
|
|
59
71
|
const decoder = new TextDecoder();
|
|
60
72
|
let fullContent = '';
|
|
61
73
|
let buffer = '';
|
|
74
|
+
const functionCalls = [];
|
|
62
75
|
try {
|
|
63
76
|
while (true) {
|
|
64
77
|
const { done, value } = await reader.read();
|
|
65
78
|
if (done)
|
|
66
79
|
break;
|
|
67
80
|
buffer += decoder.decode(value, { stream: true });
|
|
68
|
-
|
|
69
|
-
|
|
70
|
-
|
|
81
|
+
const lines = buffer.split('\n');
|
|
82
|
+
buffer = lines.pop() || '';
|
|
83
|
+
for (const line of lines) {
|
|
84
|
+
const trimmed = line.trim();
|
|
85
|
+
if (!trimmed || !trimmed.startsWith('data: '))
|
|
86
|
+
continue;
|
|
71
87
|
try {
|
|
72
|
-
const data = JSON.parse(
|
|
73
|
-
const
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
88
|
+
const data = JSON.parse(trimmed.slice(6));
|
|
89
|
+
const parts = data.candidates?.[0]?.content?.parts || [];
|
|
90
|
+
for (const part of parts) {
|
|
91
|
+
if (part.text) {
|
|
92
|
+
fullContent += part.text;
|
|
93
|
+
yield { type: 'text', content: part.text };
|
|
94
|
+
}
|
|
95
|
+
if (part.functionCall) {
|
|
96
|
+
functionCalls.push({
|
|
97
|
+
id: `call_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`,
|
|
98
|
+
type: 'function',
|
|
99
|
+
function: {
|
|
100
|
+
name: part.functionCall.name,
|
|
101
|
+
arguments: JSON.stringify(part.functionCall.args || {}),
|
|
102
|
+
},
|
|
103
|
+
});
|
|
104
|
+
}
|
|
77
105
|
}
|
|
78
106
|
}
|
|
79
107
|
catch {
|
|
80
108
|
// Skip invalid JSON
|
|
81
109
|
}
|
|
82
|
-
buffer = buffer.slice(jsonMatch.index + jsonMatch[0].length);
|
|
83
110
|
}
|
|
84
111
|
}
|
|
85
112
|
}
|
|
86
113
|
finally {
|
|
87
114
|
reader.releaseLock();
|
|
88
115
|
}
|
|
89
|
-
|
|
116
|
+
if (functionCalls.length > 0) {
|
|
117
|
+
yield { type: 'tool_calls', toolCalls: functionCalls };
|
|
118
|
+
}
|
|
119
|
+
return { content: fullContent, role: 'assistant', tool_calls: functionCalls.length > 0 ? functionCalls : undefined };
|
|
90
120
|
}
|
|
91
|
-
buildRequest(messages, options) {
|
|
92
|
-
const
|
|
93
|
-
|
|
94
|
-
|
|
121
|
+
buildRequest(messages, options, tools) {
|
|
122
|
+
const systemMsg = messages.find(m => m.role === 'system');
|
|
123
|
+
const nonSystemMsgs = messages.filter(m => m.role !== 'system');
|
|
124
|
+
const contents = nonSystemMsgs.map(m => ({
|
|
95
125
|
role: m.role === 'assistant' ? 'model' : 'user',
|
|
96
126
|
parts: [{ text: m.content }],
|
|
97
127
|
}));
|
|
98
|
-
|
|
128
|
+
const body = {
|
|
99
129
|
contents,
|
|
130
|
+
...(systemMsg && {
|
|
131
|
+
systemInstruction: { parts: [{ text: systemMsg.content }] },
|
|
132
|
+
}),
|
|
100
133
|
generationConfig: {
|
|
101
134
|
temperature: options?.temperature ?? this.config.temperature,
|
|
102
135
|
maxOutputTokens: options?.maxTokens ?? this.config.maxTokens,
|
|
103
136
|
},
|
|
104
137
|
};
|
|
138
|
+
if (tools && tools.length > 0) {
|
|
139
|
+
body.tools = this.buildTools(tools);
|
|
140
|
+
}
|
|
141
|
+
return body;
|
|
105
142
|
}
|
|
106
143
|
}
|
|
107
144
|
//# sourceMappingURL=gemini.js.map
|
package/dist/llm/gemini.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../src/llm/gemini.ts"],"names":[],"mappings":"AAAA;;GAEG;AAMH,MAAM,OAAO,cAAc;IAChB,IAAI,GAAG,QAAQ,CAAC;
|
|
1
|
+
{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../src/llm/gemini.ts"],"names":[],"mappings":"AAAA;;GAEG;AAMH,MAAM,OAAO,cAAc;IAChB,IAAI,GAAG,QAAQ,CAAC;IACf,MAAM,CAAiB;IAEjC,YAAY,MAAsB;QAChC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,UAAU,CAAC,KAAa;QACtB,OAAO,CAAC;gBACN,oBAAoB,EAAE,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBACvC,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,WAAW,EAAE,IAAI,CAAC,WAAW;oBAC7B,UAAU,EAAE,IAAI,CAAC,WAAW,IAAI;wBAC9B,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,MAAM,CAAC,WAAW,CAC5B,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CACjF;wBACD,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;qBACnE;iBACF,CAAC,CAAC;aACJ,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,QAAuB,EAAE,OAAqB,EAAE,KAAc;QACvE,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;QACzD,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,WAAW,IAAI,CAAC,MAAM,CAAC,KAAK,wBAAwB,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QAE3G,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAChC,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;YACnD,MAAM,IAAI,KAAK,CAAC,oBAAoB,QAAQ,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC,CAAC;QAClE,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnC,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,KAAK,GAAG,SAAS,EAAE,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC;QAC9C,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEpE,MAAM,aAAa,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;YAC9E,EAAE,EAAE,QAAQ,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;YAClE,IAAI,EAAE,UAAmB;YACzB,QAAQ,EAAE;gBACR,IAAI,EAAE,CAAC,CAAC,YAAY,CAAC,IAAI;gBACzB,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,IAAI,EAAE,CAAC;aACrD;SACF,CAAC,CAAC,CAAC;QAEJ,OAAO;YACL,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,WAAW;YACjB,KAAK,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC;gBAC1B,aAAa,EAAE,IAAI,CAAC,aAAa,CAAC,gBAAgB,IAAI,CAAC;gBACvD,iBAAiB,EAAE,IAAI,CAAC,aAAa,CAAC,oBAAoB,IAAI,CAAC;gBAC/D,YAAY,EAAE,IAAI,CAAC,aAAa,CAAC,eAAe,IAAI,CAAC;aACtD,CAAC,CAAC,CAAC,SAAS;YACb,UAAU,EAAE,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS;SACjE,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,CAAC,UAAU,CAAC,QAAuB,EAAE,OAAqB,EAAE,KAAc;QAC9E,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;QACzD,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,WAAW,IAAI,CAAC,MAAM,CAAC,KAAK,sCAAsC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QAEzH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAChC,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;YACnD,MAAM,IAAI,KAAK,CAAC,oBAAoB,QAAQ,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC,CAAC;QAClE,CAAC;QAED,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;QACzC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;QAClC,IAAI,WAAW,GAAG,EAAE,CAAC;QACrB,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,MAAM,aAAa,GAAe,EAAE,CAAC;QAErC,IAAI,CAAC;YACH,OAAO,IAAI,EAAE,CAAC;gBACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5C,IAAI,IAAI;oBAAE,MAAM;gBAEhB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBAClD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;gBAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACzB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;oBAC5B,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC;wBAAE,SAAS;oBAExD,IAAI,CAAC;wBACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;wBAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC;wBACzD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;4BACzB,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gCACd,WAAW,IAAI,IAAI,CAAC,IAAI,CAAC;gCACzB,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;4BAC7C,CAAC;4BACD,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gCACtB,aAAa,CAAC,IAAI,CAAC;oCACjB,EAAE,EAAE,QAAQ,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;oCAClE,IAAI,EAAE,UAAU;oCAChB,QAAQ,EAAE;wCACR,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;wCAC5B,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,IAAI,EAAE,CAAC;qCACxD;iCACF,CAAC,CAAC;4BACL,CAAC;wBACH,CAAC;oBACH,CAAC;oBAAC,MAAM,CAAC;wBACP,oBAAoB;oBACtB,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,CAAC;QAED,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;QACzD,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,WAAW,EAAE,UAAU,EAAE,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;IACvH,CAAC;IAEO,YAAY,CAAC,QAAuB,EAAE,OAAqB,EAAE,KAAc;QACjF,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QAC1D,MAAM,aAAa,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QAEhE,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACvC,IAAI,EAAE,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;YAC/C,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;SAC7B,CAAC,CAAC,CAAC;QAEJ,MAAM,IAAI,GAAQ;YAChB,QAAQ;YACR,GAAG,CAAC,SAAS,IAAI;gBACf,iBAAiB,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,CAAC,OAAO,EAAE,CAAC,EAAE;aAC5D,CAAC;YACF,gBAAgB,EAAE;gBAChB,WAAW,EAAE,OAAO,EAAE,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW;gBAC5D,eAAe,EAAE,OAAO,EAAE,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS;aAC7D;SACF,CAAC;QAEF,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACtC,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}
|
package/dist/llm/index.d.ts
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
/**
|
|
2
2
|
* LLM Service - Factory for creating providers
|
|
3
|
-
*
|
|
3
|
+
* Supports all providers + custom relay stations (中转站)
|
|
4
4
|
*/
|
|
5
5
|
import type { ChatMessage, ChatResponse, ChatOptions } from '../types.js';
|
|
6
6
|
import type { Tool } from '../tools/types.js';
|
|
@@ -9,39 +9,26 @@ import type { AIConfig } from '../config/types.js';
|
|
|
9
9
|
export declare class LLMService {
|
|
10
10
|
private provider;
|
|
11
11
|
private tools;
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
* Create LLMService from AIConfig
|
|
15
|
-
*/
|
|
12
|
+
private apiKey;
|
|
13
|
+
constructor(provider: LLMProvider, apiKey: string);
|
|
16
14
|
static fromConfig(config: AIConfig): LLMService;
|
|
17
|
-
/**
|
|
18
|
-
* Register tools for function calling
|
|
19
|
-
*/
|
|
20
15
|
registerTools(tools: Tool[]): void;
|
|
21
|
-
/**
|
|
22
|
-
* Get provider name
|
|
23
|
-
*/
|
|
24
16
|
getProviderName(): string;
|
|
25
|
-
|
|
26
|
-
* Non-streaming chat
|
|
27
|
-
*/
|
|
17
|
+
hasApiKey(): boolean;
|
|
28
18
|
chat(messages: ChatMessage[], options?: ChatOptions): Promise<ChatResponse>;
|
|
29
19
|
/**
|
|
30
20
|
* Streaming chat with agent loop (tool call support)
|
|
21
|
+
* - Streams text to the user
|
|
22
|
+
* - Detects structured tool calls from the API
|
|
23
|
+
* - Executes tools and loops back for more
|
|
31
24
|
*/
|
|
32
25
|
chatStream(messages: ChatMessage[], options?: ChatOptions & {
|
|
33
26
|
maxIterations?: number;
|
|
34
27
|
}): AsyncGenerator<string, ChatResponse>;
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
*/
|
|
38
|
-
private hasApiKey;
|
|
39
|
-
/**
|
|
40
|
-
* Mock response when no API key is set
|
|
41
|
-
*/
|
|
42
|
-
private getMockResponse;
|
|
28
|
+
private truncateArgs;
|
|
29
|
+
private getNoKeyMessage;
|
|
43
30
|
}
|
|
44
|
-
export type { LLMProvider, ProviderConfig } from './provider.js';
|
|
31
|
+
export type { LLMProvider, ProviderConfig, StreamChunk } from './provider.js';
|
|
45
32
|
export { OpenAIProvider } from './openai.js';
|
|
46
33
|
export { AnthropicProvider } from './anthropic.js';
|
|
47
34
|
export { GeminiProvider } from './gemini.js';
|
package/dist/llm/index.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/llm/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,WAAW,EAAY,MAAM,aAAa,CAAC;AACpF,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,mBAAmB,CAAC;AAC9C,OAAO,KAAK,EAAE,WAAW,
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/llm/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,WAAW,EAAY,MAAM,aAAa,CAAC;AACpF,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,mBAAmB,CAAC;AAC9C,OAAO,KAAK,EAAE,WAAW,EAAe,MAAM,eAAe,CAAC;AAC9D,OAAO,KAAK,EAAE,QAAQ,EAAgB,MAAM,oBAAoB,CAAC;AAMjE,qBAAa,UAAU;IACrB,OAAO,CAAC,QAAQ,CAAc;IAC9B,OAAO,CAAC,KAAK,CAAgC;IAC7C,OAAO,CAAC,MAAM,CAAS;gBAEX,QAAQ,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM;IAKjD,MAAM,CAAC,UAAU,CAAC,MAAM,EAAE,QAAQ,GAAG,UAAU;IA6B/C,aAAa,CAAC,KAAK,EAAE,IAAI,EAAE,GAAG,IAAI;IAMlC,eAAe,IAAI,MAAM;IACzB,SAAS,IAAI,OAAO;IAEd,IAAI,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,OAAO,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;IAMjF;;;;;OAKG;IACI,UAAU,CACf,QAAQ,EAAE,WAAW,EAAE,EACvB,OAAO,CAAC,EAAE,WAAW,GAAG;QAAE,aAAa,CAAC,EAAE,MAAM,CAAA;KAAE,GACjD,cAAc,CAAC,MAAM,EAAE,YAAY,CAAC;IA2FvC,OAAO,CAAC,YAAY;IAcpB,OAAO,CAAC,eAAe;CAYxB;AAED,YAAY,EAAE,WAAW,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAC9E,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC"}
|