Fix/node change price (#157)

* 修复节点信息切换不加载价格问题

* 修复模型定价 改为不只是更改模型才会更改定价

* 添加版本号

Author: topfunplusnew

bizyui/js/handle_load_nodes.js

@@ -5,9 +5,9 @@ import { addPriceBadgeToNode, hasModelInput } from "./model_price.js";
 const BIZYAIR_MODEL_TYPE_KEY = "bizyair_model_type";
 
 // 获取node 的模型配置input
-export async function applyBadgeToNode(_this) {
-  // 优先从节点属性中恢复模型类型（用于工作流加载场景）
-  if (_this.properties && _this.properties[BIZYAIR_MODEL_TYPE_KEY]) {
+export async function applyBadgeToNode(_this, forceRefresh = false) {
+  // 如果不是强制刷新，优先从节点属性中恢复模型类型（用于工作流加载场景）
+  if (!forceRefresh && _this.properties && _this.properties[BIZYAIR_MODEL_TYPE_KEY]) {
     const modelType = _this.properties[BIZYAIR_MODEL_TYPE_KEY];
     await addPriceBadgeToNode(_this, modelType);
     return;
@@ -26,8 +26,8 @@ export async function applyBadgeToNode(_this) {
       return;
     }
 
-    // 解析模型类型
-    const modelType = getModelTypeFromHiddenInput(hiddenOutput);
+    // 解析模型类型（传入节点对象以获取用户选择的model）
+    const modelType = getModelTypeFromHiddenInput(hiddenOutput, _this);
 
     // 将hiddenOutput临时存储在节点上，方便后续使用
     _this._bizyairHiddenOutput = hiddenOutput;
@@ -48,10 +48,30 @@ export async function applyBadgeToNode(_this) {
 }
 
 // 从hiddenInput中获取模型类型
-function getModelTypeFromHiddenInput(hiddenOutput){
+function getModelTypeFromHiddenInput(hiddenOutput, node){
   const modelJson = JSON.parse(hiddenOutput.type);
+  
+  // 从节点的widgets中获取用户选择的model
+  let selectedModel = null;
+  const possibleWidgetNames = ["model", "model_name"];
+  
+  if (node.widgets && Array.isArray(node.widgets)) {
+    for (const widget of node.widgets) {
+      if (possibleWidgetNames.includes(widget.name) && widget.value) {
+        selectedModel = widget.value;
+        break;
+      }
+    }
+  }
+  
+  // 如果找到了用户选择的model，使用它作为key
+  if (selectedModel && modelJson[selectedModel]) {
+    return modelJson[selectedModel];
+  }
+  
+  // 如果没找到或key不存在，回退到使用第一个键（兼容旧逻辑）
   const modelsList = Object.keys(modelJson);
-  return modelJson[modelsList[0]]
+  return modelJson[modelsList[0]];
 }
 
 // 自定义节点创建处理函数

bizyui/js/handle_node_configure.js

@@ -3,16 +3,16 @@ import { applyBadgeToNode } from "./handle_load_nodes.js";
 app.registerExtension({
   name: "bizyair.handle.node.configure",
   nodeCreated(node, app) {
-    const possibleWidgetNames = ["model_name", "model"];
+    // const possibleWidgetNames = ["model_name", "model"];
     // 在这里可以拿到变化之后的值，并且也可以拿到node，这时候给node切换badge即可
     if (node && node.widgets && Array.isArray(node.widgets)) {
       // 只有model选择的widget才注册callback函数
+      // 不仅仅是切换model才会修改模型定价，比如切换输入参数也会修改模型定价
       node.widgets.forEach((widget) => {
-        if (possibleWidgetNames.includes(widget.name)) {
-          widget.callback = async function () {
-            await applyBadgeToNode(node);
-          };
-        }
+        widget.callback = async function () {
+          // 用户手动修改widget时，强制刷新badge（不使用缓存的模型类型）
+          await applyBadgeToNode(node, true);
+        };
       });
     }
   },

bizyui/js/model_price.js

@@ -146,14 +146,12 @@ export function getBadgeConfigByCoinType(coinType, priceText) {
 
 /**
  * 为节点添加价格徽章
- * TODO 重新添加价格徽章的时候 节点元素不能重新render
  * @param {Object} node - 节点对象，包含 widgets 和其他节点信息
+ * @param {string} modelName - 模型名称
  */
 export async function addPriceBadgeToNode(node, modelName = "") {
   try {
-    console.log(`[modelPrice] addPriceBadgeToNode`, node);
-    // 从节点 widgets 中提取价格计算所需的数据 这个也可以封装一个函数
-    // NODE 上的type 考虑是否也封装一个函数
+    // 从节点 widgets 中提取价格计算所需的数据
     const nodeInputs = {};
     if (node.widgets && Array.isArray(node.widgets)) {
       node.widgets.forEach((widget) => {
@@ -177,11 +175,13 @@ export async function addPriceBadgeToNode(node, modelName = "") {
         priceResult.data.coin_type,
         priceResult.data.result
       );
+      
       // 添加价格徽章
       addCustomBadge(node, badgeConfig);
     }
   } catch (error) {
     console.error("添加价格徽章失败:", error);
+    
     // 添加错误徽章
     addCustomBadge(node, {
       text: "价格获取失败",
@@ -219,6 +219,7 @@ export async function fetchNodePrice(model, nodeInputs) {
     if (!nodeInputs) {
       throw new Error("节点输入信息为空，无法获取价格");
     }
+    
     // 调用bizyengine价格查询接口
     const response = await fetch(
       `/bizyair/trd_api_pricing?model=${encodeURIComponent(model)}`,
@@ -230,7 +231,7 @@ export async function fetchNodePrice(model, nodeInputs) {
         body: JSON.stringify(nodeInputs),
       }
     );
-
+    
     if (!response.ok) {
       throw new Error(`HTTP error! status: ${response.status}`);
     }

version.txt

@@ -1 +1 @@
-1.2.84
+1.2.85