mslxdff 0.1.88 → 0.1.90

package/src/transport/retry.js

@@ -0,0 +1,24 @@
+export function backoffDelay(baseMs, attempt) {
+  // attempt 0 => base, 1 => base*2, 2 => base*4
+  return baseMs * (1 << attempt);
+}
+
+export function resolveRetry(status, attempt, config) {
+  const key = status === "network" ? "network" : String(status);
+  const entry = config?.[key];
+  if (!entry) return { shouldRetry: false, delayMs: 0 };
+  const attempts = Number(entry.attempts) || 0;
+  const base = Number(entry.delayMs) || 0;
+  if (attempt < attempts) {
+    return { shouldRetry: true, delayMs: backoffDelay(base, attempt) };
+  }
+  return { shouldRetry: false, delayMs: 0 };
+}
+
+export function shouldRetry(status, attempt, config) {
+  return resolveRetry(status, attempt, config).shouldRetry;
+}
+
+export function sleep(ms) {
+  return new Promise((r) => setTimeout(r, ms));
+}

package/src/transport/sse.js

@@ -0,0 +1,93 @@
+/**
+ * SSE 行协议深模块 — 纯函数 + 流式
+ * 负责 `data: / : / event: / [DONE]` 的解析与粘包 remain 处理
+ */
+
+export function parseSseChunk(buf) {
+  const events = [];
+  let remain = String(buf || "");
+  // 按 \n\n 分帧，\r\n 也兼容
+  // 保留最后一帧若不以 \n\n 结尾则为 remain
+  const normalized = remain.replace(/\r\n/g, "\n");
+  const parts = normalized.split("\n\n");
+  // 最后一部分若原串不以 \n\n 结尾则为 remain
+  const endsWithDelim = normalized.endsWith("\n\n") || normalized.endsWith("\n\r\n");
+  // parts 最后一项在非结尾时为未完成帧
+  const complete = endsWithDelim ? parts : parts.slice(0, -1);
+  remain = endsWithDelim ? "" : parts[parts.length - 1] || "";
+  for (const frame of complete) {
+    if (!frame) continue; // 空帧
+    const lines = frame.split("\n");
+    const dataLines = [];
+    for (const line of lines) {
+      if (!line) continue;
+      if (line.startsWith(":")) continue; // 注释/keepalive
+      if (line.startsWith("event:")) continue; // 事件类型忽略
+      if (line.startsWith("data:")) {
+        let v = line.slice(5);
+        if (v.startsWith(" ")) v = v.slice(1);
+        dataLines.push(v);
+      }
+    }
+    if (dataLines.length === 0) continue;
+    // 多行 data 按 \n 拼接（SSE 规范）
+    const payload = dataLines.join("\n");
+    // 空 data: 仍产生事件，由上层过滤；此处保留以便测试可见
+    events.push(payload);
+  }
+  return { events, remain };
+}
+
+export function createSseParser() {
+  let buf = "";
+  return {
+    push(chunk) {
+      buf += String(chunk || "");
+      const { events, remain } = parseSseChunk(buf);
+      buf = remain;
+      return events;
+    },
+    flush() {
+      // 未形成完整帧的不发（避免半包误触发）
+      if (!buf.trim()) { buf = ""; return []; }
+      // 若残留包含完整 data: 行但缺结尾，不强行发
+      return [];
+    },
+    getRemain() { return buf; },
+  };
+}
+
+export async function* streamFromResponse(res) {
+  if (!res || !res.body) return;
+  const parser = createSseParser();
+  const reader = res.body.getReader ? res.body.getReader() : null;
+  const decoder = new TextDecoder();
+  if (reader) {
+    while (true) {
+      const { done, value } = await reader.read();
+      if (done) break;
+      const text = decoder.decode(value, { stream: true });
+      const evs = parser.push(text);
+      for (const e of evs) {
+        if (e === "[DONE]") return;
+        if (e === "") continue;
+        yield e;
+      }
+    }
+    // flush 残留
+    const rest = parser.flush();
+    for (const e of rest) {
+      if (e === "[DONE]") return;
+      if (e) yield e;
+    }
+  } else if (typeof res.text === "function") {
+    // 回退：一次性 text
+    const txt = await res.text();
+    const evs = parser.push(txt);
+    for (const e of evs) {
+      if (e === "[DONE]") return;
+      if (e === "") continue;
+      yield e;
+    }
+  }
+}

package/src/upstream-responses.js

@@ -0,0 +1,64 @@
+/**
+ * responses 转换层 — 从 upstream.js 抽出的 muse-spark 专用形状转换。
+ * chat ⇄ responses 互转纯函数，无网络、无副作用。
+ */
+export function isResponsesModel(model) {
+  return String(model || "").toLowerCase().startsWith("muse-spark");
+}
+
+export function chatToResponsesBody(chatBody) {
+  const msgs = Array.isArray(chatBody?.messages) ? chatBody.messages : [];
+  const system = msgs.filter((m) => m.role === "system").map((m) => String(m.content || "")).join("\n");
+  const nonSystem = msgs.filter((m) => m.role !== "system");
+  const inputParts = nonSystem.map((m) => {
+    const c = m.content;
+    if (typeof c === "string") return `${m.role}: ${c}`;
+    if (Array.isArray(c)) return `${m.role}: ${c.map((x) => x.text || "").join("")}`;
+    return `${m.role}: ${String(c || "")}`;
+  });
+  const input = inputParts.join("\n\n") || "hi";
+  const out = { model: chatBody.model, input, stream: false };
+  if (system) out.instructions = system;
+  if (chatBody.tools) out.tools = chatBody.tools;
+  if (chatBody.tool_choice) out.tool_choice = chatBody.tool_choice;
+  if (chatBody.temperature != null) out.temperature = chatBody.temperature;
+  if (chatBody.max_tokens != null) out.max_output_tokens = chatBody.max_tokens;
+  return out;
+}
+
+export function responsesToChatJson(respJson) {
+  let text = "";
+  for (const item of respJson.output || []) {
+    if (item.type === "message" && item.role === "assistant") {
+      for (const c of item.content || []) {
+        if (c.type === "output_text") text += c.text || "";
+        else if (c.type === "text") text += c.text || "";
+      }
+    }
+  }
+  if (!text) {
+    for (const item of respJson.output || []) {
+      if (item.type === "message") {
+        const t = item.content?.[0]?.text;
+        if (t) { text = t; break; }
+      }
+    }
+  }
+  const chatJson = {
+    id: respJson.id || `resp_${Date.now()}`,
+    object: "chat.completion",
+    created: Math.floor((respJson.created_at || Date.now() / 1000)),
+    model: respJson.model,
+    choices: [{ index: 0, finish_reason: respJson.status === "completed" ? "stop" : "length", message: { role: "assistant", content: text } }],
+    usage: respJson.usage || { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 },
+  };
+  return chatJson;
+}
+
+/** responses 成功 Response 转回 chat 形状（anon 兜底与主路径复用） */
+export function toChatResponse(res, respJson) {
+  const chatJson = responsesToChatJson(respJson);
+  const headers = new Headers(res.headers);
+  headers.set("content-type", "application/json");
+  return new Response(JSON.stringify(chatJson), { status: res.status, headers });
+}