@sulala/agent 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +120 -0
- package/bin/sulala.mjs +7 -0
- package/dist/agent/loop.d.ts +38 -0
- package/dist/agent/loop.d.ts.map +1 -0
- package/dist/agent/loop.js +384 -0
- package/dist/agent/loop.js.map +1 -0
- package/dist/agent/session-queue.d.ts +2 -0
- package/dist/agent/session-queue.d.ts.map +1 -0
- package/dist/agent/session-queue.js +13 -0
- package/dist/agent/session-queue.js.map +1 -0
- package/dist/agent/skill-install.d.ts +41 -0
- package/dist/agent/skill-install.d.ts.map +1 -0
- package/dist/agent/skill-install.js +211 -0
- package/dist/agent/skill-install.js.map +1 -0
- package/dist/agent/skills-config.d.ts +24 -0
- package/dist/agent/skills-config.d.ts.map +1 -0
- package/dist/agent/skills-config.js +126 -0
- package/dist/agent/skills-config.js.map +1 -0
- package/dist/agent/skills-watcher.d.ts +7 -0
- package/dist/agent/skills-watcher.d.ts.map +1 -0
- package/dist/agent/skills-watcher.js +32 -0
- package/dist/agent/skills-watcher.js.map +1 -0
- package/dist/agent/skills.d.ts +32 -0
- package/dist/agent/skills.d.ts.map +1 -0
- package/dist/agent/skills.js +208 -0
- package/dist/agent/skills.js.map +1 -0
- package/dist/agent/skills.test.d.ts +2 -0
- package/dist/agent/skills.test.d.ts.map +1 -0
- package/dist/agent/skills.test.js +59 -0
- package/dist/agent/skills.test.js.map +1 -0
- package/dist/agent/tools.d.ts +8 -0
- package/dist/agent/tools.d.ts.map +1 -0
- package/dist/agent/tools.js +147 -0
- package/dist/agent/tools.js.map +1 -0
- package/dist/ai/orchestrator.d.ts +38 -0
- package/dist/ai/orchestrator.d.ts.map +1 -0
- package/dist/ai/orchestrator.js +360 -0
- package/dist/ai/orchestrator.js.map +1 -0
- package/dist/ai/orchestrator.test.d.ts +2 -0
- package/dist/ai/orchestrator.test.d.ts.map +1 -0
- package/dist/ai/orchestrator.test.js +29 -0
- package/dist/ai/orchestrator.test.js.map +1 -0
- package/dist/cli.d.ts +3 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +278 -0
- package/dist/cli.js.map +1 -0
- package/dist/config.d.ts +4 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +77 -0
- package/dist/config.js.map +1 -0
- package/dist/config.test.d.ts +2 -0
- package/dist/config.test.d.ts.map +1 -0
- package/dist/config.test.js +16 -0
- package/dist/config.test.js.map +1 -0
- package/dist/db/index.d.ts +42 -0
- package/dist/db/index.d.ts.map +1 -0
- package/dist/db/index.js +121 -0
- package/dist/db/index.js.map +1 -0
- package/dist/db/schema.sql +74 -0
- package/dist/gateway/server.d.ts +13 -0
- package/dist/gateway/server.d.ts.map +1 -0
- package/dist/gateway/server.js +566 -0
- package/dist/gateway/server.js.map +1 -0
- package/dist/index.d.ts +2 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +78 -0
- package/dist/index.js.map +1 -0
- package/dist/plugins/index.d.ts +52 -0
- package/dist/plugins/index.d.ts.map +1 -0
- package/dist/plugins/index.js +176 -0
- package/dist/plugins/index.js.map +1 -0
- package/dist/scheduler/cron.d.ts +8 -0
- package/dist/scheduler/cron.d.ts.map +1 -0
- package/dist/scheduler/cron.js +31 -0
- package/dist/scheduler/cron.js.map +1 -0
- package/dist/scheduler/queue.d.ts +13 -0
- package/dist/scheduler/queue.d.ts.map +1 -0
- package/dist/scheduler/queue.js +75 -0
- package/dist/scheduler/queue.js.map +1 -0
- package/dist/scheduler/queue.test.d.ts +2 -0
- package/dist/scheduler/queue.test.d.ts.map +1 -0
- package/dist/scheduler/queue.test.js +41 -0
- package/dist/scheduler/queue.test.js.map +1 -0
- package/dist/types.d.ts +149 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/dist/watcher/index.d.ts +15 -0
- package/dist/watcher/index.d.ts.map +1 -0
- package/dist/watcher/index.js +87 -0
- package/dist/watcher/index.js.map +1 -0
- package/dist/webhooks.d.ts +2 -0
- package/dist/webhooks.d.ts.map +1 -0
- package/dist/webhooks.js +38 -0
- package/dist/webhooks.js.map +1 -0
- package/package.json +62 -0
- package/src/db/schema.sql +74 -0
- package/src/index.ts +83 -0
|
@@ -0,0 +1,360 @@
|
|
|
1
|
+
import { log, saveAiResult } from '../db/index.js';
|
|
2
|
+
const providers = new Map();
|
|
3
|
+
const defaultProvider = process.env.AI_DEFAULT_PROVIDER || 'openai';
|
|
4
|
+
export function registerProvider(name, adapter) {
|
|
5
|
+
providers.set(name, adapter);
|
|
6
|
+
}
|
|
7
|
+
export function getProvider(name) {
|
|
8
|
+
const key = name ?? defaultProvider;
|
|
9
|
+
const p = providers.get(key);
|
|
10
|
+
if (!p)
|
|
11
|
+
throw new Error(`Unknown AI provider: ${key}. Registered: ${[...providers.keys()].join(', ')}`);
|
|
12
|
+
return p;
|
|
13
|
+
}
|
|
14
|
+
/** Stream completion (OpenAI and OpenRouter). Falls back to non-streaming for other providers. */
|
|
15
|
+
export async function completeStream(options, onChunk) {
|
|
16
|
+
const provider = options.provider ?? defaultProvider;
|
|
17
|
+
const messages = options.messages ?? [];
|
|
18
|
+
const model = options.model;
|
|
19
|
+
const max_tokens = options.max_tokens ?? 1024;
|
|
20
|
+
const tools = options.tools;
|
|
21
|
+
const streamClient = provider === 'openai' ? openAiStreamClient : provider === 'openrouter' ? openRouterStreamClient : null;
|
|
22
|
+
const streamDefaultModel = provider === 'openai' ? OPENAI_DEFAULT : provider === 'openrouter' ? OPENROUTER_DEFAULT : '';
|
|
23
|
+
if (!streamClient || !streamDefaultModel) {
|
|
24
|
+
const result = await complete({ ...options, provider, messages, model, max_tokens, tools, signal: options.signal });
|
|
25
|
+
onChunk({ type: 'finish', content: result.content ?? '', tool_calls: result.tool_calls, usage: result.usage });
|
|
26
|
+
return { content: result.content ?? '', tool_calls: result.tool_calls, usage: result.usage };
|
|
27
|
+
}
|
|
28
|
+
const openAiMessages = messages.map((m) => {
|
|
29
|
+
if (!m.tool_calls?.length)
|
|
30
|
+
return m;
|
|
31
|
+
return {
|
|
32
|
+
...m,
|
|
33
|
+
tool_calls: m.tool_calls.map((tc) => ({
|
|
34
|
+
id: tc.id,
|
|
35
|
+
type: 'function',
|
|
36
|
+
function: { name: tc.name, arguments: tc.arguments ?? '' },
|
|
37
|
+
})),
|
|
38
|
+
};
|
|
39
|
+
});
|
|
40
|
+
const createOpts = {
|
|
41
|
+
model: model || streamDefaultModel,
|
|
42
|
+
messages: openAiMessages,
|
|
43
|
+
max_tokens,
|
|
44
|
+
stream: true,
|
|
45
|
+
};
|
|
46
|
+
if (tools?.length) {
|
|
47
|
+
createOpts.tools = tools.map((t) => ({
|
|
48
|
+
type: 'function',
|
|
49
|
+
function: { name: t.name, description: t.description, parameters: t.parameters ?? {} },
|
|
50
|
+
}));
|
|
51
|
+
}
|
|
52
|
+
const createOptions = options.signal ? { signal: options.signal } : {};
|
|
53
|
+
const stream = await streamClient.chat.completions.create(createOpts, createOptions);
|
|
54
|
+
let content = '';
|
|
55
|
+
const toolCallByIndex = {};
|
|
56
|
+
let usage;
|
|
57
|
+
for await (const chunk of stream) {
|
|
58
|
+
const choice = chunk.choices?.[0];
|
|
59
|
+
const delta = choice?.delta;
|
|
60
|
+
if (chunk.usage)
|
|
61
|
+
usage = chunk.usage;
|
|
62
|
+
if (delta?.content) {
|
|
63
|
+
content += delta.content;
|
|
64
|
+
onChunk({ type: 'delta', content: delta.content });
|
|
65
|
+
}
|
|
66
|
+
if (delta?.tool_calls) {
|
|
67
|
+
for (const tc of delta.tool_calls) {
|
|
68
|
+
const idx = tc.index;
|
|
69
|
+
if (!toolCallByIndex[idx])
|
|
70
|
+
toolCallByIndex[idx] = { id: tc.id ?? `tc_${idx}`, name: '', arguments: '' };
|
|
71
|
+
if (tc.id)
|
|
72
|
+
toolCallByIndex[idx].id = tc.id;
|
|
73
|
+
if (tc.function?.name)
|
|
74
|
+
toolCallByIndex[idx].name += tc.function.name;
|
|
75
|
+
if (tc.function?.arguments)
|
|
76
|
+
toolCallByIndex[idx].arguments += tc.function.arguments;
|
|
77
|
+
}
|
|
78
|
+
}
|
|
79
|
+
}
|
|
80
|
+
const orderedToolCalls = Object.keys(toolCallByIndex)
|
|
81
|
+
.map(Number)
|
|
82
|
+
.sort((a, b) => a - b)
|
|
83
|
+
.map((i) => toolCallByIndex[i]);
|
|
84
|
+
onChunk({ type: 'finish', content, tool_calls: orderedToolCalls.length ? orderedToolCalls : undefined, usage });
|
|
85
|
+
return { content, tool_calls: orderedToolCalls.length ? orderedToolCalls : undefined, usage };
|
|
86
|
+
}
|
|
87
|
+
let openAiStreamClient = null;
|
|
88
|
+
let openRouterStreamClient = null;
|
|
89
|
+
export async function complete(options = {}) {
|
|
90
|
+
const { provider = defaultProvider, model, messages, max_tokens = 1024, task_id = null, tools, signal, } = options;
|
|
91
|
+
const start = Date.now();
|
|
92
|
+
const adapter = getProvider(provider);
|
|
93
|
+
const id = `ai_${Date.now()}_${Math.random().toString(36).slice(2, 9)}`;
|
|
94
|
+
let result;
|
|
95
|
+
try {
|
|
96
|
+
result = await adapter.complete({ model, messages: messages ?? [], max_tokens, tools, signal });
|
|
97
|
+
}
|
|
98
|
+
catch (err) {
|
|
99
|
+
log('ai', 'error', `Completion failed: ${err instanceof Error ? err.message : err}`, {
|
|
100
|
+
provider,
|
|
101
|
+
stack: err instanceof Error ? err.stack : undefined,
|
|
102
|
+
});
|
|
103
|
+
throw err;
|
|
104
|
+
}
|
|
105
|
+
const meta = { provider, model, latency_ms: Date.now() - start };
|
|
106
|
+
saveAiResult({
|
|
107
|
+
id,
|
|
108
|
+
provider,
|
|
109
|
+
model: model || adapter.defaultModel,
|
|
110
|
+
task_id,
|
|
111
|
+
request_meta: { messages_count: messages?.length },
|
|
112
|
+
response_meta: meta,
|
|
113
|
+
});
|
|
114
|
+
log('ai', 'info', 'Completion done', meta);
|
|
115
|
+
return { id, ...result, meta };
|
|
116
|
+
}
|
|
117
|
+
function stubAdapter(defaultModel = 'stub') {
|
|
118
|
+
return {
|
|
119
|
+
defaultModel,
|
|
120
|
+
async complete({ messages }) {
|
|
121
|
+
const last = messages?.[messages.length - 1];
|
|
122
|
+
const content = typeof last?.content === 'string' ? last.content : '';
|
|
123
|
+
return {
|
|
124
|
+
content: `[stub] You said: ${content.slice(0, 80)}...`,
|
|
125
|
+
usage: { prompt_tokens: 0, completion_tokens: 0 },
|
|
126
|
+
};
|
|
127
|
+
},
|
|
128
|
+
};
|
|
129
|
+
}
|
|
130
|
+
const OPENAI_DEFAULT = process.env.AI_OPENAI_DEFAULT_MODEL || 'gpt-4o-mini';
|
|
131
|
+
const OPENROUTER_DEFAULT = process.env.AI_OPENROUTER_DEFAULT_MODEL || 'openai/gpt-4o-mini';
|
|
132
|
+
registerProvider('stub', stubAdapter('stub'));
|
|
133
|
+
if (process.env.OPENAI_API_KEY) {
|
|
134
|
+
try {
|
|
135
|
+
const mod = await import('openai').catch(() => ({}));
|
|
136
|
+
const OpenAI = mod.OpenAI;
|
|
137
|
+
if (OpenAI) {
|
|
138
|
+
const client = new OpenAI({ apiKey: process.env.OPENAI_API_KEY });
|
|
139
|
+
openAiStreamClient = client;
|
|
140
|
+
registerProvider('openai', {
|
|
141
|
+
defaultModel: OPENAI_DEFAULT,
|
|
142
|
+
async complete({ model, messages, max_tokens, tools: toolsOpt, signal }) {
|
|
143
|
+
// OpenAI expects tool_calls as { id, type: "function", function: { name, arguments } }
|
|
144
|
+
const openAiMessages = (messages ?? []).map((m) => {
|
|
145
|
+
if (!m.tool_calls?.length)
|
|
146
|
+
return m;
|
|
147
|
+
return {
|
|
148
|
+
...m,
|
|
149
|
+
tool_calls: m.tool_calls.map((tc) => ({
|
|
150
|
+
id: tc.id,
|
|
151
|
+
type: 'function',
|
|
152
|
+
function: { name: tc.name, arguments: tc.arguments ?? '' },
|
|
153
|
+
})),
|
|
154
|
+
};
|
|
155
|
+
});
|
|
156
|
+
const createOpts = {
|
|
157
|
+
model: model || OPENAI_DEFAULT,
|
|
158
|
+
messages: openAiMessages,
|
|
159
|
+
max_tokens: max_tokens || 1024,
|
|
160
|
+
};
|
|
161
|
+
if (toolsOpt?.length) {
|
|
162
|
+
createOpts.tools = toolsOpt.map((t) => ({
|
|
163
|
+
type: 'function',
|
|
164
|
+
function: { name: t.name, description: t.description, parameters: t.parameters ?? {} },
|
|
165
|
+
}));
|
|
166
|
+
}
|
|
167
|
+
const res = await client.chat.completions.create(createOpts, signal ? { signal } : undefined);
|
|
168
|
+
const choice = res.choices?.[0];
|
|
169
|
+
const msg = choice?.message;
|
|
170
|
+
const tool_calls = msg?.tool_calls?.map((tc) => ({
|
|
171
|
+
id: tc.id,
|
|
172
|
+
name: tc.function?.name ?? '',
|
|
173
|
+
arguments: tc.function?.arguments ?? '',
|
|
174
|
+
}));
|
|
175
|
+
return {
|
|
176
|
+
content: msg?.content ?? '',
|
|
177
|
+
usage: res.usage ?? {},
|
|
178
|
+
...(tool_calls?.length ? { tool_calls } : {}),
|
|
179
|
+
};
|
|
180
|
+
},
|
|
181
|
+
});
|
|
182
|
+
}
|
|
183
|
+
}
|
|
184
|
+
catch {
|
|
185
|
+
// import failed
|
|
186
|
+
}
|
|
187
|
+
}
|
|
188
|
+
if (process.env.OPENROUTER_API_KEY) {
|
|
189
|
+
try {
|
|
190
|
+
const mod = await import('openai').catch(() => ({}));
|
|
191
|
+
const OpenAI = mod.OpenAI;
|
|
192
|
+
if (OpenAI) {
|
|
193
|
+
const client = new OpenAI({
|
|
194
|
+
apiKey: process.env.OPENROUTER_API_KEY,
|
|
195
|
+
baseURL: 'https://openrouter.ai/api/v1',
|
|
196
|
+
});
|
|
197
|
+
openRouterStreamClient = client;
|
|
198
|
+
registerProvider('openrouter', {
|
|
199
|
+
defaultModel: OPENROUTER_DEFAULT,
|
|
200
|
+
async complete({ model, messages, max_tokens, tools: toolsOpt, signal }) {
|
|
201
|
+
const openAiMessages = (messages ?? []).map((m) => {
|
|
202
|
+
if (!m.tool_calls?.length)
|
|
203
|
+
return m;
|
|
204
|
+
return {
|
|
205
|
+
...m,
|
|
206
|
+
tool_calls: m.tool_calls.map((tc) => ({
|
|
207
|
+
id: tc.id,
|
|
208
|
+
type: 'function',
|
|
209
|
+
function: { name: tc.name, arguments: tc.arguments ?? '' },
|
|
210
|
+
})),
|
|
211
|
+
};
|
|
212
|
+
});
|
|
213
|
+
const createOpts = {
|
|
214
|
+
model: model || OPENROUTER_DEFAULT,
|
|
215
|
+
messages: openAiMessages,
|
|
216
|
+
max_tokens: max_tokens || 1024,
|
|
217
|
+
};
|
|
218
|
+
if (toolsOpt?.length) {
|
|
219
|
+
createOpts.tools = toolsOpt.map((t) => ({
|
|
220
|
+
type: 'function',
|
|
221
|
+
function: { name: t.name, description: t.description, parameters: t.parameters ?? {} },
|
|
222
|
+
}));
|
|
223
|
+
}
|
|
224
|
+
const res = await client.chat.completions.create(createOpts, signal ? { signal } : undefined);
|
|
225
|
+
const choice = res.choices?.[0];
|
|
226
|
+
const msg = choice?.message;
|
|
227
|
+
const tool_calls = msg?.tool_calls?.map((tc) => ({
|
|
228
|
+
id: tc.id,
|
|
229
|
+
name: tc.function?.name ?? '',
|
|
230
|
+
arguments: tc.function?.arguments ?? '',
|
|
231
|
+
}));
|
|
232
|
+
return {
|
|
233
|
+
content: msg?.content ?? '',
|
|
234
|
+
usage: res.usage ?? {},
|
|
235
|
+
...(tool_calls?.length ? { tool_calls } : {}),
|
|
236
|
+
};
|
|
237
|
+
},
|
|
238
|
+
});
|
|
239
|
+
}
|
|
240
|
+
}
|
|
241
|
+
catch {
|
|
242
|
+
// import failed
|
|
243
|
+
}
|
|
244
|
+
}
|
|
245
|
+
if (!providers.has('openai')) {
|
|
246
|
+
registerProvider('openai', stubAdapter(OPENAI_DEFAULT));
|
|
247
|
+
}
|
|
248
|
+
const CLAUDE_DEFAULT = process.env.AI_CLAUDE_DEFAULT_MODEL || 'claude-sonnet-4-6';
|
|
249
|
+
if (process.env.ANTHROPIC_API_KEY) {
|
|
250
|
+
registerProvider('claude', {
|
|
251
|
+
defaultModel: CLAUDE_DEFAULT,
|
|
252
|
+
async complete({ model, messages, max_tokens }) {
|
|
253
|
+
const key = process.env.ANTHROPIC_API_KEY;
|
|
254
|
+
const systemMsg = messages.find((m) => m.role === 'system');
|
|
255
|
+
const chatMessages = messages.filter((m) => m.role !== 'system').map((m) => ({
|
|
256
|
+
role: m.role === 'assistant' ? 'assistant' : 'user',
|
|
257
|
+
content: typeof m.content === 'string' ? m.content : JSON.stringify(m.content),
|
|
258
|
+
}));
|
|
259
|
+
const body = {
|
|
260
|
+
model: model || CLAUDE_DEFAULT,
|
|
261
|
+
max_tokens: max_tokens || 1024,
|
|
262
|
+
messages: chatMessages,
|
|
263
|
+
...(systemMsg && { system: typeof systemMsg.content === 'string' ? systemMsg.content : JSON.stringify(systemMsg.content) }),
|
|
264
|
+
};
|
|
265
|
+
const res = await fetch('https://api.anthropic.com/v1/messages', {
|
|
266
|
+
method: 'POST',
|
|
267
|
+
headers: {
|
|
268
|
+
'Content-Type': 'application/json',
|
|
269
|
+
'x-api-key': key,
|
|
270
|
+
'anthropic-version': '2023-06-01',
|
|
271
|
+
},
|
|
272
|
+
body: JSON.stringify(body),
|
|
273
|
+
});
|
|
274
|
+
if (!res.ok) {
|
|
275
|
+
const err = await res.text();
|
|
276
|
+
throw new Error(`Anthropic API: ${res.status} ${err}`);
|
|
277
|
+
}
|
|
278
|
+
const data = (await res.json());
|
|
279
|
+
const content = data.content?.find((c) => c.type === 'text')?.text || '';
|
|
280
|
+
const usage = data.usage || {};
|
|
281
|
+
return { content, usage: { prompt_tokens: usage.input_tokens ?? 0, completion_tokens: usage.output_tokens ?? 0 } };
|
|
282
|
+
},
|
|
283
|
+
});
|
|
284
|
+
}
|
|
285
|
+
else {
|
|
286
|
+
registerProvider('claude', stubAdapter(CLAUDE_DEFAULT));
|
|
287
|
+
}
|
|
288
|
+
const GEMINI_DEFAULT = process.env.AI_GEMINI_DEFAULT_MODEL || 'gemini-2.5-flash';
|
|
289
|
+
const GEMINI_KEY = process.env.GOOGLE_GEMINI_API_KEY || process.env.GEMINI_API_KEY;
|
|
290
|
+
if (GEMINI_KEY) {
|
|
291
|
+
registerProvider('gemini', {
|
|
292
|
+
defaultModel: GEMINI_DEFAULT,
|
|
293
|
+
async complete({ model, messages, max_tokens }) {
|
|
294
|
+
const modelId = model || GEMINI_DEFAULT;
|
|
295
|
+
const url = `https://generativelanguage.googleapis.com/v1beta/models/${modelId}:generateContent?key=${GEMINI_KEY}`;
|
|
296
|
+
const contents = messages
|
|
297
|
+
.filter((m) => m.role !== 'system')
|
|
298
|
+
.map((m) => ({
|
|
299
|
+
role: (m.role === 'assistant' ? 'model' : 'user'),
|
|
300
|
+
parts: [{ text: typeof m.content === 'string' ? m.content : JSON.stringify(m.content) }],
|
|
301
|
+
}));
|
|
302
|
+
const systemInstruction = messages.find((m) => m.role === 'system');
|
|
303
|
+
const body = {
|
|
304
|
+
contents: contents.length ? contents : [{ role: 'user', parts: [{ text: '' }] }],
|
|
305
|
+
generationConfig: { maxOutputTokens: max_tokens || 1024 },
|
|
306
|
+
...(systemInstruction && {
|
|
307
|
+
systemInstruction: { parts: [{ text: typeof systemInstruction.content === 'string' ? systemInstruction.content : JSON.stringify(systemInstruction.content) }] },
|
|
308
|
+
}),
|
|
309
|
+
};
|
|
310
|
+
const res = await fetch(url, {
|
|
311
|
+
method: 'POST',
|
|
312
|
+
headers: { 'Content-Type': 'application/json' },
|
|
313
|
+
body: JSON.stringify(body),
|
|
314
|
+
});
|
|
315
|
+
if (!res.ok) {
|
|
316
|
+
const err = await res.text();
|
|
317
|
+
throw new Error(`Gemini API: ${res.status} ${err}`);
|
|
318
|
+
}
|
|
319
|
+
const data = (await res.json());
|
|
320
|
+
const text = data.candidates?.[0]?.content?.parts?.[0]?.text ?? '';
|
|
321
|
+
const usage = data.usageMetadata || {};
|
|
322
|
+
return {
|
|
323
|
+
content: text,
|
|
324
|
+
usage: { prompt_tokens: usage.promptTokenCount ?? 0, completion_tokens: usage.candidatesTokenCount ?? 0 },
|
|
325
|
+
};
|
|
326
|
+
},
|
|
327
|
+
});
|
|
328
|
+
}
|
|
329
|
+
else {
|
|
330
|
+
registerProvider('gemini', stubAdapter(GEMINI_DEFAULT));
|
|
331
|
+
}
|
|
332
|
+
const OLLAMA_DEFAULT = process.env.AI_OLLAMA_DEFAULT_MODEL || 'llama3.2';
|
|
333
|
+
const OLLAMA_BASE = process.env.OLLAMA_BASE_URL || 'http://localhost:11434';
|
|
334
|
+
registerProvider('ollama', {
|
|
335
|
+
defaultModel: OLLAMA_DEFAULT,
|
|
336
|
+
async complete({ model, messages, max_tokens }) {
|
|
337
|
+
const url = `${OLLAMA_BASE}/api/chat`;
|
|
338
|
+
const body = {
|
|
339
|
+
model: model || OLLAMA_DEFAULT,
|
|
340
|
+
messages: messages.map((m) => ({ role: m.role, content: typeof m.content === 'string' ? m.content : JSON.stringify(m.content) })),
|
|
341
|
+
options: { num_predict: max_tokens || 1024 },
|
|
342
|
+
stream: false,
|
|
343
|
+
};
|
|
344
|
+
const res = await fetch(url, {
|
|
345
|
+
method: 'POST',
|
|
346
|
+
headers: { 'Content-Type': 'application/json' },
|
|
347
|
+
body: JSON.stringify(body),
|
|
348
|
+
});
|
|
349
|
+
if (!res.ok) {
|
|
350
|
+
const err = await res.text();
|
|
351
|
+
throw new Error(`Ollama: ${res.status} ${err}`);
|
|
352
|
+
}
|
|
353
|
+
const data = (await res.json());
|
|
354
|
+
const content = data.message?.content || '';
|
|
355
|
+
return { content, usage: data.eval_count != null ? { completion_tokens: data.eval_count } : undefined };
|
|
356
|
+
},
|
|
357
|
+
});
|
|
358
|
+
if (!providers.has('llama'))
|
|
359
|
+
registerProvider('llama', getProvider('ollama'));
|
|
360
|
+
//# sourceMappingURL=orchestrator.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"orchestrator.js","sourceRoot":"","sources":["../../src/ai/orchestrator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAGnD,MAAM,SAAS,GAAG,IAAI,GAAG,EAAqB,CAAC;AAC/C,MAAM,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,QAAQ,CAAC;AAEpE,MAAM,UAAU,gBAAgB,CAAC,IAAY,EAAE,OAAkB;IAC/D,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;AAC/B,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,IAAoB;IAC9C,MAAM,GAAG,GAAG,IAAI,IAAI,eAAe,CAAC;IACpC,MAAM,CAAC,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAC7B,IAAI,CAAC,CAAC;QAAE,MAAM,IAAI,KAAK,CAAC,wBAAwB,GAAG,iBAAiB,CAAC,GAAG,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACxG,OAAO,CAAC,CAAC;AACX,CAAC;AAMD,kGAAkG;AAClG,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,OAAwB,EACxB,OAAuC;IAEvC,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,eAAe,CAAC;IACrD,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC;IACxC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;IAC5B,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,IAAI,CAAC;IAC9C,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;IAE5B,MAAM,YAAY,GAAG,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,QAAQ,KAAK,YAAY,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,IAAI,CAAC;IAC5H,MAAM,kBAAkB,GAAG,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,QAAQ,KAAK,YAAY,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,CAAC;IACxH,IAAI,CAAC,YAAY,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACzC,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,EAAE,GAAG,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QACpH,OAAO,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,EAAE,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;QAC/G,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,EAAE,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;IAC/F,CAAC;IAED,MAAM,cAAc,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QACxC,IAAI,CAAC,CAAC,CAAC,UAAU,EAAE,MAAM;YAAE,OAAO,CAAC,CAAC;QACpC,OAAO;YACL,GAAG,CAAC;YACJ,UAAU,EAAE,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBACpC,EAAE,EAAE,EAAE,CAAC,EAAE;gBACT,IAAI,EAAE,UAAmB;gBACzB,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,SAAS,IAAI,EAAE,EAAE;aAC3D,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC,CAAC,CAAC;IACH,MAAM,UAAU,GAA4B;QAC1C,KAAK,EAAE,KAAK,IAAI,kBAAkB;QAClC,QAAQ,EAAE,cAAc;QACxB,UAAU;QACV,MAAM,EAAE,IAAI;KACb,CAAC;IACF,IAAI,KAAK,EAAE,MAAM,EAAE,CAAC;QAClB,UAAU,CAAC,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACnC,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,CAAC,UAAU,IAAI,EAAE,EAAE;SACvF,CAAC,CAAC,CAAC;IACN,CAAC;IAED,MAAM,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IACvE,MAAM,MAAM,GAAG,MAAO,YAAY,CAAC,IAAI,CAAC,WAAW,CAAC,MAA+F,CAAC,UAAU,EAAE,aAAa,CAG3K,CAAC;IACH,IAAI,OAAO,GAAG,EAAE,CAAC;IACjB,MAAM,eAAe,GAAoE,EAAE,CAAC;IAC5F,IAAI,KAAyC,CAAC;IAE9C,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QACjC,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QAClC,MAAM,KAAK,GAAG,MAAM,EAAE,KAAK,CAAC;QAC5B,IAAK,KAA4C,CAAC,KAAK;YAAE,KAAK,GAAI,KAA2C,CAAC,KAAK,CAAC;QACpH,IAAI,KAAK,EAAE,OAAO,EAAE,CAAC;YACnB,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC;YACzB,OAAO,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QACrD,CAAC;QACD,IAAI,KAAK,EAAE,UAAU,EAAE,CAAC;YACtB,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;gBAClC,MAAM,GAAG,GAAG,EAAE,CAAC,KAAK,CAAC;gBACrB,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC;oBAAE,eAAe,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,MAAM,GAAG,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;gBACxG,IAAI,EAAE,CAAC,EAAE;oBAAE,eAAe,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;gBAC3C,IAAI,EAAE,CAAC,QAAQ,EAAE,IAAI;oBAAE,eAAe,CAAC,GAAG,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACrE,IAAI,EAAE,CAAC,QAAQ,EAAE,SAAS;oBAAE,eAAe,CAAC,GAAG,CAAC,CAAC,SAAS,IAAI,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;YACtF,CAAC;QACH,CAAC;IACH,CAAC;IACD,MAAM,gBAAgB,GAAG,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC;SAClD,GAAG,CAAC,MAAM,CAAC;SACX,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;SACrB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;IAClC,OAAO,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;IAChH,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,SAAS,EAAE,KAAK,EAAE,CAAC;AAChG,CAAC;AAGD,IAAI,kBAAkB,GAA8B,IAAI,CAAC;AACzD,IAAI,sBAAsB,GAA8B,IAAI,CAAC;AAE7D,MAAM,CAAC,KAAK,UAAU,QAAQ,CAAC,UAA2B,EAAE;IAO1D,MAAM,EACJ,QAAQ,GAAG,eAAe,EAC1B,KAAK,EACL,QAAQ,EACR,UAAU,GAAG,IAAI,EACjB,OAAO,GAAG,IAAI,EACd,KAAK,EACL,MAAM,GACP,GAAG,OAAO,CAAC;IACZ,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACzB,MAAM,OAAO,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;IACtC,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;IACxE,IAAI,MAAM,CAAC;IACX,IAAI,CAAC;QACH,MAAM,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,IAAI,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;IAClG,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,sBAAsB,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,EAAE;YACnF,QAAQ;YACR,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;SACpD,CAAC,CAAC;QACH,MAAM,GAAG,CAAC;IACZ,CAAC;IACD,MAAM,IAAI,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,EAAE,CAAC;IACjE,YAAY,CAAC;QACX,EAAE;QACF,QAAQ;QACR,KAAK,EAAE,KAAK,IAAI,OAAO,CAAC,YAAY;QACpC,OAAO;QACP,YAAY,EAAE,EAAE,cAAc,EAAE,QAAQ,EAAE,MAAM,EAAE;QAClD,aAAa,EAAE,IAAI;KACpB,CAAC,CAAC;IACH,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,iBAAiB,EAAE,IAAI,CAAC,CAAC;IAC3C,OAAO,EAAE,EAAE,EAAE,GAAG,MAAM,EAAE,IAAI,EAAE,CAAC;AACjC,CAAC;AAED,SAAS,WAAW,CAAC,YAAY,GAAG,MAAM;IACxC,OAAO;QACL,YAAY;QACZ,KAAK,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE;YACzB,MAAM,IAAI,GAAG,QAAQ,EAAE,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC7C,MAAM,OAAO,GAAG,OAAO,IAAI,EAAE,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;YACtE,OAAO;gBACL,OAAO,EAAE,oBAAoB,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK;gBACtD,KAAK,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE;aAClD,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,aAAa,CAAC;AAC5E,MAAM,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,2BAA2B,IAAI,oBAAoB,CAAC;AAE3F,gBAAgB,CAAC,MAAM,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;AAC9C,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC;IAC/B,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACrD,MAAM,MAAM,GAAI,GAAwM,CAAC,MAAM,CAAC;QAChO,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC,CAAC;YAClE,kBAAkB,GAAG,MAAuC,CAAC;YAC7D,gBAAgB,CAAC,QAAQ,EAAE;gBACzB,YAAY,EAAE,cAAc;gBAC5B,KAAK,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE;oBACrE,uFAAuF;oBACvF,MAAM,cAAc,GAAG,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;wBAChD,IAAI,CAAC,CAAC,CAAC,UAAU,EAAE,MAAM;4BAAE,OAAO,CAAC,CAAC;wBACpC,OAAO;4BACL,GAAG,CAAC;4BACJ,UAAU,EAAE,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gCACpC,EAAE,EAAE,EAAE,CAAC,EAAE;gCACT,IAAI,EAAE,UAAmB;gCACzB,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,SAAS,IAAI,EAAE,EAAE;6BAC3D,CAAC,CAAC;yBACJ,CAAC;oBACJ,CAAC,CAAC,CAAC;oBACH,MAAM,UAAU,GAA4B;wBAC1C,KAAK,EAAE,KAAK,IAAI,cAAc;wBAC9B,QAAQ,EAAE,cAAc;wBACxB,UAAU,EAAE,UAAU,IAAI,IAAI;qBAC/B,CAAC;oBACF,IAAI,QAAQ,EAAE,MAAM,EAAE,CAAC;wBACrB,UAAU,CAAC,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;4BACtC,IAAI,EAAE,UAAU;4BAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,CAAC,UAAU,IAAI,EAAE,EAAE;yBACvF,CAAC,CAAC,CAAC;oBACN,CAAC;oBACD,MAAM,GAAG,GAAG,MAAO,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAgF,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS,CAA0K,CAAC;oBAClV,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;oBAChC,MAAM,GAAG,GAAG,MAAM,EAAE,OAAO,CAAC;oBAC5B,MAAM,UAAU,GAAG,GAAG,EAAE,UAAU,EAAE,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;wBAC/C,EAAE,EAAE,EAAE,CAAC,EAAE;wBACT,IAAI,EAAE,EAAE,CAAC,QAAQ,EAAE,IAAI,IAAI,EAAE;wBAC7B,SAAS,EAAE,EAAE,CAAC,QAAQ,EAAE,SAAS,IAAI,EAAE;qBACxC,CAAC,CAAC,CAAC;oBACJ,OAAO;wBACL,OAAO,EAAE,GAAG,EAAE,OAAO,IAAI,EAAE;wBAC3B,KAAK,EAAE,GAAG,CAAC,KAAK,IAAI,EAAE;wBACtB,GAAG,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBAC9C,CAAC;gBACJ,CAAC;aACF,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,gBAAgB;IAClB,CAAC;AACH,CAAC;AAED,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,CAAC;IACnC,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACrD,MAAM,MAAM,GAAI,GAA6S,CAAC,MAAM,CAAC;QACrU,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC;gBACxB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,kBAAkB;gBACtC,OAAO,EAAE,8BAA8B;aACxC,CAAC,CAAC;YACH,sBAAsB,GAAG,MAAuC,CAAC;YACjE,gBAAgB,CAAC,YAAY,EAAE;gBAC7B,YAAY,EAAE,kBAAkB;gBAChC,KAAK,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE;oBACrE,MAAM,cAAc,GAAG,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;wBAChD,IAAI,CAAC,CAAC,CAAC,UAAU,EAAE,MAAM;4BAAE,OAAO,CAAC,CAAC;wBACpC,OAAO;4BACL,GAAG,CAAC;4BACJ,UAAU,EAAE,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gCACpC,EAAE,EAAE,EAAE,CAAC,EAAE;gCACT,IAAI,EAAE,UAAmB;gCACzB,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,SAAS,IAAI,EAAE,EAAE;6BAC3D,CAAC,CAAC;yBACJ,CAAC;oBACJ,CAAC,CAAC,CAAC;oBACH,MAAM,UAAU,GAA4B;wBAC1C,KAAK,EAAE,KAAK,IAAI,kBAAkB;wBAClC,QAAQ,EAAE,cAAc;wBACxB,UAAU,EAAE,UAAU,IAAI,IAAI;qBAC/B,CAAC;oBACF,IAAI,QAAQ,EAAE,MAAM,EAAE,CAAC;wBACrB,UAAU,CAAC,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;4BACtC,IAAI,EAAE,UAAU;4BAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,CAAC,UAAU,IAAI,EAAE,EAAE;yBACvF,CAAC,CAAC,CAAC;oBACN,CAAC;oBACD,MAAM,GAAG,GAAG,MAAO,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAgF,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS,CAA0K,CAAC;oBAClV,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;oBAChC,MAAM,GAAG,GAAG,MAAM,EAAE,OAAO,CAAC;oBAC5B,MAAM,UAAU,GAAG,GAAG,EAAE,UAAU,EAAE,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;wBAC/C,EAAE,EAAE,EAAE,CAAC,EAAE;wBACT,IAAI,EAAE,EAAE,CAAC,QAAQ,EAAE,IAAI,IAAI,EAAE;wBAC7B,SAAS,EAAE,EAAE,CAAC,QAAQ,EAAE,SAAS,IAAI,EAAE;qBACxC,CAAC,CAAC,CAAC;oBACJ,OAAO;wBACL,OAAO,EAAE,GAAG,EAAE,OAAO,IAAI,EAAE;wBAC3B,KAAK,EAAE,GAAG,CAAC,KAAK,IAAI,EAAE;wBACtB,GAAG,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBAC9C,CAAC;gBACJ,CAAC;aACF,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,gBAAgB;IAClB,CAAC;AACH,CAAC;AAED,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;IAC7B,gBAAgB,CAAC,QAAQ,EAAE,WAAW,CAAC,cAAc,CAAC,CAAC,CAAC;AAC1D,CAAC;AAED,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,mBAAmB,CAAC;AAClF,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC;IAClC,gBAAgB,CAAC,QAAQ,EAAE;QACzB,YAAY,EAAE,cAAc;QAC5B,KAAK,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE;YAC5C,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,iBAAkB,CAAC;YAC3C,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;YAC5D,MAAM,YAAY,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC3E,IAAI,EAAE,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,WAAoB,CAAC,CAAC,CAAC,MAAe;gBACrE,OAAO,EAAE,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC;aAC/E,CAAC,CAAC,CAAC;YACJ,MAAM,IAAI,GAAG;gBACX,KAAK,EAAE,KAAK,IAAI,cAAc;gBAC9B,UAAU,EAAE,UAAU,IAAI,IAAI;gBAC9B,QAAQ,EAAE,YAAY;gBACtB,GAAG,CAAC,SAAS,IAAI,EAAE,MAAM,EAAE,OAAO,SAAS,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;aAC5H,CAAC;YACF,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,uCAAuC,EAAE;gBAC/D,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,WAAW,EAAE,GAAG;oBAChB,mBAAmB,EAAE,YAAY;iBAClC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,CAAC,CAAC;YACH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;gBACZ,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CAAC,kBAAkB,GAAG,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC,CAAC;YACzD,CAAC;YACD,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA+G,CAAC;YAC9I,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;YACzE,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;YAC/B,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,aAAa,EAAE,KAAK,CAAC,YAAY,IAAI,CAAC,EAAE,iBAAiB,EAAE,KAAK,CAAC,aAAa,IAAI,CAAC,EAAE,EAAE,CAAC;QACrH,CAAC;KACF,CAAC,CAAC;AACL,CAAC;KAAM,CAAC;IACN,gBAAgB,CAAC,QAAQ,EAAE,WAAW,CAAC,cAAc,CAAC,CAAC,CAAC;AAC1D,CAAC;AAED,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,kBAAkB,CAAC;AACjF,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;AACnF,IAAI,UAAU,EAAE,CAAC;IACf,gBAAgB,CAAC,QAAQ,EAAE;QACzB,YAAY,EAAE,cAAc;QAC5B,KAAK,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE;YAC5C,MAAM,OAAO,GAAG,KAAK,IAAI,cAAc,CAAC;YACxC,MAAM,GAAG,GAAG,2DAA2D,OAAO,wBAAwB,UAAU,EAAE,CAAC;YACnH,MAAM,QAAQ,GAAG,QAAQ;iBACtB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;iBAClC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACX,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAqB;gBACrE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;aACzF,CAAC,CAAC,CAAC;YACN,MAAM,iBAAiB,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;YACpE,MAAM,IAAI,GAAG;gBACX,QAAQ,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;gBACzF,gBAAgB,EAAE,EAAE,eAAe,EAAE,UAAU,IAAI,IAAI,EAAE;gBACzD,GAAG,CAAC,iBAAiB,IAAI;oBACvB,iBAAiB,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,iBAAiB,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE;iBAChK,CAAC;aACH,CAAC;YACF,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAC3B,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,CAAC,CAAC;YACH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;gBACZ,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CAAC,eAAe,GAAG,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC,CAAC;YACtD,CAAC;YACD,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAG7B,CAAC;YACF,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;YACnE,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,IAAI,EAAE,CAAC;YACvC,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,KAAK,EAAE,EAAE,aAAa,EAAE,KAAK,CAAC,gBAAgB,IAAI,CAAC,EAAE,iBAAiB,EAAE,KAAK,CAAC,oBAAoB,IAAI,CAAC,EAAE;aAC1G,CAAC;QACJ,CAAC;KACF,CAAC,CAAC;AACL,CAAC;KAAM,CAAC;IACN,gBAAgB,CAAC,QAAQ,EAAE,WAAW,CAAC,cAAc,CAAC,CAAC,CAAC;AAC1D,CAAC;AAED,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,UAAU,CAAC;AACzE,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,wBAAwB,CAAC;AAC5E,gBAAgB,CAAC,QAAQ,EAAE;IACzB,YAAY,EAAE,cAAc;IAC5B,KAAK,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE;QAC5C,MAAM,GAAG,GAAG,GAAG,WAAW,WAAW,CAAC;QACtC,MAAM,IAAI,GAAG;YACX,KAAK,EAAE,KAAK,IAAI,cAAc;YAC9B,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACjI,OAAO,EAAE,EAAE,WAAW,EAAE,UAAU,IAAI,IAAI,EAAE;YAC5C,MAAM,EAAE,KAAK;SACd,CAAC;QACF,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAC3B,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CAAC,CAAC;QACH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YACZ,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,WAAW,GAAG,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC,CAAC;QAClD,CAAC;QACD,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA4D,CAAC;QAC3F,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QAC5C,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;IAC1G,CAAC;CACF,CAAC,CAAC;AACH,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC;IAAE,gBAAgB,CAAC,OAAO,EAAE,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"orchestrator.test.d.ts","sourceRoot":"","sources":["../../src/ai/orchestrator.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,29 @@
|
|
|
1
|
+
import { describe, it, expect, vi, beforeEach } from 'vitest';
|
|
2
|
+
vi.mock('../db/index.js', () => ({
|
|
3
|
+
log: vi.fn(),
|
|
4
|
+
saveAiResult: vi.fn(),
|
|
5
|
+
}));
|
|
6
|
+
describe('orchestrator', () => {
|
|
7
|
+
beforeEach(async () => {
|
|
8
|
+
const { registerProvider } = await import('./orchestrator.js');
|
|
9
|
+
const stub = {
|
|
10
|
+
defaultModel: 'stub',
|
|
11
|
+
async complete({ messages }) {
|
|
12
|
+
return { content: messages?.[0]?.content || 'ok', usage: { completion_tokens: 0 } };
|
|
13
|
+
},
|
|
14
|
+
};
|
|
15
|
+
registerProvider('test-provider', stub);
|
|
16
|
+
});
|
|
17
|
+
it('getProvider throws for unknown provider', async () => {
|
|
18
|
+
const { getProvider: getProv } = await import('./orchestrator.js');
|
|
19
|
+
expect(() => getProv('nonexistent')).toThrow(/Unknown AI provider/);
|
|
20
|
+
});
|
|
21
|
+
it('getProvider returns registered adapter', async () => {
|
|
22
|
+
const { getProvider } = await import('./orchestrator.js');
|
|
23
|
+
const adapter = getProvider('test-provider');
|
|
24
|
+
expect(adapter.defaultModel).toBe('stub');
|
|
25
|
+
const out = await adapter.complete({ messages: [{ role: 'user', content: 'hi' }] });
|
|
26
|
+
expect(out.content).toBe('hi');
|
|
27
|
+
});
|
|
28
|
+
});
|
|
29
|
+
//# sourceMappingURL=orchestrator.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"orchestrator.test.js","sourceRoot":"","sources":["../../src/ai/orchestrator.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAG9D,EAAE,CAAC,IAAI,CAAC,gBAAgB,EAAE,GAAG,EAAE,CAAC,CAAC;IAC/B,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;IACZ,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;CACtB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,EAAE,gBAAgB,EAAE,GAAG,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC;QAC/D,MAAM,IAAI,GAAc;YACtB,YAAY,EAAE,MAAM;YACpB,KAAK,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE;gBACzB,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,OAAiB,IAAI,IAAI,EAAE,KAAK,EAAE,EAAE,iBAAiB,EAAE,CAAC,EAAE,EAAE,CAAC;YAChG,CAAC;SACF,CAAC;QACF,gBAAgB,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,GAAG,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC;QACnE,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC;QAC1D,MAAM,OAAO,GAAG,WAAW,CAAC,eAAe,CAAC,CAAC;QAC7C,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1C,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QACpF,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
package/dist/cli.d.ts
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cli.d.ts","sourceRoot":"","sources":["../src/cli.ts"],"names":[],"mappings":""}
|