@axiom-lattice/client-sdk 2.1.56 → 2.1.58
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/ChunkMessageMerger.d.ts.map +1 -1
- package/dist/ChunkMessageMerger.js +42 -103
- package/dist/ChunkMessageMerger.js.map +1 -1
- package/dist/abstract-client.d.ts +98 -1
- package/dist/abstract-client.d.ts.map +1 -1
- package/dist/abstract-client.js +139 -0
- package/dist/abstract-client.js.map +1 -1
- package/dist/index.d.ts +98 -1
- package/dist/index.js +174 -46
- package/dist/index.js.map +1 -1
- package/dist/index.mjs +174 -46
- package/dist/index.mjs.map +1 -1
- package/dist/types.d.ts.map +1 -1
- package/dist/types.js.map +1 -1
- package/package.json +2 -2
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ChunkMessageMerger.d.ts","sourceRoot":"","sources":["../src/ChunkMessageMerger.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAGL,YAAY,EACb,MAAM,0BAA0B,CAAC;AAGlC;;;GAGG;AACH,wBAAgB,yBAAyB;
|
|
1
|
+
{"version":3,"file":"ChunkMessageMerger.d.ts","sourceRoot":"","sources":["../src/ChunkMessageMerger.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAGL,YAAY,EACb,MAAM,0BAA0B,CAAC;AAGlC;;;GAGG;AACH,wBAAgB,yBAAyB;kBAqJlB,YAAY;4BAbF,GAAG,EAAE;uBA4FZ,GAAG,EAAE;uCAQW,GAAG,EAAE;mCAaP,MAAM,KAAG,OAAO;;EAyCvD"}
|
|
@@ -53,7 +53,7 @@ export function createSimpleMessageMerger() {
|
|
|
53
53
|
* Ensures a message exists, creating it if needed
|
|
54
54
|
* @param id - Message ID
|
|
55
55
|
* @param role - Message role
|
|
56
|
-
* @returns The message object
|
|
56
|
+
* @returns The message object (may not be in the array for tool messages)
|
|
57
57
|
*/
|
|
58
58
|
function ensureMessage(id, role) {
|
|
59
59
|
const index = messageMap.get(id);
|
|
@@ -86,22 +86,30 @@ export function createSimpleMessageMerger() {
|
|
|
86
86
|
return newMessage;
|
|
87
87
|
}
|
|
88
88
|
/**
|
|
89
|
-
*
|
|
90
|
-
*
|
|
89
|
+
* Replaces a message in the array with an updated shallow clone.
|
|
90
|
+
* Returns the new message object, or null if the message is not in the array.
|
|
91
91
|
*/
|
|
92
|
-
function
|
|
93
|
-
|
|
94
|
-
|
|
95
|
-
|
|
92
|
+
function replaceMessage(id, updates) {
|
|
93
|
+
const index = messageMap.get(id);
|
|
94
|
+
if (index === undefined)
|
|
95
|
+
return null;
|
|
96
|
+
const updated = { ...messages[index], ...updates };
|
|
97
|
+
messages[index] = updated;
|
|
98
|
+
return updated;
|
|
99
|
+
}
|
|
100
|
+
/**
|
|
101
|
+
* Builds tool calls from incremental chunks for a message.
|
|
102
|
+
* Returns the tool calls array, or null if no complete tool calls exist.
|
|
103
|
+
*/
|
|
104
|
+
function buildToolCalls(messageId) {
|
|
105
|
+
const builders = toolBuilders.get(messageId);
|
|
96
106
|
if (!builders || builders.size === 0)
|
|
97
|
-
return;
|
|
107
|
+
return null;
|
|
98
108
|
const toolCalls = [];
|
|
99
|
-
for (const [
|
|
109
|
+
for (const [, builder] of Array.from(builders.entries())) {
|
|
100
110
|
if (builder.name && builder.id) {
|
|
101
|
-
// Try to parse arguments as JSON object
|
|
102
111
|
const args = builder.args.trim();
|
|
103
112
|
const parsedArgs = safeJsonParse(args);
|
|
104
|
-
// Create ToolCall object
|
|
105
113
|
toolCalls.push({
|
|
106
114
|
id: builder.id,
|
|
107
115
|
name: builder.name,
|
|
@@ -110,10 +118,7 @@ export function createSimpleMessageMerger() {
|
|
|
110
118
|
});
|
|
111
119
|
}
|
|
112
120
|
}
|
|
113
|
-
|
|
114
|
-
if (toolCalls.length > 0) {
|
|
115
|
-
message.tool_calls = toolCalls;
|
|
116
|
-
}
|
|
121
|
+
return toolCalls.length > 0 ? toolCalls : null;
|
|
117
122
|
}
|
|
118
123
|
/**
|
|
119
124
|
* Initialize messages
|
|
@@ -126,25 +131,33 @@ export function createSimpleMessageMerger() {
|
|
|
126
131
|
});
|
|
127
132
|
}
|
|
128
133
|
/**
|
|
129
|
-
* Processes a message chunk
|
|
134
|
+
* Processes a message chunk.
|
|
135
|
+
* All updates use immutable patterns — message objects in the array
|
|
136
|
+
* are replaced with shallow clones when any field changes.
|
|
130
137
|
* @param chunk - Message chunk to process
|
|
131
138
|
*/
|
|
132
139
|
function push(chunk) {
|
|
133
|
-
// console.log("push chunk", chunk);
|
|
134
140
|
const role = normalizeRole(chunk.type);
|
|
135
|
-
|
|
141
|
+
let message = ensureMessage(chunk.data.id, role);
|
|
142
|
+
const isInArray = message.role !== "tool";
|
|
136
143
|
// Accumulate content
|
|
137
144
|
if (chunk.data.content) {
|
|
138
|
-
|
|
145
|
+
const newContent = (message.content || "") + chunk.data.content;
|
|
146
|
+
if (isInArray) {
|
|
147
|
+
message = replaceMessage(chunk.data.id, { content: newContent }) || message;
|
|
148
|
+
}
|
|
149
|
+
else {
|
|
150
|
+
message.content = newContent;
|
|
151
|
+
}
|
|
139
152
|
}
|
|
140
|
-
// Handle
|
|
153
|
+
// Handle complete tool calls
|
|
141
154
|
if (message.role === "ai" &&
|
|
142
155
|
chunk.data.tool_calls &&
|
|
143
156
|
chunk.data.tool_calls.length > 0) {
|
|
144
157
|
for (const toolCall of chunk.data.tool_calls) {
|
|
145
158
|
toolId_MessageIdMap.set(toolCall.id, message.id);
|
|
146
159
|
}
|
|
147
|
-
message
|
|
160
|
+
message = replaceMessage(chunk.data.id, { tool_calls: [...chunk.data.tool_calls] }) || message;
|
|
148
161
|
}
|
|
149
162
|
// Handle tool call chunks (incremental)
|
|
150
163
|
if (chunk.data.tool_call_chunks && message.role === "ai") {
|
|
@@ -166,8 +179,12 @@ export function createSimpleMessageMerger() {
|
|
|
166
179
|
if (chunk_item.args)
|
|
167
180
|
builder.args += chunk_item.args;
|
|
168
181
|
}
|
|
169
|
-
|
|
182
|
+
const toolCalls = buildToolCalls(message.id);
|
|
183
|
+
if (toolCalls) {
|
|
184
|
+
message = replaceMessage(chunk.data.id, { tool_calls: toolCalls }) || message;
|
|
185
|
+
}
|
|
170
186
|
}
|
|
187
|
+
// Handle tool responses — merge into the AI message's tool_calls
|
|
171
188
|
if (message.role === "tool" && chunk.data.tool_call_id) {
|
|
172
189
|
const original_tool_message_id = chunk.data.id;
|
|
173
190
|
const messageId = toolId_MessageIdMap.get(chunk.data.tool_call_id);
|
|
@@ -175,58 +192,16 @@ export function createSimpleMessageMerger() {
|
|
|
175
192
|
return;
|
|
176
193
|
const messageIndex = messageMap.get(messageId);
|
|
177
194
|
if (messageIndex !== undefined) {
|
|
178
|
-
const
|
|
179
|
-
|
|
195
|
+
const aiMessage = messages[messageIndex];
|
|
196
|
+
const updatedToolCalls = aiMessage.tool_calls?.map((tc) => {
|
|
180
197
|
if (tc.id === chunk.data.tool_call_id) {
|
|
181
198
|
return { ...tc, response: chunk.data.content, status: "success", original_tool_message_id };
|
|
182
199
|
}
|
|
183
200
|
return tc;
|
|
184
201
|
});
|
|
185
|
-
messages[messageIndex] =
|
|
202
|
+
messages[messageIndex] = { ...aiMessage, tool_calls: updatedToolCalls };
|
|
186
203
|
}
|
|
187
204
|
}
|
|
188
|
-
// // Handle complete tool calls (overwrite)
|
|
189
|
-
// if (chunk.data.tool_calls && message.role === "ai") {
|
|
190
|
-
// const toolCalls: any[] = chunk.data.tool_calls
|
|
191
|
-
// .filter((tc) => tc.name && tc.id)
|
|
192
|
-
// .map((tc) => ({
|
|
193
|
-
// id: tc.id,
|
|
194
|
-
// name: tc.name,
|
|
195
|
-
// args: tc.args,
|
|
196
|
-
// type: "tool_call",
|
|
197
|
-
// response: tc.response,
|
|
198
|
-
// status: "success",
|
|
199
|
-
// }));
|
|
200
|
-
// if (toolCalls.length > 0) {
|
|
201
|
-
// message.tool_calls = toolCalls;
|
|
202
|
-
// }
|
|
203
|
-
// }
|
|
204
|
-
// Handle tool calls in additional_kwargs
|
|
205
|
-
// if (
|
|
206
|
-
// chunk.data.additional_kwargs?.tool_calls &&
|
|
207
|
-
// message.role === "ai" &&
|
|
208
|
-
// !message.tool_calls
|
|
209
|
-
// ) {
|
|
210
|
-
// const toolCalls: any[] = [];
|
|
211
|
-
// for (const tc of chunk.data.additional_kwargs.tool_calls) {
|
|
212
|
-
// if (tc.function?.name && tc.id) {
|
|
213
|
-
// // Convert to our format
|
|
214
|
-
// const args =
|
|
215
|
-
// typeof tc.function.arguments === "string"
|
|
216
|
-
// ? safeJsonParse(tc.function.arguments)
|
|
217
|
-
// : tc.function.arguments || {};
|
|
218
|
-
// toolCalls.push({
|
|
219
|
-
// id: tc.id,
|
|
220
|
-
// name: tc.function.name,
|
|
221
|
-
// args,
|
|
222
|
-
// type: "tool_call",
|
|
223
|
-
// });
|
|
224
|
-
// }
|
|
225
|
-
// }
|
|
226
|
-
// if (toolCalls.length > 0) {
|
|
227
|
-
// message.tool_calls = toolCalls;
|
|
228
|
-
// }
|
|
229
|
-
// }
|
|
230
205
|
}
|
|
231
206
|
/**
|
|
232
207
|
* Gets all messages with tool responses merged into tool calls
|
|
@@ -234,42 +209,6 @@ export function createSimpleMessageMerger() {
|
|
|
234
209
|
*/
|
|
235
210
|
function getMessages() {
|
|
236
211
|
return [...messages];
|
|
237
|
-
// Create a map to store tool responses by their tool_call_id
|
|
238
|
-
const toolResponsesMap = {};
|
|
239
|
-
// First pass: collect all tool responses
|
|
240
|
-
messages.forEach((message) => {
|
|
241
|
-
if (message.role === "tool" && message.tool_call_id) {
|
|
242
|
-
toolResponsesMap[message.tool_call_id] = message;
|
|
243
|
-
}
|
|
244
|
-
});
|
|
245
|
-
// Second pass: merge tool responses with their corresponding AI messages
|
|
246
|
-
return messages
|
|
247
|
-
.filter((message) => {
|
|
248
|
-
// Keep all messages except tool messages
|
|
249
|
-
return message.role !== "tool";
|
|
250
|
-
})
|
|
251
|
-
.map((message) => {
|
|
252
|
-
if (message.role === "ai" && message.tool_calls) {
|
|
253
|
-
// Create a new message object with merged tool responses
|
|
254
|
-
const mergedMessage = { ...message };
|
|
255
|
-
// Add tool responses to the corresponding tool calls
|
|
256
|
-
if (mergedMessage.tool_calls) {
|
|
257
|
-
mergedMessage.tool_calls = mergedMessage.tool_calls.map((toolCall) => {
|
|
258
|
-
const toolResponse = toolResponsesMap[toolCall.id];
|
|
259
|
-
if (toolResponse) {
|
|
260
|
-
return {
|
|
261
|
-
...toolCall,
|
|
262
|
-
response: toolResponse.content,
|
|
263
|
-
status: "success",
|
|
264
|
-
};
|
|
265
|
-
}
|
|
266
|
-
return toolCall;
|
|
267
|
-
});
|
|
268
|
-
}
|
|
269
|
-
return mergedMessage;
|
|
270
|
-
}
|
|
271
|
-
return message;
|
|
272
|
-
});
|
|
273
212
|
}
|
|
274
213
|
/**
|
|
275
214
|
* Gets messages without tool calls
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ChunkMessageMerger.js","sourceRoot":"","sources":["../src/ChunkMessageMerger.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAOH,OAAO,EAAE,KAAK,EAAE,MAAM,yBAAyB,CAAC;AAEhD;;;GAGG;AACH,MAAM,UAAU,yBAAyB;IACvC,IAAI,QAAQ,GAAc,EAAE,CAAC;IAC7B,MAAM,UAAU,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC7C,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAkB,CAAC;IAEtD,uDAAuD;IACvD,MAAM,YAAY,GAAG,IAAI,GAAG,EAUzB,CAAC;IAEJ;;;;OAIG;IACH,SAAS,aAAa,CAAC,IAAY;QACjC,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,IAAI;gBACP,OAAO,IAAI,CAAC;YACd,KAAK,OAAO,CAAC;YACb,KAAK,MAAM;gBACT,OAAO,OAAO,CAAC;YACjB,KAAK,QAAQ;gBACX,OAAO,QAAQ,CAAC;YAClB,KAAK,MAAM;gBACT,OAAO,MAAM,CAAC;YAChB,KAAK,WAAW;gBACd,OAAO,WAAW,CAAC;YACrB;gBACE,OAAO,IAAI,CAAC;QAChB,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,SAAS,aAAa,CAAC,GAAW;QAChC,IAAI,CAAC;YACH,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC;QACpB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,GAAG,CAAC,CAAC,0CAA0C;QACxD,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,SAAS,aAAa,CAAC,EAAU,EAAE,IAAY;QAC7C,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACjC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC;QACzB,CAAC;QAED,qBAAqB;QACrB,IAAI,UAAe,CAAC;QACpB,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,OAAO;gBACV,UAAU,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;gBAChD,MAAM;YACR,KAAK,QAAQ;gBACX,UAAU,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;gBACjD,MAAM;YACR,KAAK,MAAM;gBACT,UAAU,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC;gBACjE,MAAM;YACR,KAAK,WAAW;gBACd,UAAU,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;gBACpD,MAAM;YACR;gBACE,UAAU,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACjD,CAAC;QAED,IAAI,UAAU,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC/B,+CAA+C;YAC/C,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC1B,UAAU,CAAC,GAAG,CAAC,EAAE,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC1C,CAAC;QACD,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;;OAGG;IACH,SAAS,
|
|
1
|
+
{"version":3,"file":"ChunkMessageMerger.js","sourceRoot":"","sources":["../src/ChunkMessageMerger.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAOH,OAAO,EAAE,KAAK,EAAE,MAAM,yBAAyB,CAAC;AAEhD;;;GAGG;AACH,MAAM,UAAU,yBAAyB;IACvC,IAAI,QAAQ,GAAc,EAAE,CAAC;IAC7B,MAAM,UAAU,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC7C,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAkB,CAAC;IAEtD,uDAAuD;IACvD,MAAM,YAAY,GAAG,IAAI,GAAG,EAUzB,CAAC;IAEJ;;;;OAIG;IACH,SAAS,aAAa,CAAC,IAAY;QACjC,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,IAAI;gBACP,OAAO,IAAI,CAAC;YACd,KAAK,OAAO,CAAC;YACb,KAAK,MAAM;gBACT,OAAO,OAAO,CAAC;YACjB,KAAK,QAAQ;gBACX,OAAO,QAAQ,CAAC;YAClB,KAAK,MAAM;gBACT,OAAO,MAAM,CAAC;YAChB,KAAK,WAAW;gBACd,OAAO,WAAW,CAAC;YACrB;gBACE,OAAO,IAAI,CAAC;QAChB,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,SAAS,aAAa,CAAC,GAAW;QAChC,IAAI,CAAC;YACH,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC;QACpB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,GAAG,CAAC,CAAC,0CAA0C;QACxD,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,SAAS,aAAa,CAAC,EAAU,EAAE,IAAY;QAC7C,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACjC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC;QACzB,CAAC;QAED,qBAAqB;QACrB,IAAI,UAAe,CAAC;QACpB,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,OAAO;gBACV,UAAU,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;gBAChD,MAAM;YACR,KAAK,QAAQ;gBACX,UAAU,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;gBACjD,MAAM;YACR,KAAK,MAAM;gBACT,UAAU,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC;gBACjE,MAAM;YACR,KAAK,WAAW;gBACd,UAAU,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;gBACpD,MAAM;YACR;gBACE,UAAU,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACjD,CAAC;QAED,IAAI,UAAU,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC/B,+CAA+C;YAC/C,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC1B,UAAU,CAAC,GAAG,CAAC,EAAE,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC1C,CAAC;QACD,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;;OAGG;IACH,SAAS,cAAc,CAAC,EAAU,EAAE,OAAqB;QACvD,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACjC,IAAI,KAAK,KAAK,SAAS;YAAE,OAAO,IAAI,CAAC;QACrC,MAAM,OAAO,GAAG,EAAE,GAAG,QAAQ,CAAC,KAAK,CAAC,EAAE,GAAG,OAAO,EAAE,CAAC;QACnD,QAAQ,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC;QAC1B,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;;OAGG;IACH,SAAS,cAAc,CAAC,SAAiB;QACvC,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,IAAI,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAElD,MAAM,SAAS,GAAU,EAAE,CAAC;QAC5B,KAAK,MAAM,CAAC,EAAE,OAAO,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;YACzD,IAAI,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,EAAE,EAAE,CAAC;gBAC/B,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACjC,MAAM,UAAU,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;gBAEvC,SAAS,CAAC,IAAI,CAAC;oBACb,EAAE,EAAE,OAAO,CAAC,EAAE;oBACd,IAAI,EAAE,OAAO,CAAC,IAAI;oBAClB,IAAI,EACF,OAAO,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE;oBACrE,IAAI,EAAE,WAAW;iBAClB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC;IACjD,CAAC;IAED;;;OAGG;IACH,SAAS,eAAe,CAAC,IAAW;QAClC,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YACnB,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;OAKG;IACH,SAAS,IAAI,CAAC,KAAmB;QAC/B,MAAM,IAAI,GAAG,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACvC,IAAI,OAAO,GAAG,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;QAEjD,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC;QAE1C,qBAAqB;QACrB,IAAI,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YACvB,MAAM,UAAU,GAAG,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC;YAChE,IAAI,SAAS,EAAE,CAAC;gBACd,OAAO,GAAG,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,IAAI,OAAO,CAAC;YAC9E,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,OAAO,GAAG,UAAU,CAAC;YAC/B,CAAC;QACH,CAAC;QAED,6BAA6B;QAC7B,IACE,OAAO,CAAC,IAAI,KAAK,IAAI;YACrB,KAAK,CAAC,IAAI,CAAC,UAAU;YACrB,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAChC,CAAC;YACD,KAAK,MAAM,QAAQ,IAAI,KAAK,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC7C,mBAAmB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;YACnD,CAAC;YAED,OAAO,GAAG,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,IAAI,OAAO,CAAC;QACjG,CAAC;QAED,wCAAwC;QACxC,IAAI,KAAK,CAAC,IAAI,CAAC,gBAAgB,IAAI,OAAO,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YACzD,IAAI,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC5C,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,QAAQ,GAAG,IAAI,GAAG,EAAE,CAAC;gBACrB,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;YACzC,CAAC;YAED,KAAK,MAAM,UAAU,IAAI,KAAK,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACrD,IAAI,OAAO,GAAG,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;gBAC7C,IAAI,CAAC,OAAO,EAAE,CAAC;oBACb,OAAO,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;oBACvB,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;gBAC1C,CAAC;gBAED,IAAI,UAAU,CAAC,IAAI;oBAAE,OAAO,CAAC,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC;gBACpD,IAAI,UAAU,CAAC,EAAE;oBAAE,OAAO,CAAC,EAAE,GAAG,UAAU,CAAC,EAAE,CAAC;gBAC9C,IAAI,UAAU,CAAC,IAAI;oBAAE,OAAO,CAAC,IAAI,IAAI,UAAU,CAAC,IAAI,CAAC;YACvD,CAAC;YAED,MAAM,SAAS,GAAG,cAAc,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC7C,IAAI,SAAS,EAAE,CAAC;gBACd,OAAO,GAAG,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,IAAI,OAAO,CAAC;YAChF,CAAC;QACH,CAAC;QAED,iEAAiE;QACjE,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvD,MAAM,wBAAwB,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;YAC/C,MAAM,SAAS,GAAG,mBAAmB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACnE,IAAI,CAAC,SAAS;gBAAE,OAAO;YACvB,MAAM,YAAY,GAAG,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAE/C,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;gBAC/B,MAAM,SAAS,GAAG,QAAQ,CAAC,YAAY,CAAqB,CAAC;gBAC7D,MAAM,gBAAgB,GAAG,SAAS,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE;oBACxD,IAAI,EAAE,CAAC,EAAE,KAAK,KAAK,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;wBACtC,OAAO,EAAE,GAAG,EAAE,EAAE,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,wBAAwB,EAAE,CAAC;oBAC9F,CAAC;oBACD,OAAO,EAAE,CAAC;gBACZ,CAAC,CAAC,CAAC;gBACH,QAAQ,CAAC,YAAY,CAAC,GAAG,EAAE,GAAG,SAAS,EAAE,UAAU,EAAE,gBAAgB,EAAE,CAAC;YAC1E,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,SAAS,WAAW;QAClB,OAAO,CAAC,GAAG,QAAQ,CAAC,CAAC;IACvB,CAAC;IAED;;;OAGG;IACH,SAAS,2BAA2B;QAClC,OAAO,QAAQ,CAAC,MAAM,CACpB,CAAC,OAAO,EAAE,EAAE,CACV,OAAO,CAAC,IAAI,KAAK,MAAM;YACvB,CAAC,OAAO,CAAC,IAAI,KAAK,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CACjD,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACH,SAAS,iBAAiB,CAAC,SAAiB;QAC1C,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACxC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,OAAO,KAAK,CAAC;QACf,CAAC;QAED,oCAAoC;QACpC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAE1B,yBAAyB;QACzB,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAE7B,wDAAwD;QACxD,KAAK,MAAM,CAAC,EAAE,EAAE,GAAG,CAAC,IAAI,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC;YAC7C,IAAI,GAAG,GAAG,KAAK,EAAE,CAAC;gBAChB,UAAU,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;QACD,yCAAyC;QACzC,YAAY,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC/B,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,SAAS,KAAK;QACZ,QAAQ,GAAG,EAAE,CAAC;QACd,UAAU,CAAC,KAAK,EAAE,CAAC;QACnB,YAAY,CAAC,KAAK,EAAE,CAAC;QACrB,mBAAmB,CAAC,KAAK,EAAE,CAAC;IAC9B,CAAC;IAED,OAAO;QACL,IAAI;QACJ,eAAe;QACf,WAAW;QACX,2BAA2B;QAC3B,iBAAiB;QACjB,KAAK;KACN,CAAC;AACJ,CAAC"}
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import { Message, MessageChunk } from "@axiom-lattice/protocols";
|
|
2
2
|
import { AgentState, Assistant, ChatResponse, ChatSendOptions, ChatStreamOptions, ClientConfig, CreateAssistantOptions, CreateThreadOptions, GetMessagesOptions, GetScheduledTasksOptions, ListThreadsOptions, RegisterToolOptions, RemovePendingMessageOptions, ResumeStreamOptions, RunOptions, ScheduledTask, Thread, UpdateAssistantOptions, UpdateThreadOptions } from "./types";
|
|
3
|
-
import type { Skill, CreateSkillRequest, DataSource, DatabaseConfigEntry, CreateDatabaseConfigRequest, UpdateDatabaseConfigRequest, MetricsServerConfigEntry, CreateMetricsServerConfigRequest, UpdateMetricsServerConfigRequest, McpServerConfigEntry, CreateMcpServerConfigRequest, UpdateMcpServerConfigRequest } from "@axiom-lattice/protocols";
|
|
3
|
+
import type { Skill, CreateSkillRequest, DataSource, DatabaseConfigEntry, CreateDatabaseConfigRequest, UpdateDatabaseConfigRequest, MetricsServerConfigEntry, CreateMetricsServerConfigRequest, UpdateMetricsServerConfigRequest, McpServerConfigEntry, CreateMcpServerConfigRequest, UpdateMcpServerConfigRequest, ConnectionEntry } from "@axiom-lattice/protocols";
|
|
4
4
|
import type { Tool } from "./types";
|
|
5
5
|
import type { TaskItem, CreateTaskRequest, UpdateTaskRequest } from "@axiom-lattice/protocols";
|
|
6
6
|
import type { TestConnectionResponse, TestMcpServerResponse, AbortAgentParams } from "./types";
|
|
@@ -57,6 +57,29 @@ export declare abstract class AbstractClient {
|
|
|
57
57
|
update: (id: string, data: UpdateMcpServerConfigRequest) => Promise<McpServerConfigEntry>;
|
|
58
58
|
delete: (id: string) => Promise<void>;
|
|
59
59
|
test: (id: string) => Promise<TestMcpServerResponse>;
|
|
60
|
+
connect: (id: string) => Promise<McpServerConfigEntry>;
|
|
61
|
+
disconnect: (id: string) => Promise<McpServerConfigEntry>;
|
|
62
|
+
};
|
|
63
|
+
/**
|
|
64
|
+
* Connections namespace for managing plugin connections
|
|
65
|
+
*/
|
|
66
|
+
connections: {
|
|
67
|
+
list: (type: string) => Promise<ConnectionEntry[]>;
|
|
68
|
+
create: (data: {
|
|
69
|
+
type: string;
|
|
70
|
+
key: string;
|
|
71
|
+
name: string;
|
|
72
|
+
description?: string;
|
|
73
|
+
config: Record<string, unknown>;
|
|
74
|
+
}) => Promise<{
|
|
75
|
+
success: boolean;
|
|
76
|
+
data: ConnectionEntry;
|
|
77
|
+
}>;
|
|
78
|
+
update: (type: string, key: string, data: Record<string, unknown>) => Promise<{
|
|
79
|
+
success: boolean;
|
|
80
|
+
data: ConnectionEntry;
|
|
81
|
+
}>;
|
|
82
|
+
delete: (type: string, key: string) => Promise<void>;
|
|
60
83
|
};
|
|
61
84
|
/**
|
|
62
85
|
* Tasks namespace for managing tasks
|
|
@@ -339,6 +362,80 @@ export declare abstract class AbstractClient {
|
|
|
339
362
|
*/
|
|
340
363
|
delete: (id: string) => Promise<void>;
|
|
341
364
|
};
|
|
365
|
+
/**
|
|
366
|
+
* Collections namespace for managing vector collections
|
|
367
|
+
*/
|
|
368
|
+
collections: {
|
|
369
|
+
/**
|
|
370
|
+
* Lists all available collections
|
|
371
|
+
* @returns A promise that resolves to the list of collections
|
|
372
|
+
*/
|
|
373
|
+
list: () => Promise<any[]>;
|
|
374
|
+
/**
|
|
375
|
+
* Retrieves a single collection by name
|
|
376
|
+
* @param name - Collection name
|
|
377
|
+
* @returns A promise that resolves to the collection information
|
|
378
|
+
*/
|
|
379
|
+
get: (name: string) => Promise<any>;
|
|
380
|
+
/**
|
|
381
|
+
* Creates a new collection
|
|
382
|
+
* @param options - Options for creating a collection
|
|
383
|
+
* @returns A promise that resolves to the created collection
|
|
384
|
+
*/
|
|
385
|
+
create: (options: {
|
|
386
|
+
name: string;
|
|
387
|
+
label: string;
|
|
388
|
+
embeddingKey: string;
|
|
389
|
+
schema?: {
|
|
390
|
+
fields: Array<{
|
|
391
|
+
key: string;
|
|
392
|
+
type: string;
|
|
393
|
+
enumValues?: string[];
|
|
394
|
+
required?: boolean;
|
|
395
|
+
}>;
|
|
396
|
+
};
|
|
397
|
+
}) => Promise<any>;
|
|
398
|
+
/**
|
|
399
|
+
* Updates an existing collection by name
|
|
400
|
+
* @param name - Collection name
|
|
401
|
+
* @param options - Options for updating a collection
|
|
402
|
+
* @returns A promise that resolves to the updated collection
|
|
403
|
+
*/
|
|
404
|
+
update: (name: string, options: {
|
|
405
|
+
label?: string;
|
|
406
|
+
embeddingKey?: string;
|
|
407
|
+
schema?: any;
|
|
408
|
+
}) => Promise<any>;
|
|
409
|
+
/**
|
|
410
|
+
* Deletes a collection by name
|
|
411
|
+
* @param name - Collection name
|
|
412
|
+
* @returns A promise that resolves when the collection is deleted
|
|
413
|
+
*/
|
|
414
|
+
delete: (name: string) => Promise<void>;
|
|
415
|
+
/**
|
|
416
|
+
* Entries sub-namespace for a specific collection.
|
|
417
|
+
* Returns entry CRUD + search methods scoped to the given collection.
|
|
418
|
+
* @param collectionName - The collection name
|
|
419
|
+
* @returns Entry operation methods
|
|
420
|
+
*/
|
|
421
|
+
entries: (collectionName: string) => {
|
|
422
|
+
list: () => Promise<any[]>;
|
|
423
|
+
create: (data: {
|
|
424
|
+
content: string;
|
|
425
|
+
metadata?: Record<string, unknown>;
|
|
426
|
+
}) => Promise<any>;
|
|
427
|
+
get: (id: string) => Promise<any>;
|
|
428
|
+
update: (id: string, data: {
|
|
429
|
+
content?: string;
|
|
430
|
+
metadata?: Record<string, unknown>;
|
|
431
|
+
}) => Promise<any>;
|
|
432
|
+
delete: (id: string) => Promise<void>;
|
|
433
|
+
search: (query: string, opts?: {
|
|
434
|
+
filter?: Record<string, unknown>;
|
|
435
|
+
top_k?: number;
|
|
436
|
+
}) => Promise<any[]>;
|
|
437
|
+
};
|
|
438
|
+
};
|
|
342
439
|
/**
|
|
343
440
|
* Threads namespace for managing threads for the current assistant
|
|
344
441
|
*/
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"abstract-client.d.ts","sourceRoot":"","sources":["../src/abstract-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AACjE,OAAO,EACL,UAAU,EAEV,SAAS,EAGT,YAAY,EACZ,eAAe,EACf,iBAAiB,EACjB,YAAY,EACZ,sBAAsB,EACtB,mBAAmB,EACnB,kBAAkB,EAClB,wBAAwB,EACxB,kBAAkB,EAClB,mBAAmB,EACnB,2BAA2B,EAC3B,mBAAmB,EACnB,UAAU,EACV,aAAa,EAEb,MAAM,EAGN,sBAAsB,EACtB,mBAAmB,EACpB,MAAM,SAAS,CAAC;AACjB,OAAO,KAAK,EACV,KAAK,EACL,kBAAkB,EAClB,UAAU,EACV,mBAAmB,EACnB,2BAA2B,EAC3B,2BAA2B,EAC3B,wBAAwB,EACxB,gCAAgC,EAChC,gCAAgC,EAChC,oBAAoB,EACpB,4BAA4B,EAC5B,4BAA4B,
|
|
1
|
+
{"version":3,"file":"abstract-client.d.ts","sourceRoot":"","sources":["../src/abstract-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AACjE,OAAO,EACL,UAAU,EAEV,SAAS,EAGT,YAAY,EACZ,eAAe,EACf,iBAAiB,EACjB,YAAY,EACZ,sBAAsB,EACtB,mBAAmB,EACnB,kBAAkB,EAClB,wBAAwB,EACxB,kBAAkB,EAClB,mBAAmB,EACnB,2BAA2B,EAC3B,mBAAmB,EACnB,UAAU,EACV,aAAa,EAEb,MAAM,EAGN,sBAAsB,EACtB,mBAAmB,EACpB,MAAM,SAAS,CAAC;AACjB,OAAO,KAAK,EACV,KAAK,EACL,kBAAkB,EAClB,UAAU,EACV,mBAAmB,EACnB,2BAA2B,EAC3B,2BAA2B,EAC3B,wBAAwB,EACxB,gCAAgC,EAChC,gCAAgC,EAChC,oBAAoB,EACpB,4BAA4B,EAC5B,4BAA4B,EAC5B,eAAe,EAChB,MAAM,0BAA0B,CAAC;AAClC,OAAO,KAAK,EACV,IAAI,EACL,MAAM,SAAS,CAAC;AACjB,OAAO,KAAK,EACV,QAAQ,EACR,iBAAiB,EACjB,iBAAiB,EAClB,MAAM,0BAA0B,CAAC;AAClC,OAAO,KAAK,EAKV,sBAAsB,EAMtB,qBAAqB,EACrB,gBAAgB,EAGjB,MAAM,SAAS,CAAC;AAEjB;;;GAGG;AACH,8BAAsB,cAAc;IAClC,SAAS,CAAC,MAAM,EAAE,YAAY,CAAC;IACxB,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,SAAM;IACrB,SAAS,CAAC,eAAe,EAAE,GAAG,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAAa;IACxE,OAAO,CAAC,qBAAqB,CAAC,CAAa;IAE3C;;OAEG;IACH,WAAW,EAAE;QACX,IAAI,EAAE,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QAC5B,GAAG,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;QACnC,MAAM,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;QACrC,MAAM,EAAE,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;KACvC,CAAC;IAEF;;OAEG;IACH,eAAe,EAAE;QACf,IAAI,EAAE,MAAM,OAAO,CAAC,mBAAmB,EAAE,CAAC,CAAC;QAC3C,GAAG,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,mBAAmB,CAAC,CAAC;QAClD,MAAM,EAAE,CAAC,IAAI,EAAE,2BAA2B,KAAK,OAAO,CAAC,mBAAmB,CAAC,CAAC;QAC5E,MAAM,EAAE,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,2BAA2B,KAAK,OAAO,CAAC,mBAAmB,CAAC,CAAC;QACxF,MAAM,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,sBAAsB,CAAC,CAAC;KACvD,CAAC;IAEF;;OAEG;IACH,cAAc,EAAE;QACd,IAAI,EAAE,MAAM,OAAO,CAAC,wBAAwB,EAAE,CAAC,CAAC;QAChD,GAAG,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,wBAAwB,CAAC,CAAC;QACvD,MAAM,EAAE,CAAC,IAAI,EAAE,gCAAgC,KAAK,OAAO,CAAC,wBAAwB,CAAC,CAAC;QACtF,MAAM,EAAE,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,gCAAgC,KAAK,OAAO,CAAC,wBAAwB,CAAC,CAAC;QAClG,MAAM,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,sBAAsB,CAAC,CAAC;QACtD,cAAc,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;KACvD,CAAC;IAEF;;OAEG;IACH,UAAU,EAAE;QACV,IAAI,EAAE,MAAM,OAAO,CAAC,oBAAoB,EAAE,CAAC,CAAC;QAC5C,GAAG,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,oBAAoB,CAAC,CAAC;QACnD,MAAM,EAAE,CAAC,IAAI,EAAE,4BAA4B,KAAK,OAAO,CAAC,oBAAoB,CAAC,CAAC;QAC9E,MAAM,EAAE,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,4BAA4B,KAAK,OAAO,CAAC,oBAAoB,CAAC,CAAC;QAC1F,MAAM,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,qBAAqB,CAAC,CAAC;QACrD,OAAO,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,oBAAoB,CAAC,CAAC;QACvD,UAAU,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,oBAAoB,CAAC,CAAC;KAC3D,CAAC;IAEF;;OAEG;IACH,WAAW,EAAE;QACX,IAAI,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC;QACnD,MAAM,EAAE,CAAC,IAAI,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,GAAG,EAAE,MAAM,CAAC;YAAC,IAAI,EAAE,MAAM,CAAC;YAAC,WAAW,CAAC,EAAE,MAAM,CAAC;YAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;SAAE,KAAK,OAAO,CAAC;YAAE,OAAO,EAAE,OAAO,CAAC;YAAC,IAAI,EAAE,eAAe,CAAA;SAAE,CAAC,CAAC;QAC3K,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,OAAO,CAAC;YAAE,OAAO,EAAE,OAAO,CAAC;YAAC,IAAI,EAAE,eAAe,CAAA;SAAE,CAAC,CAAC;QAC3H,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;KACtD,CAAC;IAEF;;OAEG;IACH,KAAK,EAAE;QACL,IAAI,EAAE,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC/D,GAAG,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,QAAQ,CAAC,CAAC;QACvC,MAAM,EAAE,CAAC,IAAI,EAAE,iBAAiB,KAAK,OAAO,CAAC,QAAQ,CAAC,CAAC;QACvD,MAAM,EAAE,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,KAAK,OAAO,CAAC,QAAQ,CAAC,CAAC;QACnE,MAAM,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;QACtC,QAAQ,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,QAAQ,CAAC,CAAC;KAC7C,CAAC;IAEF;;OAEG;IACH,IAAI,EAAE;QACJ,IAAI,EAAE,CAAC,UAAU,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;QAC9C,GAAG,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;QAClC,MAAM,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;QACpC,MAAM,EAAE,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC;QAChD,MAAM,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;KACvC,CAAC;IAEF;;;OAGG;IACH,sBAAsB,CAAC,OAAO,EAAE,MAAM,IAAI,GAAG,IAAI;IAIjD;;;OAGG;IACH,SAAS,CAAC,sBAAsB,IAAI,CAAC,MAAM,IAAI,CAAC,GAAG,SAAS;IAI5D;;;OAGG;gBACS,MAAM,EAAE,YAAY;IA6MhC;;;;;;OAMG;IACG,OAAO,CAAC,CAAC,EACb,GAAG,EAAE,MAAM,EACX,OAAO,CAAC,EAAE;QACR,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,IAAI,CAAC,EAAE,GAAG,CAAC;QACX,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;KAClC,GACA,OAAO,CAAC,CAAC,CAAC;IAIb;;;OAGG;IACH,QAAQ,CAAC,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAE5C;;;OAGG;IACH,cAAc,CAAC,WAAW,EAAE,MAAM,GAAG,IAAI;IAIzC;;;;OAIG;IACH,SAAS,CAAC,QAAQ,CAAC,cAAc,CAAC,MAAM,EAAE,YAAY,GAAG,cAAc;IAEvE;;;;OAIG;IACH,KAAK,CAAC,SAAS,CAAC,EAAE,OAAO,CAAC,YAAY,CAAC,GAAG,cAAc;IAwBxD;;;OAGG;IACH,SAAS,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,EAC9B,GAAG,EAAE,MAAM,EACX,OAAO,CAAC,EAAE;QACR,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,IAAI,CAAC,EAAE,GAAG,CAAC;QACX,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;KAClC,GACA,OAAO,CAAC,CAAC,CAAC;IAEb;;;OAGG;IACH;;;;OAIG;IACH,SAAS,CAAC,wBAAwB,CAAC,OAAO,EAAE,UAAU,GAAG;QACvD,GAAG,EAAE,MAAM,CAAC;QACZ,MAAM,EAAE,MAAM,CAAC;QACf,IAAI,EAAE,GAAG,CAAC;KACX;IAqBD,SAAS,CAAC,QAAQ,CAAC,aAAa,CAC9B,OAAO,EAAE,UAAU,EACnB,OAAO,EAAE,CAAC,KAAK,EAAE,YAAY,KAAK,IAAI,EACtC,UAAU,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,UAAU,KAAK,IAAI,EACzC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,GAC/B,MAAM,IAAI;IAEb;;;;;OAKG;IACG,YAAY,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,MAAM,CAAC;IAcjE;;;;;OAKG;IACG,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAUlD;;;;;OAKG;IACG,WAAW,CAAC,OAAO,CAAC,EAAE,kBAAkB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAOlE;;;;;OAKG;IACG,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAOnD;;;;OAIG;IACG,WAAW,CAAC,OAAO,EAAE,kBAAkB,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IA0BlE;;;;OAIG;IACG,aAAa,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC;IAM1D;;;OAGG;IACG,aAAa,IAAI,OAAO,CAAC,MAAM,CAAC;IAOtC;;;;OAIG;IACG,GAAG,CAAC,OAAO,EAAE,UAAU,GAAG,OAAO,CAAC,YAAY,CAAC;IAkCrD;;OAEG;IACH,IAAI;QACF;;;;WAIG;wBACmB,eAAe,KAAG,OAAO,CAAC,YAAY,CAAC;QAsB7D;;;;;;;WAOG;0BAEQ,iBAAiB,WACjB,CAAC,KAAK,EAAE,YAAY,KAAK,IAAI,eACzB,CAAC,KAAK,CAAC,EAAE,UAAU,KAAK,IAAI,YAC/B,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,KAC/B,CAAC,MAAM,IAAI,CAAC;QAuCf;;;;WAIG;wCACmC,2BAA2B,KAAG,OAAO,CAAC;YAAE,OAAO,EAAE,OAAO,CAAA;SAAE,CAAC;QAUjG;;;;WAIG;wBACmB,gBAAgB,KAAG,OAAO,CAAC;YAAE,OAAO,EAAE,OAAO,CAAC;YAAC,OAAO,EAAE,MAAM,CAAA;SAAE,CAAC;MAQtF;IAEH;;;;;;;OAOG;IACH,QAAQ,CAAC,YAAY,CACnB,OAAO,EAAE,mBAAmB,EAC5B,OAAO,EAAE,CAAC,KAAK,EAAE,YAAY,KAAK,IAAI,EACtC,UAAU,CAAC,EAAE,MAAM,IAAI,EACvB,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,GAC/B,MAAM,IAAI;IAEb;;OAEG;IACH,KAAK;QACH;;;WAGG;4BACiB,mBAAmB,KAAG,IAAI;QAS9C;;;WAGG;2BACgB,MAAM,KAAG,IAAI;MAGhC;IAEF;;OAEG;IACH,UAAU;QACR;;;;WAIG;oBACa,OAAO,CAAC,SAAS,EAAE,CAAC;QAOpC;;;;;WAKG;kBACa,MAAM,KAAG,OAAO,CAAC,SAAS,CAAC;QAU3C;;;;;WAKG;0BACqB,sBAAsB,KAAG,OAAO,CAAC,SAAS,CAAC;QAcnE;;;;;;WAMG;qBAEG,MAAM,WACD,sBAAsB,KAC9B,OAAO,CAAC,SAAS,CAAC;QAcrB;;;;;WAKG;qBACgB,MAAM,KAAG,OAAO,CAAC,IAAI,CAAC;MAQzC;IAEF;;OAEG;IACH,MAAM;QACJ;;;WAGG;oBACa,OAAO,CAAC,KAAK,EAAE,CAAC;QAYhC;;;;WAIG;kBACa,MAAM,KAAG,OAAO,CAAC,KAAK,CAAC;QAYvC;;;;WAIG;0BACqB,kBAAkB,KAAG,OAAO,CAAC,KAAK,CAAC;QAe3D;;;;;WAKG;qBAEG,MAAM,WACD,OAAO,CAAC,kBAAkB,CAAC,KACnC,OAAO,CAAC,KAAK,CAAC;QAejB;;;;WAIG;qBACgB,MAAM,KAAG,OAAO,CAAC,IAAI,CAAC;MAQzC;IAEF;;OAEG;IACH,WAAW;QACT;;;WAGG;oBACa,OAAO,CAAC,GAAG,EAAE,CAAC;QAY9B;;;;WAIG;oBACe,MAAM,KAAG,OAAO,CAAC,GAAG,CAAC;QAYvC;;;;WAIG;0BACqB;YACtB,IAAI,EAAE,MAAM,CAAC;YACb,KAAK,EAAE,MAAM,CAAC;YACd,YAAY,EAAE,MAAM,CAAC;YACrB,MAAM,CAAC,EAAE;gBAAE,MAAM,EAAE,KAAK,CAAC;oBAAE,GAAG,EAAE,MAAM,CAAC;oBAAC,IAAI,EAAE,MAAM,CAAC;oBAAC,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;oBAAC,QAAQ,CAAC,EAAE,OAAO,CAAA;iBAAE,CAAC,CAAA;aAAE,CAAC;SACtG,KAAG,OAAO,CAAC,GAAG,CAAC;QAehB;;;;;WAKG;uBAEK,MAAM,WACH;YAAE,KAAK,CAAC,EAAE,MAAM,CAAC;YAAC,YAAY,CAAC,EAAE,MAAM,CAAC;YAAC,MAAM,CAAC,EAAE,GAAG,CAAA;SAAE,KAC/D,OAAO,CAAC,GAAG,CAAC;QAef;;;;WAIG;uBACkB,MAAM,KAAG,OAAO,CAAC,IAAI,CAAC;QAS3C;;;;;WAKG;kCACuB,MAAM;wBACd,OAAO,CAAC,GAAG,EAAE,CAAC;2BAMT;gBAAE,OAAO,EAAE,MAAM,CAAC;gBAAC,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;aAAE,KAAG,OAAO,CAAC,GAAG,CAAC;sBAU3E,MAAM,KAAG,OAAO,CAAC,GAAG,CAAC;yBAOlB,MAAM,QAAQ;gBAAE,OAAO,CAAC,EAAE,MAAM,CAAC;gBAAC,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;aAAE,KAAG,OAAO,CAAC,GAAG,CAAC;yBAUrF,MAAM,KAAG,OAAO,CAAC,IAAI,CAAC;4BAMnB,MAAM,SAAS;gBAAE,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;gBAAC,KAAK,CAAC,EAAE,MAAM,CAAA;aAAE,KAAG,OAAO,CAAC,GAAG,EAAE,CAAC;;MAU5G;IAEF;;OAEG;IACH,OAAO;QACL;;;WAGG;oBACa,OAAO,CAAC,MAAM,EAAE,CAAC;QAOjC;;;;WAIG;wBACmB,MAAM,KAAG,OAAO,CAAC,MAAM,CAAC;QAU9C;;;;WAIG;0BACqB,mBAAmB,KAAG,OAAO,CAAC,MAAM,CAAC;QAc7D;;;;;WAKG;2BAES,MAAM,WACP,mBAAmB,KAC3B,OAAO,CAAC,MAAM,CAAC;QAclB;;;;WAIG;2BACsB,MAAM,KAAG,OAAO,CAAC,IAAI,CAAC;MAQ/C;IAEF;;OAEG;IACH,SAAS;QACP;;;;WAIG;+BAEQ,wBAAwB,KAChC,OAAO,CAAC,aAAa,EAAE,CAAC;QAyB3B;;;;WAIG;sBACiB,MAAM,KAAG,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC;QAQ1D;;;;WAIG;yBACoB,MAAM,KAAG,OAAO,CAAC,OAAO,CAAC;QAUhD;;;;WAIG;wBACmB,MAAM,KAAG,OAAO,CAAC,OAAO,CAAC;QAU/C;;;;WAIG;yBACoB,MAAM,KAAG,OAAO,CAAC,OAAO,CAAC;MAShD;CACH"}
|
package/dist/abstract-client.js
CHANGED
|
@@ -261,6 +261,118 @@ export class AbstractClient {
|
|
|
261
261
|
});
|
|
262
262
|
},
|
|
263
263
|
};
|
|
264
|
+
/**
|
|
265
|
+
* Collections namespace for managing vector collections
|
|
266
|
+
*/
|
|
267
|
+
this.collections = {
|
|
268
|
+
/**
|
|
269
|
+
* Lists all available collections
|
|
270
|
+
* @returns A promise that resolves to the list of collections
|
|
271
|
+
*/
|
|
272
|
+
list: async () => {
|
|
273
|
+
const response = await this.makeRequest("/api/collections");
|
|
274
|
+
return response.data.records;
|
|
275
|
+
},
|
|
276
|
+
/**
|
|
277
|
+
* Retrieves a single collection by name
|
|
278
|
+
* @param name - Collection name
|
|
279
|
+
* @returns A promise that resolves to the collection information
|
|
280
|
+
*/
|
|
281
|
+
get: async (name) => {
|
|
282
|
+
const response = await this.makeRequest(`/api/collections/${name}`);
|
|
283
|
+
if (!response.data) {
|
|
284
|
+
throw new ApiError("Collection not found", 404);
|
|
285
|
+
}
|
|
286
|
+
return response.data;
|
|
287
|
+
},
|
|
288
|
+
/**
|
|
289
|
+
* Creates a new collection
|
|
290
|
+
* @param options - Options for creating a collection
|
|
291
|
+
* @returns A promise that resolves to the created collection
|
|
292
|
+
*/
|
|
293
|
+
create: async (options) => {
|
|
294
|
+
const response = await this.makeRequest("/api/collections", {
|
|
295
|
+
method: "POST",
|
|
296
|
+
body: options,
|
|
297
|
+
});
|
|
298
|
+
if (!response.data) {
|
|
299
|
+
throw new ApiError("Failed to create collection", 500);
|
|
300
|
+
}
|
|
301
|
+
return response.data;
|
|
302
|
+
},
|
|
303
|
+
/**
|
|
304
|
+
* Updates an existing collection by name
|
|
305
|
+
* @param name - Collection name
|
|
306
|
+
* @param options - Options for updating a collection
|
|
307
|
+
* @returns A promise that resolves to the updated collection
|
|
308
|
+
*/
|
|
309
|
+
update: async (name, options) => {
|
|
310
|
+
const response = await this.makeRequest(`/api/collections/${name}`, {
|
|
311
|
+
method: "PUT",
|
|
312
|
+
body: options,
|
|
313
|
+
});
|
|
314
|
+
if (!response.data) {
|
|
315
|
+
throw new ApiError("Failed to update collection", 500);
|
|
316
|
+
}
|
|
317
|
+
return response.data;
|
|
318
|
+
},
|
|
319
|
+
/**
|
|
320
|
+
* Deletes a collection by name
|
|
321
|
+
* @param name - Collection name
|
|
322
|
+
* @returns A promise that resolves when the collection is deleted
|
|
323
|
+
*/
|
|
324
|
+
delete: async (name) => {
|
|
325
|
+
await this.makeRequest(`/api/collections/${name}`, {
|
|
326
|
+
method: "DELETE",
|
|
327
|
+
});
|
|
328
|
+
},
|
|
329
|
+
/**
|
|
330
|
+
* Entries sub-namespace for a specific collection.
|
|
331
|
+
* Returns entry CRUD + search methods scoped to the given collection.
|
|
332
|
+
* @param collectionName - The collection name
|
|
333
|
+
* @returns Entry operation methods
|
|
334
|
+
*/
|
|
335
|
+
entries: (collectionName) => ({
|
|
336
|
+
list: async () => {
|
|
337
|
+
const response = await this.makeRequest(`/api/collections/${collectionName}/entries`);
|
|
338
|
+
return response.data.records;
|
|
339
|
+
},
|
|
340
|
+
create: async (data) => {
|
|
341
|
+
const response = await this.makeRequest(`/api/collections/${collectionName}/entries`, {
|
|
342
|
+
method: "POST",
|
|
343
|
+
body: data,
|
|
344
|
+
});
|
|
345
|
+
if (!response.data)
|
|
346
|
+
throw new ApiError("Failed to create entry", 500);
|
|
347
|
+
return response.data;
|
|
348
|
+
},
|
|
349
|
+
get: async (id) => {
|
|
350
|
+
const response = await this.makeRequest(`/api/collections/${collectionName}/entries/${id}`);
|
|
351
|
+
if (!response.data)
|
|
352
|
+
throw new ApiError("Entry not found", 404);
|
|
353
|
+
return response.data;
|
|
354
|
+
},
|
|
355
|
+
update: async (id, data) => {
|
|
356
|
+
const response = await this.makeRequest(`/api/collections/${collectionName}/entries/${id}`, {
|
|
357
|
+
method: "PUT",
|
|
358
|
+
body: data,
|
|
359
|
+
});
|
|
360
|
+
if (!response.data)
|
|
361
|
+
throw new ApiError("Failed to update entry", 500);
|
|
362
|
+
return response.data;
|
|
363
|
+
},
|
|
364
|
+
delete: async (id) => {
|
|
365
|
+
await this.makeRequest(`/api/collections/${collectionName}/entries/${id}`, { method: "DELETE" });
|
|
366
|
+
},
|
|
367
|
+
search: async (query, opts) => {
|
|
368
|
+
const response = await this.makeRequest(`/api/collections/${collectionName}/search`, {
|
|
369
|
+
method: "POST",
|
|
370
|
+
body: { query, ...opts },
|
|
371
|
+
});
|
|
372
|
+
return response.data.records;
|
|
373
|
+
},
|
|
374
|
+
}),
|
|
375
|
+
};
|
|
264
376
|
/**
|
|
265
377
|
* Threads namespace for managing threads for the current assistant
|
|
266
378
|
*/
|
|
@@ -523,6 +635,33 @@ export class AbstractClient {
|
|
|
523
635
|
test: async (id) => {
|
|
524
636
|
return await this.makeRequest(`/api/mcp-servers/${id}/test`, { method: "POST" });
|
|
525
637
|
},
|
|
638
|
+
connect: async (id) => {
|
|
639
|
+
const response = await this.makeRequest(`/api/mcp-servers/${id}/connect`, { method: "POST" });
|
|
640
|
+
if (!response.data)
|
|
641
|
+
throw new ApiError("Failed to connect MCP server", 500);
|
|
642
|
+
return response.data;
|
|
643
|
+
},
|
|
644
|
+
disconnect: async (id) => {
|
|
645
|
+
const response = await this.makeRequest(`/api/mcp-servers/${id}/disconnect`, { method: "POST" });
|
|
646
|
+
if (!response.data)
|
|
647
|
+
throw new ApiError("Failed to disconnect MCP server", 500);
|
|
648
|
+
return response.data;
|
|
649
|
+
},
|
|
650
|
+
};
|
|
651
|
+
this.connections = {
|
|
652
|
+
list: async (type) => {
|
|
653
|
+
const response = await this.makeRequest(`/api/connections?type=${encodeURIComponent(type)}`);
|
|
654
|
+
return response.data.records;
|
|
655
|
+
},
|
|
656
|
+
create: async (data) => {
|
|
657
|
+
return await this.makeRequest("/api/connections", { method: "POST", body: data });
|
|
658
|
+
},
|
|
659
|
+
update: async (type, key, data) => {
|
|
660
|
+
return await this.makeRequest(`/api/connections/${encodeURIComponent(key)}?type=${encodeURIComponent(type)}`, { method: "PUT", body: data });
|
|
661
|
+
},
|
|
662
|
+
delete: async (type, key) => {
|
|
663
|
+
await this.makeRequest(`/api/connections/${encodeURIComponent(key)}?type=${encodeURIComponent(type)}`, { method: "DELETE" });
|
|
664
|
+
},
|
|
526
665
|
};
|
|
527
666
|
this.tasks = {
|
|
528
667
|
list: async (params) => {
|