orbitx 0.1.1 → 0.1.3
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 +65 -9
- package/dist/core/anthropic-provider.d.ts +15 -0
- package/dist/core/anthropic-provider.d.ts.map +1 -0
- package/dist/core/anthropic-provider.js +208 -0
- package/dist/core/anthropic-provider.js.map +1 -0
- package/dist/core/deepseek-provider.d.ts +1 -4
- package/dist/core/deepseek-provider.d.ts.map +1 -1
- package/dist/core/deepseek-provider.js +15 -15
- package/dist/core/deepseek-provider.js.map +1 -1
- package/dist/core/mcp-client.d.ts +0 -8
- package/dist/core/mcp-client.d.ts.map +1 -1
- package/dist/core/mcp-client.js +14 -20
- package/dist/core/mcp-client.js.map +1 -1
- package/dist/core/mcp-computer.d.ts +3 -5
- package/dist/core/mcp-computer.d.ts.map +1 -1
- package/dist/core/mcp-computer.js +71 -28
- package/dist/core/mcp-computer.js.map +1 -1
- package/dist/core/mcp-connection.d.ts +17 -0
- package/dist/core/mcp-connection.d.ts.map +1 -1
- package/dist/core/mcp-connection.js +24 -2
- package/dist/core/mcp-connection.js.map +1 -1
- package/dist/core/mcp-ipc-connection.d.ts +13 -14
- package/dist/core/mcp-ipc-connection.d.ts.map +1 -1
- package/dist/core/mcp-ipc-connection.js +75 -64
- package/dist/core/mcp-ipc-connection.js.map +1 -1
- package/dist/core/mcp-server.js +4 -4
- package/dist/core/mcp-server.js.map +1 -1
- package/dist/core/mcp-ws-connection.d.ts +45 -0
- package/dist/core/mcp-ws-connection.d.ts.map +1 -0
- package/dist/core/mcp-ws-connection.js +165 -0
- package/dist/core/mcp-ws-connection.js.map +1 -0
- package/dist/core/ollama-provider.d.ts +1 -4
- package/dist/core/ollama-provider.d.ts.map +1 -1
- package/dist/core/ollama-provider.js +16 -16
- package/dist/core/ollama-provider.js.map +1 -1
- package/dist/core/openai-provider.d.ts +15 -0
- package/dist/core/openai-provider.d.ts.map +1 -0
- package/dist/core/openai-provider.js +180 -0
- package/dist/core/openai-provider.js.map +1 -0
- package/dist/core/tool-schema-translator.d.ts +12 -2
- package/dist/core/tool-schema-translator.d.ts.map +1 -1
- package/dist/core/tool-schema-translator.js +29 -2
- package/dist/core/tool-schema-translator.js.map +1 -1
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +3 -0
- package/dist/index.js.map +1 -1
- package/dist/sandbox.js +19 -3
- package/dist/sandbox.js.map +1 -1
- package/package.json +5 -2
|
@@ -0,0 +1,165 @@
|
|
|
1
|
+
import * as http from "http";
|
|
2
|
+
import * as https from "https";
|
|
3
|
+
import { WebSocket, WebSocketServer } from "ws";
|
|
4
|
+
import MCPConnection from "./mcp-connection.js";
|
|
5
|
+
const TOKEN_HEADER = "x-mcp-ws-token";
|
|
6
|
+
export class MCPWSConnection extends MCPConnection {
|
|
7
|
+
#options;
|
|
8
|
+
#ws = null;
|
|
9
|
+
#wss = null;
|
|
10
|
+
#httpServer = null;
|
|
11
|
+
#writeBuffer = [];
|
|
12
|
+
#closed = false;
|
|
13
|
+
constructor(options) {
|
|
14
|
+
super(true);
|
|
15
|
+
this.#options = options;
|
|
16
|
+
if (options.mode === "client" && !options.url) {
|
|
17
|
+
throw new Error("MCPWSConnection: 'url' is required when mode is 'client'");
|
|
18
|
+
}
|
|
19
|
+
if (options.mode === "server" && !options.port) {
|
|
20
|
+
throw new Error("MCPWSConnection: 'port' is required when mode is 'server'");
|
|
21
|
+
}
|
|
22
|
+
this.on("write_to_server", (data) => {
|
|
23
|
+
if (this.#options.mode === "client")
|
|
24
|
+
this.#sendJSON(data);
|
|
25
|
+
});
|
|
26
|
+
this.on("write_to_client", (data) => {
|
|
27
|
+
if (this.#options.mode === "server")
|
|
28
|
+
this.#sendJSON(data);
|
|
29
|
+
});
|
|
30
|
+
if (options.mode === "server") {
|
|
31
|
+
this.#startServer();
|
|
32
|
+
}
|
|
33
|
+
else {
|
|
34
|
+
this.#connectClient();
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
close() {
|
|
38
|
+
this.#closed = true;
|
|
39
|
+
this.#ws?.close();
|
|
40
|
+
this.#wss?.close();
|
|
41
|
+
this.#httpServer?.close();
|
|
42
|
+
}
|
|
43
|
+
#startServer() {
|
|
44
|
+
this.emit("role", "server");
|
|
45
|
+
const useTLS = !!this.#options.tls;
|
|
46
|
+
const httpServer = useTLS
|
|
47
|
+
? https.createServer({
|
|
48
|
+
key: this.#options.tls.key,
|
|
49
|
+
cert: this.#options.tls.cert,
|
|
50
|
+
ca: this.#options.tls.ca,
|
|
51
|
+
})
|
|
52
|
+
: http.createServer();
|
|
53
|
+
this.#httpServer = httpServer;
|
|
54
|
+
const wss = new WebSocketServer({ noServer: true });
|
|
55
|
+
this.#wss = wss;
|
|
56
|
+
httpServer.on("upgrade", (request, socket, head) => {
|
|
57
|
+
if (!this.#isTokenValid(request.headers[TOKEN_HEADER])) {
|
|
58
|
+
socket.write("HTTP/1.1 401 Unauthorized\r\n\r\n");
|
|
59
|
+
socket.destroy();
|
|
60
|
+
return;
|
|
61
|
+
}
|
|
62
|
+
wss.handleUpgrade(request, socket, head, (ws) => {
|
|
63
|
+
wss.emit("connection", ws, request);
|
|
64
|
+
});
|
|
65
|
+
});
|
|
66
|
+
wss.on("connection", (socket) => {
|
|
67
|
+
this.#attachSocket(socket);
|
|
68
|
+
});
|
|
69
|
+
wss.on("error", (err) => this.emit("error", err));
|
|
70
|
+
httpServer.on("error", (err) => this.emit("error", err));
|
|
71
|
+
httpServer.listen(this.#options.port, this.#options.host, () => {
|
|
72
|
+
this.emit("listening", this.#options.port);
|
|
73
|
+
});
|
|
74
|
+
}
|
|
75
|
+
#isTokenValid(headerValue) {
|
|
76
|
+
const expected = this.#options.token;
|
|
77
|
+
if (!expected)
|
|
78
|
+
return true; // no token configured => no auth check
|
|
79
|
+
const provided = Array.isArray(headerValue) ? headerValue[0] : headerValue;
|
|
80
|
+
return provided === expected;
|
|
81
|
+
}
|
|
82
|
+
#connectClient() {
|
|
83
|
+
this.emit("role", "client");
|
|
84
|
+
const headers = {};
|
|
85
|
+
if (this.#options.token) {
|
|
86
|
+
headers[TOKEN_HEADER] = this.#options.token;
|
|
87
|
+
}
|
|
88
|
+
const wsOptions = { headers };
|
|
89
|
+
if (this.#options.tls?.ca) {
|
|
90
|
+
wsOptions.ca = this.#options.tls.ca;
|
|
91
|
+
}
|
|
92
|
+
const socket = new WebSocket(this.#options.url, wsOptions);
|
|
93
|
+
socket.once("open", () => {
|
|
94
|
+
this.#attachSocket(socket);
|
|
95
|
+
});
|
|
96
|
+
socket.once("error", (err) => {
|
|
97
|
+
this.emit("error", err);
|
|
98
|
+
this.#maybeReconnect();
|
|
99
|
+
});
|
|
100
|
+
}
|
|
101
|
+
#attachSocket(sock) {
|
|
102
|
+
this.#ws = sock;
|
|
103
|
+
for (const line of this.#writeBuffer) {
|
|
104
|
+
sock.send(line);
|
|
105
|
+
}
|
|
106
|
+
this.#writeBuffer = [];
|
|
107
|
+
sock.on("message", (data) => {
|
|
108
|
+
this.#handleIncoming(data.toString("utf8"));
|
|
109
|
+
});
|
|
110
|
+
sock.on("close", () => {
|
|
111
|
+
this.emit("disconnected");
|
|
112
|
+
if (this.#ws === sock)
|
|
113
|
+
this.#ws = null;
|
|
114
|
+
this.#maybeReconnect();
|
|
115
|
+
});
|
|
116
|
+
sock.on("error", (err) => {
|
|
117
|
+
this.emit("error", err);
|
|
118
|
+
});
|
|
119
|
+
this.emit("connected", this.#options.mode);
|
|
120
|
+
}
|
|
121
|
+
#sendJSON(data) {
|
|
122
|
+
let line;
|
|
123
|
+
try {
|
|
124
|
+
line = JSON.stringify(data);
|
|
125
|
+
}
|
|
126
|
+
catch (err) {
|
|
127
|
+
this.emit("error", new Error(`MCPWSConnection: cannot serialise data ${err}`));
|
|
128
|
+
return;
|
|
129
|
+
}
|
|
130
|
+
if (this.#ws && this.#ws.readyState === WebSocket.OPEN) {
|
|
131
|
+
this.#ws.send(line);
|
|
132
|
+
}
|
|
133
|
+
else {
|
|
134
|
+
this.#writeBuffer.push(line);
|
|
135
|
+
}
|
|
136
|
+
}
|
|
137
|
+
#handleIncoming(raw) {
|
|
138
|
+
const trimmed = raw.trim();
|
|
139
|
+
if (!trimmed)
|
|
140
|
+
return;
|
|
141
|
+
try {
|
|
142
|
+
const parsed = JSON.parse(trimmed);
|
|
143
|
+
if (this.#options.mode === "server") {
|
|
144
|
+
this.emit("server_read", parsed);
|
|
145
|
+
}
|
|
146
|
+
else {
|
|
147
|
+
this.emit("client_read", parsed);
|
|
148
|
+
}
|
|
149
|
+
}
|
|
150
|
+
catch {
|
|
151
|
+
this.emit("error", new Error(`MCPWSConnection: malformed JSON – ${trimmed}`));
|
|
152
|
+
}
|
|
153
|
+
}
|
|
154
|
+
#maybeReconnect() {
|
|
155
|
+
if (this.#closed)
|
|
156
|
+
return;
|
|
157
|
+
if (this.#options.mode === "client") {
|
|
158
|
+
setTimeout(() => {
|
|
159
|
+
this.#connectClient();
|
|
160
|
+
}, this.#options.reconnectDelay ?? 500);
|
|
161
|
+
}
|
|
162
|
+
}
|
|
163
|
+
}
|
|
164
|
+
export default MCPWSConnection;
|
|
165
|
+
//# sourceMappingURL=mcp-ws-connection.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-ws-connection.js","sourceRoot":"","sources":["../../src/core/mcp-ws-connection.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,KAAK,KAAK,MAAM,OAAO,CAAC;AAC/B,OAAO,EAAE,SAAS,EAAE,eAAe,EAAsB,MAAM,IAAI,CAAC;AACpE,OAAO,aAAa,MAAM,qBAAqB,CAAC;AAyChD,MAAM,YAAY,GAAG,gBAAgB,CAAC;AAEtC,MAAM,OAAO,eAAgB,SAAQ,aAAa;IAC9C,QAAQ,CAAyB;IACjC,GAAG,GAAqB,IAAI,CAAC;IAC7B,IAAI,GAA2B,IAAI,CAAC;IACpC,WAAW,GAAsC,IAAI,CAAC;IACtD,YAAY,GAAa,EAAE,CAAC;IAC5B,OAAO,GAAG,KAAK,CAAC;IAEhB,YAAY,OAA+B;QACvC,KAAK,CAAC,IAAI,CAAC,CAAC;QACZ,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;QAExB,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;YAC5C,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;QAChF,CAAC;QACD,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;YAC7C,MAAM,IAAI,KAAK,CAAC,2DAA2D,CAAC,CAAC;QACjF,CAAC;QAED,IAAI,CAAC,EAAE,CAAC,iBAAiB,EAAE,CAAC,IAAa,EAAE,EAAE;YACzC,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,KAAK,QAAQ;gBAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,EAAE,CAAC,iBAAiB,EAAE,CAAC,IAAa,EAAE,EAAE;YACzC,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,KAAK,QAAQ;gBAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC5B,IAAI,CAAC,YAAY,EAAE,CAAC;QACxB,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,cAAc,EAAE,CAAC;QAC1B,CAAC;IACL,CAAC;IAED,KAAK;QACD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,CAAC,GAAG,EAAE,KAAK,EAAE,CAAC;QAClB,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC;QACnB,IAAI,CAAC,WAAW,EAAE,KAAK,EAAE,CAAC;IAC9B,CAAC;IAED,YAAY;QACR,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAE5B,MAAM,MAAM,GAAG,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC;QAEnC,MAAM,UAAU,GAAG,MAAM;YACrB,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC;gBACf,GAAG,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAI,CAAC,GAAG;gBAC3B,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAI,CAAC,IAAI;gBAC7B,EAAE,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAI,CAAC,EAAE;aAC5B,CAAC;YACJ,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;QAE1B,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;QAE9B,MAAM,GAAG,GAAG,IAAI,eAAe,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;QACpD,IAAI,CAAC,IAAI,GAAG,GAAG,CAAC;QAEhB,UAAU,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE;YAC/C,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC;gBACrD,MAAM,CAAC,KAAK,CAAC,mCAAmC,CAAC,CAAC;gBAClD,MAAM,CAAC,OAAO,EAAE,CAAC;gBACjB,OAAO;YACX,CAAC;YAED,GAAG,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE;gBAC5C,GAAG,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC;YACxC,CAAC,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;QAEH,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,MAAM,EAAE,EAAE;YAC5B,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QAClD,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QAEzD,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,GAAG,EAAE;YAC3D,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;IACP,CAAC;IAED,aAAa,CAAC,WAA0C;QACpD,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;QACrC,IAAI,CAAC,QAAQ;YAAE,OAAO,IAAI,CAAC,CAAC,uCAAuC;QAEnE,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC;QAC3E,OAAO,QAAQ,KAAK,QAAQ,CAAC;IACjC,CAAC;IAED,cAAc;QACV,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAE5B,MAAM,OAAO,GAA2B,EAAE,CAAC;QAC3C,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;YACtB,OAAO,CAAC,YAAY,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;QAChD,CAAC;QAED,MAAM,SAAS,GAAkB,EAAE,OAAO,EAAE,CAAC;QAC7C,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC;YACxB,SAAS,CAAC,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QACxC,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAI,EAAE,SAAS,CAAC,CAAC;QAE5D,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE;YACrB,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;YACzB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;YACxB,IAAI,CAAC,eAAe,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;IACP,CAAC;IAED,aAAa,CAAC,IAAe;QACzB,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC;QAEhB,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACnC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpB,CAAC;QACD,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;QAEvB,IAAI,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE;YACxB,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YAClB,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC1B,IAAI,IAAI,CAAC,GAAG,KAAK,IAAI;gBAAE,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC;YACvC,IAAI,CAAC,eAAe,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;YACrB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IAC/C,CAAC;IAED,SAAS,CAAC,IAAa;QACnB,IAAI,IAAY,CAAC;QACjB,IAAI,CAAC;YACD,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACX,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,0CAA0C,GAAG,EAAE,CAAC,CAAC,CAAC;YAC/E,OAAO;QACX,CAAC;QAED,IAAI,IAAI,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YACrD,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxB,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjC,CAAC;IACL,CAAC;IAED,eAAe,CAAC,GAAW;QACvB,MAAM,OAAO,GAAG,GAAG,CAAC,IAAI,EAAE,CAAC;QAC3B,IAAI,CAAC,OAAO;YAAE,OAAO;QAErB,IAAI,CAAC;YACD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACnC,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAClC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;YACrC,CAAC;iBAAM,CAAC;gBACJ,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;YACrC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACL,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,qCAAqC,OAAO,EAAE,CAAC,CAAC,CAAC;QAClF,CAAC;IACL,CAAC;IAED,eAAe;QACX,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAClC,UAAU,CAAC,GAAG,EAAE;gBACZ,IAAI,CAAC,cAAc,EAAE,CAAC;YAC1B,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,cAAc,IAAI,GAAG,CAAC,CAAC;QAC5C,CAAC;IACL,CAAC;CACJ;AAED,eAAe,eAAe,CAAC"}
|
|
@@ -1,10 +1,7 @@
|
|
|
1
1
|
import type { Message, ChatResponse, StreamCallback, ToolSchema, ProviderCapabilities } from "./ai-provider.js";
|
|
2
2
|
import AIProvider from "./ai-provider.js";
|
|
3
3
|
export declare class OllamaProvider extends AIProvider {
|
|
4
|
-
private
|
|
5
|
-
private model;
|
|
6
|
-
private supportsTools;
|
|
7
|
-
private contextWindow;
|
|
4
|
+
#private;
|
|
8
5
|
constructor(model: string, host?: string, options?: {
|
|
9
6
|
supportsTools?: boolean;
|
|
10
7
|
contextWindow?: number;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ollama-provider.d.ts","sourceRoot":"","sources":["../../src/core/ollama-provider.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,UAAU,EAAmB,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACjI,OAAO,UAAU,MAAM,kBAAkB,CAAC;AAuD1C,qBAAa,cAAe,SAAQ,UAAU
|
|
1
|
+
{"version":3,"file":"ollama-provider.d.ts","sourceRoot":"","sources":["../../src/core/ollama-provider.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,UAAU,EAAmB,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACjI,OAAO,UAAU,MAAM,kBAAkB,CAAC;AAuD1C,qBAAa,cAAe,SAAQ,UAAU;;gBAW9B,KAAK,EAAE,MAAM,EAAE,IAAI,GAAE,MAAiC,EAAE,OAAO,GAAE;QAAE,aAAa,CAAC,EAAE,OAAO,CAAC;QAAC,aAAa,CAAC,EAAE,MAAM,CAAA;KAAO;IAQrI,eAAe,IAAI,oBAAoB;IASjC,IAAI,CACN,QAAQ,EAAE,OAAO,EAAE,EACnB,cAAc,CAAC,EAAE,cAAc,EAC/B,KAAK,CAAC,EAAE,UAAU,EAAE,GACrB,OAAO,CAAC,YAAY,CAAC;CAqE3B;AAED,eAAe,cAAc,CAAC"}
|
|
@@ -48,35 +48,35 @@ function fromOllamaToolCalls(toolCalls) {
|
|
|
48
48
|
}));
|
|
49
49
|
}
|
|
50
50
|
export class OllamaProvider extends AIProvider {
|
|
51
|
-
client;
|
|
52
|
-
model;
|
|
51
|
+
#client;
|
|
52
|
+
#model;
|
|
53
53
|
// Whether the currently-configured model supports Ollama's native
|
|
54
54
|
// function-calling API. Ollama throws if `tools` is sent to a model
|
|
55
55
|
// that doesn't support it, so this must be known ahead of time rather
|
|
56
56
|
// than discovered by trial and error — callers should set this based on
|
|
57
57
|
// the model card (most recent tool-capable families default to true).
|
|
58
|
-
supportsTools;
|
|
59
|
-
contextWindow;
|
|
58
|
+
#supportsTools;
|
|
59
|
+
#contextWindow;
|
|
60
60
|
constructor(model, host = "http://localhost:11434", options = {}) {
|
|
61
61
|
super();
|
|
62
|
-
this
|
|
63
|
-
this
|
|
64
|
-
this
|
|
65
|
-
this
|
|
62
|
+
this.#client = new Ollama({ host });
|
|
63
|
+
this.#model = model;
|
|
64
|
+
this.#supportsTools = options.supportsTools ?? false;
|
|
65
|
+
this.#contextWindow = options.contextWindow ?? DEFAULT_CONTEXT_WINDOW;
|
|
66
66
|
}
|
|
67
67
|
getCapabilities() {
|
|
68
68
|
return {
|
|
69
|
-
supportsTools: this
|
|
69
|
+
supportsTools: this.#supportsTools,
|
|
70
70
|
supportsImages: true,
|
|
71
|
-
contextWindow: this
|
|
71
|
+
contextWindow: this.#contextWindow,
|
|
72
72
|
safeUsageRatio: 0.5,
|
|
73
73
|
};
|
|
74
74
|
}
|
|
75
75
|
async chat(messages, streamCallback, tools) {
|
|
76
|
-
const formattedMessages = toOllamaMessages(messages, this
|
|
76
|
+
const formattedMessages = toOllamaMessages(messages, this.#supportsTools);
|
|
77
77
|
// Never send `tools` to a model we don't know supports it — Ollama
|
|
78
78
|
// throws rather than silently ignoring an unsupported `tools` field.
|
|
79
|
-
const formattedTools = tools && tools.length > 0 && this
|
|
79
|
+
const formattedTools = tools && tools.length > 0 && this.#supportsTools
|
|
80
80
|
? toOpenAIFunctionTools(tools)
|
|
81
81
|
: undefined;
|
|
82
82
|
if (streamCallback) {
|
|
@@ -84,8 +84,8 @@ export class OllamaProvider extends AIProvider {
|
|
|
84
84
|
// Per-attempt accumulators live inside this closure so a
|
|
85
85
|
// failed/partial stream from a previous attempt never leaks
|
|
86
86
|
// into a retry's output — each attempt starts clean.
|
|
87
|
-
const stream = await this
|
|
88
|
-
model: this
|
|
87
|
+
const stream = await this.#client.chat({
|
|
88
|
+
model: this.#model,
|
|
89
89
|
messages: formattedMessages,
|
|
90
90
|
...(formattedTools ? { tools: formattedTools } : {}),
|
|
91
91
|
stream: true
|
|
@@ -121,8 +121,8 @@ export class OllamaProvider extends AIProvider {
|
|
|
121
121
|
}
|
|
122
122
|
else {
|
|
123
123
|
return withRetry(async () => {
|
|
124
|
-
const response = await this
|
|
125
|
-
model: this
|
|
124
|
+
const response = await this.#client.chat({
|
|
125
|
+
model: this.#model,
|
|
126
126
|
messages: formattedMessages,
|
|
127
127
|
...(formattedTools ? { tools: formattedTools } : {}),
|
|
128
128
|
});
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ollama-provider.js","sourceRoot":"","sources":["../../src/core/ollama-provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAEhC,OAAO,UAAU,MAAM,kBAAkB,CAAC;AAC1C,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAEvC,MAAM,sBAAsB,GAAG,MAAM,CAAC;AAEtC,SAAS,gBAAgB,CAAC,QAAmB,EAAE,aAAsB;IACjE,OAAO,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;QACtB,oEAAoE;QACpE,mEAAmE;QACnE,mEAAmE;QACnE,mEAAmE;QACnE,mEAAmE;QACnE,gCAAgC;QAChC,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;YACxC,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;QAC1D,CAAC;QAED,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACtB,OAAO;gBACH,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE;gBAC1B,SAAS,EAAE,GAAG,CAAC,QAAQ;aAC1B,CAAC;QACN,CAAC;QAED,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,IAAI,GAAG,CAAC,SAAS,EAAE,MAAM,IAAI,aAAa,EAAE,CAAC;YACrE,OAAO;gBACH,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE;gBAC1B,UAAU,EAAE,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;oBACjC,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,MAAM,EAAE;iBACpD,CAAC,CAAC;aACN,CAAC;QACN,CAAC;QAED,IAAI,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC;YACpB,MAAM,IAAI,GAAG,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAE,CAAS,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC3F,MAAM,MAAM,GAAG,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAE,CAAS,CAAC,KAAK,CAAC,CAAC;YACpF,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;QACnF,CAAC;QAED,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;IAC1D,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,mBAAmB,CAAC,SAA4B;IACrD,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IAC3D,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,EAAO,EAAE,GAAW,EAAE,EAAE,CAAC,CAAC;QAC5C,EAAE,EAAE,GAAG,GAAG,EAAE;QACZ,IAAI,EAAE,EAAE,CAAC,QAAQ,EAAE,IAAI,IAAI,EAAE;QAC7B,MAAM,EAAE,EAAE,CAAC,QAAQ,EAAE,SAAS,IAAI,EAAE;KACvC,CAAC,CAAC,CAAC;AACR,CAAC;AAED,MAAM,OAAO,cAAe,SAAQ,UAAU;
|
|
1
|
+
{"version":3,"file":"ollama-provider.js","sourceRoot":"","sources":["../../src/core/ollama-provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAEhC,OAAO,UAAU,MAAM,kBAAkB,CAAC;AAC1C,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAEvC,MAAM,sBAAsB,GAAG,MAAM,CAAC;AAEtC,SAAS,gBAAgB,CAAC,QAAmB,EAAE,aAAsB;IACjE,OAAO,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;QACtB,oEAAoE;QACpE,mEAAmE;QACnE,mEAAmE;QACnE,mEAAmE;QACnE,mEAAmE;QACnE,gCAAgC;QAChC,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;YACxC,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;QAC1D,CAAC;QAED,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACtB,OAAO;gBACH,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE;gBAC1B,SAAS,EAAE,GAAG,CAAC,QAAQ;aAC1B,CAAC;QACN,CAAC;QAED,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,IAAI,GAAG,CAAC,SAAS,EAAE,MAAM,IAAI,aAAa,EAAE,CAAC;YACrE,OAAO;gBACH,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE;gBAC1B,UAAU,EAAE,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;oBACjC,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,MAAM,EAAE;iBACpD,CAAC,CAAC;aACN,CAAC;QACN,CAAC;QAED,IAAI,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC;YACpB,MAAM,IAAI,GAAG,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAE,CAAS,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC3F,MAAM,MAAM,GAAG,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAE,CAAS,CAAC,KAAK,CAAC,CAAC;YACpF,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;QACnF,CAAC;QAED,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;IAC1D,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,mBAAmB,CAAC,SAA4B;IACrD,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IAC3D,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,EAAO,EAAE,GAAW,EAAE,EAAE,CAAC,CAAC;QAC5C,EAAE,EAAE,GAAG,GAAG,EAAE;QACZ,IAAI,EAAE,EAAE,CAAC,QAAQ,EAAE,IAAI,IAAI,EAAE;QAC7B,MAAM,EAAE,EAAE,CAAC,QAAQ,EAAE,SAAS,IAAI,EAAE;KACvC,CAAC,CAAC,CAAC;AACR,CAAC;AAED,MAAM,OAAO,cAAe,SAAQ,UAAU;IAC1C,OAAO,CAAS;IAChB,MAAM,CAAS;IACf,kEAAkE;IAClE,oEAAoE;IACpE,sEAAsE;IACtE,wEAAwE;IACxE,sEAAsE;IACtE,cAAc,CAAU;IACxB,cAAc,CAAS;IAEvB,YAAY,KAAa,EAAE,OAAe,wBAAwB,EAAE,UAA+D,EAAE;QACjI,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,OAAO,GAAG,IAAI,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;QACpC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,aAAa,IAAI,KAAK,CAAC;QACrD,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,aAAa,IAAI,sBAAsB,CAAC;IAC1E,CAAC;IAED,eAAe;QACX,OAAO;YACH,aAAa,EAAE,IAAI,CAAC,cAAc;YAClC,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI,CAAC,cAAc;YAClC,cAAc,EAAE,GAAG;SACtB,CAAC;IACN,CAAC;IAED,KAAK,CAAC,IAAI,CACN,QAAmB,EACnB,cAA+B,EAC/B,KAAoB;QAEpB,MAAM,iBAAiB,GAAG,gBAAgB,CAAC,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;QAC1E,mEAAmE;QACnE,qEAAqE;QACrE,MAAM,cAAc,GAAG,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,cAAc;YACnE,CAAC,CAAC,qBAAqB,CAAC,KAAK,CAAC;YAC9B,CAAC,CAAC,SAAS,CAAC;QAEhB,IAAI,cAAc,EAAE,CAAC;YACjB,OAAO,SAAS,CAAC,KAAK,IAAI,EAAE;gBACxB,yDAAyD;gBACzD,4DAA4D;gBAC5D,qDAAqD;gBACrD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;oBACnC,KAAK,EAAE,IAAI,CAAC,MAAM;oBAClB,QAAQ,EAAE,iBAAwB;oBAClC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBACpD,MAAM,EAAE,IAAI;iBACf,CAAC,CAAC;gBAEH,IAAI,WAAW,GAAG,EAAE,CAAC;gBACrB,IAAI,eAAe,GAAG,CAAC,CAAC;gBACxB,IAAI,SAAS,GAAG,CAAC,CAAC;gBAClB,IAAI,SAAwC,CAAC;gBAE7C,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;oBAC/B,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;oBAC7C,IAAI,OAAO,EAAE,CAAC;wBACV,WAAW,IAAI,OAAO,CAAC;wBACvB,cAAc,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;oBAChE,CAAC;oBAED,IAAI,KAAK,CAAC,OAAO,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC;wBACpC,SAAS,GAAG,mBAAmB,CAAC,KAAK,CAAC,OAAO,CAAC,UAAiB,CAAC,CAAC;oBACrE,CAAC;oBAED,IAAI,KAAK,CAAC,iBAAiB,KAAK,SAAS,EAAE,CAAC;wBACxC,eAAe,GAAG,KAAK,CAAC,iBAAiB,CAAC;oBAC9C,CAAC;oBACD,IAAI,KAAK,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;wBACjC,SAAS,GAAG,KAAK,CAAC,UAAU,CAAC;oBACjC,CAAC;gBACL,CAAC;gBAED,cAAc,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;gBACpG,OAAO;oBACH,OAAO,EAAE,WAAW;oBACpB,WAAW,EAAE,eAAe;oBAC5B,YAAY,EAAE,SAAS;oBACvB,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACtC,CAAC;YACN,CAAC,CAAC,CAAC;QACP,CAAC;aAAM,CAAC;YACJ,OAAO,SAAS,CAAC,KAAK,IAAI,EAAE;gBACxB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;oBACrC,KAAK,EAAE,IAAI,CAAC,MAAM;oBAClB,QAAQ,EAAE,iBAAwB;oBAClC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACvD,CAAC,CAAC;gBAEH,OAAO;oBACH,OAAO,EAAE,QAAQ,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE;oBACxC,WAAW,EAAE,QAAQ,CAAC,iBAAiB,IAAI,CAAC;oBAC5C,YAAY,EAAE,QAAQ,CAAC,UAAU,IAAI,CAAC;oBACtC,GAAG,CAAC,mBAAmB,CAAC,QAAQ,CAAC,OAAO,EAAE,UAAiB,CAAC,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,mBAAmB,CAAC,QAAQ,CAAC,OAAO,EAAE,UAAiB,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC/I,CAAC;YACN,CAAC,CAAC,CAAC;QACP,CAAC;IACL,CAAC;CACJ;AAED,eAAe,cAAc,CAAC"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import type { Message, ChatResponse, StreamCallback, ToolSchema, ProviderCapabilities } from "./ai-provider.js";
|
|
2
|
+
import AIProvider from "./ai-provider.js";
|
|
3
|
+
export declare class OpenAIProvider extends AIProvider {
|
|
4
|
+
#private;
|
|
5
|
+
constructor(apiKey: string, model?: string, options?: {
|
|
6
|
+
supportsTools?: boolean;
|
|
7
|
+
supportsImages?: boolean;
|
|
8
|
+
contextWindow?: number;
|
|
9
|
+
baseURL?: string;
|
|
10
|
+
});
|
|
11
|
+
getCapabilities(): ProviderCapabilities;
|
|
12
|
+
chat(messages: Message[], streamCallback?: StreamCallback, tools?: ToolSchema[]): Promise<ChatResponse>;
|
|
13
|
+
}
|
|
14
|
+
export default OpenAIProvider;
|
|
15
|
+
//# sourceMappingURL=openai-provider.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai-provider.d.ts","sourceRoot":"","sources":["../../src/core/openai-provider.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,UAAU,EAAmB,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACjI,OAAO,UAAU,MAAM,kBAAkB,CAAC;AAwE1C,qBAAa,cAAe,SAAQ,UAAU;;gBAO9B,MAAM,EAAE,MAAM,EAAE,KAAK,GAAE,MAAgB,EAAE,OAAO,GAAE;QAAE,aAAa,CAAC,EAAE,OAAO,CAAC;QAAC,cAAc,CAAC,EAAE,OAAO,CAAC;QAAC,aAAa,CAAC,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,MAAM,CAAA;KAAO;IAYlK,eAAe,IAAI,oBAAoB;IASjC,IAAI,CACN,QAAQ,EAAE,OAAO,EAAE,EACnB,cAAc,CAAC,EAAE,cAAc,EAC/B,KAAK,CAAC,EAAE,UAAU,EAAE,GACrB,OAAO,CAAC,YAAY,CAAC;CAqF3B;AAED,eAAe,cAAc,CAAC"}
|
|
@@ -0,0 +1,180 @@
|
|
|
1
|
+
import OpenAI from "openai";
|
|
2
|
+
import AIProvider from "./ai-provider.js";
|
|
3
|
+
import { toOpenAIFunctionTools } from "./tool-schema-translator.js";
|
|
4
|
+
import { withRetry } from "./retry.js";
|
|
5
|
+
const MODEL_CONTEXT_WINDOWS = {
|
|
6
|
+
"gpt-5": 400_000,
|
|
7
|
+
"gpt-5-mini": 400_000,
|
|
8
|
+
"gpt-5-nano": 400_000,
|
|
9
|
+
"gpt-4.1": 1_047_576,
|
|
10
|
+
"gpt-4.1-mini": 1_047_576,
|
|
11
|
+
"gpt-4o": 128_000,
|
|
12
|
+
"gpt-4o-mini": 128_000,
|
|
13
|
+
"o3": 200_000,
|
|
14
|
+
"o4-mini": 200_000,
|
|
15
|
+
};
|
|
16
|
+
const DEFAULT_CONTEXT_WINDOW = 128_000;
|
|
17
|
+
// Models whose Chat Completions image support is text-only / unsupported —
|
|
18
|
+
// kept as an explicit denylist so new models default to "supports images"
|
|
19
|
+
// rather than silently dropping multimodal input.
|
|
20
|
+
const NO_IMAGE_MODELS = new Set(["o3-mini", "o1-mini"]);
|
|
21
|
+
function toOpenAIMessages(messages) {
|
|
22
|
+
return messages.map(msg => {
|
|
23
|
+
if (msg.role === "tool") {
|
|
24
|
+
return {
|
|
25
|
+
role: "tool",
|
|
26
|
+
content: msg.content || "",
|
|
27
|
+
tool_call_id: msg.toolCallId,
|
|
28
|
+
};
|
|
29
|
+
}
|
|
30
|
+
if (msg.role === "assistant" && msg.toolCalls?.length) {
|
|
31
|
+
return {
|
|
32
|
+
role: "assistant",
|
|
33
|
+
content: msg.content || null,
|
|
34
|
+
tool_calls: msg.toolCalls.map(tc => ({
|
|
35
|
+
id: tc.id,
|
|
36
|
+
type: "function",
|
|
37
|
+
function: { name: tc.name, arguments: JSON.stringify(tc.inputs) },
|
|
38
|
+
})),
|
|
39
|
+
};
|
|
40
|
+
}
|
|
41
|
+
if (msg.parts?.length) {
|
|
42
|
+
return {
|
|
43
|
+
role: msg.role,
|
|
44
|
+
content: msg.parts.map(p => p.type === "text"
|
|
45
|
+
? { type: "text", text: p.text }
|
|
46
|
+
: { type: "image_url", image_url: { url: `data:${p.mimeType || "image/png"};base64,${p.image}` } }),
|
|
47
|
+
};
|
|
48
|
+
}
|
|
49
|
+
return { role: msg.role, content: msg.content || "" };
|
|
50
|
+
});
|
|
51
|
+
}
|
|
52
|
+
function fromOpenAIToolCalls(toolCalls) {
|
|
53
|
+
if (!toolCalls || toolCalls.length === 0)
|
|
54
|
+
return undefined;
|
|
55
|
+
return toolCalls
|
|
56
|
+
.filter((tc) => tc.type === "function" || tc.function)
|
|
57
|
+
.map((tc) => {
|
|
58
|
+
let inputs = {};
|
|
59
|
+
try {
|
|
60
|
+
inputs = JSON.parse(tc.function.arguments || "{}");
|
|
61
|
+
}
|
|
62
|
+
catch {
|
|
63
|
+
inputs = {};
|
|
64
|
+
}
|
|
65
|
+
return { id: tc.id, name: tc.function.name, inputs };
|
|
66
|
+
});
|
|
67
|
+
}
|
|
68
|
+
export class OpenAIProvider extends AIProvider {
|
|
69
|
+
#client;
|
|
70
|
+
#model;
|
|
71
|
+
#supportsTools;
|
|
72
|
+
#supportsImages;
|
|
73
|
+
#contextWindow;
|
|
74
|
+
constructor(apiKey, model = "gpt-5", options = {}) {
|
|
75
|
+
super();
|
|
76
|
+
this.#client = new OpenAI({
|
|
77
|
+
apiKey: apiKey,
|
|
78
|
+
...(options.baseURL ? { baseURL: options.baseURL } : {}),
|
|
79
|
+
});
|
|
80
|
+
this.#model = model;
|
|
81
|
+
this.#supportsTools = options.supportsTools ?? true;
|
|
82
|
+
this.#supportsImages = options.supportsImages ?? !NO_IMAGE_MODELS.has(model);
|
|
83
|
+
this.#contextWindow = options.contextWindow ?? MODEL_CONTEXT_WINDOWS[model] ?? DEFAULT_CONTEXT_WINDOW;
|
|
84
|
+
}
|
|
85
|
+
getCapabilities() {
|
|
86
|
+
return {
|
|
87
|
+
supportsTools: this.#supportsTools,
|
|
88
|
+
supportsImages: this.#supportsImages,
|
|
89
|
+
contextWindow: this.#contextWindow,
|
|
90
|
+
safeUsageRatio: 0.5,
|
|
91
|
+
};
|
|
92
|
+
}
|
|
93
|
+
async chat(messages, streamCallback, tools) {
|
|
94
|
+
const formattedMessages = toOpenAIMessages(messages);
|
|
95
|
+
const formattedTools = tools && tools.length > 0 && this.#supportsTools
|
|
96
|
+
? toOpenAIFunctionTools(tools)
|
|
97
|
+
: undefined;
|
|
98
|
+
if (streamCallback) {
|
|
99
|
+
return withRetry(async () => {
|
|
100
|
+
// Per-attempt accumulators live inside this closure so a
|
|
101
|
+
// failed/partial stream from a previous attempt never leaks
|
|
102
|
+
// into a retry's output — each attempt starts clean.
|
|
103
|
+
const stream = await this.#client.chat.completions.create({
|
|
104
|
+
model: this.#model,
|
|
105
|
+
messages: formattedMessages,
|
|
106
|
+
...(formattedTools ? { tools: formattedTools } : {}),
|
|
107
|
+
stream: true,
|
|
108
|
+
stream_options: { include_usage: true },
|
|
109
|
+
});
|
|
110
|
+
let fullContent = "";
|
|
111
|
+
let inputTokens = 0;
|
|
112
|
+
let outputTokens = 0;
|
|
113
|
+
const toolCallChunks = {};
|
|
114
|
+
for await (const chunk of stream) {
|
|
115
|
+
const delta = chunk.choices[0]?.delta;
|
|
116
|
+
const content = delta?.content || "";
|
|
117
|
+
if (content) {
|
|
118
|
+
fullContent += content;
|
|
119
|
+
streamCallback({ role: "assistant", content, done: false });
|
|
120
|
+
}
|
|
121
|
+
if (delta?.tool_calls) {
|
|
122
|
+
for (const tc of delta.tool_calls) {
|
|
123
|
+
const idx = tc.index ?? 0;
|
|
124
|
+
if (!toolCallChunks[idx])
|
|
125
|
+
toolCallChunks[idx] = { arguments: "" };
|
|
126
|
+
if (tc.id)
|
|
127
|
+
toolCallChunks[idx].id = tc.id;
|
|
128
|
+
if (tc.function?.name)
|
|
129
|
+
toolCallChunks[idx].name = tc.function.name;
|
|
130
|
+
if (tc.function?.arguments)
|
|
131
|
+
toolCallChunks[idx].arguments += tc.function.arguments;
|
|
132
|
+
}
|
|
133
|
+
}
|
|
134
|
+
if (chunk.usage) {
|
|
135
|
+
inputTokens = chunk.usage.prompt_tokens || 0;
|
|
136
|
+
outputTokens = chunk.usage.completion_tokens || 0;
|
|
137
|
+
}
|
|
138
|
+
}
|
|
139
|
+
const toolCalls = Object.keys(toolCallChunks).length > 0
|
|
140
|
+
? Object.values(toolCallChunks).map(tc => {
|
|
141
|
+
let inputs = {};
|
|
142
|
+
try {
|
|
143
|
+
inputs = JSON.parse(tc.arguments || "{}");
|
|
144
|
+
}
|
|
145
|
+
catch {
|
|
146
|
+
inputs = {};
|
|
147
|
+
}
|
|
148
|
+
return { id: tc.id || "", name: tc.name || "", inputs };
|
|
149
|
+
})
|
|
150
|
+
: undefined;
|
|
151
|
+
streamCallback({ role: "assistant", content: "", done: true, ...(toolCalls ? { toolCalls } : {}) });
|
|
152
|
+
return {
|
|
153
|
+
content: fullContent,
|
|
154
|
+
inputTokens,
|
|
155
|
+
outputTokens,
|
|
156
|
+
...(toolCalls ? { toolCalls } : {}),
|
|
157
|
+
};
|
|
158
|
+
});
|
|
159
|
+
}
|
|
160
|
+
else {
|
|
161
|
+
return withRetry(async () => {
|
|
162
|
+
const response = await this.#client.chat.completions.create({
|
|
163
|
+
model: this.#model,
|
|
164
|
+
messages: formattedMessages,
|
|
165
|
+
...(formattedTools ? { tools: formattedTools } : {}),
|
|
166
|
+
});
|
|
167
|
+
const message = response.choices[0]?.message;
|
|
168
|
+
const toolCalls = fromOpenAIToolCalls(message?.tool_calls);
|
|
169
|
+
return {
|
|
170
|
+
content: message?.content || "",
|
|
171
|
+
inputTokens: response.usage?.prompt_tokens || 0,
|
|
172
|
+
outputTokens: response.usage?.completion_tokens || 0,
|
|
173
|
+
...(toolCalls ? { toolCalls } : {}),
|
|
174
|
+
};
|
|
175
|
+
});
|
|
176
|
+
}
|
|
177
|
+
}
|
|
178
|
+
}
|
|
179
|
+
export default OpenAIProvider;
|
|
180
|
+
//# sourceMappingURL=openai-provider.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai-provider.js","sourceRoot":"","sources":["../../src/core/openai-provider.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,OAAO,UAAU,MAAM,kBAAkB,CAAC;AAC1C,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAEvC,MAAM,qBAAqB,GAA2B;IAClD,OAAO,EAAE,OAAO;IAChB,YAAY,EAAE,OAAO;IACrB,YAAY,EAAE,OAAO;IACrB,SAAS,EAAE,SAAS;IACpB,cAAc,EAAE,SAAS;IACzB,QAAQ,EAAE,OAAO;IACjB,aAAa,EAAE,OAAO;IACtB,IAAI,EAAE,OAAO;IACb,SAAS,EAAE,OAAO;CACrB,CAAC;AACF,MAAM,sBAAsB,GAAG,OAAO,CAAC;AAEvC,2EAA2E;AAC3E,0EAA0E;AAC1E,kDAAkD;AAClD,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC;AAExD,SAAS,gBAAgB,CAAC,QAAmB;IACzC,OAAO,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;QACtB,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACtB,OAAO;gBACH,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE;gBAC1B,YAAY,EAAE,GAAG,CAAC,UAAU;aAC/B,CAAC;QACN,CAAC;QAED,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,IAAI,GAAG,CAAC,SAAS,EAAE,MAAM,EAAE,CAAC;YACpD,OAAO;gBACH,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,IAAI;gBAC5B,UAAU,EAAE,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;oBACjC,EAAE,EAAE,EAAE,CAAC,EAAE;oBACT,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,MAAM,CAAC,EAAE;iBACpE,CAAC,CAAC;aACN,CAAC;QACN,CAAC;QAED,IAAI,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC;YACpB,OAAO;gBACH,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,OAAO,EAAE,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM;oBACzC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE;oBAChC,CAAC,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,SAAS,EAAE,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAC,QAAQ,IAAI,WAAW,WAAW,CAAC,CAAC,KAAK,EAAE,EAAE,EAAE,CAAC;aAC1G,CAAC;QACN,CAAC;QAED,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;IAC1D,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,mBAAmB,CAAC,SAA4B;IACrD,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IAC3D,OAAO,SAAS;SACX,MAAM,CAAC,CAAC,EAAO,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,KAAK,UAAU,IAAI,EAAE,CAAC,QAAQ,CAAC;SAC1D,GAAG,CAAC,CAAC,EAAO,EAAE,EAAE;QACb,IAAI,MAAM,GAAwB,EAAE,CAAC;QACrC,IAAI,CAAC;YACD,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC;QACvD,CAAC;QAAC,MAAM,CAAC;YACL,MAAM,GAAG,EAAE,CAAC;QAChB,CAAC;QACD,OAAO,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC;IACzD,CAAC,CAAC,CAAC;AACX,CAAC;AAED,MAAM,OAAO,cAAe,SAAQ,UAAU;IAC1C,OAAO,CAAS;IAChB,MAAM,CAAS;IACf,cAAc,CAAU;IACxB,eAAe,CAAU;IACzB,cAAc,CAAS;IAEvB,YAAY,MAAc,EAAE,QAAgB,OAAO,EAAE,UAA2G,EAAE;QAC9J,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,OAAO,GAAG,IAAI,MAAM,CAAC;YACtB,MAAM,EAAE,MAAM;YACd,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC3D,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,aAAa,IAAI,IAAI,CAAC;QACpD,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,cAAc,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC7E,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,aAAa,IAAI,qBAAqB,CAAC,KAAK,CAAC,IAAI,sBAAsB,CAAC;IAC1G,CAAC;IAED,eAAe;QACX,OAAO;YACH,aAAa,EAAE,IAAI,CAAC,cAAc;YAClC,cAAc,EAAE,IAAI,CAAC,eAAe;YACpC,aAAa,EAAE,IAAI,CAAC,cAAc;YAClC,cAAc,EAAE,GAAG;SACtB,CAAC;IACN,CAAC;IAED,KAAK,CAAC,IAAI,CACN,QAAmB,EACnB,cAA+B,EAC/B,KAAoB;QAEpB,MAAM,iBAAiB,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACrD,MAAM,cAAc,GAAG,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,cAAc;YACnE,CAAC,CAAC,qBAAqB,CAAC,KAAK,CAAC;YAC9B,CAAC,CAAC,SAAS,CAAC;QAEhB,IAAI,cAAc,EAAE,CAAC;YACjB,OAAO,SAAS,CAAC,KAAK,IAAI,EAAE;gBACxB,yDAAyD;gBACzD,4DAA4D;gBAC5D,qDAAqD;gBACrD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;oBACtD,KAAK,EAAE,IAAI,CAAC,MAAM;oBAClB,QAAQ,EAAE,iBAAwB;oBAClC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBACpD,MAAM,EAAE,IAAI;oBACZ,cAAc,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;iBAC1C,CAAC,CAAC;gBAEH,IAAI,WAAW,GAAG,EAAE,CAAC;gBACrB,IAAI,WAAW,GAAG,CAAC,CAAC;gBACpB,IAAI,YAAY,GAAG,CAAC,CAAC;gBACrB,MAAM,cAAc,GAAsE,EAAE,CAAC;gBAE7F,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;oBAC/B,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAY,CAAC;oBAC7C,MAAM,OAAO,GAAG,KAAK,EAAE,OAAO,IAAI,EAAE,CAAC;oBACrC,IAAI,OAAO,EAAE,CAAC;wBACV,WAAW,IAAI,OAAO,CAAC;wBACvB,cAAc,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;oBAChE,CAAC;oBAED,IAAI,KAAK,EAAE,UAAU,EAAE,CAAC;wBACpB,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;4BAChC,MAAM,GAAG,GAAG,EAAE,CAAC,KAAK,IAAI,CAAC,CAAC;4BAC1B,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC;gCAAE,cAAc,CAAC,GAAG,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;4BAClE,IAAI,EAAE,CAAC,EAAE;gCAAE,cAAc,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;4BAC1C,IAAI,EAAE,CAAC,QAAQ,EAAE,IAAI;gCAAE,cAAc,CAAC,GAAG,CAAC,CAAC,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC;4BACnE,IAAI,EAAE,CAAC,QAAQ,EAAE,SAAS;gCAAE,cAAc,CAAC,GAAG,CAAC,CAAC,SAAS,IAAI,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;wBACvF,CAAC;oBACL,CAAC;oBAED,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;wBACd,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;wBAC7C,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC,iBAAiB,IAAI,CAAC,CAAC;oBACtD,CAAC;gBACL,CAAC;gBAED,MAAM,SAAS,GAAkC,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,MAAM,GAAG,CAAC;oBACnF,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;wBACrC,IAAI,MAAM,GAAwB,EAAE,CAAC;wBACrC,IAAI,CAAC;4BAAC,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC;wBAAC,CAAC;wBAAC,MAAM,CAAC;4BAAC,MAAM,GAAG,EAAE,CAAC;wBAAC,CAAC;wBACzE,OAAO,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,IAAI,EAAE,EAAE,MAAM,EAAE,CAAC;oBAC5D,CAAC,CAAC;oBACF,CAAC,CAAC,SAAS,CAAC;gBAEhB,cAAc,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;gBAEpG,OAAO;oBACH,OAAO,EAAE,WAAW;oBACpB,WAAW;oBACX,YAAY;oBACZ,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACtC,CAAC;YACN,CAAC,CAAC,CAAC;QACP,CAAC;aAAM,CAAC;YACJ,OAAO,SAAS,CAAC,KAAK,IAAI,EAAE;gBACxB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;oBACxD,KAAK,EAAE,IAAI,CAAC,MAAM;oBAClB,QAAQ,EAAE,iBAAwB;oBAClC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACvD,CAAC,CAAC;gBAEH,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC;gBAC7C,MAAM,SAAS,GAAG,mBAAmB,CAAC,OAAO,EAAE,UAAiB,CAAC,CAAC;gBAElE,OAAO;oBACH,OAAO,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE;oBAC/B,WAAW,EAAE,QAAQ,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;oBAC/C,YAAY,EAAE,QAAQ,CAAC,KAAK,EAAE,iBAAiB,IAAI,CAAC;oBACpD,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACtC,CAAC;YACN,CAAC,CAAC,CAAC;QACP,CAAC;IACL,CAAC;CACJ;AAED,eAAe,cAAc,CAAC"}
|
|
@@ -2,9 +2,19 @@ import type { ToolSchema } from "./ai-provider.js";
|
|
|
2
2
|
/**
|
|
3
3
|
* Translates our standard, provider-agnostic ToolSchema[] into the
|
|
4
4
|
* OpenAI-compatible `tools: [{type:"function", function:{...}}]` shape.
|
|
5
|
-
* Both the OpenAI SDK (DeepSeekProvider) and Ollama's
|
|
6
|
-
* use this same shape, so it's shared rather than
|
|
5
|
+
* Both the OpenAI SDK (DeepSeekProvider, OpenAIProvider) and Ollama's
|
|
6
|
+
* function-calling API use this same shape, so it's shared rather than
|
|
7
|
+
* duplicated per provider.
|
|
7
8
|
*/
|
|
8
9
|
export declare function toOpenAIFunctionTools(tools: ToolSchema[]): any[];
|
|
10
|
+
/**
|
|
11
|
+
* Translates our standard, provider-agnostic ToolSchema[] into Anthropic's
|
|
12
|
+
* `tools: [{name, description, input_schema:{...}}]` shape. Anthropic uses
|
|
13
|
+
* a flat JSON-Schema `input_schema` rather than OpenAI's nested
|
|
14
|
+
* `function.parameters`, so this can't reuse toOpenAIFunctionTools — kept
|
|
15
|
+
* as its own function but mirroring the same property-bag construction for
|
|
16
|
+
* consistency with the OpenAI-shape translator above.
|
|
17
|
+
*/
|
|
18
|
+
export declare function toAnthropicTools(tools: ToolSchema[]): any[];
|
|
9
19
|
export default toOpenAIFunctionTools;
|
|
10
20
|
//# sourceMappingURL=tool-schema-translator.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"tool-schema-translator.d.ts","sourceRoot":"","sources":["../../src/core/tool-schema-translator.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAEnD
|
|
1
|
+
{"version":3,"file":"tool-schema-translator.d.ts","sourceRoot":"","sources":["../../src/core/tool-schema-translator.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAEnD;;;;;;GAMG;AACH,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,UAAU,EAAE,GAAG,GAAG,EAAE,CAoBhE;AAED;;;;;;;GAOG;AACH,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,UAAU,EAAE,GAAG,GAAG,EAAE,CAiB3D;AAED,eAAe,qBAAqB,CAAC"}
|
|
@@ -1,8 +1,9 @@
|
|
|
1
1
|
/**
|
|
2
2
|
* Translates our standard, provider-agnostic ToolSchema[] into the
|
|
3
3
|
* OpenAI-compatible `tools: [{type:"function", function:{...}}]` shape.
|
|
4
|
-
* Both the OpenAI SDK (DeepSeekProvider) and Ollama's
|
|
5
|
-
* use this same shape, so it's shared rather than
|
|
4
|
+
* Both the OpenAI SDK (DeepSeekProvider, OpenAIProvider) and Ollama's
|
|
5
|
+
* function-calling API use this same shape, so it's shared rather than
|
|
6
|
+
* duplicated per provider.
|
|
6
7
|
*/
|
|
7
8
|
export function toOpenAIFunctionTools(tools) {
|
|
8
9
|
return tools.map(t => ({
|
|
@@ -25,5 +26,31 @@ export function toOpenAIFunctionTools(tools) {
|
|
|
25
26
|
},
|
|
26
27
|
}));
|
|
27
28
|
}
|
|
29
|
+
/**
|
|
30
|
+
* Translates our standard, provider-agnostic ToolSchema[] into Anthropic's
|
|
31
|
+
* `tools: [{name, description, input_schema:{...}}]` shape. Anthropic uses
|
|
32
|
+
* a flat JSON-Schema `input_schema` rather than OpenAI's nested
|
|
33
|
+
* `function.parameters`, so this can't reuse toOpenAIFunctionTools — kept
|
|
34
|
+
* as its own function but mirroring the same property-bag construction for
|
|
35
|
+
* consistency with the OpenAI-shape translator above.
|
|
36
|
+
*/
|
|
37
|
+
export function toAnthropicTools(tools) {
|
|
38
|
+
return tools.map(t => ({
|
|
39
|
+
name: t.name,
|
|
40
|
+
description: t.description,
|
|
41
|
+
input_schema: {
|
|
42
|
+
type: "object",
|
|
43
|
+
properties: Object.fromEntries(t.inputs.map(i => [
|
|
44
|
+
i.name,
|
|
45
|
+
{
|
|
46
|
+
type: i.type === "array" ? "array" : i.type === "object" ? "object" : i.type,
|
|
47
|
+
description: i.description,
|
|
48
|
+
...(i.default !== undefined ? { default: i.default } : {}),
|
|
49
|
+
}
|
|
50
|
+
])),
|
|
51
|
+
required: t.inputs.filter(i => i.required).map(i => i.name),
|
|
52
|
+
},
|
|
53
|
+
}));
|
|
54
|
+
}
|
|
28
55
|
export default toOpenAIFunctionTools;
|
|
29
56
|
//# sourceMappingURL=tool-schema-translator.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"tool-schema-translator.js","sourceRoot":"","sources":["../../src/core/tool-schema-translator.ts"],"names":[],"mappings":"AAEA
|
|
1
|
+
{"version":3,"file":"tool-schema-translator.js","sourceRoot":"","sources":["../../src/core/tool-schema-translator.ts"],"names":[],"mappings":"AAEA;;;;;;GAMG;AACH,MAAM,UAAU,qBAAqB,CAAC,KAAmB;IACrD,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACnB,IAAI,EAAE,UAAU;QAChB,QAAQ,EAAE;YACN,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,WAAW,EAAE,CAAC,CAAC,WAAW;YAC1B,UAAU,EAAE;gBACR,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC7C,CAAC,CAAC,IAAI;oBACN;wBACI,IAAI,EAAE,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI;wBAC5E,WAAW,EAAE,CAAC,CAAC,WAAW;wBAC1B,GAAG,CAAC,CAAC,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBAC7D;iBACJ,CAAC,CAAC;gBACH,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;aAC9D;SACJ;KACJ,CAAC,CAAC,CAAC;AACR,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,gBAAgB,CAAC,KAAmB;IAChD,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACnB,IAAI,EAAE,CAAC,CAAC,IAAI;QACZ,WAAW,EAAE,CAAC,CAAC,WAAW;QAC1B,YAAY,EAAE;YACV,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC7C,CAAC,CAAC,IAAI;gBACN;oBACI,IAAI,EAAE,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI;oBAC5E,WAAW,EAAE,CAAC,CAAC,WAAW;oBAC1B,GAAG,CAAC,CAAC,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC7D;aACJ,CAAC,CAAC;YACH,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;SAC9D;KACJ,CAAC,CAAC,CAAC;AACR,CAAC;AAED,eAAe,qBAAqB,CAAC"}
|
package/dist/index.d.ts
CHANGED
|
@@ -3,9 +3,12 @@ export type { ChatResponse, Message, StreamCallback, ToolSchema, ToolCallRequest
|
|
|
3
3
|
export { BaseAgent } from "./core/base-agent.js";
|
|
4
4
|
export type { ExtractedSegment, ParsedToolCall } from "./core/base-agent.js";
|
|
5
5
|
export { DeepSeekProvider } from "./core/deepseek-provider.js";
|
|
6
|
+
export { OpenAIProvider } from "./core/openai-provider.js";
|
|
7
|
+
export { AnthropicProvider } from "./core/anthropic-provider.js";
|
|
6
8
|
export { MCPClient } from "./core/mcp-client.js";
|
|
7
9
|
export { MCPConnection } from "./core/mcp-connection.js";
|
|
8
10
|
export { MCPIPCConnection } from "./core/mcp-ipc-connection.js";
|
|
11
|
+
export { MCPWSConnection } from "./core/mcp-ws-connection.js";
|
|
9
12
|
export { MCPServer } from "./core/mcp-server.js";
|
|
10
13
|
export { MCPTool, MCPCustomClass, generateRefId, normalizeToolOutput } from "./core/mcp.js";
|
|
11
14
|
export type { MCP, MCPToolOutput } from "./core/mcp.js";
|
package/dist/index.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACnD,YAAY,EAAE,YAAY,EAAE,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,eAAe,EAAE,kBAAkB,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAC1J,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,YAAY,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAC7E,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AACzD,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAChE,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAE,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AAC5F,YAAY,EAAE,GAAG,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAC3D,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAC;AACxC,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAC3D,OAAO,EAAE,qBAAqB,EAAE,MAAM,kCAAkC,CAAC;AACzE,OAAO,EAAE,qBAAqB,EAAE,MAAM,2BAA2B,CAAC;AAClE,YAAY,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,MAAM,2BAA2B,CAAC;AAEpI,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAGxD,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAEpD,OAAO,EAAE,kBAAkB,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAElF,OAAO,EACH,kBAAkB,EAClB,kBAAkB,EAClB,gBAAgB,EAChB,eAAe,EACf,kBAAkB,EAClB,kBAAkB,EAClB,iBAAiB,EACjB,SAAS,EACT,eAAe,GAClB,MAAM,uBAAuB,CAAC;AAC/B,YAAY,EAAE,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAE3E,OAAO,EACH,wBAAwB,EACxB,wBAAwB,EACxB,sBAAsB,EACtB,mBAAmB,EACnB,kBAAkB,EAClB,iBAAiB,EACjB,eAAe,EACf,gBAAgB,EAChB,eAAe,EACf,qBAAqB,EACrB,iBAAiB,EACjB,wBAAwB,EACxB,kBAAkB,EAClB,qBAAqB,EACrB,qBAAqB,EACrB,YAAY,EACZ,kBAAkB,GACrB,MAAM,0BAA0B,CAAC;AAClC,YAAY,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAE7D,OAAO,EACH,cAAc,EACd,eAAe,EACf,cAAc,EACd,aAAa,EACb,eAAe,EACf,YAAY,EACZ,UAAU,EACV,UAAU,EACV,OAAO,EACP,aAAa,GAChB,MAAM,qBAAqB,CAAC;AAC7B,YAAY,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAEnD,OAAO,EACH,WAAW,EACX,YAAY,EACZ,YAAY,EACZ,YAAY,EACZ,kBAAkB,EAClB,iBAAiB,EACjB,SAAS,EACT,eAAe,GAClB,MAAM,uBAAuB,CAAC;AAC/B,YAAY,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAEvD,OAAO,EACH,cAAc,EACd,gBAAgB,EAChB,kBAAkB,EAClB,YAAY,EACZ,gBAAgB,EAChB,kBAAkB,EACrB,MAAM,0BAA0B,CAAC;AAClC,YAAY,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAE7D,OAAO,EACH,kBAAkB,EAClB,mBAAmB,EACnB,yBAAyB,EAC5B,MAAM,kCAAkC,CAAC;AAC1C,YAAY,EAAE,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;AAE5E,OAAO,EAAE,oBAAoB,EAAE,MAAM,8BAA8B,CAAC;AACpE,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,EAAE,qBAAqB,EAAE,MAAM,+BAA+B,CAAC;AACtE,OAAO,EAAE,uBAAuB,EAAE,MAAM,kCAAkC,CAAC;AAC3E,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,mBAAmB,EAAE,MAAM,6BAA6B,CAAC;AAClE,OAAO,EAAE,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAChE,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAC3D,OAAO,EAAE,oBAAoB,EAAE,MAAM,iCAAiC,CAAC"}
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACnD,YAAY,EAAE,YAAY,EAAE,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,eAAe,EAAE,kBAAkB,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAC1J,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,YAAY,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAC7E,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAC3D,OAAO,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AACjE,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AACzD,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,6BAA6B,CAAC;AAC9D,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EAAE,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AAC5F,YAAY,EAAE,GAAG,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAC3D,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAC;AACxC,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAC3D,OAAO,EAAE,qBAAqB,EAAE,MAAM,kCAAkC,CAAC;AACzE,OAAO,EAAE,qBAAqB,EAAE,MAAM,2BAA2B,CAAC;AAClE,YAAY,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,MAAM,2BAA2B,CAAC;AAEpI,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAGxD,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAEpD,OAAO,EAAE,kBAAkB,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAElF,OAAO,EACH,kBAAkB,EAClB,kBAAkB,EAClB,gBAAgB,EAChB,eAAe,EACf,kBAAkB,EAClB,kBAAkB,EAClB,iBAAiB,EACjB,SAAS,EACT,eAAe,GAClB,MAAM,uBAAuB,CAAC;AAC/B,YAAY,EAAE,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAE3E,OAAO,EACH,wBAAwB,EACxB,wBAAwB,EACxB,sBAAsB,EACtB,mBAAmB,EACnB,kBAAkB,EAClB,iBAAiB,EACjB,eAAe,EACf,gBAAgB,EAChB,eAAe,EACf,qBAAqB,EACrB,iBAAiB,EACjB,wBAAwB,EACxB,kBAAkB,EAClB,qBAAqB,EACrB,qBAAqB,EACrB,YAAY,EACZ,kBAAkB,GACrB,MAAM,0BAA0B,CAAC;AAClC,YAAY,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAE7D,OAAO,EACH,cAAc,EACd,eAAe,EACf,cAAc,EACd,aAAa,EACb,eAAe,EACf,YAAY,EACZ,UAAU,EACV,UAAU,EACV,OAAO,EACP,aAAa,GAChB,MAAM,qBAAqB,CAAC;AAC7B,YAAY,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAEnD,OAAO,EACH,WAAW,EACX,YAAY,EACZ,YAAY,EACZ,YAAY,EACZ,kBAAkB,EAClB,iBAAiB,EACjB,SAAS,EACT,eAAe,GAClB,MAAM,uBAAuB,CAAC;AAC/B,YAAY,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAEvD,OAAO,EACH,cAAc,EACd,gBAAgB,EAChB,kBAAkB,EAClB,YAAY,EACZ,gBAAgB,EAChB,kBAAkB,EACrB,MAAM,0BAA0B,CAAC;AAClC,YAAY,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAE7D,OAAO,EACH,kBAAkB,EAClB,mBAAmB,EACnB,yBAAyB,EAC5B,MAAM,kCAAkC,CAAC;AAC1C,YAAY,EAAE,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;AAE5E,OAAO,EAAE,oBAAoB,EAAE,MAAM,8BAA8B,CAAC;AACpE,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,EAAE,qBAAqB,EAAE,MAAM,+BAA+B,CAAC;AACtE,OAAO,EAAE,uBAAuB,EAAE,MAAM,kCAAkC,CAAC;AAC3E,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,mBAAmB,EAAE,MAAM,6BAA6B,CAAC;AAClE,OAAO,EAAE,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAChE,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAC3D,OAAO,EAAE,oBAAoB,EAAE,MAAM,iCAAiC,CAAC"}
|
package/dist/index.js
CHANGED
|
@@ -1,9 +1,12 @@
|
|
|
1
1
|
export { AIProvider } from "./core/ai-provider.js";
|
|
2
2
|
export { BaseAgent } from "./core/base-agent.js";
|
|
3
3
|
export { DeepSeekProvider } from "./core/deepseek-provider.js";
|
|
4
|
+
export { OpenAIProvider } from "./core/openai-provider.js";
|
|
5
|
+
export { AnthropicProvider } from "./core/anthropic-provider.js";
|
|
4
6
|
export { MCPClient } from "./core/mcp-client.js";
|
|
5
7
|
export { MCPConnection } from "./core/mcp-connection.js";
|
|
6
8
|
export { MCPIPCConnection } from "./core/mcp-ipc-connection.js";
|
|
9
|
+
export { MCPWSConnection } from "./core/mcp-ws-connection.js";
|
|
7
10
|
export { MCPServer } from "./core/mcp-server.js";
|
|
8
11
|
export { MCPTool, MCPCustomClass, generateRefId, normalizeToolOutput } from "./core/mcp.js";
|
|
9
12
|
export { OllamaProvider } from "./core/ollama-provider.js";
|