elyth-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/dist/agent.d.ts +3 -0
- package/dist/agent.d.ts.map +1 -0
- package/dist/agent.js +99 -0
- package/dist/agent.js.map +1 -0
- package/dist/cli.d.ts +2 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +360 -0
- package/dist/cli.js.map +1 -0
- package/dist/config.d.ts +16 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +92 -0
- package/dist/config.js.map +1 -0
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +7 -0
- package/dist/index.js.map +1 -0
- package/dist/logger.d.ts +20 -0
- package/dist/logger.d.ts.map +1 -0
- package/dist/logger.js +116 -0
- package/dist/logger.js.map +1 -0
- package/dist/mcp-client.d.ts +14 -0
- package/dist/mcp-client.d.ts.map +1 -0
- package/dist/mcp-client.js +49 -0
- package/dist/mcp-client.js.map +1 -0
- package/dist/prompt/build-prompt.d.ts +2 -0
- package/dist/prompt/build-prompt.d.ts.map +1 -0
- package/dist/prompt/build-prompt.js +16 -0
- package/dist/prompt/build-prompt.js.map +1 -0
- package/dist/providers/claude.d.ts +9 -0
- package/dist/providers/claude.d.ts.map +1 -0
- package/dist/providers/claude.js +82 -0
- package/dist/providers/claude.js.map +1 -0
- package/dist/providers/gemini.d.ts +15 -0
- package/dist/providers/gemini.d.ts.map +1 -0
- package/dist/providers/gemini.js +181 -0
- package/dist/providers/gemini.js.map +1 -0
- package/dist/providers/index.d.ts +3 -0
- package/dist/providers/index.d.ts.map +1 -0
- package/dist/providers/index.js +14 -0
- package/dist/providers/index.js.map +1 -0
- package/dist/providers/openai.d.ts +12 -0
- package/dist/providers/openai.d.ts.map +1 -0
- package/dist/providers/openai.js +118 -0
- package/dist/providers/openai.js.map +1 -0
- package/dist/providers/types.d.ts +42 -0
- package/dist/providers/types.d.ts.map +1 -0
- package/dist/providers/types.js +2 -0
- package/dist/providers/types.js.map +1 -0
- package/dist/scheduler.d.ts +3 -0
- package/dist/scheduler.d.ts.map +1 -0
- package/dist/scheduler.js +37 -0
- package/dist/scheduler.js.map +1 -0
- package/package.json +31 -0
package/dist/logger.js
ADDED
|
@@ -0,0 +1,116 @@
|
|
|
1
|
+
import fs from 'node:fs';
|
|
2
|
+
import path from 'node:path';
|
|
3
|
+
const COLORS = {
|
|
4
|
+
reset: '\x1b[0m',
|
|
5
|
+
dim: '\x1b[2m',
|
|
6
|
+
cyan: '\x1b[36m',
|
|
7
|
+
green: '\x1b[32m',
|
|
8
|
+
yellow: '\x1b[33m',
|
|
9
|
+
red: '\x1b[31m',
|
|
10
|
+
magenta: '\x1b[35m',
|
|
11
|
+
blue: '\x1b[34m',
|
|
12
|
+
};
|
|
13
|
+
export class Logger {
|
|
14
|
+
logFile;
|
|
15
|
+
stream;
|
|
16
|
+
constructor(logDir) {
|
|
17
|
+
if (!fs.existsSync(logDir)) {
|
|
18
|
+
fs.mkdirSync(logDir, { recursive: true });
|
|
19
|
+
}
|
|
20
|
+
const timestamp = new Date()
|
|
21
|
+
.toISOString()
|
|
22
|
+
.replace(/[:.]/g, '-')
|
|
23
|
+
.slice(0, 19);
|
|
24
|
+
this.logFile = path.join(logDir, `${timestamp}.jsonl`);
|
|
25
|
+
this.stream = fs.createWriteStream(this.logFile, { flags: 'a' });
|
|
26
|
+
}
|
|
27
|
+
write(entry) {
|
|
28
|
+
this.stream.write(JSON.stringify(entry) + '\n');
|
|
29
|
+
}
|
|
30
|
+
now() {
|
|
31
|
+
return new Date().toISOString();
|
|
32
|
+
}
|
|
33
|
+
logTickStart(provider, model) {
|
|
34
|
+
this.write({
|
|
35
|
+
type: 'tick_start',
|
|
36
|
+
timestamp: this.now(),
|
|
37
|
+
provider,
|
|
38
|
+
model,
|
|
39
|
+
});
|
|
40
|
+
console.log(`${COLORS.cyan}[tick_start]${COLORS.reset} provider=${provider} model=${model}`);
|
|
41
|
+
}
|
|
42
|
+
logToolCall(call) {
|
|
43
|
+
this.write({
|
|
44
|
+
type: 'tool_call',
|
|
45
|
+
timestamp: this.now(),
|
|
46
|
+
name: call.name,
|
|
47
|
+
input: call.input,
|
|
48
|
+
});
|
|
49
|
+
console.log(`${COLORS.yellow}[tool_call]${COLORS.reset} ${call.name}(${JSON.stringify(call.input)})`);
|
|
50
|
+
}
|
|
51
|
+
logToolResult(call, result) {
|
|
52
|
+
this.write({
|
|
53
|
+
type: 'tool_result',
|
|
54
|
+
timestamp: this.now(),
|
|
55
|
+
name: call.name,
|
|
56
|
+
content: result.content,
|
|
57
|
+
isError: result.isError,
|
|
58
|
+
});
|
|
59
|
+
const color = result.isError ? COLORS.red : COLORS.green;
|
|
60
|
+
const preview = result.content.length > 200
|
|
61
|
+
? result.content.slice(0, 200) + '...'
|
|
62
|
+
: result.content;
|
|
63
|
+
console.log(`${color}[tool_result]${COLORS.reset} ${call.name} → ${preview}`);
|
|
64
|
+
}
|
|
65
|
+
logResponse(res) {
|
|
66
|
+
this.write({
|
|
67
|
+
type: 'llm_response',
|
|
68
|
+
timestamp: this.now(),
|
|
69
|
+
content: res.content,
|
|
70
|
+
stopReason: res.stopReason,
|
|
71
|
+
toolCallCount: res.toolCalls.length,
|
|
72
|
+
});
|
|
73
|
+
if (res.content) {
|
|
74
|
+
console.log(`${COLORS.magenta}[llm]${COLORS.reset} ${res.content}`);
|
|
75
|
+
}
|
|
76
|
+
if (res.toolCalls.length > 0) {
|
|
77
|
+
console.log(`${COLORS.dim} (${res.toolCalls.length} tool calls, stopReason=${res.stopReason})${COLORS.reset}`);
|
|
78
|
+
}
|
|
79
|
+
}
|
|
80
|
+
logTickEnd(turns, durationMs) {
|
|
81
|
+
this.write({
|
|
82
|
+
type: 'tick_end',
|
|
83
|
+
timestamp: this.now(),
|
|
84
|
+
turns,
|
|
85
|
+
durationMs,
|
|
86
|
+
});
|
|
87
|
+
console.log(`${COLORS.cyan}[tick_end]${COLORS.reset} turns=${turns} duration=${(durationMs / 1000).toFixed(1)}s`);
|
|
88
|
+
console.log(`${COLORS.dim} log: ${this.logFile}${COLORS.reset}`);
|
|
89
|
+
}
|
|
90
|
+
logError(error) {
|
|
91
|
+
const message = error instanceof Error ? error.message : String(error);
|
|
92
|
+
this.write({
|
|
93
|
+
type: 'error',
|
|
94
|
+
timestamp: this.now(),
|
|
95
|
+
message,
|
|
96
|
+
});
|
|
97
|
+
console.error(`${COLORS.red}[error]${COLORS.reset} ${message}`);
|
|
98
|
+
}
|
|
99
|
+
close() {
|
|
100
|
+
this.stream.end();
|
|
101
|
+
}
|
|
102
|
+
static cleanOldLogs(logDir, maxAgeDays = 7) {
|
|
103
|
+
if (!fs.existsSync(logDir))
|
|
104
|
+
return;
|
|
105
|
+
const cutoff = Date.now() - maxAgeDays * 24 * 60 * 60 * 1000;
|
|
106
|
+
for (const file of fs.readdirSync(logDir)) {
|
|
107
|
+
const filePath = path.join(logDir, file);
|
|
108
|
+
const stat = fs.statSync(filePath);
|
|
109
|
+
if (stat.isFile() && stat.mtimeMs < cutoff) {
|
|
110
|
+
fs.unlinkSync(filePath);
|
|
111
|
+
console.log(`${COLORS.dim} Deleted old log: ${file}${COLORS.reset}`);
|
|
112
|
+
}
|
|
113
|
+
}
|
|
114
|
+
}
|
|
115
|
+
}
|
|
116
|
+
//# sourceMappingURL=logger.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"logger.js","sourceRoot":"","sources":["../src/logger.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAS7B,MAAM,MAAM,GAAG;IACb,KAAK,EAAE,SAAS;IAChB,GAAG,EAAE,SAAS;IACd,IAAI,EAAE,UAAU;IAChB,KAAK,EAAE,UAAU;IACjB,MAAM,EAAE,UAAU;IAClB,GAAG,EAAE,UAAU;IACf,OAAO,EAAE,UAAU;IACnB,IAAI,EAAE,UAAU;CACjB,CAAC;AAEF,MAAM,OAAO,MAAM;IACT,OAAO,CAAS;IAChB,MAAM,CAAiB;IAE/B,YAAY,MAAc;QACxB,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3B,EAAE,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5C,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE;aACzB,WAAW,EAAE;aACb,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC;aACrB,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAChB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,SAAS,QAAQ,CAAC,CAAC;QACvD,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;IACnE,CAAC;IAEO,KAAK,CAAC,KAAe;QAC3B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,CAAC;IAClD,CAAC;IAEO,GAAG;QACT,OAAO,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAClC,CAAC;IAED,YAAY,CAAC,QAAgB,EAAE,KAAa;QAC1C,IAAI,CAAC,KAAK,CAAC;YACT,IAAI,EAAE,YAAY;YAClB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,QAAQ;YACR,KAAK;SACN,CAAC,CAAC;QACH,OAAO,CAAC,GAAG,CACT,GAAG,MAAM,CAAC,IAAI,eAAe,MAAM,CAAC,KAAK,aAAa,QAAQ,UAAU,KAAK,EAAE,CAChF,CAAC;IACJ,CAAC;IAED,WAAW,CAAC,IAAc;QACxB,IAAI,CAAC,KAAK,CAAC;YACT,IAAI,EAAE,WAAW;YACjB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAC,CAAC;QACH,OAAO,CAAC,GAAG,CACT,GAAG,MAAM,CAAC,MAAM,cAAc,MAAM,CAAC,KAAK,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CACzF,CAAC;IACJ,CAAC;IAED,aAAa,CACX,IAAc,EACd,MAA6C;QAE7C,IAAI,CAAC,KAAK,CAAC;YACT,IAAI,EAAE,aAAa;YACnB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,OAAO,EAAE,MAAM,CAAC,OAAO;SACxB,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;QACzD,MAAM,OAAO,GACX,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,GAAG;YACzB,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,KAAK;YACtC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,OAAO,CAAC,GAAG,CACT,GAAG,KAAK,gBAAgB,MAAM,CAAC,KAAK,IAAI,IAAI,CAAC,IAAI,MAAM,OAAO,EAAE,CACjE,CAAC;IACJ,CAAC;IAED,WAAW,CAAC,GAAgB;QAC1B,IAAI,CAAC,KAAK,CAAC;YACT,IAAI,EAAE,cAAc;YACpB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,UAAU,EAAE,GAAG,CAAC,UAAU;YAC1B,aAAa,EAAE,GAAG,CAAC,SAAS,CAAC,MAAM;SACpC,CAAC,CAAC;QACH,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;YAChB,OAAO,CAAC,GAAG,CACT,GAAG,MAAM,CAAC,OAAO,QAAQ,MAAM,CAAC,KAAK,IAAI,GAAG,CAAC,OAAO,EAAE,CACvD,CAAC;QACJ,CAAC;QACD,IAAI,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,OAAO,CAAC,GAAG,CACT,GAAG,MAAM,CAAC,GAAG,MAAM,GAAG,CAAC,SAAS,CAAC,MAAM,2BAA2B,GAAG,CAAC,UAAU,IAAI,MAAM,CAAC,KAAK,EAAE,CACnG,CAAC;QACJ,CAAC;IACH,CAAC;IAED,UAAU,CAAC,KAAa,EAAE,UAAkB;QAC1C,IAAI,CAAC,KAAK,CAAC;YACT,IAAI,EAAE,UAAU;YAChB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,KAAK;YACL,UAAU;SACX,CAAC,CAAC;QACH,OAAO,CAAC,GAAG,CACT,GAAG,MAAM,CAAC,IAAI,aAAa,MAAM,CAAC,KAAK,UAAU,KAAK,aAAa,CAAC,UAAU,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CACrG,CAAC;QACF,OAAO,CAAC,GAAG,CACT,GAAG,MAAM,CAAC,GAAG,UAAU,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,KAAK,EAAE,CACrD,CAAC;IACJ,CAAC;IAED,QAAQ,CAAC,KAAc;QACrB,MAAM,OAAO,GACX,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACzD,IAAI,CAAC,KAAK,CAAC;YACT,IAAI,EAAE,OAAO;YACb,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,OAAO;SACR,CAAC,CAAC;QACH,OAAO,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,GAAG,UAAU,MAAM,CAAC,KAAK,IAAI,OAAO,EAAE,CAAC,CAAC;IAClE,CAAC;IAED,KAAK;QACH,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;IACpB,CAAC;IAED,MAAM,CAAC,YAAY,CAAC,MAAc,EAAE,aAAqB,CAAC;QACxD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC;YAAE,OAAO;QACnC,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,UAAU,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;QAC7D,KAAK,MAAM,IAAI,IAAI,EAAE,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YACzC,MAAM,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACnC,IAAI,IAAI,CAAC,MAAM,EAAE,IAAI,IAAI,CAAC,OAAO,GAAG,MAAM,EAAE,CAAC;gBAC3C,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;gBACxB,OAAO,CAAC,GAAG,CACT,GAAG,MAAM,CAAC,GAAG,sBAAsB,IAAI,GAAG,MAAM,CAAC,KAAK,EAAE,CACzD,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import type { ToolDefinition } from './providers/types.js';
|
|
2
|
+
export declare class McpClient {
|
|
3
|
+
private client;
|
|
4
|
+
private transport;
|
|
5
|
+
constructor();
|
|
6
|
+
connect(apiKey: string, apiBase: string): Promise<void>;
|
|
7
|
+
getTools(): Promise<ToolDefinition[]>;
|
|
8
|
+
callTool(name: string, args: Record<string, unknown>): Promise<{
|
|
9
|
+
content: string;
|
|
10
|
+
isError: boolean;
|
|
11
|
+
}>;
|
|
12
|
+
disconnect(): Promise<void>;
|
|
13
|
+
}
|
|
14
|
+
//# sourceMappingURL=mcp-client.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-client.d.ts","sourceRoot":"","sources":["../src/mcp-client.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAE3D,qBAAa,SAAS;IACpB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,SAAS,CAAqC;;IAShD,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAkBvD,QAAQ,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;IASrC,QAAQ,CACZ,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC5B,OAAO,CAAC;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC;IAiB3C,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;CAGlC"}
|
|
@@ -0,0 +1,49 @@
|
|
|
1
|
+
import { Client } from '@modelcontextprotocol/sdk/client/index.js';
|
|
2
|
+
import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js';
|
|
3
|
+
export class McpClient {
|
|
4
|
+
client;
|
|
5
|
+
transport = null;
|
|
6
|
+
constructor() {
|
|
7
|
+
this.client = new Client({ name: 'elyth-agent', version: '0.1.0' }, { capabilities: {} });
|
|
8
|
+
}
|
|
9
|
+
async connect(apiKey, apiBase) {
|
|
10
|
+
const isWindows = process.platform === 'win32';
|
|
11
|
+
this.transport = new StdioClientTransport({
|
|
12
|
+
command: isWindows ? 'cmd' : 'npx',
|
|
13
|
+
args: isWindows
|
|
14
|
+
? ['/c', 'npx', '-y', 'elyth-mcp-server']
|
|
15
|
+
: ['-y', 'elyth-mcp-server'],
|
|
16
|
+
env: {
|
|
17
|
+
...process.env,
|
|
18
|
+
ELYTH_API_KEY: apiKey,
|
|
19
|
+
ELYTH_API_BASE: apiBase,
|
|
20
|
+
},
|
|
21
|
+
});
|
|
22
|
+
await this.client.connect(this.transport);
|
|
23
|
+
}
|
|
24
|
+
async getTools() {
|
|
25
|
+
const result = await this.client.listTools();
|
|
26
|
+
return result.tools.map((tool) => ({
|
|
27
|
+
name: tool.name,
|
|
28
|
+
description: tool.description || '',
|
|
29
|
+
inputSchema: tool.inputSchema,
|
|
30
|
+
}));
|
|
31
|
+
}
|
|
32
|
+
async callTool(name, args) {
|
|
33
|
+
const result = await this.client.callTool({ name, arguments: args });
|
|
34
|
+
const textParts = [];
|
|
35
|
+
let isError = result.isError === true;
|
|
36
|
+
if (Array.isArray(result.content)) {
|
|
37
|
+
for (const block of result.content) {
|
|
38
|
+
if (block.type === 'text') {
|
|
39
|
+
textParts.push(block.text);
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
return { content: textParts.join('\n'), isError };
|
|
44
|
+
}
|
|
45
|
+
async disconnect() {
|
|
46
|
+
await this.client.close();
|
|
47
|
+
}
|
|
48
|
+
}
|
|
49
|
+
//# sourceMappingURL=mcp-client.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-client.js","sourceRoot":"","sources":["../src/mcp-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AAGjF,MAAM,OAAO,SAAS;IACZ,MAAM,CAAS;IACf,SAAS,GAAgC,IAAI,CAAC;IAEtD;QACE,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CACtB,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,OAAO,EAAE,EACzC,EAAE,YAAY,EAAE,EAAE,EAAE,CACrB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,OAAe;QAC3C,MAAM,SAAS,GAAG,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC;QAE/C,IAAI,CAAC,SAAS,GAAG,IAAI,oBAAoB,CAAC;YACxC,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK;YAClC,IAAI,EAAE,SAAS;gBACb,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,kBAAkB,CAAC;gBACzC,CAAC,CAAC,CAAC,IAAI,EAAE,kBAAkB,CAAC;YAC9B,GAAG,EAAE;gBACH,GAAG,OAAO,CAAC,GAAG;gBACd,aAAa,EAAE,MAAM;gBACrB,cAAc,EAAE,OAAO;aACxB;SACF,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;QAC7C,OAAO,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACjC,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;YACnC,WAAW,EAAE,IAAI,CAAC,WAAsC;SACzD,CAAC,CAAC,CAAC;IACN,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,IAAY,EACZ,IAA6B;QAE7B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAErE,MAAM,SAAS,GAAa,EAAE,CAAC;QAC/B,IAAI,OAAO,GAAG,MAAM,CAAC,OAAO,KAAK,IAAI,CAAC;QAEtC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;YAClC,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBAC1B,SAAS,CAAC,IAAI,CAAE,KAAwC,CAAC,IAAI,CAAC,CAAC;gBACjE,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IAC5B,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"build-prompt.d.ts","sourceRoot":"","sources":["../../src/prompt/build-prompt.ts"],"names":[],"mappings":"AAEA,wBAAgB,WAAW,CACzB,WAAW,EAAE,MAAM,EACnB,SAAS,EAAE,MAAM,EACjB,cAAc,EAAE,MAAM,GACrB,MAAM,CAiBR"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
import fs from 'node:fs';
|
|
2
|
+
export function buildPrompt(personaPath, rulesPath, systemBasePath) {
|
|
3
|
+
const persona = fs.readFileSync(personaPath, 'utf-8').trim();
|
|
4
|
+
let rules = '';
|
|
5
|
+
if (fs.existsSync(rulesPath)) {
|
|
6
|
+
rules = fs.readFileSync(rulesPath, 'utf-8').trim();
|
|
7
|
+
}
|
|
8
|
+
const systemBase = fs.readFileSync(systemBasePath, 'utf-8').trim();
|
|
9
|
+
const parts = [persona];
|
|
10
|
+
if (rules) {
|
|
11
|
+
parts.push(rules);
|
|
12
|
+
}
|
|
13
|
+
parts.push(systemBase);
|
|
14
|
+
return parts.join('\n\n---\n\n');
|
|
15
|
+
}
|
|
16
|
+
//# sourceMappingURL=build-prompt.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"build-prompt.js","sourceRoot":"","sources":["../../src/prompt/build-prompt.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AAEzB,MAAM,UAAU,WAAW,CACzB,WAAmB,EACnB,SAAiB,EACjB,cAAsB;IAEtB,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC;IAE7D,IAAI,KAAK,GAAG,EAAE,CAAC;IACf,IAAI,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC7B,KAAK,GAAG,EAAE,CAAC,YAAY,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC;IACrD,CAAC;IAED,MAAM,UAAU,GAAG,EAAE,CAAC,YAAY,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC;IAEnE,MAAM,KAAK,GAAG,CAAC,OAAO,CAAC,CAAC;IACxB,IAAI,KAAK,EAAE,CAAC;QACV,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACpB,CAAC;IACD,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAEvB,OAAO,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;AACnC,CAAC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import type { LLMProvider, LLMResponse, Message, ToolDefinition } from './types.js';
|
|
2
|
+
export declare class ClaudeProvider implements LLMProvider {
|
|
3
|
+
private client;
|
|
4
|
+
private model;
|
|
5
|
+
constructor(apiKey: string, model: string);
|
|
6
|
+
chat(systemPrompt: string, messages: Message[], tools: ToolDefinition[]): Promise<LLMResponse>;
|
|
7
|
+
private convertContent;
|
|
8
|
+
}
|
|
9
|
+
//# sourceMappingURL=claude.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"claude.d.ts","sourceRoot":"","sources":["../../src/providers/claude.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,WAAW,EACX,WAAW,EACX,OAAO,EACP,cAAc,EAGf,MAAM,YAAY,CAAC;AAEpB,qBAAa,cAAe,YAAW,WAAW;IAChD,OAAO,CAAC,MAAM,CAAY;IAC1B,OAAO,CAAC,KAAK,CAAS;gBAEV,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM;IAKnC,IAAI,CACR,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,OAAO,EAAE,EACnB,KAAK,EAAE,cAAc,EAAE,GACtB,OAAO,CAAC,WAAW,CAAC;IAsDvB,OAAO,CAAC,cAAc;CA0BvB"}
|
|
@@ -0,0 +1,82 @@
|
|
|
1
|
+
import Anthropic from '@anthropic-ai/sdk';
|
|
2
|
+
export class ClaudeProvider {
|
|
3
|
+
client;
|
|
4
|
+
model;
|
|
5
|
+
constructor(apiKey, model) {
|
|
6
|
+
this.client = new Anthropic({ apiKey });
|
|
7
|
+
this.model = model;
|
|
8
|
+
}
|
|
9
|
+
async chat(systemPrompt, messages, tools) {
|
|
10
|
+
const anthropicMessages = messages.map((m) => ({
|
|
11
|
+
role: m.role,
|
|
12
|
+
content: this.convertContent(m.content),
|
|
13
|
+
}));
|
|
14
|
+
const anthropicTools = tools.map((t) => ({
|
|
15
|
+
name: t.name,
|
|
16
|
+
description: t.description,
|
|
17
|
+
input_schema: t.inputSchema,
|
|
18
|
+
}));
|
|
19
|
+
const response = await this.client.messages.create({
|
|
20
|
+
model: this.model,
|
|
21
|
+
max_tokens: 4096,
|
|
22
|
+
system: systemPrompt,
|
|
23
|
+
messages: anthropicMessages,
|
|
24
|
+
tools: anthropicTools.length > 0 ? anthropicTools : undefined,
|
|
25
|
+
});
|
|
26
|
+
const textParts = [];
|
|
27
|
+
const toolCalls = [];
|
|
28
|
+
for (const block of response.content) {
|
|
29
|
+
if (block.type === 'text') {
|
|
30
|
+
textParts.push(block.text);
|
|
31
|
+
}
|
|
32
|
+
else if (block.type === 'tool_use') {
|
|
33
|
+
toolCalls.push({
|
|
34
|
+
id: block.id,
|
|
35
|
+
name: block.name,
|
|
36
|
+
input: block.input,
|
|
37
|
+
});
|
|
38
|
+
}
|
|
39
|
+
}
|
|
40
|
+
let stopReason;
|
|
41
|
+
switch (response.stop_reason) {
|
|
42
|
+
case 'tool_use':
|
|
43
|
+
stopReason = 'tool_use';
|
|
44
|
+
break;
|
|
45
|
+
case 'max_tokens':
|
|
46
|
+
stopReason = 'max_tokens';
|
|
47
|
+
break;
|
|
48
|
+
default:
|
|
49
|
+
stopReason = 'end';
|
|
50
|
+
}
|
|
51
|
+
return {
|
|
52
|
+
content: textParts.length > 0 ? textParts.join('\n') : null,
|
|
53
|
+
toolCalls,
|
|
54
|
+
stopReason,
|
|
55
|
+
};
|
|
56
|
+
}
|
|
57
|
+
convertContent(content) {
|
|
58
|
+
if (typeof content === 'string')
|
|
59
|
+
return content;
|
|
60
|
+
return content.map((block) => {
|
|
61
|
+
switch (block.type) {
|
|
62
|
+
case 'text':
|
|
63
|
+
return { type: 'text', text: block.text };
|
|
64
|
+
case 'tool_use':
|
|
65
|
+
return {
|
|
66
|
+
type: 'tool_use',
|
|
67
|
+
id: block.id,
|
|
68
|
+
name: block.name,
|
|
69
|
+
input: block.input,
|
|
70
|
+
};
|
|
71
|
+
case 'tool_result':
|
|
72
|
+
return {
|
|
73
|
+
type: 'tool_result',
|
|
74
|
+
tool_use_id: block.tool_use_id,
|
|
75
|
+
content: block.content,
|
|
76
|
+
is_error: block.is_error,
|
|
77
|
+
};
|
|
78
|
+
}
|
|
79
|
+
});
|
|
80
|
+
}
|
|
81
|
+
}
|
|
82
|
+
//# sourceMappingURL=claude.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"claude.js","sourceRoot":"","sources":["../../src/providers/claude.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,MAAM,mBAAmB,CAAC;AAU1C,MAAM,OAAO,cAAc;IACjB,MAAM,CAAY;IAClB,KAAK,CAAS;IAEtB,YAAY,MAAc,EAAE,KAAa;QACvC,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;QACxC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,IAAI,CACR,YAAoB,EACpB,QAAmB,EACnB,KAAuB;QAEvB,MAAM,iBAAiB,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC7C,IAAI,EAAE,CAAC,CAAC,IAA4B;YACpC,OAAO,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,OAAO,CAAC;SACxC,CAAC,CAAC,CAAC;QAEJ,MAAM,cAAc,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACvC,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,WAAW,EAAE,CAAC,CAAC,WAAW;YAC1B,YAAY,EAAE,CAAC,CAAC,WAA6C;SAC9D,CAAC,CAAC,CAAC;QAEJ,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;YACjD,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,UAAU,EAAE,IAAI;YAChB,MAAM,EAAE,YAAY;YACpB,QAAQ,EAAE,iBAAiB;YAC3B,KAAK,EAAE,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,SAAS;SAC9D,CAAC,CAAC;QAEH,MAAM,SAAS,GAAa,EAAE,CAAC;QAC/B,MAAM,SAAS,GAAe,EAAE,CAAC;QAEjC,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACrC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC1B,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC7B,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBACrC,SAAS,CAAC,IAAI,CAAC;oBACb,EAAE,EAAE,KAAK,CAAC,EAAE;oBACZ,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,KAAK,EAAE,KAAK,CAAC,KAAgC;iBAC9C,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,IAAI,UAAqC,CAAC;QAC1C,QAAQ,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC7B,KAAK,UAAU;gBACb,UAAU,GAAG,UAAU,CAAC;gBACxB,MAAM;YACR,KAAK,YAAY;gBACf,UAAU,GAAG,YAAY,CAAC;gBAC1B,MAAM;YACR;gBACE,UAAU,GAAG,KAAK,CAAC;QACvB,CAAC;QAED,OAAO;YACL,OAAO,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI;YAC3D,SAAS;YACT,UAAU;SACX,CAAC;IACJ,CAAC;IAEO,cAAc,CACpB,OAAgC;QAEhC,IAAI,OAAO,OAAO,KAAK,QAAQ;YAAE,OAAO,OAAO,CAAC;QAEhD,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;YAC3B,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;gBACnB,KAAK,MAAM;oBACT,OAAO,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;gBACrD,KAAK,UAAU;oBACb,OAAO;wBACL,IAAI,EAAE,UAAmB;wBACzB,EAAE,EAAE,KAAK,CAAC,EAAE;wBACZ,IAAI,EAAE,KAAK,CAAC,IAAI;wBAChB,KAAK,EAAE,KAAK,CAAC,KAAK;qBACnB,CAAC;gBACJ,KAAK,aAAa;oBAChB,OAAO;wBACL,IAAI,EAAE,aAAsB;wBAC5B,WAAW,EAAE,KAAK,CAAC,WAAW;wBAC9B,OAAO,EAAE,KAAK,CAAC,OAAO;wBACtB,QAAQ,EAAE,KAAK,CAAC,QAAQ;qBACzB,CAAC;YACN,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;CACF"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import type { LLMProvider, LLMResponse, Message, ToolDefinition } from './types.js';
|
|
2
|
+
export declare class GeminiProvider implements LLMProvider {
|
|
3
|
+
private genAI;
|
|
4
|
+
private model;
|
|
5
|
+
constructor(apiKey: string, model: string);
|
|
6
|
+
chat(systemPrompt: string, messages: Message[], tools: ToolDefinition[]): Promise<LLMResponse>;
|
|
7
|
+
private convertSchema;
|
|
8
|
+
/** Recursively remove JSON Schema fields that Gemini API doesn't accept */
|
|
9
|
+
private stripUnsupportedFields;
|
|
10
|
+
private convertMessages;
|
|
11
|
+
/** Scan all messages to build a tool_use_id → name mapping */
|
|
12
|
+
private buildToolNameMap;
|
|
13
|
+
private messageToParts;
|
|
14
|
+
}
|
|
15
|
+
//# sourceMappingURL=gemini.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../src/providers/gemini.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EACV,WAAW,EACX,WAAW,EACX,OAAO,EACP,cAAc,EAGf,MAAM,YAAY,CAAC;AAEpB,qBAAa,cAAe,YAAW,WAAW;IAChD,OAAO,CAAC,KAAK,CAAqB;IAClC,OAAO,CAAC,KAAK,CAAS;gBAEV,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM;IAKnC,IAAI,CACR,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,OAAO,EAAE,EACnB,KAAK,EAAE,cAAc,EAAE,GACtB,OAAO,CAAC,WAAW,CAAC;IAsFvB,OAAO,CAAC,aAAa;IAMrB,2EAA2E;IAC3E,OAAO,CAAC,sBAAsB;IAmB9B,OAAO,CAAC,eAAe;IAqBvB,8DAA8D;IAC9D,OAAO,CAAC,gBAAgB;IAaxB,OAAO,CAAC,cAAc;CA2CvB"}
|
|
@@ -0,0 +1,181 @@
|
|
|
1
|
+
import { GoogleGenerativeAI, } from '@google/generative-ai';
|
|
2
|
+
export class GeminiProvider {
|
|
3
|
+
genAI;
|
|
4
|
+
model;
|
|
5
|
+
constructor(apiKey, model) {
|
|
6
|
+
this.genAI = new GoogleGenerativeAI(apiKey);
|
|
7
|
+
this.model = model;
|
|
8
|
+
}
|
|
9
|
+
async chat(systemPrompt, messages, tools) {
|
|
10
|
+
const geminiTools = tools.length > 0
|
|
11
|
+
? [
|
|
12
|
+
{
|
|
13
|
+
functionDeclarations: tools.map((t) => ({
|
|
14
|
+
name: t.name,
|
|
15
|
+
description: t.description,
|
|
16
|
+
parameters: this.convertSchema(t.inputSchema),
|
|
17
|
+
})),
|
|
18
|
+
},
|
|
19
|
+
]
|
|
20
|
+
: undefined;
|
|
21
|
+
const model = this.genAI.getGenerativeModel({
|
|
22
|
+
model: this.model,
|
|
23
|
+
systemInstruction: systemPrompt,
|
|
24
|
+
tools: geminiTools,
|
|
25
|
+
});
|
|
26
|
+
const toolNameMap = this.buildToolNameMap(messages);
|
|
27
|
+
const lastMessage = messages[messages.length - 1];
|
|
28
|
+
const lastHasFunctionResponse = Array.isArray(lastMessage.content) &&
|
|
29
|
+
lastMessage.content.some((b) => b.type === 'tool_result');
|
|
30
|
+
let result;
|
|
31
|
+
if (lastHasFunctionResponse) {
|
|
32
|
+
// sendMessage assumes role 'user', but functionResponse needs role 'function'.
|
|
33
|
+
// Include all messages in history and use sendMessage with empty-ish prompt
|
|
34
|
+
// to trigger the model to continue.
|
|
35
|
+
const geminiHistory = this.convertMessages(messages);
|
|
36
|
+
const chat = model.startChat({ history: geminiHistory });
|
|
37
|
+
result = await chat.sendMessage('Continue based on the tool results above.');
|
|
38
|
+
}
|
|
39
|
+
else {
|
|
40
|
+
const geminiHistory = this.convertMessages(messages.slice(0, -1));
|
|
41
|
+
const chat = model.startChat({ history: geminiHistory });
|
|
42
|
+
const lastParts = this.messageToParts(lastMessage, toolNameMap);
|
|
43
|
+
result = await chat.sendMessage(lastParts);
|
|
44
|
+
}
|
|
45
|
+
const response = result.response;
|
|
46
|
+
const candidate = response.candidates?.[0];
|
|
47
|
+
const textParts = [];
|
|
48
|
+
const toolCalls = [];
|
|
49
|
+
if (candidate?.content?.parts) {
|
|
50
|
+
for (const part of candidate.content.parts) {
|
|
51
|
+
if (part.text) {
|
|
52
|
+
textParts.push(part.text);
|
|
53
|
+
}
|
|
54
|
+
if (part.functionCall) {
|
|
55
|
+
const rawPart = part;
|
|
56
|
+
const metadata = {};
|
|
57
|
+
// Preserve thought_signature for Gemini conversation history
|
|
58
|
+
if (rawPart.thoughtSignature) {
|
|
59
|
+
metadata.thoughtSignature = rawPart.thoughtSignature;
|
|
60
|
+
}
|
|
61
|
+
toolCalls.push({
|
|
62
|
+
id: `gemini_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`,
|
|
63
|
+
name: part.functionCall.name,
|
|
64
|
+
input: part.functionCall.args || {},
|
|
65
|
+
metadata: Object.keys(metadata).length > 0 ? metadata : undefined,
|
|
66
|
+
});
|
|
67
|
+
}
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
const finishReason = candidate?.finishReason;
|
|
71
|
+
let stopReason;
|
|
72
|
+
if (toolCalls.length > 0) {
|
|
73
|
+
stopReason = 'tool_use';
|
|
74
|
+
}
|
|
75
|
+
else if (finishReason === 'MAX_TOKENS') {
|
|
76
|
+
stopReason = 'max_tokens';
|
|
77
|
+
}
|
|
78
|
+
else {
|
|
79
|
+
stopReason = 'end';
|
|
80
|
+
}
|
|
81
|
+
return {
|
|
82
|
+
content: textParts.length > 0 ? textParts.join('\n') : null,
|
|
83
|
+
toolCalls,
|
|
84
|
+
stopReason,
|
|
85
|
+
};
|
|
86
|
+
}
|
|
87
|
+
convertSchema(schema) {
|
|
88
|
+
return this.stripUnsupportedFields(schema);
|
|
89
|
+
}
|
|
90
|
+
/** Recursively remove JSON Schema fields that Gemini API doesn't accept */
|
|
91
|
+
stripUnsupportedFields(obj) {
|
|
92
|
+
if (Array.isArray(obj)) {
|
|
93
|
+
return obj.map((item) => this.stripUnsupportedFields(item));
|
|
94
|
+
}
|
|
95
|
+
if (obj !== null && typeof obj === 'object') {
|
|
96
|
+
const result = {};
|
|
97
|
+
const UNSUPPORTED = new Set([
|
|
98
|
+
'$schema', 'additionalProperties', '$id', '$ref',
|
|
99
|
+
'$comment', '$defs', 'definitions',
|
|
100
|
+
]);
|
|
101
|
+
for (const [key, value] of Object.entries(obj)) {
|
|
102
|
+
if (UNSUPPORTED.has(key))
|
|
103
|
+
continue;
|
|
104
|
+
result[key] = this.stripUnsupportedFields(value);
|
|
105
|
+
}
|
|
106
|
+
return result;
|
|
107
|
+
}
|
|
108
|
+
return obj;
|
|
109
|
+
}
|
|
110
|
+
convertMessages(messages) {
|
|
111
|
+
// Build a mapping of tool_use_id → function name from all messages
|
|
112
|
+
const toolNameMap = this.buildToolNameMap(messages);
|
|
113
|
+
return messages.map((m) => {
|
|
114
|
+
let role;
|
|
115
|
+
if (m.role === 'assistant') {
|
|
116
|
+
role = 'model';
|
|
117
|
+
}
|
|
118
|
+
else if (Array.isArray(m.content) &&
|
|
119
|
+
m.content.some((b) => b.type === 'tool_result')) {
|
|
120
|
+
// Gemini requires functionResponse parts to have role 'function'
|
|
121
|
+
role = 'function';
|
|
122
|
+
}
|
|
123
|
+
else {
|
|
124
|
+
role = 'user';
|
|
125
|
+
}
|
|
126
|
+
return { role, parts: this.messageToParts(m, toolNameMap) };
|
|
127
|
+
});
|
|
128
|
+
}
|
|
129
|
+
/** Scan all messages to build a tool_use_id → name mapping */
|
|
130
|
+
buildToolNameMap(messages) {
|
|
131
|
+
const map = new Map();
|
|
132
|
+
for (const msg of messages) {
|
|
133
|
+
if (typeof msg.content === 'string')
|
|
134
|
+
continue;
|
|
135
|
+
for (const block of msg.content) {
|
|
136
|
+
if (block.type === 'tool_use') {
|
|
137
|
+
map.set(block.id, block.name);
|
|
138
|
+
}
|
|
139
|
+
}
|
|
140
|
+
}
|
|
141
|
+
return map;
|
|
142
|
+
}
|
|
143
|
+
messageToParts(msg, toolNameMap = new Map()) {
|
|
144
|
+
if (typeof msg.content === 'string') {
|
|
145
|
+
return [{ text: msg.content }];
|
|
146
|
+
}
|
|
147
|
+
const blocks = msg.content;
|
|
148
|
+
const parts = [];
|
|
149
|
+
for (const block of blocks) {
|
|
150
|
+
switch (block.type) {
|
|
151
|
+
case 'text':
|
|
152
|
+
parts.push({ text: block.text });
|
|
153
|
+
break;
|
|
154
|
+
case 'tool_use': {
|
|
155
|
+
const fcPart = {
|
|
156
|
+
functionCall: {
|
|
157
|
+
name: block.name,
|
|
158
|
+
args: block.input,
|
|
159
|
+
},
|
|
160
|
+
};
|
|
161
|
+
// Restore thought_signature if preserved
|
|
162
|
+
if (block.metadata?.thoughtSignature) {
|
|
163
|
+
fcPart.thoughtSignature = block.metadata.thoughtSignature;
|
|
164
|
+
}
|
|
165
|
+
parts.push(fcPart);
|
|
166
|
+
break;
|
|
167
|
+
}
|
|
168
|
+
case 'tool_result':
|
|
169
|
+
parts.push({
|
|
170
|
+
functionResponse: {
|
|
171
|
+
name: toolNameMap.get(block.tool_use_id) ?? 'unknown',
|
|
172
|
+
response: { result: block.content },
|
|
173
|
+
},
|
|
174
|
+
});
|
|
175
|
+
break;
|
|
176
|
+
}
|
|
177
|
+
}
|
|
178
|
+
return parts.length > 0 ? parts : [{ text: '' }];
|
|
179
|
+
}
|
|
180
|
+
}
|
|
181
|
+
//# sourceMappingURL=gemini.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../src/providers/gemini.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,kBAAkB,GAInB,MAAM,uBAAuB,CAAC;AAU/B,MAAM,OAAO,cAAc;IACjB,KAAK,CAAqB;IAC1B,KAAK,CAAS;IAEtB,YAAY,MAAc,EAAE,KAAa;QACvC,IAAI,CAAC,KAAK,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAC5C,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,IAAI,CACR,YAAoB,EACpB,QAAmB,EACnB,KAAuB;QAEvB,MAAM,WAAW,GACf,KAAK,CAAC,MAAM,GAAG,CAAC;YACd,CAAC,CAAC;gBACE;oBACE,oBAAoB,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBACtC,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;wBAC1B,UAAU,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,WAAW,CAAC;qBAC9C,CAAC,CAAC;iBACJ;aACF;YACH,CAAC,CAAC,SAAS,CAAC;QAEhB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,iBAAiB,EAAE,YAAY;YAC/B,KAAK,EAAE,WAAW;SACnB,CAAC,CAAC;QAEH,MAAM,WAAW,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACpD,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAClD,MAAM,uBAAuB,GAC3B,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC;YAClC,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAkB,CAAC,IAAI,KAAK,aAAa,CAAC,CAAC;QAE9E,IAAI,MAAM,CAAC;QACX,IAAI,uBAAuB,EAAE,CAAC;YAC5B,+EAA+E;YAC/E,4EAA4E;YAC5E,oCAAoC;YACpC,MAAM,aAAa,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;YACrD,MAAM,IAAI,GAAG,KAAK,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC;YACzD,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,2CAA2C,CAAC,CAAC;QAC/E,CAAC;aAAM,CAAC;YACN,MAAM,aAAa,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAClE,MAAM,IAAI,GAAG,KAAK,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC;YACzD,MAAM,SAAS,GAAG,IAAI,CAAC,cAAc,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC;YAChE,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;QAC7C,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QACjC,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;QAE3C,MAAM,SAAS,GAAa,EAAE,CAAC;QAC/B,MAAM,SAAS,GAAe,EAAE,CAAC;QAEjC,IAAI,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YAC9B,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;gBAC3C,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;oBACd,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC5B,CAAC;gBACD,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;oBACtB,MAAM,OAAO,GAAG,IAA0C,CAAC;oBAC3D,MAAM,QAAQ,GAA4B,EAAE,CAAC;oBAC7C,6DAA6D;oBAC7D,IAAI,OAAO,CAAC,gBAAgB,EAAE,CAAC;wBAC7B,QAAQ,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC;oBACvD,CAAC;oBACD,SAAS,CAAC,IAAI,CAAC;wBACb,EAAE,EAAE,UAAU,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;wBACpE,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;wBAC5B,KAAK,EAAG,IAAI,CAAC,YAAY,CAAC,IAAgC,IAAI,EAAE;wBAChE,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;qBAClE,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,YAAY,GAAG,SAAS,EAAE,YAAY,CAAC;QAC7C,IAAI,UAAqC,CAAC;QAC1C,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,UAAU,GAAG,UAAU,CAAC;QAC1B,CAAC;aAAM,IAAI,YAAY,KAAK,YAAY,EAAE,CAAC;YACzC,UAAU,GAAG,YAAY,CAAC;QAC5B,CAAC;aAAM,CAAC;YACN,UAAU,GAAG,KAAK,CAAC;QACrB,CAAC;QAED,OAAO;YACL,OAAO,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI;YAC3D,SAAS;YACT,UAAU;SACX,CAAC;IACJ,CAAC;IAEO,aAAa,CACnB,MAA+B;QAE/B,OAAO,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAA8B,CAAC;IAC1E,CAAC;IAED,2EAA2E;IACnE,sBAAsB,CAAC,GAAY;QACzC,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;YACvB,OAAO,GAAG,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAC,CAAC;QAC9D,CAAC;QACD,IAAI,GAAG,KAAK,IAAI,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;YAC5C,MAAM,MAAM,GAA4B,EAAE,CAAC;YAC3C,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC;gBAC1B,SAAS,EAAE,sBAAsB,EAAE,KAAK,EAAE,MAAM;gBAChD,UAAU,EAAE,OAAO,EAAE,aAAa;aACnC,CAAC,CAAC;YACH,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAA8B,CAAC,EAAE,CAAC;gBAC1E,IAAI,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC;oBAAE,SAAS;gBACnC,MAAM,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC;YACnD,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAEO,eAAe,CAAC,QAAmB;QACzC,mEAAmE;QACnE,MAAM,WAAW,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAEpD,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACxB,IAAI,IAAY,CAAC;YACjB,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBAC3B,IAAI,GAAG,OAAO,CAAC;YACjB,CAAC;iBAAM,IACL,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;gBACxB,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAkB,CAAC,IAAI,KAAK,aAAa,CAAC,EACjE,CAAC;gBACD,iEAAiE;gBACjE,IAAI,GAAG,UAAU,CAAC;YACpB,CAAC;iBAAM,CAAC;gBACN,IAAI,GAAG,MAAM,CAAC;YAChB,CAAC;YACD,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC,EAAE,WAAW,CAAC,EAAE,CAAC;QAC9D,CAAC,CAAC,CAAC;IACL,CAAC;IAED,8DAA8D;IACtD,gBAAgB,CAAC,QAAmB;QAC1C,MAAM,GAAG,GAAG,IAAI,GAAG,EAAkB,CAAC;QACtC,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ;gBAAE,SAAS;YAC9C,KAAK,MAAM,KAAK,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;gBAChC,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAC9B,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;gBAChC,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAEO,cAAc,CACpB,GAAY,EACZ,cAAmC,IAAI,GAAG,EAAE;QAE5C,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YACpC,OAAO,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;QACjC,CAAC;QAED,MAAM,MAAM,GAAG,GAAG,CAAC,OAAyB,CAAC;QAC7C,MAAM,KAAK,GAAW,EAAE,CAAC;QAEzB,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;gBACnB,KAAK,MAAM;oBACT,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;oBACjC,MAAM;gBACR,KAAK,UAAU,CAAC,CAAC,CAAC;oBAChB,MAAM,MAAM,GAA4B;wBACtC,YAAY,EAAE;4BACZ,IAAI,EAAE,KAAK,CAAC,IAAI;4BAChB,IAAI,EAAE,KAAK,CAAC,KAAK;yBAClB;qBACF,CAAC;oBACF,yCAAyC;oBACzC,IAAI,KAAK,CAAC,QAAQ,EAAE,gBAAgB,EAAE,CAAC;wBACrC,MAAM,CAAC,gBAAgB,GAAG,KAAK,CAAC,QAAQ,CAAC,gBAAgB,CAAC;oBAC5D,CAAC;oBACD,KAAK,CAAC,IAAI,CAAC,MAAyB,CAAC,CAAC;oBACtC,MAAM;gBACR,CAAC;gBACD,KAAK,aAAa;oBAChB,KAAK,CAAC,IAAI,CAAC;wBACT,gBAAgB,EAAE;4BAChB,IAAI,EAAE,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,SAAS;4BACrD,QAAQ,EAAE,EAAE,MAAM,EAAE,KAAK,CAAC,OAAO,EAAE;yBACpC;qBACF,CAAC,CAAC;oBACH,MAAM;YACV,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;IACnD,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAK9C,wBAAgB,cAAc,CAC5B,QAAQ,EAAE,QAAQ,GAAG,QAAQ,GAAG,QAAQ,EACxC,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,MAAM,GACb,WAAW,CASb"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import { ClaudeProvider } from './claude.js';
|
|
2
|
+
import { OpenAIProvider } from './openai.js';
|
|
3
|
+
import { GeminiProvider } from './gemini.js';
|
|
4
|
+
export function createProvider(provider, model, apiKey) {
|
|
5
|
+
switch (provider) {
|
|
6
|
+
case 'claude':
|
|
7
|
+
return new ClaudeProvider(apiKey, model);
|
|
8
|
+
case 'openai':
|
|
9
|
+
return new OpenAIProvider(apiKey, model);
|
|
10
|
+
case 'gemini':
|
|
11
|
+
return new GeminiProvider(apiKey, model);
|
|
12
|
+
}
|
|
13
|
+
}
|
|
14
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAE7C,MAAM,UAAU,cAAc,CAC5B,QAAwC,EACxC,KAAa,EACb,MAAc;IAEd,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,QAAQ;YACX,OAAO,IAAI,cAAc,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAC3C,KAAK,QAAQ;YACX,OAAO,IAAI,cAAc,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAC3C,KAAK,QAAQ;YACX,OAAO,IAAI,cAAc,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAC7C,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { LLMProvider, LLMResponse, Message, ToolDefinition } from './types.js';
|
|
2
|
+
export declare class OpenAIProvider implements LLMProvider {
|
|
3
|
+
private client;
|
|
4
|
+
private model;
|
|
5
|
+
constructor(apiKey: string, model: string);
|
|
6
|
+
chat(systemPrompt: string, messages: Message[], tools: ToolDefinition[]): Promise<LLMResponse>;
|
|
7
|
+
/** Convert one internal Message into one or more OpenAI messages.
|
|
8
|
+
* A user message containing multiple tool_results expands to
|
|
9
|
+
* one `role: 'tool'` message per result (OpenAI requirement). */
|
|
10
|
+
private convertMessage;
|
|
11
|
+
}
|
|
12
|
+
//# sourceMappingURL=openai.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../src/providers/openai.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,WAAW,EACX,WAAW,EACX,OAAO,EACP,cAAc,EAGf,MAAM,YAAY,CAAC;AAEpB,qBAAa,cAAe,YAAW,WAAW;IAChD,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAS;gBAEV,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM;IAKnC,IAAI,CACR,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,OAAO,EAAE,EACnB,KAAK,EAAE,cAAc,EAAE,GACtB,OAAO,CAAC,WAAW,CAAC;IAuDvB;;sEAEkE;IAClE,OAAO,CAAC,cAAc;CA2DvB"}
|