fusion-agent 1.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 +467 -0
- package/REQUIREMENTS.md +158 -0
- package/dist/cli.d.ts +3 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +500 -0
- package/dist/cli.js.map +1 -0
- package/dist/cluster-monitor/cluster-monitor.d.ts +49 -0
- package/dist/cluster-monitor/cluster-monitor.d.ts.map +1 -0
- package/dist/cluster-monitor/cluster-monitor.js +374 -0
- package/dist/cluster-monitor/cluster-monitor.js.map +1 -0
- package/dist/cluster-monitor/index.d.ts +9 -0
- package/dist/cluster-monitor/index.d.ts.map +1 -0
- package/dist/cluster-monitor/index.js +19 -0
- package/dist/cluster-monitor/index.js.map +1 -0
- package/dist/cluster-monitor/kubernetes-connector.d.ts +21 -0
- package/dist/cluster-monitor/kubernetes-connector.d.ts.map +1 -0
- package/dist/cluster-monitor/kubernetes-connector.js +109 -0
- package/dist/cluster-monitor/kubernetes-connector.js.map +1 -0
- package/dist/cluster-monitor/notifications.d.ts +26 -0
- package/dist/cluster-monitor/notifications.d.ts.map +1 -0
- package/dist/cluster-monitor/notifications.js +182 -0
- package/dist/cluster-monitor/notifications.js.map +1 -0
- package/dist/cluster-monitor/remediation.d.ts +33 -0
- package/dist/cluster-monitor/remediation.d.ts.map +1 -0
- package/dist/cluster-monitor/remediation.js +134 -0
- package/dist/cluster-monitor/remediation.js.map +1 -0
- package/dist/cluster-monitor/rules.d.ts +26 -0
- package/dist/cluster-monitor/rules.d.ts.map +1 -0
- package/dist/cluster-monitor/rules.js +133 -0
- package/dist/cluster-monitor/rules.js.map +1 -0
- package/dist/cluster-monitor/types.d.ts +184 -0
- package/dist/cluster-monitor/types.d.ts.map +1 -0
- package/dist/cluster-monitor/types.js +3 -0
- package/dist/cluster-monitor/types.js.map +1 -0
- package/dist/index.d.ts +47 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +115 -0
- package/dist/index.js.map +1 -0
- package/dist/live-debugger/index.d.ts +32 -0
- package/dist/live-debugger/index.d.ts.map +1 -0
- package/dist/live-debugger/index.js +119 -0
- package/dist/live-debugger/index.js.map +1 -0
- package/dist/live-debugger/log-watcher.d.ts +19 -0
- package/dist/live-debugger/log-watcher.d.ts.map +1 -0
- package/dist/live-debugger/log-watcher.js +77 -0
- package/dist/live-debugger/log-watcher.js.map +1 -0
- package/dist/live-debugger/service-connector.d.ts +33 -0
- package/dist/live-debugger/service-connector.d.ts.map +1 -0
- package/dist/live-debugger/service-connector.js +123 -0
- package/dist/live-debugger/service-connector.js.map +1 -0
- package/dist/providers/anthropic.d.ts +9 -0
- package/dist/providers/anthropic.d.ts.map +1 -0
- package/dist/providers/anthropic.js +71 -0
- package/dist/providers/anthropic.js.map +1 -0
- package/dist/providers/base.d.ts +29 -0
- package/dist/providers/base.d.ts.map +1 -0
- package/dist/providers/base.js +14 -0
- package/dist/providers/base.js.map +1 -0
- package/dist/providers/gemini.d.ts +11 -0
- package/dist/providers/gemini.d.ts.map +1 -0
- package/dist/providers/gemini.js +73 -0
- package/dist/providers/gemini.js.map +1 -0
- package/dist/providers/index.d.ts +14 -0
- package/dist/providers/index.d.ts.map +1 -0
- package/dist/providers/index.js +38 -0
- package/dist/providers/index.js.map +1 -0
- package/dist/providers/openai.d.ts +9 -0
- package/dist/providers/openai.d.ts.map +1 -0
- package/dist/providers/openai.js +59 -0
- package/dist/providers/openai.js.map +1 -0
- package/dist/session/guardrails.d.ts +20 -0
- package/dist/session/guardrails.d.ts.map +1 -0
- package/dist/session/guardrails.js +83 -0
- package/dist/session/guardrails.js.map +1 -0
- package/dist/session/index.d.ts +6 -0
- package/dist/session/index.d.ts.map +1 -0
- package/dist/session/index.js +12 -0
- package/dist/session/index.js.map +1 -0
- package/dist/session/session-manager.d.ts +15 -0
- package/dist/session/session-manager.d.ts.map +1 -0
- package/dist/session/session-manager.js +89 -0
- package/dist/session/session-manager.js.map +1 -0
- package/dist/session/session.d.ts +63 -0
- package/dist/session/session.d.ts.map +1 -0
- package/dist/session/session.js +123 -0
- package/dist/session/session.js.map +1 -0
- package/dist/speckits/base.d.ts +9 -0
- package/dist/speckits/base.d.ts.map +1 -0
- package/dist/speckits/base.js +3 -0
- package/dist/speckits/base.js.map +1 -0
- package/dist/speckits/cluster-debugger.d.ts +3 -0
- package/dist/speckits/cluster-debugger.d.ts.map +1 -0
- package/dist/speckits/cluster-debugger.js +48 -0
- package/dist/speckits/cluster-debugger.js.map +1 -0
- package/dist/speckits/code-review.d.ts +3 -0
- package/dist/speckits/code-review.d.ts.map +1 -0
- package/dist/speckits/code-review.js +41 -0
- package/dist/speckits/code-review.js.map +1 -0
- package/dist/speckits/debugger.d.ts +3 -0
- package/dist/speckits/debugger.d.ts.map +1 -0
- package/dist/speckits/debugger.js +38 -0
- package/dist/speckits/debugger.js.map +1 -0
- package/dist/speckits/index.d.ts +11 -0
- package/dist/speckits/index.d.ts.map +1 -0
- package/dist/speckits/index.js +40 -0
- package/dist/speckits/index.js.map +1 -0
- package/dist/speckits/more-speckits.d.ts +6 -0
- package/dist/speckits/more-speckits.d.ts.map +1 -0
- package/dist/speckits/more-speckits.js +136 -0
- package/dist/speckits/more-speckits.js.map +1 -0
- package/dist/speckits/vibe-coder.d.ts +3 -0
- package/dist/speckits/vibe-coder.d.ts.map +1 -0
- package/dist/speckits/vibe-coder.js +32 -0
- package/dist/speckits/vibe-coder.js.map +1 -0
- package/dist/utils/config.d.ts +18 -0
- package/dist/utils/config.d.ts.map +1 -0
- package/dist/utils/config.js +107 -0
- package/dist/utils/config.js.map +1 -0
- package/dist/utils/file-ops.d.ts +15 -0
- package/dist/utils/file-ops.d.ts.map +1 -0
- package/dist/utils/file-ops.js +109 -0
- package/dist/utils/file-ops.js.map +1 -0
- package/dist/utils/logger.d.ts +4 -0
- package/dist/utils/logger.d.ts.map +1 -0
- package/dist/utils/logger.js +25 -0
- package/dist/utils/logger.js.map +1 -0
- package/dist/web/public/css/app.css +370 -0
- package/dist/web/public/index.html +91 -0
- package/dist/web/public/js/app.js +212 -0
- package/dist/web/routes/sessions.d.ts +4 -0
- package/dist/web/routes/sessions.d.ts.map +1 -0
- package/dist/web/routes/sessions.js +51 -0
- package/dist/web/routes/sessions.js.map +1 -0
- package/dist/web/routes/settings.d.ts +3 -0
- package/dist/web/routes/settings.d.ts.map +1 -0
- package/dist/web/routes/settings.js +33 -0
- package/dist/web/routes/settings.js.map +1 -0
- package/dist/web/server.d.ts +16 -0
- package/dist/web/server.d.ts.map +1 -0
- package/dist/web/server.js +76 -0
- package/dist/web/server.js.map +1 -0
- package/package.json +86 -0
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
import { EventEmitter } from 'events';
|
|
2
|
+
export type ServiceConnectionType = 'process' | 'docker' | 'http-poll';
|
|
3
|
+
export interface ProcessConnectionOptions {
|
|
4
|
+
type: 'process';
|
|
5
|
+
command: string;
|
|
6
|
+
args?: string[];
|
|
7
|
+
cwd?: string;
|
|
8
|
+
}
|
|
9
|
+
export interface DockerConnectionOptions {
|
|
10
|
+
type: 'docker';
|
|
11
|
+
container: string;
|
|
12
|
+
tail?: number;
|
|
13
|
+
}
|
|
14
|
+
export interface HttpPollConnectionOptions {
|
|
15
|
+
type: 'http-poll';
|
|
16
|
+
url: string;
|
|
17
|
+
intervalMs?: number;
|
|
18
|
+
headers?: Record<string, string>;
|
|
19
|
+
}
|
|
20
|
+
export type ServiceConnectionOptions = ProcessConnectionOptions | DockerConnectionOptions | HttpPollConnectionOptions;
|
|
21
|
+
export declare class ServiceConnector extends EventEmitter {
|
|
22
|
+
private options;
|
|
23
|
+
private child?;
|
|
24
|
+
private pollInterval?;
|
|
25
|
+
private running;
|
|
26
|
+
constructor(options: ServiceConnectionOptions);
|
|
27
|
+
start(): void;
|
|
28
|
+
private startProcess;
|
|
29
|
+
private startDocker;
|
|
30
|
+
private startHttpPoll;
|
|
31
|
+
stop(): void;
|
|
32
|
+
}
|
|
33
|
+
//# sourceMappingURL=service-connector.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"service-connector.d.ts","sourceRoot":"","sources":["../../src/live-debugger/service-connector.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAKtC,MAAM,MAAM,qBAAqB,GAAG,SAAS,GAAG,QAAQ,GAAG,WAAW,CAAC;AAEvE,MAAM,WAAW,wBAAwB;IACvC,IAAI,EAAE,SAAS,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,GAAG,CAAC,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,uBAAuB;IACtC,IAAI,EAAE,QAAQ,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,yBAAyB;IACxC,IAAI,EAAE,WAAW,CAAC;IAClB,GAAG,EAAE,MAAM,CAAC;IACZ,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAClC;AAED,MAAM,MAAM,wBAAwB,GAChC,wBAAwB,GACxB,uBAAuB,GACvB,yBAAyB,CAAC;AAE9B,qBAAa,gBAAiB,SAAQ,YAAY;IAChD,OAAO,CAAC,OAAO,CAA2B;IAC1C,OAAO,CAAC,KAAK,CAAC,CAAe;IAC7B,OAAO,CAAC,YAAY,CAAC,CAAiC;IACtD,OAAO,CAAC,OAAO,CAAS;gBAEZ,OAAO,EAAE,wBAAwB;IAK7C,KAAK,IAAI,IAAI;IAiBb,OAAO,CAAC,YAAY;IAiCpB,OAAO,CAAC,WAAW;IAwBnB,OAAO,CAAC,aAAa;IA0BrB,IAAI,IAAI,IAAI;CAWb"}
|
|
@@ -0,0 +1,123 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.ServiceConnector = void 0;
|
|
7
|
+
const events_1 = require("events");
|
|
8
|
+
const child_process_1 = require("child_process");
|
|
9
|
+
const axios_1 = __importDefault(require("axios"));
|
|
10
|
+
const logger_1 = require("../utils/logger");
|
|
11
|
+
class ServiceConnector extends events_1.EventEmitter {
|
|
12
|
+
constructor(options) {
|
|
13
|
+
super();
|
|
14
|
+
this.running = false;
|
|
15
|
+
this.options = options;
|
|
16
|
+
}
|
|
17
|
+
start() {
|
|
18
|
+
if (this.running)
|
|
19
|
+
return;
|
|
20
|
+
this.running = true;
|
|
21
|
+
switch (this.options.type) {
|
|
22
|
+
case 'process':
|
|
23
|
+
this.startProcess(this.options);
|
|
24
|
+
break;
|
|
25
|
+
case 'docker':
|
|
26
|
+
this.startDocker(this.options);
|
|
27
|
+
break;
|
|
28
|
+
case 'http-poll':
|
|
29
|
+
this.startHttpPoll(this.options);
|
|
30
|
+
break;
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
startProcess(opts) {
|
|
34
|
+
logger_1.logger.debug(`Attaching to process: ${opts.command} ${(opts.args || []).join(' ')}`);
|
|
35
|
+
this.child = (0, child_process_1.spawn)(opts.command, opts.args || [], {
|
|
36
|
+
cwd: opts.cwd,
|
|
37
|
+
stdio: ['ignore', 'pipe', 'pipe'],
|
|
38
|
+
shell: true,
|
|
39
|
+
});
|
|
40
|
+
let stdoutBuffer = '';
|
|
41
|
+
let stderrBuffer = '';
|
|
42
|
+
const processBuffer = (buffer, data) => {
|
|
43
|
+
const combined = buffer + data.toString('utf-8');
|
|
44
|
+
const lines = combined.split('\n');
|
|
45
|
+
const remainder = lines.pop() || '';
|
|
46
|
+
for (const line of lines) {
|
|
47
|
+
if (line.trim())
|
|
48
|
+
this.emit('line', line);
|
|
49
|
+
}
|
|
50
|
+
return remainder;
|
|
51
|
+
};
|
|
52
|
+
this.child.stdout?.on('data', (data) => {
|
|
53
|
+
stdoutBuffer = processBuffer(stdoutBuffer, data);
|
|
54
|
+
});
|
|
55
|
+
this.child.stderr?.on('data', (data) => {
|
|
56
|
+
stderrBuffer = processBuffer(stderrBuffer, data);
|
|
57
|
+
});
|
|
58
|
+
this.child.on('exit', (code) => {
|
|
59
|
+
this.emit('exit', code);
|
|
60
|
+
this.running = false;
|
|
61
|
+
});
|
|
62
|
+
}
|
|
63
|
+
startDocker(opts) {
|
|
64
|
+
const tail = opts.tail ?? 100;
|
|
65
|
+
logger_1.logger.debug(`Attaching to Docker container: ${opts.container}`);
|
|
66
|
+
this.child = (0, child_process_1.spawn)('docker', ['logs', '-f', '--tail', String(tail), opts.container], {
|
|
67
|
+
stdio: ['ignore', 'pipe', 'pipe'],
|
|
68
|
+
});
|
|
69
|
+
let buffer = '';
|
|
70
|
+
const processData = (data) => {
|
|
71
|
+
buffer += data.toString('utf-8');
|
|
72
|
+
const lines = buffer.split('\n');
|
|
73
|
+
buffer = lines.pop() || '';
|
|
74
|
+
for (const line of lines) {
|
|
75
|
+
if (line.trim())
|
|
76
|
+
this.emit('line', line);
|
|
77
|
+
}
|
|
78
|
+
};
|
|
79
|
+
this.child.stdout?.on('data', processData);
|
|
80
|
+
this.child.stderr?.on('data', processData);
|
|
81
|
+
this.child.on('exit', (code) => {
|
|
82
|
+
this.emit('exit', code);
|
|
83
|
+
this.running = false;
|
|
84
|
+
});
|
|
85
|
+
}
|
|
86
|
+
startHttpPoll(opts) {
|
|
87
|
+
const interval = opts.intervalMs ?? 5000;
|
|
88
|
+
logger_1.logger.debug(`HTTP polling: ${opts.url} every ${interval}ms`);
|
|
89
|
+
let lastStatus = '';
|
|
90
|
+
const poll = async () => {
|
|
91
|
+
try {
|
|
92
|
+
const response = await axios_1.default.get(opts.url, {
|
|
93
|
+
headers: opts.headers,
|
|
94
|
+
timeout: interval - 500,
|
|
95
|
+
});
|
|
96
|
+
const statusLine = `HTTP ${response.status} - ${JSON.stringify(response.data).slice(0, 200)}`;
|
|
97
|
+
if (statusLine !== lastStatus) {
|
|
98
|
+
this.emit('line', statusLine);
|
|
99
|
+
lastStatus = statusLine;
|
|
100
|
+
}
|
|
101
|
+
}
|
|
102
|
+
catch (err) {
|
|
103
|
+
const msg = err instanceof Error ? err.message : String(err);
|
|
104
|
+
this.emit('line', `HTTP ERROR: ${msg}`);
|
|
105
|
+
}
|
|
106
|
+
};
|
|
107
|
+
void poll();
|
|
108
|
+
this.pollInterval = setInterval(() => { void poll(); }, interval);
|
|
109
|
+
}
|
|
110
|
+
stop() {
|
|
111
|
+
this.running = false;
|
|
112
|
+
if (this.child) {
|
|
113
|
+
this.child.kill('SIGTERM');
|
|
114
|
+
this.child = undefined;
|
|
115
|
+
}
|
|
116
|
+
if (this.pollInterval) {
|
|
117
|
+
clearInterval(this.pollInterval);
|
|
118
|
+
this.pollInterval = undefined;
|
|
119
|
+
}
|
|
120
|
+
}
|
|
121
|
+
}
|
|
122
|
+
exports.ServiceConnector = ServiceConnector;
|
|
123
|
+
//# sourceMappingURL=service-connector.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"service-connector.js","sourceRoot":"","sources":["../../src/live-debugger/service-connector.ts"],"names":[],"mappings":";;;;;;AAAA,mCAAsC;AACtC,iDAAoD;AACpD,kDAA0B;AAC1B,4CAAyC;AA6BzC,MAAa,gBAAiB,SAAQ,qBAAY;IAMhD,YAAY,OAAiC;QAC3C,KAAK,EAAE,CAAC;QAHF,YAAO,GAAG,KAAK,CAAC;QAItB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,KAAK;QACH,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QAEpB,QAAQ,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;YAC1B,KAAK,SAAS;gBACZ,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAChC,MAAM;YACR,KAAK,QAAQ;gBACX,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC/B,MAAM;YACR,KAAK,WAAW;gBACd,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACjC,MAAM;QACV,CAAC;IACH,CAAC;IAEO,YAAY,CAAC,IAA8B;QACjD,eAAM,CAAC,KAAK,CAAC,yBAAyB,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACrF,IAAI,CAAC,KAAK,GAAG,IAAA,qBAAK,EAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,IAAI,IAAI,EAAE,EAAE;YAChD,GAAG,EAAE,IAAI,CAAC,GAAG;YACb,KAAK,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC;YACjC,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QAEH,IAAI,YAAY,GAAG,EAAE,CAAC;QACtB,IAAI,YAAY,GAAG,EAAE,CAAC;QAEtB,MAAM,aAAa,GAAG,CAAC,MAAc,EAAE,IAAY,EAAU,EAAE;YAC7D,MAAM,QAAQ,GAAG,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YACjD,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACnC,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;YACpC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,IAAI,CAAC,IAAI,EAAE;oBAAE,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YAC3C,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC,CAAC;QAEF,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;YAC7C,YAAY,GAAG,aAAa,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;YAC7C,YAAY,GAAG,aAAa,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YAC7B,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YACxB,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;QACvB,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,WAAW,CAAC,IAA6B;QAC/C,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,GAAG,CAAC;QAC9B,eAAM,CAAC,KAAK,CAAC,kCAAkC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;QACjE,IAAI,CAAC,KAAK,GAAG,IAAA,qBAAK,EAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE;YACnF,KAAK,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC;SAClC,CAAC,CAAC;QAEH,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,MAAM,WAAW,GAAG,CAAC,IAAY,EAAQ,EAAE;YACzC,MAAM,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YACjC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;YAC3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,IAAI,CAAC,IAAI,EAAE;oBAAE,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAC3C,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAC3C,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YAC7B,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YACxB,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;QACvB,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,aAAa,CAAC,IAA+B;QACnD,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC;QACzC,eAAM,CAAC,KAAK,CAAC,iBAAiB,IAAI,CAAC,GAAG,UAAU,QAAQ,IAAI,CAAC,CAAC;QAE9D,IAAI,UAAU,GAAG,EAAE,CAAC;QACpB,MAAM,IAAI,GAAG,KAAK,IAAmB,EAAE;YACrC,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,eAAK,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE;oBACzC,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,OAAO,EAAE,QAAQ,GAAG,GAAG;iBACxB,CAAC,CAAC;gBACH,MAAM,UAAU,GAAG,QAAQ,QAAQ,CAAC,MAAM,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC;gBAC9F,IAAI,UAAU,KAAK,UAAU,EAAE,CAAC;oBAC9B,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;oBAC9B,UAAU,GAAG,UAAU,CAAC;gBAC1B,CAAC;YACH,CAAC;YAAC,OAAO,GAAY,EAAE,CAAC;gBACtB,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBAC7D,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,eAAe,GAAG,EAAE,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC,CAAC;QAEF,KAAK,IAAI,EAAE,CAAC;QACZ,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;IACpE,CAAC;IAED,IAAI;QACF,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;QACrB,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC3B,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;QACzB,CAAC;QACD,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACjC,IAAI,CAAC,YAAY,GAAG,SAAS,CAAC;QAChC,CAAC;IACH,CAAC;CACF;AA1HD,4CA0HC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import { BaseProvider, CompletionOptions, CompletionResult } from './base';
|
|
2
|
+
export declare const ANTHROPIC_MODELS: readonly ["claude-3-5-sonnet-20241022", "claude-3-5-haiku-20241022", "claude-3-opus-20240229"];
|
|
3
|
+
export declare const DEFAULT_ANTHROPIC_MODEL = "claude-3-5-sonnet-20241022";
|
|
4
|
+
export declare class AnthropicProvider extends BaseProvider {
|
|
5
|
+
private client;
|
|
6
|
+
constructor(apiKey: string, model?: string);
|
|
7
|
+
complete(options: CompletionOptions): Promise<CompletionResult>;
|
|
8
|
+
}
|
|
9
|
+
//# sourceMappingURL=anthropic.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../src/providers/anthropic.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,iBAAiB,EAAE,gBAAgB,EAAW,MAAM,QAAQ,CAAC;AAEpF,eAAO,MAAM,gBAAgB,gGAAiG,CAAC;AAC/H,eAAO,MAAM,uBAAuB,+BAA+B,CAAC;AAEpE,qBAAa,iBAAkB,SAAQ,YAAY;IACjD,OAAO,CAAC,MAAM,CAAY;gBAEd,MAAM,EAAE,MAAM,EAAE,KAAK,SAA0B;IAKrD,QAAQ,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC;CA0DtE"}
|
|
@@ -0,0 +1,71 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.AnthropicProvider = exports.DEFAULT_ANTHROPIC_MODEL = exports.ANTHROPIC_MODELS = void 0;
|
|
7
|
+
const sdk_1 = __importDefault(require("@anthropic-ai/sdk"));
|
|
8
|
+
const base_1 = require("./base");
|
|
9
|
+
exports.ANTHROPIC_MODELS = ['claude-3-5-sonnet-20241022', 'claude-3-5-haiku-20241022', 'claude-3-opus-20240229'];
|
|
10
|
+
exports.DEFAULT_ANTHROPIC_MODEL = 'claude-3-5-sonnet-20241022';
|
|
11
|
+
class AnthropicProvider extends base_1.BaseProvider {
|
|
12
|
+
constructor(apiKey, model = exports.DEFAULT_ANTHROPIC_MODEL) {
|
|
13
|
+
super(apiKey, model);
|
|
14
|
+
this.client = new sdk_1.default({ apiKey });
|
|
15
|
+
}
|
|
16
|
+
async complete(options) {
|
|
17
|
+
const model = options.model || this.model;
|
|
18
|
+
const maxTokens = options.maxTokens || 4096;
|
|
19
|
+
const temperature = options.temperature ?? 0.7;
|
|
20
|
+
// Anthropic separates system prompt from conversation messages
|
|
21
|
+
const systemMessages = options.messages.filter((m) => m.role === 'system');
|
|
22
|
+
const conversationMessages = options.messages.filter((m) => m.role !== 'system');
|
|
23
|
+
const systemPrompt = systemMessages.map((m) => m.content).join('\n\n');
|
|
24
|
+
if (options.stream && options.onChunk) {
|
|
25
|
+
const stream = await this.client.messages.stream({
|
|
26
|
+
model,
|
|
27
|
+
max_tokens: maxTokens,
|
|
28
|
+
temperature,
|
|
29
|
+
system: systemPrompt || undefined,
|
|
30
|
+
messages: conversationMessages,
|
|
31
|
+
});
|
|
32
|
+
let fullContent = '';
|
|
33
|
+
for await (const chunk of stream) {
|
|
34
|
+
if (chunk.type === 'content_block_delta' && chunk.delta.type === 'text_delta') {
|
|
35
|
+
fullContent += chunk.delta.text;
|
|
36
|
+
options.onChunk(chunk.delta.text);
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
const finalMsg = await stream.finalMessage();
|
|
40
|
+
return {
|
|
41
|
+
content: fullContent,
|
|
42
|
+
model: finalMsg.model,
|
|
43
|
+
usage: {
|
|
44
|
+
promptTokens: finalMsg.usage.input_tokens,
|
|
45
|
+
completionTokens: finalMsg.usage.output_tokens,
|
|
46
|
+
totalTokens: finalMsg.usage.input_tokens + finalMsg.usage.output_tokens,
|
|
47
|
+
},
|
|
48
|
+
};
|
|
49
|
+
}
|
|
50
|
+
const response = await this.client.messages.create({
|
|
51
|
+
model,
|
|
52
|
+
max_tokens: maxTokens,
|
|
53
|
+
temperature,
|
|
54
|
+
system: systemPrompt || undefined,
|
|
55
|
+
messages: conversationMessages,
|
|
56
|
+
});
|
|
57
|
+
const textBlock = response.content.find((b) => b.type === 'text');
|
|
58
|
+
const content = textBlock && textBlock.type === 'text' ? textBlock.text : '';
|
|
59
|
+
return {
|
|
60
|
+
content,
|
|
61
|
+
model: response.model,
|
|
62
|
+
usage: {
|
|
63
|
+
promptTokens: response.usage.input_tokens,
|
|
64
|
+
completionTokens: response.usage.output_tokens,
|
|
65
|
+
totalTokens: response.usage.input_tokens + response.usage.output_tokens,
|
|
66
|
+
},
|
|
67
|
+
};
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
exports.AnthropicProvider = AnthropicProvider;
|
|
71
|
+
//# sourceMappingURL=anthropic.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../src/providers/anthropic.ts"],"names":[],"mappings":";;;;;;AAAA,4DAA0C;AAC1C,iCAAoF;AAEvE,QAAA,gBAAgB,GAAG,CAAC,4BAA4B,EAAE,2BAA2B,EAAE,wBAAwB,CAAU,CAAC;AAClH,QAAA,uBAAuB,GAAG,4BAA4B,CAAC;AAEpE,MAAa,iBAAkB,SAAQ,mBAAY;IAGjD,YAAY,MAAc,EAAE,KAAK,GAAG,+BAAuB;QACzD,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,IAAI,aAAS,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAA0B;QACvC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC;QAC1C,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC;QAC5C,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,GAAG,CAAC;QAE/C,+DAA+D;QAC/D,MAAM,cAAc,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QACpF,MAAM,oBAAoB,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QAC1F,MAAM,YAAY,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEhF,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAC/C,KAAK;gBACL,UAAU,EAAE,SAAS;gBACrB,WAAW;gBACX,MAAM,EAAE,YAAY,IAAI,SAAS;gBACjC,QAAQ,EAAE,oBAAgD;aAC3D,CAAC,CAAC;YAEH,IAAI,WAAW,GAAG,EAAE,CAAC;YACrB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBACjC,IAAI,KAAK,CAAC,IAAI,KAAK,qBAAqB,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;oBAC9E,WAAW,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC;oBAChC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACpC,CAAC;YACH,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,YAAY,EAAE,CAAC;YAC7C,OAAO;gBACL,OAAO,EAAE,WAAW;gBACpB,KAAK,EAAE,QAAQ,CAAC,KAAK;gBACrB,KAAK,EAAE;oBACL,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;oBACzC,gBAAgB,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa;oBAC9C,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,aAAa;iBACxE;aACF,CAAC;QACJ,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;YACjD,KAAK;YACL,UAAU,EAAE,SAAS;YACrB,WAAW;YACX,MAAM,EAAE,YAAY,IAAI,SAAS;YACjC,QAAQ,EAAE,oBAAgD;SAC3D,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;QAClE,MAAM,OAAO,GAAG,SAAS,IAAI,SAAS,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;QAC7E,OAAO;YACL,OAAO;YACP,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,KAAK,EAAE;gBACL,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;gBACzC,gBAAgB,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa;gBAC9C,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,aAAa;aACxE;SACF,CAAC;IACJ,CAAC;CACF;AAlED,8CAkEC"}
|
|
@@ -0,0 +1,29 @@
|
|
|
1
|
+
export interface Message {
|
|
2
|
+
role: 'system' | 'user' | 'assistant';
|
|
3
|
+
content: string;
|
|
4
|
+
}
|
|
5
|
+
export interface CompletionOptions {
|
|
6
|
+
messages: Message[];
|
|
7
|
+
model?: string;
|
|
8
|
+
maxTokens?: number;
|
|
9
|
+
temperature?: number;
|
|
10
|
+
stream?: boolean;
|
|
11
|
+
onChunk?: (chunk: string) => void;
|
|
12
|
+
}
|
|
13
|
+
export interface CompletionResult {
|
|
14
|
+
content: string;
|
|
15
|
+
model: string;
|
|
16
|
+
usage?: {
|
|
17
|
+
promptTokens: number;
|
|
18
|
+
completionTokens: number;
|
|
19
|
+
totalTokens: number;
|
|
20
|
+
};
|
|
21
|
+
}
|
|
22
|
+
export declare abstract class BaseProvider {
|
|
23
|
+
protected model: string;
|
|
24
|
+
protected apiKey: string;
|
|
25
|
+
constructor(apiKey: string, model: string);
|
|
26
|
+
abstract complete(options: CompletionOptions): Promise<CompletionResult>;
|
|
27
|
+
getModel(): string;
|
|
28
|
+
}
|
|
29
|
+
//# sourceMappingURL=base.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../src/providers/base.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,OAAO;IACtB,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,CAAC;IACtC,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,iBAAiB;IAChC,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;CACnC;AAED,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE;QACN,YAAY,EAAE,MAAM,CAAC;QACrB,gBAAgB,EAAE,MAAM,CAAC;QACzB,WAAW,EAAE,MAAM,CAAC;KACrB,CAAC;CACH;AAED,8BAAsB,YAAY;IAChC,SAAS,CAAC,KAAK,EAAE,MAAM,CAAC;IACxB,SAAS,CAAC,MAAM,EAAE,MAAM,CAAC;gBAEb,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM;IAKzC,QAAQ,CAAC,QAAQ,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAExE,QAAQ,IAAI,MAAM;CAGnB"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.BaseProvider = void 0;
|
|
4
|
+
class BaseProvider {
|
|
5
|
+
constructor(apiKey, model) {
|
|
6
|
+
this.apiKey = apiKey;
|
|
7
|
+
this.model = model;
|
|
8
|
+
}
|
|
9
|
+
getModel() {
|
|
10
|
+
return this.model;
|
|
11
|
+
}
|
|
12
|
+
}
|
|
13
|
+
exports.BaseProvider = BaseProvider;
|
|
14
|
+
//# sourceMappingURL=base.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/providers/base.ts"],"names":[],"mappings":";;;AAwBA,MAAsB,YAAY;IAIhC,YAAY,MAAc,EAAE,KAAa;QACvC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAID,QAAQ;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;CACF;AAdD,oCAcC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import { BaseProvider, CompletionOptions, CompletionResult } from './base';
|
|
2
|
+
export declare const GEMINI_MODELS: readonly ["gemini-1.5-pro", "gemini-1.5-flash", "gemini-2.0-flash"];
|
|
3
|
+
export declare const DEFAULT_GEMINI_MODEL = "gemini-1.5-pro";
|
|
4
|
+
export declare class GeminiProvider extends BaseProvider {
|
|
5
|
+
private client;
|
|
6
|
+
private generativeModel;
|
|
7
|
+
constructor(apiKey: string, model?: string);
|
|
8
|
+
private convertMessages;
|
|
9
|
+
complete(options: CompletionOptions): Promise<CompletionResult>;
|
|
10
|
+
}
|
|
11
|
+
//# sourceMappingURL=gemini.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../src/providers/gemini.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,iBAAiB,EAAE,gBAAgB,EAAW,MAAM,QAAQ,CAAC;AAEpF,eAAO,MAAM,aAAa,qEAAsE,CAAC;AACjG,eAAO,MAAM,oBAAoB,mBAAmB,CAAC;AAErD,qBAAa,cAAe,SAAQ,YAAY;IAC9C,OAAO,CAAC,MAAM,CAAqB;IACnC,OAAO,CAAC,eAAe,CAAkB;gBAE7B,MAAM,EAAE,MAAM,EAAE,KAAK,SAAuB;IAMxD,OAAO,CAAC,eAAe;IAoBjB,QAAQ,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAgDtE"}
|
|
@@ -0,0 +1,73 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.GeminiProvider = exports.DEFAULT_GEMINI_MODEL = exports.GEMINI_MODELS = void 0;
|
|
4
|
+
const generative_ai_1 = require("@google/generative-ai");
|
|
5
|
+
const base_1 = require("./base");
|
|
6
|
+
exports.GEMINI_MODELS = ['gemini-1.5-pro', 'gemini-1.5-flash', 'gemini-2.0-flash'];
|
|
7
|
+
exports.DEFAULT_GEMINI_MODEL = 'gemini-1.5-pro';
|
|
8
|
+
class GeminiProvider extends base_1.BaseProvider {
|
|
9
|
+
constructor(apiKey, model = exports.DEFAULT_GEMINI_MODEL) {
|
|
10
|
+
super(apiKey, model);
|
|
11
|
+
this.client = new generative_ai_1.GoogleGenerativeAI(apiKey);
|
|
12
|
+
this.generativeModel = this.client.getGenerativeModel({ model });
|
|
13
|
+
}
|
|
14
|
+
convertMessages(messages) {
|
|
15
|
+
const systemMessages = messages.filter((m) => m.role === 'system');
|
|
16
|
+
const system = systemMessages.map((m) => m.content).join('\n\n');
|
|
17
|
+
const conversationMessages = messages.filter((m) => m.role !== 'system');
|
|
18
|
+
if (conversationMessages.length === 0) {
|
|
19
|
+
return { system, history: [], lastUserMsg: '' };
|
|
20
|
+
}
|
|
21
|
+
const lastMsg = conversationMessages[conversationMessages.length - 1];
|
|
22
|
+
const historyMessages = conversationMessages.slice(0, -1);
|
|
23
|
+
const history = historyMessages.map((m) => ({
|
|
24
|
+
role: m.role === 'assistant' ? 'model' : 'user',
|
|
25
|
+
parts: [{ text: m.content }],
|
|
26
|
+
}));
|
|
27
|
+
return { system, history, lastUserMsg: lastMsg.content };
|
|
28
|
+
}
|
|
29
|
+
async complete(options) {
|
|
30
|
+
const modelName = options.model || this.model;
|
|
31
|
+
const maxTokens = options.maxTokens || 4096;
|
|
32
|
+
const temperature = options.temperature ?? 0.7;
|
|
33
|
+
// Re-create model in case model override is different
|
|
34
|
+
const genModel = modelName !== this.model
|
|
35
|
+
? this.client.getGenerativeModel({ model: modelName })
|
|
36
|
+
: this.generativeModel;
|
|
37
|
+
const { system, history, lastUserMsg } = this.convertMessages(options.messages);
|
|
38
|
+
const fullPrompt = system ? `${system}\n\n${lastUserMsg}` : lastUserMsg;
|
|
39
|
+
const chat = genModel.startChat({
|
|
40
|
+
history,
|
|
41
|
+
generationConfig: {
|
|
42
|
+
maxOutputTokens: maxTokens,
|
|
43
|
+
temperature,
|
|
44
|
+
},
|
|
45
|
+
});
|
|
46
|
+
if (options.stream && options.onChunk) {
|
|
47
|
+
const result = await chat.sendMessageStream(fullPrompt);
|
|
48
|
+
let fullContent = '';
|
|
49
|
+
for await (const chunk of result.stream) {
|
|
50
|
+
const text = chunk.text();
|
|
51
|
+
fullContent += text;
|
|
52
|
+
options.onChunk(text);
|
|
53
|
+
}
|
|
54
|
+
return { content: fullContent, model: modelName };
|
|
55
|
+
}
|
|
56
|
+
const result = await chat.sendMessage(fullPrompt);
|
|
57
|
+
const content = result.response.text();
|
|
58
|
+
const usage = result.response.usageMetadata;
|
|
59
|
+
return {
|
|
60
|
+
content,
|
|
61
|
+
model: modelName,
|
|
62
|
+
usage: usage
|
|
63
|
+
? {
|
|
64
|
+
promptTokens: usage.promptTokenCount || 0,
|
|
65
|
+
completionTokens: usage.candidatesTokenCount || 0,
|
|
66
|
+
totalTokens: usage.totalTokenCount || 0,
|
|
67
|
+
}
|
|
68
|
+
: undefined,
|
|
69
|
+
};
|
|
70
|
+
}
|
|
71
|
+
}
|
|
72
|
+
exports.GeminiProvider = GeminiProvider;
|
|
73
|
+
//# sourceMappingURL=gemini.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../src/providers/gemini.ts"],"names":[],"mappings":";;;AAAA,yDAAqF;AACrF,iCAAoF;AAEvE,QAAA,aAAa,GAAG,CAAC,gBAAgB,EAAE,kBAAkB,EAAE,kBAAkB,CAAU,CAAC;AACpF,QAAA,oBAAoB,GAAG,gBAAgB,CAAC;AAErD,MAAa,cAAe,SAAQ,mBAAY;IAI9C,YAAY,MAAc,EAAE,KAAK,GAAG,4BAAoB;QACtD,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,IAAI,kCAAkB,CAAC,MAAM,CAAC,CAAC;QAC7C,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IACnE,CAAC;IAEO,eAAe,CAAC,QAAmB;QACzC,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QACnE,MAAM,MAAM,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjE,MAAM,oBAAoB,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QAEzE,IAAI,oBAAoB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;QAClD,CAAC;QAED,MAAM,OAAO,GAAG,oBAAoB,CAAC,oBAAoB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACtE,MAAM,eAAe,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAE1D,MAAM,OAAO,GAAc,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACrD,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,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAA0B;QACvC,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC;QAC9C,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC;QAC5C,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,GAAG,CAAC;QAE/C,sDAAsD;QACtD,MAAM,QAAQ,GAAG,SAAS,KAAK,IAAI,CAAC,KAAK;YACvC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;YACtD,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC;QAEzB,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAEhF,MAAM,UAAU,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,OAAO,WAAW,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC;QAExE,MAAM,IAAI,GAAG,QAAQ,CAAC,SAAS,CAAC;YAC9B,OAAO;YACP,gBAAgB,EAAE;gBAChB,eAAe,EAAE,SAAS;gBAC1B,WAAW;aACZ;SACF,CAAC,CAAC;QAEH,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YACxD,IAAI,WAAW,GAAG,EAAE,CAAC;YACrB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBACxC,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;gBAC1B,WAAW,IAAI,IAAI,CAAC;gBACpB,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QACpD,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;QAClD,MAAM,OAAO,GAAG,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QACvC,MAAM,KAAK,GAAG,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC;QAC5C,OAAO;YACL,OAAO;YACP,KAAK,EAAE,SAAS;YAChB,KAAK,EAAE,KAAK;gBACV,CAAC,CAAC;oBACE,YAAY,EAAE,KAAK,CAAC,gBAAgB,IAAI,CAAC;oBACzC,gBAAgB,EAAE,KAAK,CAAC,oBAAoB,IAAI,CAAC;oBACjD,WAAW,EAAE,KAAK,CAAC,eAAe,IAAI,CAAC;iBACxC;gBACH,CAAC,CAAC,SAAS;SACd,CAAC;IACJ,CAAC;CACF;AA9ED,wCA8EC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import { BaseProvider } from './base';
|
|
2
|
+
import { OpenAIProvider } from './openai';
|
|
3
|
+
import { AnthropicProvider } from './anthropic';
|
|
4
|
+
import { GeminiProvider } from './gemini';
|
|
5
|
+
export type ProviderName = 'openai' | 'anthropic' | 'gemini';
|
|
6
|
+
export interface ProviderOptions {
|
|
7
|
+
provider: ProviderName;
|
|
8
|
+
apiKey?: string;
|
|
9
|
+
model?: string;
|
|
10
|
+
}
|
|
11
|
+
export declare function createProvider(options: ProviderOptions): BaseProvider;
|
|
12
|
+
export { BaseProvider, OpenAIProvider, AnthropicProvider, GeminiProvider };
|
|
13
|
+
export type { CompletionOptions, CompletionResult, Message } from './base';
|
|
14
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,cAAc,EAAwB,MAAM,UAAU,CAAC;AAChE,OAAO,EAAE,iBAAiB,EAA2B,MAAM,aAAa,CAAC;AACzE,OAAO,EAAE,cAAc,EAAwB,MAAM,UAAU,CAAC;AAEhE,MAAM,MAAM,YAAY,GAAG,QAAQ,GAAG,WAAW,GAAG,QAAQ,CAAC;AAE7D,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,YAAY,CAAC;IACvB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,wBAAgB,cAAc,CAAC,OAAO,EAAE,eAAe,GAAG,YAAY,CAsBrE;AAED,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,iBAAiB,EAAE,cAAc,EAAE,CAAC;AAC3E,YAAY,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,OAAO,EAAE,MAAM,QAAQ,CAAC"}
|
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.GeminiProvider = exports.AnthropicProvider = exports.OpenAIProvider = exports.BaseProvider = void 0;
|
|
4
|
+
exports.createProvider = createProvider;
|
|
5
|
+
const base_1 = require("./base");
|
|
6
|
+
Object.defineProperty(exports, "BaseProvider", { enumerable: true, get: function () { return base_1.BaseProvider; } });
|
|
7
|
+
const openai_1 = require("./openai");
|
|
8
|
+
Object.defineProperty(exports, "OpenAIProvider", { enumerable: true, get: function () { return openai_1.OpenAIProvider; } });
|
|
9
|
+
const anthropic_1 = require("./anthropic");
|
|
10
|
+
Object.defineProperty(exports, "AnthropicProvider", { enumerable: true, get: function () { return anthropic_1.AnthropicProvider; } });
|
|
11
|
+
const gemini_1 = require("./gemini");
|
|
12
|
+
Object.defineProperty(exports, "GeminiProvider", { enumerable: true, get: function () { return gemini_1.GeminiProvider; } });
|
|
13
|
+
function createProvider(options) {
|
|
14
|
+
const { provider, apiKey, model } = options;
|
|
15
|
+
switch (provider) {
|
|
16
|
+
case 'openai': {
|
|
17
|
+
const key = apiKey || process.env.OPENAI_API_KEY || '';
|
|
18
|
+
if (!key)
|
|
19
|
+
throw new Error('OpenAI API key is required. Set OPENAI_API_KEY env variable.');
|
|
20
|
+
return new openai_1.OpenAIProvider(key, model || openai_1.DEFAULT_OPENAI_MODEL);
|
|
21
|
+
}
|
|
22
|
+
case 'anthropic': {
|
|
23
|
+
const key = apiKey || process.env.ANTHROPIC_API_KEY || '';
|
|
24
|
+
if (!key)
|
|
25
|
+
throw new Error('Anthropic API key is required. Set ANTHROPIC_API_KEY env variable.');
|
|
26
|
+
return new anthropic_1.AnthropicProvider(key, model || anthropic_1.DEFAULT_ANTHROPIC_MODEL);
|
|
27
|
+
}
|
|
28
|
+
case 'gemini': {
|
|
29
|
+
const key = apiKey || process.env.GEMINI_API_KEY || '';
|
|
30
|
+
if (!key)
|
|
31
|
+
throw new Error('Gemini API key is required. Set GEMINI_API_KEY env variable.');
|
|
32
|
+
return new gemini_1.GeminiProvider(key, model || gemini_1.DEFAULT_GEMINI_MODEL);
|
|
33
|
+
}
|
|
34
|
+
default:
|
|
35
|
+
throw new Error(`Unknown provider: ${provider}`);
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":";;;AAaA,wCAsBC;AAnCD,iCAAsC;AAqC7B,6FArCA,mBAAY,OAqCA;AApCrB,qCAAgE;AAoCzC,+FApCd,uBAAc,OAoCc;AAnCrC,2CAAyE;AAmClC,kGAnC9B,6BAAiB,OAmC8B;AAlCxD,qCAAgE;AAkCN,+FAlCjD,uBAAc,OAkCiD;AAxBxE,SAAgB,cAAc,CAAC,OAAwB;IACrD,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC;IAE5C,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,MAAM,GAAG,GAAG,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,EAAE,CAAC;YACvD,IAAI,CAAC,GAAG;gBAAE,MAAM,IAAI,KAAK,CAAC,8DAA8D,CAAC,CAAC;YAC1F,OAAO,IAAI,uBAAc,CAAC,GAAG,EAAE,KAAK,IAAI,6BAAoB,CAAC,CAAC;QAChE,CAAC;QACD,KAAK,WAAW,CAAC,CAAC,CAAC;YACjB,MAAM,GAAG,GAAG,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,EAAE,CAAC;YAC1D,IAAI,CAAC,GAAG;gBAAE,MAAM,IAAI,KAAK,CAAC,oEAAoE,CAAC,CAAC;YAChG,OAAO,IAAI,6BAAiB,CAAC,GAAG,EAAE,KAAK,IAAI,mCAAuB,CAAC,CAAC;QACtE,CAAC;QACD,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,MAAM,GAAG,GAAG,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,EAAE,CAAC;YACvD,IAAI,CAAC,GAAG;gBAAE,MAAM,IAAI,KAAK,CAAC,8DAA8D,CAAC,CAAC;YAC1F,OAAO,IAAI,uBAAc,CAAC,GAAG,EAAE,KAAK,IAAI,6BAAoB,CAAC,CAAC;QAChE,CAAC;QACD;YACE,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;IACrD,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import { BaseProvider, CompletionOptions, CompletionResult } from './base';
|
|
2
|
+
export declare const OPENAI_MODELS: readonly ["gpt-4o", "gpt-4o-mini", "gpt-4-turbo", "gpt-3.5-turbo"];
|
|
3
|
+
export declare const DEFAULT_OPENAI_MODEL = "gpt-4o";
|
|
4
|
+
export declare class OpenAIProvider extends BaseProvider {
|
|
5
|
+
private client;
|
|
6
|
+
constructor(apiKey: string, model?: string);
|
|
7
|
+
complete(options: CompletionOptions): Promise<CompletionResult>;
|
|
8
|
+
}
|
|
9
|
+
//# 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,EAAE,YAAY,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,MAAM,QAAQ,CAAC;AAE3E,eAAO,MAAM,aAAa,oEAAqE,CAAC;AAChG,eAAO,MAAM,oBAAoB,WAAW,CAAC;AAE7C,qBAAa,cAAe,SAAQ,YAAY;IAC9C,OAAO,CAAC,MAAM,CAAS;gBAEX,MAAM,EAAE,MAAM,EAAE,KAAK,SAAuB;IAKlD,QAAQ,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC;CA6CtE"}
|
|
@@ -0,0 +1,59 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.OpenAIProvider = exports.DEFAULT_OPENAI_MODEL = exports.OPENAI_MODELS = void 0;
|
|
7
|
+
const openai_1 = __importDefault(require("openai"));
|
|
8
|
+
const base_1 = require("./base");
|
|
9
|
+
exports.OPENAI_MODELS = ['gpt-4o', 'gpt-4o-mini', 'gpt-4-turbo', 'gpt-3.5-turbo'];
|
|
10
|
+
exports.DEFAULT_OPENAI_MODEL = 'gpt-4o';
|
|
11
|
+
class OpenAIProvider extends base_1.BaseProvider {
|
|
12
|
+
constructor(apiKey, model = exports.DEFAULT_OPENAI_MODEL) {
|
|
13
|
+
super(apiKey, model);
|
|
14
|
+
this.client = new openai_1.default({ apiKey });
|
|
15
|
+
}
|
|
16
|
+
async complete(options) {
|
|
17
|
+
const model = options.model || this.model;
|
|
18
|
+
const maxTokens = options.maxTokens || 4096;
|
|
19
|
+
const temperature = options.temperature ?? 0.7;
|
|
20
|
+
if (options.stream && options.onChunk) {
|
|
21
|
+
const stream = await this.client.chat.completions.create({
|
|
22
|
+
model,
|
|
23
|
+
messages: options.messages,
|
|
24
|
+
max_tokens: maxTokens,
|
|
25
|
+
temperature,
|
|
26
|
+
stream: true,
|
|
27
|
+
});
|
|
28
|
+
let fullContent = '';
|
|
29
|
+
for await (const chunk of stream) {
|
|
30
|
+
const delta = chunk.choices[0]?.delta?.content || '';
|
|
31
|
+
if (delta) {
|
|
32
|
+
fullContent += delta;
|
|
33
|
+
options.onChunk(delta);
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
return { content: fullContent, model };
|
|
37
|
+
}
|
|
38
|
+
const response = await this.client.chat.completions.create({
|
|
39
|
+
model,
|
|
40
|
+
messages: options.messages,
|
|
41
|
+
max_tokens: maxTokens,
|
|
42
|
+
temperature,
|
|
43
|
+
});
|
|
44
|
+
const content = response.choices[0]?.message?.content || '';
|
|
45
|
+
return {
|
|
46
|
+
content,
|
|
47
|
+
model: response.model,
|
|
48
|
+
usage: response.usage
|
|
49
|
+
? {
|
|
50
|
+
promptTokens: response.usage.prompt_tokens,
|
|
51
|
+
completionTokens: response.usage.completion_tokens,
|
|
52
|
+
totalTokens: response.usage.total_tokens,
|
|
53
|
+
}
|
|
54
|
+
: undefined,
|
|
55
|
+
};
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
exports.OpenAIProvider = OpenAIProvider;
|
|
59
|
+
//# sourceMappingURL=openai.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/providers/openai.ts"],"names":[],"mappings":";;;;;;AAAA,oDAA4B;AAC5B,iCAA2E;AAE9D,QAAA,aAAa,GAAG,CAAC,QAAQ,EAAE,aAAa,EAAE,aAAa,EAAE,eAAe,CAAU,CAAC;AACnF,QAAA,oBAAoB,GAAG,QAAQ,CAAC;AAE7C,MAAa,cAAe,SAAQ,mBAAY;IAG9C,YAAY,MAAc,EAAE,KAAK,GAAG,4BAAoB;QACtD,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,IAAI,gBAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAA0B;QACvC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC;QAC1C,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC;QAC5C,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,GAAG,CAAC;QAE/C,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBACvD,KAAK;gBACL,QAAQ,EAAE,OAAO,CAAC,QAAoD;gBACtE,UAAU,EAAE,SAAS;gBACrB,WAAW;gBACX,MAAM,EAAE,IAAI;aACb,CAAC,CAAC;YAEH,IAAI,WAAW,GAAG,EAAE,CAAC;YACrB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBACjC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,IAAI,EAAE,CAAC;gBACrD,IAAI,KAAK,EAAE,CAAC;oBACV,WAAW,IAAI,KAAK,CAAC;oBACrB,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBACzB,CAAC;YACH,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QACzC,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;YACzD,KAAK;YACL,QAAQ,EAAE,OAAO,CAAC,QAAoD;YACtE,UAAU,EAAE,SAAS;YACrB,WAAW;SACZ,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QAC5D,OAAO;YACL,OAAO;YACP,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,KAAK,EAAE,QAAQ,CAAC,KAAK;gBACnB,CAAC,CAAC;oBACE,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa;oBAC1C,gBAAgB,EAAE,QAAQ,CAAC,KAAK,CAAC,iBAAiB;oBAClD,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;iBACzC;gBACH,CAAC,CAAC,SAAS;SACd,CAAC;IACJ,CAAC;CACF;AArDD,wCAqDC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
export interface Guardrail {
|
|
2
|
+
id: string;
|
|
3
|
+
type: 'allow-paths' | 'deny-paths' | 'deny-operations' | 'max-tokens' | 'style' | 'custom';
|
|
4
|
+
value: string | string[] | number;
|
|
5
|
+
description?: string;
|
|
6
|
+
}
|
|
7
|
+
export interface GuardrailViolation {
|
|
8
|
+
guardrail: Guardrail;
|
|
9
|
+
message: string;
|
|
10
|
+
}
|
|
11
|
+
/**
|
|
12
|
+
* Converts guardrails into a system prompt suffix that instructs the AI.
|
|
13
|
+
*/
|
|
14
|
+
export declare function guardrailsToSystemPrompt(guardrails: Guardrail[]): string;
|
|
15
|
+
/**
|
|
16
|
+
* Validates a proposed file path against guardrails.
|
|
17
|
+
*/
|
|
18
|
+
export declare function checkFilePathGuardrails(filePath: string, guardrails: Guardrail[]): GuardrailViolation | null;
|
|
19
|
+
export declare function createGuardrail(type: Guardrail['type'], value: string | string[] | number, description?: string): Guardrail;
|
|
20
|
+
//# sourceMappingURL=guardrails.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"guardrails.d.ts","sourceRoot":"","sources":["../../src/session/guardrails.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,SAAS;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,aAAa,GAAG,YAAY,GAAG,iBAAiB,GAAG,YAAY,GAAG,OAAO,GAAG,QAAQ,CAAC;IAC3F,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,MAAM,CAAC;IAClC,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,kBAAkB;IACjC,SAAS,EAAE,SAAS,CAAC;IACrB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED;;GAEG;AACH,wBAAgB,wBAAwB,CAAC,UAAU,EAAE,SAAS,EAAE,GAAG,MAAM,CAkCxE;AAED;;GAEG;AACH,wBAAgB,uBAAuB,CACrC,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,SAAS,EAAE,GACtB,kBAAkB,GAAG,IAAI,CA2B3B;AAED,wBAAgB,eAAe,CAC7B,IAAI,EAAE,SAAS,CAAC,MAAM,CAAC,EACvB,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,MAAM,EACjC,WAAW,CAAC,EAAE,MAAM,GACnB,SAAS,CAOX"}
|