st-comp 0.0.270 → 0.0.271

package/lib/aiTools.js

@@ -1,307 +1,93 @@
-// 辅助函数：处理缓冲区剩余数据
-const tryProcessBuffer = (buffer, callback) => {
-  const lines = buffer.split("\n");
-  let currentEvent = null;
-
-  for (const line of lines) {
-    if (line.startsWith("event:")) {
-      currentEvent = line.substring(6).trim();
-      continue;
-    }
-
-    if (line.startsWith("data:")) {
-      const dataStr = line.substring(5).trim();
-      if (dataStr && dataStr !== "[DONE]") {
-        try {
-          const parsed = JSON.parse(dataStr);
-
-          // 错误响应
-          if (currentEvent === "error" || parsed.code || parsed.message) {
-            const errorMsg = parsed.message || parsed.code || "未知错误";
-            callback("error", `百炼服务端错误: ${errorMsg}`);
-            return;
-          }
-
-          // 正常响应
-          const text = parsed?.output?.text;
-          if (text && callback) callback("message", text);
-        } catch (e) {}
-      }
-      currentEvent = null;
-    }
-  }
-};
-
 /**
- * @description: 百炼dashscope直连
+ * @description: 通用 SSE 流式请求
+ * @param {Object} params
+ * @param {string} params.baseUrl - 服务地址
+ * @param {string} params.path - 接口路径
+ * @param {string} params.token - 认证token
+ * @param {string} params.value - 查询内容
+ * @param {string} params.origin - 来源标识，默认 "web"
+ * @param {Function} params.getSSE - 回调函数 (type, str) => void
+ * type: "finish" || "error" || "message"
  */
