更新

TongjiUniApp/unpackage/dist/dev/mp-weixin/tongji/utils/aiStreamRequest.js

@@ -0,0 +1,141 @@
+"use strict";
+const common_vendor = require("../../common/vendor.js");
+function joinApiUrl(base, path) {
+  const b = String(base || "").replace(/\/+$/, "");
+  const p = String(path || "").replace(/^\/+/, "");
+  return p ? `${b}/${p}` : b;
+}
+function appendQuery(url, data) {
+  if (!data || typeof data !== "object")
+    return url;
+  const keys = Object.keys(data);
+  if (!keys.length)
+    return url;
+  const qs = keys.map((k) => `${encodeURIComponent(k)}=${encodeURIComponent(data[k] ?? "")}`).join("&");
+  return url + (url.includes("?") ? "&" : "?") + qs;
+}
+function decodeChunkData(data) {
+  if (typeof data === "string")
+    return data;
+  if (!data)
+    return "";
+  if (typeof TextDecoder !== "undefined" && data instanceof ArrayBuffer) {
+    return new TextDecoder("utf-8").decode(new Uint8Array(data));
+  }
+  if (data instanceof ArrayBuffer) {
+    const bytes = new Uint8Array(data);
+    let str = "";
+    for (let i = 0; i < bytes.length; i++)
+      str += String.fromCharCode(bytes[i]);
+    try {
+      return decodeURIComponent(escape(str));
+    } catch (e) {
+      return str;
+    }
+  }
+  return "";
+}
+function consumeSseBuffer(buffer, onEvent) {
+  let rest = buffer;
+  let idx = rest.indexOf("\n\n");
+  while (idx !== -1) {
+    const block = rest.slice(0, idx);
+    rest = rest.slice(idx + 2);
+    let event = "message";
+    let dataLine = "";
+    block.split("\n").forEach((line) => {
+      if (line.startsWith("event:"))
+        event = line.slice(6).trim();
+      else if (line.startsWith("data:"))
+        dataLine += line.slice(5).trim();
+    });
+    if (dataLine) {
+      try {
+        onEvent(event, JSON.parse(dataLine));
+      } catch (e) {
+        onEvent(event, dataLine);
+      }
+    }
+    idx = rest.indexOf("\n\n");
+  }
+  return rest;
+}
+function parsePartialDietPlainText(text) {
+  const out = {};
+  if (!text)
+    return out;
+  const lineRe = /^(早餐|喝的|午餐|晚餐|提示|少碰)[：:]\s*(.*)$/gm;
+  let match;
+  while ((match = lineRe.exec(text)) !== null) {
+    const key = { "早餐": "breakfast", "喝的": "drinks", "午餐": "lunch", "晚餐": "dinner", "提示": "tips", "少碰": "avoid" }[match[1]];
+    const val = (match[2] || "").trim();
+    if (key === "avoid") {
+      out.avoid = val.split(/[、,，;；\s]+/).map((s) => s.trim()).filter(Boolean);
+    } else {
+      out[key] = val;
+    }
+  }
+  const tail = text.match(/(?:^|\n)(早餐|喝的|午餐|晚餐|提示|少碰)[：:]\s*([^\n]*)$/);
+  if (tail) {
+    const key = { "早餐": "breakfast", "喝的": "drinks", "午餐": "lunch", "晚餐": "dinner", "提示": "tips", "少碰": "avoid" }[tail[1]];
+    const val = (tail[2] || "").trim();
+    if (key === "avoid") {
+      if (val)
+        out.avoid = val.split(/[、,，;；\s]+/).map((s) => s.trim()).filter(Boolean);
+    } else {
+      out[key] = val;
+    }
+  }
+  return out;
+}
+function requestAiStream(opts) {
+  const {
+    baseUrl,
+    url,
+    method = "GET",
+    data = {},
+    onEvent,
+    fallback
+  } = opts;
+  return new Promise((resolve, reject) => {
+    const token = common_vendor.index.getStorageSync("token") || "";
+    let sseBuffer = "";
+    let requestUrl = joinApiUrl(baseUrl, url);
+    if (method === "GET") {
+      requestUrl = appendQuery(requestUrl, data);
+    }
+    const task = common_vendor.wx$1.request({
+      url: requestUrl,
+      method,
+      data: method === "POST" ? data : {},
+      enableChunked: true,
+      timeout: 6e4,
+      header: {
+        token,
+        "content-type": "application/x-www-form-urlencoded",
+        "Cache-Control": "no-cache"
+      },
+      success: (res) => {
+        if (res && res.data) {
+          sseBuffer += decodeChunkData(res.data);
+          sseBuffer = consumeSseBuffer(sseBuffer, onEvent);
+        }
+        resolve(res);
+      },
+      fail: (err) => reject(err)
+    });
+    if (task && typeof task.onChunkReceived === "function") {
+      task.onChunkReceived((res) => {
+        sseBuffer += decodeChunkData(res.data);
+        sseBuffer = consumeSseBuffer(sseBuffer, onEvent);
+      });
+    } else if (typeof fallback === "function") {
+      fallback().then(resolve).catch(reject);
+    } else {
+      reject(new Error("当前环境不支持流式请求"));
+    }
+  });
+}
+exports.parsePartialDietPlainText = parsePartialDietPlainText;
+exports.requestAiStream = requestAiStream;
+//# sourceMappingURL=../../../.sourcemap/mp-weixin/tongji/utils/aiStreamRequest.js.map