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.
@@ -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,4 @@
1
+ // Minimal type definitions for the two API shapes we translate between.
2
+ // Trimmed to the fields the proxy actually inspects or emits.
3
+ export {};
4
+ //# sourceMappingURL=types.js.map
@@ -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"}
@@ -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"}
@@ -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"}
@@ -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