st-comp 0.0.270 → 0.0.271

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "st-comp",
   "public": true,
-  "version": "0.0.270",
+  "version": "0.0.271",
   "type": "module",
   "scripts": {
     "dev": "vite",

package/packages/VarietyAiHelper/index.vue

@@ -209,7 +209,6 @@ const sendMessage = async () => {
           scrollToBottom();
           return;
         }
-
         if (type === "node") {
           if (!firstNodeReceived) {
             firstNodeReceived = true;
@@ -229,7 +228,6 @@ const sendMessage = async () => {
           scrollToBottom();
           return;
         }
-
         if (type === "message") {
           fullResponse = data;
           const lastMessage = messageList.value[messageList.value.length - 1];
@@ -238,7 +236,6 @@ const sendMessage = async () => {
             scrollToBottom();
           }
         }
-
         if (type === "finish") {
           isSending.value = false;
           showFinalResult.value = true;

package/packages/VarietySearch/components/FactorScreen/index.vue

@@ -1,13 +1,14 @@
 <!-- 因子筛选组件 -->
 <script setup name="FactorScreen">
 import { nextTick, ref, watch, inject, reactive } from "vue";
+import { getToken } from "st-func";
 import { Close, Plus, CircleCloseFilled, InfoFilled, Document } from "@element-plus/icons-vue";
 import { handleVerifyScore, extractConditionDetails, extractVariables } from "./tools.js";
-import { sendToBaiLianAppStreaming } from "../../../../public/aiTools";
+import { sendToNodeFactorAiTest } from "../../../../public/aiTools";
 import FactorDescription from "./FactorDescription.vue";
 import MonacoEditor from "../../../MonacoEditor/index.vue";
 
-const { request } = inject("stConfig"); // 组件库全局配置
+const { request, env } = inject("stConfig"); // 组件库全局配置
 
 const emit = defineEmits(["actionState"]);
 const props = defineProps({
@@ -41,6 +42,7 @@ const factorType = ref("脚本");
 const scriptTestLoading = ref(false);
 const scriptCopyLoading = ref(false);
 const scriptAILoading = ref(false);
+const scriptAIButtonLoading = ref(false);
 const testVariety = ref("");
 const testVarietyVisible = ref(false);
 const scriptTestLogVisible = ref(false);
@@ -392,29 +394,37 @@ const handleScriptTest = async (action) => {
 // 测试: AI分析
 const handleTestAi = () => {
   emit("actionState", "因子筛选", "AI分析");
-  const appId = "87be17ecc84d4d2ea3108d7155bec2ac";
-  const apiKey = "sk-d995eb26a4334bdeb2ccb4cbfaf51de8";
   scriptAILoading.value = true;
+  scriptAIButtonLoading.value = true;
   scriptTestResult.ai = "";
-  let fullResponse = "";
-  sendToBaiLianAppStreaming({
-    appId,
-    apiKey,
-    value: scriptTestResult.code,
-    callback: (type, data) => {
-      // 百炼应用错误
-      if (type === "error") return;
-      // 流式输出(正常)
-      else if (type === "message") {
-        fullResponse += data;
-        scriptTestResult.ai = fullResponse;
-      }
-      // 流式输出(完毕)
-      else if (type === "finish") {
-        scriptAILoading.value = false;
-      }
-    },
-  });
+  try {
+    sendToNodeFactorAiTest({
+      baseUrl: env.VITE_BASE_URL,
+      token: getToken(),
+      value: scriptTestResult.code,
+      callback: (type, data) => {
+        if (type === "error") {
+          scriptAILoading.value = false;
+          scriptAIButtonLoading.value = false;
+          ElMessage.error("AI分析失败: " + data);
+          return;
+        }
+        if (type === "message") {
+          scriptAILoading.value = false;
+          scriptTestResult.ai += data;
+        }
+        if (type === "finish") {
+          scriptAILoading.value = false;
+          scriptAIButtonLoading.value = false;
+        }
+      },
+    });
+  } catch (error) {
+    ElMessage.error(`AI响应异常: ${error}`);
+    scriptTestResult.ai = error;
+    scriptAILoading.value = false;
+    scriptAIButtonLoading.value = false;
+  }
 };
 
 // 监控: 窗口开关
@@ -1006,7 +1016,7 @@ watch(
       <div class="dialog-footer">
         <el-button
           type="primary"
-          :loading="scriptAILoading"
+          :loading="scriptAIButtonLoading"
           @click="handleTestAi"
         >
           AI分析
@@ -1137,8 +1147,11 @@ watch(
       word-wrap: break-word;
     }
     .ai {
-      flex: 2;
+      flex: 1;
       background-color: white;
+      :deep(.vuepress-markdown-body) {
+        padding: 0;
+      }
     }
   }
 }

package/public/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/src/App.vue

@@ -47,7 +47,7 @@ const route = useRoute();
  * @description: 组件库Demo本地
  */
 // const VITE_RUN_ENV = {
-//   VITE_PROJECT_NAME: "动量回测",
+//   VITE_PROJECT_NAME: "迈幻量化",
 //   VITE_PROJECT_CLIENTID: "7",
 //   VITE_BASE_URL: "http://127.0.0.1:7001",
 //   BASE_URL: "/",
@@ -61,7 +61,7 @@ const route = useRoute();
  * @description: 组件库Demo测试
  */
 const VITE_RUN_ENV = {
-  VITE_PROJECT_NAME: "动量回测",
+  VITE_PROJECT_NAME: "迈幻量化",
   VITE_PROJECT_CLIENTID: "7",
   VITE_BASE_URL: "http://192.168.12.38:5173",
   BASE_URL: "/",
@@ -75,7 +75,7 @@ const VITE_RUN_ENV = {
  * @description: 组件库Demo线上
  */
 // const VITE_RUN_ENV = {
-//   VITE_PROJECT_NAME: "动量回测",
+//   VITE_PROJECT_NAME: "迈幻量化",
 //   VITE_PROJECT_CLIENTID: "7",
 //   VITE_BASE_URL: "//investapi.hzyotoy.com",
 //   BASE_URL: "/",