-// 百炼应用(常规)
-export const sendToBaiLianAppNonStreaming = async ({ appId, apiKey, value }) => {
+const sendToNodeSSE = async ({ baseUrl, path, token, value, origin = "web", getSSE }) => {
   try {
-    const response = await fetch(`https://dashscope.aliyuncs.com/api/v1/apps/${appId}/completion`, {
+    // 创建请求
+    const response = await fetch(`${baseUrl}${path}`, {
       method: "POST",
       headers: {
-        Authorization: `Bearer ${apiKey}`,
         "Content-Type": "application/json",
+        token,
       },
       body: JSON.stringify({
-        input: { prompt: value },
-        parameters: {
-          incremental_output: false, // 非流式
-        },
-      }),
-    });
-    if (!response.ok) {
-      throw new Error(`HTTP ${response.status}: ${response.statusText}`);
-    }
-    const data = await response.json();
-    const outputText = data?.output?.text;
-    return outputText;
-  } catch (error) {
-    if (error.name === "AbortError") {
-      throw new Error("请求超时");
-    }
-    throw error;
-  }
-};
-// 百炼应用(流式)
-export const sendToBaiLianAppStreaming = async ({ appId, apiKey, value, callback }) => {
-  try {
-    const response = await fetch(`https://dashscope.aliyuncs.com/api/v1/apps/${appId}/completion`, {
-      method: "POST",
-      body: JSON.stringify({
-        input: { prompt: value },
-        parameters: { incremental_output: "true" },
-        debug: {},
+        query: value,
+        origin,
+        stream: true,
       }),
-      headers: {
-        Authorization: `Bearer ${apiKey}`,
-        "Content-Type": "application/json",
-        "X-DashScope-SSE": "enable",
-      },
     });
-
     if (!response.ok) {
-      callback("error", `HTTP ${response.status}: ${response.statusText}`);
+      getSSE("error", `HTTP ${response.status}: ${response.statusText}`);
       return;
     }
 
+    // 创建reader读取器和decoder解码器
     const reader = response.body.getReader();
     const decoder = new TextDecoder();
-    let buffer = "";
-    let currentEvent = null; // 记录当前 event 类型
-
-    while (true) {
-      const { done, value } = await reader.read();
-      if (done) {
-        // 处理最后可能遗留的数据
-        if (buffer.trim()) {
-          tryProcessBuffer(buffer, callback);
-        }
-        callback("finish", "");
-        break;
-      }
-
-      buffer += decoder.decode(value, { stream: true });
 
-      let newlineIndex;
-      while ((newlineIndex = buffer.indexOf("\n")) !== -1) {
-        const line = buffer.substring(0, newlineIndex).trim();
-        buffer = buffer.substring(newlineIndex + 1);
-
-        // 处理 event: 行（错误类型）
-        if (line.startsWith("event:")) {
-          currentEvent = line.substring(6).trim();
-          continue;
-        }
-
-        // 处理 data: 行
-        if (line.startsWith("data:")) {
-          const dataStr = line.substring(5).trim();
-
-          // 检查是否是错误响应
-          if (currentEvent === "error" || (dataStr && dataStr.startsWith("{"))) {
-            try {
-              const parsed = JSON.parse(dataStr);
-
-              // 处理错误响应
-              if (parsed.code || parsed.message || currentEvent === "error") {
-                const errorMsg = parsed.message || parsed.code || "未知错误";
-                callback("error", `百炼服务端错误: ${errorMsg}`);
-                return; // 终止流式处理
-              }
-
-              // 正常响应：提取 text
-              const text = parsed?.output?.text;
-              if (text && callback) {
-                callback("message", text);
-              }
-            } catch (e) {
-              console.debug("JSON 解析失败:", dataStr);
-            }
-          }
-
-          // 重置 event 类型
-          currentEvent = null;
-        }
-      }
-    }
-  } catch (error) {
-    console.error("流式请求失败:", error);
-    callback("error", error.message);
-  }
-};
-// 百炼workflow(流式)
-export const sendToBaiLianWorkflowStreaming = async ({ mode, appId, apiKey, token, value, callback }) => {
-  try {
-    const response = await fetch(`https://dashscope.aliyuncs.com/api/v1/apps/${appId}/completion`, {
-      method: "POST",
-      body: JSON.stringify({
-        input: { prompt: value, biz_params: { token, mode, origin: "web" } },
-        parameters: { incremental_output: "true", flow_stream_mode: "message_format_plus" },
-        debug: {},
-      }),
-      headers: {
-        Authorization: `Bearer ${apiKey}`,
-        "Content-Type": "application/json",
-        "X-DashScope-SSE": "enable",
-      },
-    });
-
-    if (!response.ok) {
-      callback("error", `HTTP ${response.status}: ${response.statusText}`);
-      return;
-    }
-
-    const reader = response.body.getReader();
-    const decoder = new TextDecoder();
+    // 缓冲区数据
     let buffer = "";
-    let finalResult = "";
-    // 记录已出现的节点（去重用）
-    const nodeMap = {};
+    let newlineIndex;
 
     while (true) {
       const { done, value } = await reader.read();
+      // 判断SSE是否传输结束
       if (done) {
-        if (finalResult) {
-          callback("message", finalResult);
-        }
-        callback("finish", "");
+        getSSE("finish");
         break;
       }
-
+      // 缓冲区填入数据
       buffer += decoder.decode(value, { stream: true });
       let newlineIndex;
+      // 从缓冲区中逐条提取完整的消息行
       while ((newlineIndex = buffer.indexOf("\n")) !== -1) {
+        // 提取一行
         const line = buffer.substring(0, newlineIndex).trim();
+        // 移除已提取的部分
         buffer = buffer.substring(newlineIndex + 1);
-
-        if (line.startsWith("data:")) {
-          const dataStr = line.substring(5).trim();
+        if (line.startsWith("data: ")) {
+          // 切割掉前面的data: , 总计6个字符, 并且去除空格
+          const dataStr = line.substring(6).trim();
+          // 过滤空数据, 和结束标记
           if (!dataStr || dataStr === "[DONE]") continue;
-
-          try {
-            const parsed = JSON.parse(dataStr);
-            const output = parsed?.output;
-            const wfMsg = output?.workflow_message;
-
-            // 提取节点信息
-            if (wfMsg?.node_name && wfMsg?.node_type) {
-              const nodeName = wfMsg.node_name;
-              const nodeStatus = wfMsg.node_status; // "executing" | "success"
-              const isCompleted = wfMsg.node_is_completed;
-
-              // 去重：同一节点只推送状态变化
-              const prevStatus = nodeMap[nodeName];
-              if (prevStatus !== nodeStatus || (isCompleted && prevStatus !== "success")) {
-                nodeMap[nodeName] = isCompleted ? "success" : nodeStatus;
-                callback("node", {
-                  name: nodeName,
-                  status: isCompleted ? "success" : "executing",
-                });
-              }
-            }
-
-            // 最终结果
-            if (output?.finish_reason === "stop" && output?.text) {
-              finalResult = output.text;
-            }
-
-            // 错误
-            if (output?.code || output?.message) {
-              callback("error", output.message || output.code || "未知错误");
-              return;
-            }
-          } catch (e) {
-            // 忽略非 JSON 行
-          }
+          getSSE("message", dataStr);
         }
       }
     }
   } catch (error) {
-    console.error("工作流请求失败:", error);
-    callback("error", error.message);
+    console.error("SSE请求失败:", error);
+    getSSE("error", error.message || "sendToNodeSSE检测到异常");
   }
 };
 
-/**
- * @description: Node+OpenAi
- */
-// 品种池AI助手(流式) - 品种池AI助手
+// 品种池AI查询助手
 export const sendToNodeVarietyAiHelper = async ({ baseUrl, token, value, callback }) => {
-  try {
-    const response = await fetch(`${baseUrl}/middleLayer/ai/varietyAiHelper`, {
-      method: "POST",
-      headers: {
-        "Content-Type": "application/json",
-        token: token,
-      },
-      body: JSON.stringify({
-        query: value,
-        origin: "web",
-        stream: true,
-      }),
-    });
-
-    if (!response.ok) {
-      callback("error", `HTTP ${response.status}: ${response.statusText}`);
-      return;
-    }
-
-    const reader = response.body.getReader();
-    const decoder = new TextDecoder();
-    let buffer = "";
-    let finalResult = "";
-    const nodeMap = {};
-
-    while (true) {
-      const { done, value } = await reader.read();
-      if (done) {
-        if (finalResult) {
-          callback("message", finalResult);
-        }
-        callback("finish", "");
-        break;
-      }
-
-      buffer += decoder.decode(value, { stream: true });
-      let newlineIndex;
-      while ((newlineIndex = buffer.indexOf("\n")) !== -1) {
-        const line = buffer.substring(0, newlineIndex).trim();
-        buffer = buffer.substring(newlineIndex + 1);
-
-        if (line.startsWith("data: ")) {
-          const dataStr = line.substring(6).trim();
-          if (!dataStr || dataStr === "[DONE]") continue;
-
+  // 工作流节点
+  const nodeMap = {};
+  const params = {
+    baseUrl,
+    path: "/middleLayer/ai/varietyAiHelper",
+    token,
+    value,
+    origin: "web",
+    getSSE: (type, str) => {
+      switch (type) {
+        case "message": {
           try {
-            const data = JSON.parse(dataStr);
-
-            // 节点状态更新
+            const data = JSON.parse(str);
+            // 节点状态
             if (data.node_name && data.node_status) {
               const nodeName = data.node_name;
               const nodeStatus = data.node_status === "success" ? "success" : "executing";
-
+              // 节点状态发生改变, 推送节点状态变更
               if (nodeMap[nodeName] !== nodeStatus) {
                 nodeMap[nodeName] = nodeStatus;
                 callback("node", {
@@ -310,26 +96,67 @@ export const sendToNodeVarietyAiHelper = async ({ baseUrl, token, value, callbac
                 });
               }
             }
-
-            // 最终结果
+            // 结果输出
             if (data.output?.finish_reason === "stop" && data.output?.text) {
-              finalResult = data.output.text;
-              callback("message", finalResult);
+              callback("message", data.output.text);
             }
-
             // 错误处理
             if (data.error) {
               callback("error", data.error);
-              return;
             }
-          } catch (e) {
-            // 忽略非 JSON 行
+          } catch (error) {
+            // 忽略非序列化消息
           }
+          break;
+        }
+        case "finish": {
+          callback("finish");
+          break;
+        }
+        case "error": {
+          callback("error", str);
+          break;
         }
       }
-    }
-  } catch (error) {
-    console.error("Node服务请求失败:", error);
-    callback("error", error.message);
-  }
+    },
+  };
+  return sendToNodeSSE(params);
+};
+
+// 因子脚本测试助手
+export const sendToNodeFactorAiTest = async ({ baseUrl, token, value, callback }) => {
+  const params = {
+    baseUrl,
+    path: "/middleLayer/ai/factorAiTest",
+    token,
+    value,
+    origin: "web",
+    getSSE: (type, str) => {
+      switch (type) {
+        case "message": {
+          try {
+            const data = JSON.parse(str);
+            // 结果输出
+            if (data.output) {
+              callback("message", data.output?.content);
+            }
+            // 错误处理
+            if (data.error) {
+              callback("error", data.error);
+            }
+          } catch (error) {}
+          break;
+        }
+        case "finish": {
+          callback("finish");
+          break;
+        }
+        case "error": {
+          callback("error", str);
+          break;
+        }
+      }
+    },
+  };
+  return sendToNodeSSE(params);
 };

package/lib/bundle.js

@@ -1,4 +1,4 @@
-import { i as m } from "./index-2dd72fda.js";
+import { i as m } from "./index-15c834c8.js";
 import "vue";
 import "echarts";
 export {