mimo2codex 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/AGENTS.md +124 -0
- package/LICENSE +21 -0
- package/README.md +536 -0
- package/README.zh.md +750 -0
- package/dist/cli.js +202 -0
- package/dist/cli.js.map +1 -0
- package/dist/config.js +86 -0
- package/dist/config.js.map +1 -0
- package/dist/server.js +161 -0
- package/dist/server.js.map +1 -0
- package/dist/translate/reqToChat.js +381 -0
- package/dist/translate/reqToChat.js.map +1 -0
- package/dist/translate/respToResponses.js +94 -0
- package/dist/translate/respToResponses.js.map +1 -0
- package/dist/translate/streamToSse.js +352 -0
- package/dist/translate/streamToSse.js.map +1 -0
- package/dist/translate/types.js +4 -0
- package/dist/translate/types.js.map +1 -0
- package/dist/upstream/chatStream.js +56 -0
- package/dist/upstream/chatStream.js.map +1 -0
- package/dist/upstream/mimoClient.js +99 -0
- package/dist/upstream/mimoClient.js.map +1 -0
- package/dist/util/ids.js +19 -0
- package/dist/util/ids.js.map +1 -0
- package/dist/util/log.js +32 -0
- package/dist/util/log.js.map +1 -0
- package/dist/util/sse.js +61 -0
- package/dist/util/sse.js.map +1 -0
- package/mimoskill/SKILL.md +145 -0
- package/mimoskill/assets/pet_prompt_template.md +94 -0
- package/mimoskill/references/models.md +111 -0
- package/mimoskill/references/pet_workflow.md +197 -0
- package/mimoskill/scripts/generate_pet.py +365 -0
- package/mimoskill/scripts/install_pet.sh +220 -0
- package/mimoskill/scripts/mimo_chat.py +199 -0
- package/package.json +69 -0
|
@@ -0,0 +1,352 @@
|
|
|
1
|
+
import { newFunctionCallId, newMessageId, newReasoningId, newResponseId, } from "../util/ids.js";
|
|
2
|
+
class StreamState {
|
|
3
|
+
responseId = newResponseId();
|
|
4
|
+
createdAt = Math.floor(Date.now() / 1000);
|
|
5
|
+
model;
|
|
6
|
+
outputIndex = 0;
|
|
7
|
+
sequenceNumber = 0;
|
|
8
|
+
activeKind = null;
|
|
9
|
+
activeItemId = null;
|
|
10
|
+
activeBuffer = "";
|
|
11
|
+
activeAnnotations = [];
|
|
12
|
+
toolCalls = new Map();
|
|
13
|
+
finalOutput = [];
|
|
14
|
+
finishReason = null;
|
|
15
|
+
usage = null;
|
|
16
|
+
exposeReasoning;
|
|
17
|
+
req;
|
|
18
|
+
constructor(req, exposeReasoning) {
|
|
19
|
+
this.req = req;
|
|
20
|
+
this.model = req.model;
|
|
21
|
+
this.exposeReasoning = exposeReasoning;
|
|
22
|
+
}
|
|
23
|
+
nextSeq() {
|
|
24
|
+
return this.sequenceNumber++;
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
// Each Responses SSE event MUST include `type` in the JSON payload (in addition
|
|
28
|
+
// to the SSE `event:` line) — the Codex client parses events from the data field,
|
|
29
|
+
// not the SSE event header. Missing `type` leads to "stream disconnected before
|
|
30
|
+
// completion" errors because the client never recognizes response.completed.
|
|
31
|
+
function emit(sink, state, event, data) {
|
|
32
|
+
sink.write(event, { type: event, ...data, sequence_number: state.nextSeq() });
|
|
33
|
+
}
|
|
34
|
+
function buildResponseSnapshot(state, status) {
|
|
35
|
+
return {
|
|
36
|
+
id: state.responseId,
|
|
37
|
+
object: "response",
|
|
38
|
+
created_at: state.createdAt,
|
|
39
|
+
status,
|
|
40
|
+
model: state.model,
|
|
41
|
+
output: state.finalOutput,
|
|
42
|
+
usage: state.usage,
|
|
43
|
+
parallel_tool_calls: state.req.parallel_tool_calls ?? true,
|
|
44
|
+
tool_choice: state.req.tool_choice ?? "auto",
|
|
45
|
+
reasoning: {
|
|
46
|
+
effort: state.req.reasoning?.effort ?? null,
|
|
47
|
+
summary: state.req.reasoning?.summary ?? null,
|
|
48
|
+
},
|
|
49
|
+
text: state.req.text?.format
|
|
50
|
+
? { format: state.req.text.format }
|
|
51
|
+
: { format: { type: "text" } },
|
|
52
|
+
incomplete_details: state.finishReason === "length" ? { reason: "max_output_tokens" } : null,
|
|
53
|
+
error: null,
|
|
54
|
+
metadata: state.req.metadata ?? null,
|
|
55
|
+
previous_response_id: state.req.previous_response_id ?? null,
|
|
56
|
+
instructions: state.req.instructions ?? null,
|
|
57
|
+
temperature: state.req.temperature ?? null,
|
|
58
|
+
top_p: state.req.top_p ?? null,
|
|
59
|
+
max_output_tokens: state.req.max_output_tokens ?? null,
|
|
60
|
+
tools: state.req.tools ?? [],
|
|
61
|
+
truncation: "disabled",
|
|
62
|
+
};
|
|
63
|
+
}
|
|
64
|
+
function openReasoning(sink, state) {
|
|
65
|
+
finalizeActive(sink, state);
|
|
66
|
+
state.activeKind = "reasoning";
|
|
67
|
+
state.activeItemId = newReasoningId();
|
|
68
|
+
state.activeBuffer = "";
|
|
69
|
+
const idx = state.outputIndex++;
|
|
70
|
+
emit(sink, state, "response.output_item.added", {
|
|
71
|
+
output_index: idx,
|
|
72
|
+
item: {
|
|
73
|
+
id: state.activeItemId,
|
|
74
|
+
type: "reasoning",
|
|
75
|
+
summary: [],
|
|
76
|
+
encrypted_content: null,
|
|
77
|
+
status: "in_progress",
|
|
78
|
+
},
|
|
79
|
+
});
|
|
80
|
+
emit(sink, state, "response.reasoning_summary_part.added", {
|
|
81
|
+
item_id: state.activeItemId,
|
|
82
|
+
output_index: idx,
|
|
83
|
+
summary_index: 0,
|
|
84
|
+
part: { type: "summary_text", text: "" },
|
|
85
|
+
});
|
|
86
|
+
}
|
|
87
|
+
function openMessage(sink, state) {
|
|
88
|
+
finalizeActive(sink, state);
|
|
89
|
+
state.activeKind = "message";
|
|
90
|
+
state.activeItemId = newMessageId();
|
|
91
|
+
state.activeBuffer = "";
|
|
92
|
+
state.activeAnnotations = [];
|
|
93
|
+
const idx = state.outputIndex++;
|
|
94
|
+
emit(sink, state, "response.output_item.added", {
|
|
95
|
+
output_index: idx,
|
|
96
|
+
item: {
|
|
97
|
+
id: state.activeItemId,
|
|
98
|
+
type: "message",
|
|
99
|
+
role: "assistant",
|
|
100
|
+
status: "in_progress",
|
|
101
|
+
content: [],
|
|
102
|
+
},
|
|
103
|
+
});
|
|
104
|
+
emit(sink, state, "response.content_part.added", {
|
|
105
|
+
item_id: state.activeItemId,
|
|
106
|
+
output_index: idx,
|
|
107
|
+
content_index: 0,
|
|
108
|
+
part: { type: "output_text", text: "", annotations: [] },
|
|
109
|
+
});
|
|
110
|
+
}
|
|
111
|
+
function translateAnnotation(a) {
|
|
112
|
+
return {
|
|
113
|
+
type: a.type ?? "url_citation",
|
|
114
|
+
url: a.url ?? "",
|
|
115
|
+
title: a.title ?? "",
|
|
116
|
+
...(a.summary !== undefined ? { snippet: a.summary } : {}),
|
|
117
|
+
};
|
|
118
|
+
}
|
|
119
|
+
function openToolCall(sink, state, index, id, name) {
|
|
120
|
+
finalizeActive(sink, state);
|
|
121
|
+
const itemId = newFunctionCallId();
|
|
122
|
+
const outputIndex = state.outputIndex++;
|
|
123
|
+
const callId = id ?? `call_${itemId.slice(3)}`;
|
|
124
|
+
const tc = {
|
|
125
|
+
itemId,
|
|
126
|
+
outputIndex,
|
|
127
|
+
callId,
|
|
128
|
+
name: name ?? "",
|
|
129
|
+
argsBuffer: "",
|
|
130
|
+
argsEmitted: false,
|
|
131
|
+
};
|
|
132
|
+
state.toolCalls.set(index, tc);
|
|
133
|
+
emit(sink, state, "response.output_item.added", {
|
|
134
|
+
output_index: outputIndex,
|
|
135
|
+
item: {
|
|
136
|
+
id: itemId,
|
|
137
|
+
type: "function_call",
|
|
138
|
+
call_id: callId,
|
|
139
|
+
name: tc.name,
|
|
140
|
+
arguments: "",
|
|
141
|
+
status: "in_progress",
|
|
142
|
+
},
|
|
143
|
+
});
|
|
144
|
+
return tc;
|
|
145
|
+
}
|
|
146
|
+
function finalizeActive(sink, state) {
|
|
147
|
+
if (state.activeKind === null)
|
|
148
|
+
return;
|
|
149
|
+
const itemId = state.activeItemId;
|
|
150
|
+
const buffer = state.activeBuffer;
|
|
151
|
+
const outputIndex = state.outputIndex - 1;
|
|
152
|
+
if (state.activeKind === "reasoning") {
|
|
153
|
+
emit(sink, state, "response.reasoning_summary_text.done", {
|
|
154
|
+
item_id: itemId,
|
|
155
|
+
output_index: outputIndex,
|
|
156
|
+
summary_index: 0,
|
|
157
|
+
text: buffer,
|
|
158
|
+
});
|
|
159
|
+
emit(sink, state, "response.reasoning_summary_part.done", {
|
|
160
|
+
item_id: itemId,
|
|
161
|
+
output_index: outputIndex,
|
|
162
|
+
summary_index: 0,
|
|
163
|
+
part: { type: "summary_text", text: buffer },
|
|
164
|
+
});
|
|
165
|
+
const finalItem = {
|
|
166
|
+
id: itemId,
|
|
167
|
+
type: "reasoning",
|
|
168
|
+
summary: [{ type: "summary_text", text: buffer }],
|
|
169
|
+
encrypted_content: null,
|
|
170
|
+
status: "completed",
|
|
171
|
+
};
|
|
172
|
+
state.finalOutput.push(finalItem);
|
|
173
|
+
emit(sink, state, "response.output_item.done", {
|
|
174
|
+
output_index: outputIndex,
|
|
175
|
+
item: finalItem,
|
|
176
|
+
});
|
|
177
|
+
}
|
|
178
|
+
else if (state.activeKind === "message") {
|
|
179
|
+
const annotations = state.activeAnnotations;
|
|
180
|
+
emit(sink, state, "response.output_text.done", {
|
|
181
|
+
item_id: itemId,
|
|
182
|
+
output_index: outputIndex,
|
|
183
|
+
content_index: 0,
|
|
184
|
+
text: buffer,
|
|
185
|
+
});
|
|
186
|
+
emit(sink, state, "response.content_part.done", {
|
|
187
|
+
item_id: itemId,
|
|
188
|
+
output_index: outputIndex,
|
|
189
|
+
content_index: 0,
|
|
190
|
+
part: { type: "output_text", text: buffer, annotations },
|
|
191
|
+
});
|
|
192
|
+
const finalItem = {
|
|
193
|
+
id: itemId,
|
|
194
|
+
type: "message",
|
|
195
|
+
role: "assistant",
|
|
196
|
+
status: "completed",
|
|
197
|
+
content: [{ type: "output_text", text: buffer, annotations }],
|
|
198
|
+
};
|
|
199
|
+
state.finalOutput.push(finalItem);
|
|
200
|
+
emit(sink, state, "response.output_item.done", {
|
|
201
|
+
output_index: outputIndex,
|
|
202
|
+
item: finalItem,
|
|
203
|
+
});
|
|
204
|
+
}
|
|
205
|
+
state.activeKind = null;
|
|
206
|
+
state.activeItemId = null;
|
|
207
|
+
state.activeBuffer = "";
|
|
208
|
+
state.activeAnnotations = [];
|
|
209
|
+
}
|
|
210
|
+
function finalizeToolCalls(sink, state) {
|
|
211
|
+
// Emit done events for tool calls in the order they were opened.
|
|
212
|
+
const ordered = Array.from(state.toolCalls.entries()).sort((a, b) => a[0] - b[0]);
|
|
213
|
+
for (const [, tc] of ordered) {
|
|
214
|
+
emit(sink, state, "response.function_call_arguments.done", {
|
|
215
|
+
item_id: tc.itemId,
|
|
216
|
+
output_index: tc.outputIndex,
|
|
217
|
+
arguments: tc.argsBuffer,
|
|
218
|
+
});
|
|
219
|
+
const finalItem = {
|
|
220
|
+
id: tc.itemId,
|
|
221
|
+
type: "function_call",
|
|
222
|
+
call_id: tc.callId,
|
|
223
|
+
name: tc.name,
|
|
224
|
+
arguments: tc.argsBuffer,
|
|
225
|
+
status: "completed",
|
|
226
|
+
};
|
|
227
|
+
state.finalOutput.push(finalItem);
|
|
228
|
+
emit(sink, state, "response.output_item.done", {
|
|
229
|
+
output_index: tc.outputIndex,
|
|
230
|
+
item: finalItem,
|
|
231
|
+
});
|
|
232
|
+
}
|
|
233
|
+
}
|
|
234
|
+
function processChunk(sink, state, chunk) {
|
|
235
|
+
if (chunk.usage) {
|
|
236
|
+
state.usage = {
|
|
237
|
+
input_tokens: chunk.usage.prompt_tokens,
|
|
238
|
+
output_tokens: chunk.usage.completion_tokens,
|
|
239
|
+
total_tokens: chunk.usage.total_tokens,
|
|
240
|
+
};
|
|
241
|
+
if (chunk.usage.prompt_tokens_details?.cached_tokens !== undefined) {
|
|
242
|
+
state.usage.input_tokens_details = {
|
|
243
|
+
cached_tokens: chunk.usage.prompt_tokens_details.cached_tokens,
|
|
244
|
+
};
|
|
245
|
+
}
|
|
246
|
+
if (chunk.usage.completion_tokens_details?.reasoning_tokens !== undefined) {
|
|
247
|
+
state.usage.output_tokens_details = {
|
|
248
|
+
reasoning_tokens: chunk.usage.completion_tokens_details.reasoning_tokens,
|
|
249
|
+
};
|
|
250
|
+
}
|
|
251
|
+
}
|
|
252
|
+
const choice = chunk.choices?.[0];
|
|
253
|
+
if (!choice)
|
|
254
|
+
return;
|
|
255
|
+
const delta = choice.delta;
|
|
256
|
+
if (delta.reasoning_content && state.exposeReasoning) {
|
|
257
|
+
if (state.activeKind !== "reasoning")
|
|
258
|
+
openReasoning(sink, state);
|
|
259
|
+
state.activeBuffer += delta.reasoning_content;
|
|
260
|
+
emit(sink, state, "response.reasoning_summary_text.delta", {
|
|
261
|
+
item_id: state.activeItemId,
|
|
262
|
+
output_index: state.outputIndex - 1,
|
|
263
|
+
summary_index: 0,
|
|
264
|
+
delta: delta.reasoning_content,
|
|
265
|
+
});
|
|
266
|
+
}
|
|
267
|
+
if (delta.content) {
|
|
268
|
+
if (state.activeKind !== "message")
|
|
269
|
+
openMessage(sink, state);
|
|
270
|
+
state.activeBuffer += delta.content;
|
|
271
|
+
emit(sink, state, "response.output_text.delta", {
|
|
272
|
+
item_id: state.activeItemId,
|
|
273
|
+
output_index: state.outputIndex - 1,
|
|
274
|
+
content_index: 0,
|
|
275
|
+
delta: delta.content,
|
|
276
|
+
});
|
|
277
|
+
}
|
|
278
|
+
// MiMo's web_search returns citations in the first streaming chunk's
|
|
279
|
+
// `delta.annotations`. Buffer them and emit per-annotation events so Codex
|
|
280
|
+
// can show inline citations live.
|
|
281
|
+
if (delta.annotations && delta.annotations.length > 0) {
|
|
282
|
+
if (state.activeKind !== "message")
|
|
283
|
+
openMessage(sink, state);
|
|
284
|
+
for (const a of delta.annotations) {
|
|
285
|
+
const translated = translateAnnotation(a);
|
|
286
|
+
const annotationIndex = state.activeAnnotations.length;
|
|
287
|
+
state.activeAnnotations.push(translated);
|
|
288
|
+
emit(sink, state, "response.output_text.annotation.added", {
|
|
289
|
+
item_id: state.activeItemId,
|
|
290
|
+
output_index: state.outputIndex - 1,
|
|
291
|
+
content_index: 0,
|
|
292
|
+
annotation_index: annotationIndex,
|
|
293
|
+
annotation: translated,
|
|
294
|
+
});
|
|
295
|
+
}
|
|
296
|
+
}
|
|
297
|
+
if (delta.tool_calls) {
|
|
298
|
+
for (const tcDelta of delta.tool_calls) {
|
|
299
|
+
let tc = state.toolCalls.get(tcDelta.index);
|
|
300
|
+
if (!tc) {
|
|
301
|
+
tc = openToolCall(sink, state, tcDelta.index, tcDelta.id, tcDelta.function?.name);
|
|
302
|
+
}
|
|
303
|
+
else if (tcDelta.function?.name && !tc.name) {
|
|
304
|
+
tc.name = tcDelta.function.name;
|
|
305
|
+
}
|
|
306
|
+
if (tcDelta.function?.arguments) {
|
|
307
|
+
tc.argsBuffer += tcDelta.function.arguments;
|
|
308
|
+
tc.argsEmitted = true;
|
|
309
|
+
emit(sink, state, "response.function_call_arguments.delta", {
|
|
310
|
+
item_id: tc.itemId,
|
|
311
|
+
output_index: tc.outputIndex,
|
|
312
|
+
delta: tcDelta.function.arguments,
|
|
313
|
+
});
|
|
314
|
+
}
|
|
315
|
+
}
|
|
316
|
+
}
|
|
317
|
+
if (choice.finish_reason) {
|
|
318
|
+
state.finishReason = choice.finish_reason;
|
|
319
|
+
}
|
|
320
|
+
}
|
|
321
|
+
export async function pipeChatStreamToResponses(sink, source, req, opts) {
|
|
322
|
+
const state = new StreamState(req, opts.exposeReasoning);
|
|
323
|
+
emit(sink, state, "response.created", {
|
|
324
|
+
response: buildResponseSnapshot(state, "in_progress"),
|
|
325
|
+
});
|
|
326
|
+
emit(sink, state, "response.in_progress", {
|
|
327
|
+
response: buildResponseSnapshot(state, "in_progress"),
|
|
328
|
+
});
|
|
329
|
+
try {
|
|
330
|
+
for await (const chunk of source.chunks) {
|
|
331
|
+
if (sink.closed())
|
|
332
|
+
return;
|
|
333
|
+
processChunk(sink, state, chunk);
|
|
334
|
+
}
|
|
335
|
+
}
|
|
336
|
+
catch (err) {
|
|
337
|
+
finalizeActive(sink, state);
|
|
338
|
+
finalizeToolCalls(sink, state);
|
|
339
|
+
const message = err instanceof Error ? err.message : String(err);
|
|
340
|
+
const failedSnapshot = buildResponseSnapshot(state, "failed");
|
|
341
|
+
failedSnapshot.error = { type: "upstream_error", message };
|
|
342
|
+
emit(sink, state, "response.failed", { response: failedSnapshot });
|
|
343
|
+
sink.end();
|
|
344
|
+
return;
|
|
345
|
+
}
|
|
346
|
+
finalizeActive(sink, state);
|
|
347
|
+
finalizeToolCalls(sink, state);
|
|
348
|
+
const completed = buildResponseSnapshot(state, "completed");
|
|
349
|
+
emit(sink, state, "response.completed", { response: completed });
|
|
350
|
+
sink.end();
|
|
351
|
+
}
|
|
352
|
+
//# sourceMappingURL=streamToSse.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"streamToSse.js","sourceRoot":"","sources":["../../src/translate/streamToSse.ts"],"names":[],"mappings":"AAQA,OAAO,EACL,iBAAiB,EACjB,YAAY,EACZ,cAAc,EACd,aAAa,GACd,MAAM,gBAAgB,CAAC;AAyBxB,MAAM,WAAW;IACf,UAAU,GAAG,aAAa,EAAE,CAAC;IAC7B,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;IAC1C,KAAK,CAAS;IACd,WAAW,GAAG,CAAC,CAAC;IAChB,cAAc,GAAG,CAAC,CAAC;IACnB,UAAU,GAAe,IAAI,CAAC;IAC9B,YAAY,GAAkB,IAAI,CAAC;IACnC,YAAY,GAAG,EAAE,CAAC;IAClB,iBAAiB,GAA0B,EAAE,CAAC;IAC9C,SAAS,GAAG,IAAI,GAAG,EAAyB,CAAC;IAC7C,WAAW,GAA0B,EAAE,CAAC;IACxC,YAAY,GAA+D,IAAI,CAAC;IAChF,KAAK,GAA0B,IAAI,CAAC;IACpC,eAAe,CAAU;IACzB,GAAG,CAAmB;IAEtB,YAAY,GAAqB,EAAE,eAAwB;QACzD,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC;QACvB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;IACzC,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,cAAc,EAAE,CAAC;IAC/B,CAAC;CACF;AAED,gFAAgF;AAChF,kFAAkF;AAClF,gFAAgF;AAChF,6EAA6E;AAC7E,SAAS,IAAI,CAAC,IAAa,EAAE,KAAkB,EAAE,KAAa,EAAE,IAA6B;IAC3F,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,IAAI,EAAE,eAAe,EAAE,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;AAChF,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAkB,EAAE,MAAiC;IAClF,OAAO;QACL,EAAE,EAAE,KAAK,CAAC,UAAU;QACpB,MAAM,EAAE,UAAU;QAClB,UAAU,EAAE,KAAK,CAAC,SAAS;QAC3B,MAAM;QACN,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,MAAM,EAAE,KAAK,CAAC,WAAW;QACzB,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,mBAAmB,EAAE,KAAK,CAAC,GAAG,CAAC,mBAAmB,IAAI,IAAI;QAC1D,WAAW,EAAE,KAAK,CAAC,GAAG,CAAC,WAAW,IAAI,MAAM;QAC5C,SAAS,EAAE;YACT,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,IAAI,IAAI;YAC3C,OAAO,EAAE,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,IAAI,IAAI;SAC9C;QACD,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM;YAC1B,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE;YACnC,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;QAChC,kBAAkB,EAChB,KAAK,CAAC,YAAY,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,mBAAmB,EAAE,CAAC,CAAC,CAAC,IAAI;QAC1E,KAAK,EAAE,IAAI;QACX,QAAQ,EAAE,KAAK,CAAC,GAAG,CAAC,QAAQ,IAAI,IAAI;QACpC,oBAAoB,EAAE,KAAK,CAAC,GAAG,CAAC,oBAAoB,IAAI,IAAI;QAC5D,YAAY,EAAE,KAAK,CAAC,GAAG,CAAC,YAAY,IAAI,IAAI;QAC5C,WAAW,EAAE,KAAK,CAAC,GAAG,CAAC,WAAW,IAAI,IAAI;QAC1C,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,KAAK,IAAI,IAAI;QAC9B,iBAAiB,EAAE,KAAK,CAAC,GAAG,CAAC,iBAAiB,IAAI,IAAI;QACtD,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE;QAC5B,UAAU,EAAE,UAAU;KACvB,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,IAAa,EAAE,KAAkB;IACtD,cAAc,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAC5B,KAAK,CAAC,UAAU,GAAG,WAAW,CAAC;IAC/B,KAAK,CAAC,YAAY,GAAG,cAAc,EAAE,CAAC;IACtC,KAAK,CAAC,YAAY,GAAG,EAAE,CAAC;IACxB,MAAM,GAAG,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IAChC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,4BAA4B,EAAE;QAC9C,YAAY,EAAE,GAAG;QACjB,IAAI,EAAE;YACJ,EAAE,EAAE,KAAK,CAAC,YAAY;YACtB,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,EAAE;YACX,iBAAiB,EAAE,IAAI;YACvB,MAAM,EAAE,aAAa;SACtB;KACF,CAAC,CAAC;IACH,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,uCAAuC,EAAE;QACzD,OAAO,EAAE,KAAK,CAAC,YAAY;QAC3B,YAAY,EAAE,GAAG;QACjB,aAAa,EAAE,CAAC;QAChB,IAAI,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE,EAAE,EAAE;KACzC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,WAAW,CAAC,IAAa,EAAE,KAAkB;IACpD,cAAc,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAC5B,KAAK,CAAC,UAAU,GAAG,SAAS,CAAC;IAC7B,KAAK,CAAC,YAAY,GAAG,YAAY,EAAE,CAAC;IACpC,KAAK,CAAC,YAAY,GAAG,EAAE,CAAC;IACxB,KAAK,CAAC,iBAAiB,GAAG,EAAE,CAAC;IAC7B,MAAM,GAAG,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IAChC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,4BAA4B,EAAE;QAC9C,YAAY,EAAE,GAAG;QACjB,IAAI,EAAE;YACJ,EAAE,EAAE,KAAK,CAAC,YAAY;YACtB,IAAI,EAAE,SAAS;YACf,IAAI,EAAE,WAAW;YACjB,MAAM,EAAE,aAAa;YACrB,OAAO,EAAE,EAAE;SACZ;KACF,CAAC,CAAC;IACH,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,6BAA6B,EAAE;QAC/C,OAAO,EAAE,KAAK,CAAC,YAAY;QAC3B,YAAY,EAAE,GAAG;QACjB,aAAa,EAAE,CAAC;QAChB,IAAI,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;KACzD,CAAC,CAAC;AACL,CAAC;AAED,SAAS,mBAAmB,CAAC,CAAiB;IAC5C,OAAO;QACL,IAAI,EAAE,CAAC,CAAC,IAAI,IAAI,cAAc;QAC9B,GAAG,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE;QAChB,KAAK,EAAE,CAAC,CAAC,KAAK,IAAI,EAAE;QACpB,GAAG,CAAC,CAAC,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC3D,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CACnB,IAAa,EACb,KAAkB,EAClB,KAAa,EACb,EAAsB,EACtB,IAAwB;IAExB,cAAc,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAC5B,MAAM,MAAM,GAAG,iBAAiB,EAAE,CAAC;IACnC,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IACxC,MAAM,MAAM,GAAG,EAAE,IAAI,QAAQ,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;IAC/C,MAAM,EAAE,GAAkB;QACxB,MAAM;QACN,WAAW;QACX,MAAM;QACN,IAAI,EAAE,IAAI,IAAI,EAAE;QAChB,UAAU,EAAE,EAAE;QACd,WAAW,EAAE,KAAK;KACnB,CAAC;IACF,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IAC/B,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,4BAA4B,EAAE;QAC9C,YAAY,EAAE,WAAW;QACzB,IAAI,EAAE;YACJ,EAAE,EAAE,MAAM;YACV,IAAI,EAAE,eAAe;YACrB,OAAO,EAAE,MAAM;YACf,IAAI,EAAE,EAAE,CAAC,IAAI;YACb,SAAS,EAAE,EAAE;YACb,MAAM,EAAE,aAAa;SACtB;KACF,CAAC,CAAC;IACH,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,cAAc,CAAC,IAAa,EAAE,KAAkB;IACvD,IAAI,KAAK,CAAC,UAAU,KAAK,IAAI;QAAE,OAAO;IAEtC,MAAM,MAAM,GAAG,KAAK,CAAC,YAAa,CAAC;IACnC,MAAM,MAAM,GAAG,KAAK,CAAC,YAAY,CAAC;IAClC,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,GAAG,CAAC,CAAC;IAE1C,IAAI,KAAK,CAAC,UAAU,KAAK,WAAW,EAAE,CAAC;QACrC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,sCAAsC,EAAE;YACxD,OAAO,EAAE,MAAM;YACf,YAAY,EAAE,WAAW;YACzB,aAAa,EAAE,CAAC;YAChB,IAAI,EAAE,MAAM;SACb,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,sCAAsC,EAAE;YACxD,OAAO,EAAE,MAAM;YACf,YAAY,EAAE,WAAW;YACzB,aAAa,EAAE,CAAC;YAChB,IAAI,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,EAAE;SAC7C,CAAC,CAAC;QACH,MAAM,SAAS,GAAwB;YACrC,EAAE,EAAE,MAAM;YACV,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YACjD,iBAAiB,EAAE,IAAI;YACvB,MAAM,EAAE,WAAW;SACpB,CAAC;QACF,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAClC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,2BAA2B,EAAE;YAC7C,YAAY,EAAE,WAAW;YACzB,IAAI,EAAE,SAAS;SAChB,CAAC,CAAC;IACL,CAAC;SAAM,IAAI,KAAK,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;QAC1C,MAAM,WAAW,GAAG,KAAK,CAAC,iBAAiB,CAAC;QAC5C,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,2BAA2B,EAAE;YAC7C,OAAO,EAAE,MAAM;YACf,YAAY,EAAE,WAAW;YACzB,aAAa,EAAE,CAAC;YAChB,IAAI,EAAE,MAAM;SACb,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,4BAA4B,EAAE;YAC9C,OAAO,EAAE,MAAM;YACf,YAAY,EAAE,WAAW;YACzB,aAAa,EAAE,CAAC;YAChB,IAAI,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE;SACzD,CAAC,CAAC;QACH,MAAM,SAAS,GAAwB;YACrC,EAAE,EAAE,MAAM;YACV,IAAI,EAAE,SAAS;YACf,IAAI,EAAE,WAAW;YACjB,MAAM,EAAE,WAAW;YACnB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC;SAC9D,CAAC;QACF,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAClC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,2BAA2B,EAAE;YAC7C,YAAY,EAAE,WAAW;YACzB,IAAI,EAAE,SAAS;SAChB,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC;IACxB,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC;IAC1B,KAAK,CAAC,YAAY,GAAG,EAAE,CAAC;IACxB,KAAK,CAAC,iBAAiB,GAAG,EAAE,CAAC;AAC/B,CAAC;AAED,SAAS,iBAAiB,CAAC,IAAa,EAAE,KAAkB;IAC1D,iEAAiE;IACjE,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAClF,KAAK,MAAM,CAAC,EAAE,EAAE,CAAC,IAAI,OAAO,EAAE,CAAC;QAC7B,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,uCAAuC,EAAE;YACzD,OAAO,EAAE,EAAE,CAAC,MAAM;YAClB,YAAY,EAAE,EAAE,CAAC,WAAW;YAC5B,SAAS,EAAE,EAAE,CAAC,UAAU;SACzB,CAAC,CAAC;QACH,MAAM,SAAS,GAAwB;YACrC,EAAE,EAAE,EAAE,CAAC,MAAM;YACb,IAAI,EAAE,eAAe;YACrB,OAAO,EAAE,EAAE,CAAC,MAAM;YAClB,IAAI,EAAE,EAAE,CAAC,IAAI;YACb,SAAS,EAAE,EAAE,CAAC,UAAU;YACxB,MAAM,EAAE,WAAW;SACpB,CAAC;QACF,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAClC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,2BAA2B,EAAE;YAC7C,YAAY,EAAE,EAAE,CAAC,WAAW;YAC5B,IAAI,EAAE,SAAS;SAChB,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED,SAAS,YAAY,CAAC,IAAa,EAAE,KAAkB,EAAE,KAAsB;IAC7E,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;QAChB,KAAK,CAAC,KAAK,GAAG;YACZ,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,aAAa;YACvC,aAAa,EAAE,KAAK,CAAC,KAAK,CAAC,iBAAiB;YAC5C,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,YAAY;SACvC,CAAC;QACF,IAAI,KAAK,CAAC,KAAK,CAAC,qBAAqB,EAAE,aAAa,KAAK,SAAS,EAAE,CAAC;YACnE,KAAK,CAAC,KAAK,CAAC,oBAAoB,GAAG;gBACjC,aAAa,EAAE,KAAK,CAAC,KAAK,CAAC,qBAAqB,CAAC,aAAa;aAC/D,CAAC;QACJ,CAAC;QACD,IAAI,KAAK,CAAC,KAAK,CAAC,yBAAyB,EAAE,gBAAgB,KAAK,SAAS,EAAE,CAAC;YAC1E,KAAK,CAAC,KAAK,CAAC,qBAAqB,GAAG;gBAClC,gBAAgB,EAAE,KAAK,CAAC,KAAK,CAAC,yBAAyB,CAAC,gBAAgB;aACzE,CAAC;QACJ,CAAC;IACH,CAAC;IAED,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;IAClC,IAAI,CAAC,MAAM;QAAE,OAAO;IACpB,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;IAE3B,IAAI,KAAK,CAAC,iBAAiB,IAAI,KAAK,CAAC,eAAe,EAAE,CAAC;QACrD,IAAI,KAAK,CAAC,UAAU,KAAK,WAAW;YAAE,aAAa,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACjE,KAAK,CAAC,YAAY,IAAI,KAAK,CAAC,iBAAiB,CAAC;QAC9C,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,uCAAuC,EAAE;YACzD,OAAO,EAAE,KAAK,CAAC,YAAa;YAC5B,YAAY,EAAE,KAAK,CAAC,WAAW,GAAG,CAAC;YACnC,aAAa,EAAE,CAAC;YAChB,KAAK,EAAE,KAAK,CAAC,iBAAiB;SAC/B,CAAC,CAAC;IACL,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;QAClB,IAAI,KAAK,CAAC,UAAU,KAAK,SAAS;YAAE,WAAW,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC7D,KAAK,CAAC,YAAY,IAAI,KAAK,CAAC,OAAO,CAAC;QACpC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,4BAA4B,EAAE;YAC9C,OAAO,EAAE,KAAK,CAAC,YAAa;YAC5B,YAAY,EAAE,KAAK,CAAC,WAAW,GAAG,CAAC;YACnC,aAAa,EAAE,CAAC;YAChB,KAAK,EAAE,KAAK,CAAC,OAAO;SACrB,CAAC,CAAC;IACL,CAAC;IAED,qEAAqE;IACrE,2EAA2E;IAC3E,kCAAkC;IAClC,IAAI,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtD,IAAI,KAAK,CAAC,UAAU,KAAK,SAAS;YAAE,WAAW,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC7D,KAAK,MAAM,CAAC,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;YAClC,MAAM,UAAU,GAAG,mBAAmB,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,eAAe,GAAG,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC;YACvD,KAAK,CAAC,iBAAiB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACzC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,uCAAuC,EAAE;gBACzD,OAAO,EAAE,KAAK,CAAC,YAAa;gBAC5B,YAAY,EAAE,KAAK,CAAC,WAAW,GAAG,CAAC;gBACnC,aAAa,EAAE,CAAC;gBAChB,gBAAgB,EAAE,eAAe;gBACjC,UAAU,EAAE,UAAU;aACvB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;QACrB,KAAK,MAAM,OAAO,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;YACvC,IAAI,EAAE,GAAG,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAC5C,IAAI,CAAC,EAAE,EAAE,CAAC;gBACR,EAAE,GAAG,YAAY,CAAC,IAAI,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;YACpF,CAAC;iBAAM,IAAI,OAAO,CAAC,QAAQ,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;gBAC9C,EAAE,CAAC,IAAI,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC;YAClC,CAAC;YACD,IAAI,OAAO,CAAC,QAAQ,EAAE,SAAS,EAAE,CAAC;gBAChC,EAAE,CAAC,UAAU,IAAI,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC;gBAC5C,EAAE,CAAC,WAAW,GAAG,IAAI,CAAC;gBACtB,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,wCAAwC,EAAE;oBAC1D,OAAO,EAAE,EAAE,CAAC,MAAM;oBAClB,YAAY,EAAE,EAAE,CAAC,WAAW;oBAC5B,KAAK,EAAE,OAAO,CAAC,QAAQ,CAAC,SAAS;iBAClC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;QACzB,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC,aAAa,CAAC;IAC5C,CAAC;AACH,CAAC;AAMD,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,IAAa,EACb,MAA4B,EAC5B,GAAqB,EACrB,IAAqB;IAErB,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,GAAG,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;IAEzD,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,kBAAkB,EAAE;QACpC,QAAQ,EAAE,qBAAqB,CAAC,KAAK,EAAE,aAAa,CAAC;KACtD,CAAC,CAAC;IACH,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,sBAAsB,EAAE;QACxC,QAAQ,EAAE,qBAAqB,CAAC,KAAK,EAAE,aAAa,CAAC;KACtD,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YACxC,IAAI,IAAI,CAAC,MAAM,EAAE;gBAAE,OAAO;YAC1B,YAAY,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,cAAc,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC5B,iBAAiB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC/B,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACjE,MAAM,cAAc,GAAG,qBAAqB,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QAC9D,cAAc,CAAC,KAAK,GAAG,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,CAAC;QAC3D,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,iBAAiB,EAAE,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC,CAAC;QACnE,IAAI,CAAC,GAAG,EAAE,CAAC;QACX,OAAO;IACT,CAAC;IAED,cAAc,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAC5B,iBAAiB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAE/B,MAAM,SAAS,GAAG,qBAAqB,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;IAC5D,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,oBAAoB,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;IACjE,IAAI,CAAC,GAAG,EAAE,CAAC;AACb,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/translate/types.ts"],"names":[],"mappings":"AAAA,wEAAwE;AACxE,8DAA8D"}
|
|
@@ -0,0 +1,56 @@
|
|
|
1
|
+
import { createParser } from "eventsource-parser";
|
|
2
|
+
import { log } from "../util/log.js";
|
|
3
|
+
// Convert a fetch Response with text/event-stream body into an AsyncIterable
|
|
4
|
+
// of parsed ChatStreamChunk objects. Skips [DONE] and malformed chunks.
|
|
5
|
+
export async function* iterChatStreamChunks(response) {
|
|
6
|
+
if (!response.body) {
|
|
7
|
+
throw new Error("upstream returned empty body for streaming response");
|
|
8
|
+
}
|
|
9
|
+
const queue = [];
|
|
10
|
+
let done = false;
|
|
11
|
+
const parser = createParser({
|
|
12
|
+
onEvent(event) {
|
|
13
|
+
const data = event.data;
|
|
14
|
+
if (!data || data === "[DONE]")
|
|
15
|
+
return;
|
|
16
|
+
try {
|
|
17
|
+
const obj = JSON.parse(data);
|
|
18
|
+
queue.push(obj);
|
|
19
|
+
}
|
|
20
|
+
catch (err) {
|
|
21
|
+
log.warn("failed to parse upstream SSE chunk; skipping", {
|
|
22
|
+
error: err.message,
|
|
23
|
+
data: data.slice(0, 200),
|
|
24
|
+
});
|
|
25
|
+
}
|
|
26
|
+
},
|
|
27
|
+
});
|
|
28
|
+
const reader = response.body.getReader();
|
|
29
|
+
const decoder = new TextDecoder("utf-8");
|
|
30
|
+
try {
|
|
31
|
+
while (true) {
|
|
32
|
+
// Drain anything the parser already produced.
|
|
33
|
+
while (queue.length > 0) {
|
|
34
|
+
yield queue.shift();
|
|
35
|
+
}
|
|
36
|
+
if (done)
|
|
37
|
+
return;
|
|
38
|
+
const { value, done: rDone } = await reader.read();
|
|
39
|
+
if (rDone) {
|
|
40
|
+
done = true;
|
|
41
|
+
continue;
|
|
42
|
+
}
|
|
43
|
+
const text = decoder.decode(value, { stream: true });
|
|
44
|
+
parser.feed(text);
|
|
45
|
+
}
|
|
46
|
+
}
|
|
47
|
+
finally {
|
|
48
|
+
try {
|
|
49
|
+
reader.releaseLock();
|
|
50
|
+
}
|
|
51
|
+
catch {
|
|
52
|
+
/* ignore */
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
//# sourceMappingURL=chatStream.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"chatStream.js","sourceRoot":"","sources":["../../src/upstream/chatStream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAA2B,MAAM,oBAAoB,CAAC;AAE3E,OAAO,EAAE,GAAG,EAAE,MAAM,gBAAgB,CAAC;AAErC,6EAA6E;AAC7E,wEAAwE;AACxE,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,oBAAoB,CACzC,QAAkB;IAElB,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;IACzE,CAAC;IACD,MAAM,KAAK,GAAsB,EAAE,CAAC;IACpC,IAAI,IAAI,GAAG,KAAK,CAAC;IAEjB,MAAM,MAAM,GAAG,YAAY,CAAC;QAC1B,OAAO,CAAC,KAAyB;YAC/B,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;YACxB,IAAI,CAAC,IAAI,IAAI,IAAI,KAAK,QAAQ;gBAAE,OAAO;YACvC,IAAI,CAAC;gBACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAoB,CAAC;gBAChD,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAClB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,GAAG,CAAC,IAAI,CAAC,8CAA8C,EAAE;oBACvD,KAAK,EAAG,GAAa,CAAC,OAAO;oBAC7B,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;iBACzB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;KACF,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;IACzC,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,OAAO,CAAC,CAAC;IAEzC,IAAI,CAAC;QACH,OAAO,IAAI,EAAE,CAAC;YACZ,8CAA8C;YAC9C,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxB,MAAM,KAAK,CAAC,KAAK,EAAG,CAAC;YACvB,CAAC;YACD,IAAI,IAAI;gBAAE,OAAO;YAEjB,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YACnD,IAAI,KAAK,EAAE,CAAC;gBACV,IAAI,GAAG,IAAI,CAAC;gBACZ,SAAS;YACX,CAAC;YACD,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YACrD,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpB,CAAC;IACH,CAAC;YAAS,CAAC;QACT,IAAI,CAAC;YACH,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,CAAC;QAAC,MAAM,CAAC;YACP,YAAY;QACd,CAAC;IACH,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,99 @@
|
|
|
1
|
+
import { log, redactKey } from "../util/log.js";
|
|
2
|
+
export class UpstreamError extends Error {
|
|
3
|
+
status;
|
|
4
|
+
bodySnippet;
|
|
5
|
+
code;
|
|
6
|
+
constructor(opts) {
|
|
7
|
+
super(opts.message);
|
|
8
|
+
this.name = "UpstreamError";
|
|
9
|
+
this.status = opts.status;
|
|
10
|
+
this.code = opts.code;
|
|
11
|
+
this.bodySnippet = opts.bodySnippet;
|
|
12
|
+
}
|
|
13
|
+
}
|
|
14
|
+
function buildUrl(baseUrl) {
|
|
15
|
+
const trimmed = baseUrl.replace(/\/+$/, "");
|
|
16
|
+
return `${trimmed}/chat/completions`;
|
|
17
|
+
}
|
|
18
|
+
function authHeader(apiKey) {
|
|
19
|
+
// MiMo accepts both "Authorization: Bearer" and "api-key". Bearer is more
|
|
20
|
+
// universally supported by intermediaries, so we use it.
|
|
21
|
+
return { Authorization: `Bearer ${apiKey}` };
|
|
22
|
+
}
|
|
23
|
+
async function readSnippet(res) {
|
|
24
|
+
try {
|
|
25
|
+
const text = await res.text();
|
|
26
|
+
return text.length > 800 ? `${text.slice(0, 800)}…` : text;
|
|
27
|
+
}
|
|
28
|
+
catch {
|
|
29
|
+
return undefined;
|
|
30
|
+
}
|
|
31
|
+
}
|
|
32
|
+
export async function callMimo(cfg, body, signal) {
|
|
33
|
+
const url = buildUrl(cfg.baseUrl);
|
|
34
|
+
const headers = {
|
|
35
|
+
"Content-Type": "application/json",
|
|
36
|
+
Accept: body.stream ? "text/event-stream" : "application/json",
|
|
37
|
+
"User-Agent": cfg.userAgent,
|
|
38
|
+
...authHeader(cfg.apiKey),
|
|
39
|
+
};
|
|
40
|
+
log.debug(`upstream POST ${url}`, {
|
|
41
|
+
model: body.model,
|
|
42
|
+
stream: !!body.stream,
|
|
43
|
+
messages: body.messages.length,
|
|
44
|
+
tools: body.tools?.length ?? 0,
|
|
45
|
+
apiKey: redactKey(cfg.apiKey),
|
|
46
|
+
});
|
|
47
|
+
// Full body in --verbose. Useful when MiMo returns an opaque 400 — you can
|
|
48
|
+
// see exactly what the proxy sent. No api key leaks; that's only in headers.
|
|
49
|
+
log.debug("upstream POST body", body);
|
|
50
|
+
const attempt = async () => {
|
|
51
|
+
const res = await fetch(url, {
|
|
52
|
+
method: "POST",
|
|
53
|
+
headers,
|
|
54
|
+
body: JSON.stringify(body),
|
|
55
|
+
signal,
|
|
56
|
+
});
|
|
57
|
+
return res;
|
|
58
|
+
};
|
|
59
|
+
let res;
|
|
60
|
+
try {
|
|
61
|
+
res = await attempt();
|
|
62
|
+
}
|
|
63
|
+
catch (err) {
|
|
64
|
+
if (err.name === "AbortError")
|
|
65
|
+
throw err;
|
|
66
|
+
// Retry once on transient network errors.
|
|
67
|
+
log.warn("upstream connect failed, retrying once", { error: err.message });
|
|
68
|
+
try {
|
|
69
|
+
res = await attempt();
|
|
70
|
+
}
|
|
71
|
+
catch (err2) {
|
|
72
|
+
throw new UpstreamError({
|
|
73
|
+
status: 502,
|
|
74
|
+
code: "upstream_unreachable",
|
|
75
|
+
message: `failed to reach MiMo: ${err2.message}`,
|
|
76
|
+
});
|
|
77
|
+
}
|
|
78
|
+
}
|
|
79
|
+
if (!res.ok) {
|
|
80
|
+
const snippet = await readSnippet(res);
|
|
81
|
+
const code = res.status === 401
|
|
82
|
+
? "authentication_error"
|
|
83
|
+
: res.status === 403
|
|
84
|
+
? "permission_denied"
|
|
85
|
+
: res.status === 429
|
|
86
|
+
? "rate_limit_exceeded"
|
|
87
|
+
: res.status >= 500
|
|
88
|
+
? "server_error"
|
|
89
|
+
: "bad_request";
|
|
90
|
+
throw new UpstreamError({
|
|
91
|
+
status: res.status,
|
|
92
|
+
code,
|
|
93
|
+
message: `MiMo returned ${res.status}: ${snippet ?? "(no body)"}`,
|
|
94
|
+
bodySnippet: snippet,
|
|
95
|
+
});
|
|
96
|
+
}
|
|
97
|
+
return res;
|
|
98
|
+
}
|
|
99
|
+
//# sourceMappingURL=mimoClient.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mimoClient.js","sourceRoot":"","sources":["../../src/upstream/mimoClient.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,GAAG,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAUhD,MAAM,OAAO,aAAc,SAAQ,KAAK;IACtC,MAAM,CAAS;IACf,WAAW,CAAU;IACrB,IAAI,CAAS;IAEb,YAAY,IAA6E;QACvF,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACpB,IAAI,CAAC,IAAI,GAAG,eAAe,CAAC;QAC5B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;QACtB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;IACtC,CAAC;CACF;AAED,SAAS,QAAQ,CAAC,OAAe;IAC/B,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IAC5C,OAAO,GAAG,OAAO,mBAAmB,CAAC;AACvC,CAAC;AAED,SAAS,UAAU,CAAC,MAAc;IAChC,0EAA0E;IAC1E,yDAAyD;IACzD,OAAO,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE,CAAC;AAC/C,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,GAAa;IACtC,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;QAC9B,OAAO,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;IAC7D,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC5B,GAAmB,EACnB,IAAiB,EACjB,MAAmB;IAEnB,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAClC,MAAM,OAAO,GAA2B;QACtC,cAAc,EAAE,kBAAkB;QAClC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,kBAAkB;QAC9D,YAAY,EAAE,GAAG,CAAC,SAAS;QAC3B,GAAG,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC;KAC1B,CAAC;IAEF,GAAG,CAAC,KAAK,CAAC,iBAAiB,GAAG,EAAE,EAAE;QAChC,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM;QACrB,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM;QAC9B,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,MAAM,IAAI,CAAC;QAC9B,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC;KAC9B,CAAC,CAAC;IACH,2EAA2E;IAC3E,6EAA6E;IAC7E,GAAG,CAAC,KAAK,CAAC,oBAAoB,EAAE,IAAI,CAAC,CAAC;IAEtC,MAAM,OAAO,GAAG,KAAK,IAAuB,EAAE;QAC5C,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAC3B,MAAM,EAAE,MAAM;YACd,OAAO;YACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;YAC1B,MAAM;SACP,CAAC,CAAC;QACH,OAAO,GAAG,CAAC;IACb,CAAC,CAAC;IAEF,IAAI,GAAa,CAAC;IAClB,IAAI,CAAC;QACH,GAAG,GAAG,MAAM,OAAO,EAAE,CAAC;IACxB,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAK,GAAa,CAAC,IAAI,KAAK,YAAY;YAAE,MAAM,GAAG,CAAC;QACpD,0CAA0C;QAC1C,GAAG,CAAC,IAAI,CAAC,wCAAwC,EAAE,EAAE,KAAK,EAAG,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;QACtF,IAAI,CAAC;YACH,GAAG,GAAG,MAAM,OAAO,EAAE,CAAC;QACxB,CAAC;QAAC,OAAO,IAAI,EAAE,CAAC;YACd,MAAM,IAAI,aAAa,CAAC;gBACtB,MAAM,EAAE,GAAG;gBACX,IAAI,EAAE,sBAAsB;gBAC5B,OAAO,EAAE,yBAA0B,IAAc,CAAC,OAAO,EAAE;aAC5D,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;QACZ,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,GAAG,CAAC,CAAC;QACvC,MAAM,IAAI,GACR,GAAG,CAAC,MAAM,KAAK,GAAG;YAChB,CAAC,CAAC,sBAAsB;YACxB,CAAC,CAAC,GAAG,CAAC,MAAM,KAAK,GAAG;gBAClB,CAAC,CAAC,mBAAmB;gBACrB,CAAC,CAAC,GAAG,CAAC,MAAM,KAAK,GAAG;oBAClB,CAAC,CAAC,qBAAqB;oBACvB,CAAC,CAAC,GAAG,CAAC,MAAM,IAAI,GAAG;wBACjB,CAAC,CAAC,cAAc;wBAChB,CAAC,CAAC,aAAa,CAAC;QAC1B,MAAM,IAAI,aAAa,CAAC;YACtB,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,IAAI;YACJ,OAAO,EAAE,iBAAiB,GAAG,CAAC,MAAM,KAAK,OAAO,IAAI,WAAW,EAAE;YACjE,WAAW,EAAE,OAAO;SACrB,CAAC,CAAC;IACL,CAAC;IAED,OAAO,GAAG,CAAC;AACb,CAAC"}
|
package/dist/util/ids.js
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import { customAlphabet } from "nanoid";
|
|
2
|
+
const alphabet = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";
|
|
3
|
+
const nanoid24 = customAlphabet(alphabet, 24);
|
|
4
|
+
export function newResponseId() {
|
|
5
|
+
return `resp_${nanoid24()}`;
|
|
6
|
+
}
|
|
7
|
+
export function newMessageId() {
|
|
8
|
+
return `msg_${nanoid24()}`;
|
|
9
|
+
}
|
|
10
|
+
export function newFunctionCallId() {
|
|
11
|
+
return `fc_${nanoid24()}`;
|
|
12
|
+
}
|
|
13
|
+
export function newReasoningId() {
|
|
14
|
+
return `rs_${nanoid24()}`;
|
|
15
|
+
}
|
|
16
|
+
export function newCallId() {
|
|
17
|
+
return `call_${nanoid24()}`;
|
|
18
|
+
}
|
|
19
|
+
//# sourceMappingURL=ids.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ids.js","sourceRoot":"","sources":["../../src/util/ids.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,QAAQ,CAAC;AAExC,MAAM,QAAQ,GACZ,gEAAgE,CAAC;AACnE,MAAM,QAAQ,GAAG,cAAc,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;AAE9C,MAAM,UAAU,aAAa;IAC3B,OAAO,QAAQ,QAAQ,EAAE,EAAE,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,YAAY;IAC1B,OAAO,OAAO,QAAQ,EAAE,EAAE,CAAC;AAC7B,CAAC;AAED,MAAM,UAAU,iBAAiB;IAC/B,OAAO,MAAM,QAAQ,EAAE,EAAE,CAAC;AAC5B,CAAC;AAED,MAAM,UAAU,cAAc;IAC5B,OAAO,MAAM,QAAQ,EAAE,EAAE,CAAC;AAC5B,CAAC;AAED,MAAM,UAAU,SAAS;IACvB,OAAO,QAAQ,QAAQ,EAAE,EAAE,CAAC;AAC9B,CAAC"}
|
package/dist/util/log.js
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
1
|
+
let verbose = false;
|
|
2
|
+
export function setVerbose(v) {
|
|
3
|
+
verbose = v;
|
|
4
|
+
}
|
|
5
|
+
function emit(level, msg, extra) {
|
|
6
|
+
if (level === "debug" && !verbose)
|
|
7
|
+
return;
|
|
8
|
+
const ts = new Date().toISOString();
|
|
9
|
+
const prefix = `[${ts}] ${level.toUpperCase()}`;
|
|
10
|
+
if (extra !== undefined) {
|
|
11
|
+
// eslint-disable-next-line no-console
|
|
12
|
+
console.error(prefix, msg, extra);
|
|
13
|
+
}
|
|
14
|
+
else {
|
|
15
|
+
// eslint-disable-next-line no-console
|
|
16
|
+
console.error(prefix, msg);
|
|
17
|
+
}
|
|
18
|
+
}
|
|
19
|
+
export const log = {
|
|
20
|
+
debug: (msg, extra) => emit("debug", msg, extra),
|
|
21
|
+
info: (msg, extra) => emit("info", msg, extra),
|
|
22
|
+
warn: (msg, extra) => emit("warn", msg, extra),
|
|
23
|
+
error: (msg, extra) => emit("error", msg, extra),
|
|
24
|
+
};
|
|
25
|
+
export function redactKey(k) {
|
|
26
|
+
if (!k)
|
|
27
|
+
return "<empty>";
|
|
28
|
+
if (k.length <= 8)
|
|
29
|
+
return "***";
|
|
30
|
+
return `${k.slice(0, 4)}…${k.slice(-4)}`;
|
|
31
|
+
}
|
|
32
|
+
//# sourceMappingURL=log.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"log.js","sourceRoot":"","sources":["../../src/util/log.ts"],"names":[],"mappings":"AAEA,IAAI,OAAO,GAAG,KAAK,CAAC;AAEpB,MAAM,UAAU,UAAU,CAAC,CAAU;IACnC,OAAO,GAAG,CAAC,CAAC;AACd,CAAC;AAED,SAAS,IAAI,CAAC,KAAY,EAAE,GAAW,EAAE,KAAe;IACtD,IAAI,KAAK,KAAK,OAAO,IAAI,CAAC,OAAO;QAAE,OAAO;IAC1C,MAAM,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACpC,MAAM,MAAM,GAAG,IAAI,EAAE,KAAK,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;IAChD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACxB,sCAAsC;QACtC,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC;IACpC,CAAC;SAAM,CAAC;QACN,sCAAsC;QACtC,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC7B,CAAC;AACH,CAAC;AAED,MAAM,CAAC,MAAM,GAAG,GAAG;IACjB,KAAK,EAAE,CAAC,GAAW,EAAE,KAAe,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,KAAK,CAAC;IAClE,IAAI,EAAE,CAAC,GAAW,EAAE,KAAe,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,CAAC;IAChE,IAAI,EAAE,CAAC,GAAW,EAAE,KAAe,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,CAAC;IAChE,KAAK,EAAE,CAAC,GAAW,EAAE,KAAe,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,KAAK,CAAC;CACnE,CAAC;AAEF,MAAM,UAAU,SAAS,CAAC,CAA4B;IACpD,IAAI,CAAC,CAAC;QAAE,OAAO,SAAS,CAAC;IACzB,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC;QAAE,OAAO,KAAK,CAAC;IAChC,OAAO,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AAC3C,CAAC"}
|
package/dist/util/sse.js
ADDED
|
@@ -0,0 +1,61 @@
|
|
|
1
|
+
export function makeServerResponseSink(res) {
|
|
2
|
+
res.setHeader("Content-Type", "text/event-stream; charset=utf-8");
|
|
3
|
+
res.setHeader("Cache-Control", "no-cache, no-transform");
|
|
4
|
+
res.setHeader("Connection", "keep-alive");
|
|
5
|
+
res.setHeader("X-Accel-Buffering", "no");
|
|
6
|
+
if (typeof res.flushHeaders === "function") {
|
|
7
|
+
res.flushHeaders();
|
|
8
|
+
}
|
|
9
|
+
let isClosed = false;
|
|
10
|
+
res.on("close", () => {
|
|
11
|
+
isClosed = true;
|
|
12
|
+
});
|
|
13
|
+
return {
|
|
14
|
+
write(event, data) {
|
|
15
|
+
if (isClosed)
|
|
16
|
+
return;
|
|
17
|
+
const payload = typeof data === "string" ? data : JSON.stringify(data);
|
|
18
|
+
res.write(`event: ${event}\ndata: ${payload}\n\n`);
|
|
19
|
+
},
|
|
20
|
+
comment(text) {
|
|
21
|
+
if (isClosed)
|
|
22
|
+
return;
|
|
23
|
+
res.write(`: ${text}\n\n`);
|
|
24
|
+
},
|
|
25
|
+
end() {
|
|
26
|
+
if (isClosed)
|
|
27
|
+
return;
|
|
28
|
+
isClosed = true;
|
|
29
|
+
res.end();
|
|
30
|
+
},
|
|
31
|
+
closed() {
|
|
32
|
+
return isClosed;
|
|
33
|
+
},
|
|
34
|
+
};
|
|
35
|
+
}
|
|
36
|
+
export function makeMemorySink() {
|
|
37
|
+
const events = [];
|
|
38
|
+
const comments = [];
|
|
39
|
+
let isClosed = false;
|
|
40
|
+
return {
|
|
41
|
+
events,
|
|
42
|
+
comments,
|
|
43
|
+
write(event, data) {
|
|
44
|
+
if (isClosed)
|
|
45
|
+
return;
|
|
46
|
+
events.push({ event, data });
|
|
47
|
+
},
|
|
48
|
+
comment(text) {
|
|
49
|
+
if (isClosed)
|
|
50
|
+
return;
|
|
51
|
+
comments.push(text);
|
|
52
|
+
},
|
|
53
|
+
end() {
|
|
54
|
+
isClosed = true;
|
|
55
|
+
},
|
|
56
|
+
closed() {
|
|
57
|
+
return isClosed;
|
|
58
|
+
},
|
|
59
|
+
};
|
|
60
|
+
}
|
|
61
|
+
//# sourceMappingURL=sse.js.map
|