0

src/api/yjbdc/yjbdcController.js

@@ -95,12 +95,13 @@ export async function GenerateArticle(ctx) {
             }
 
             //给用户一些比较好的体验
-            if (params.AIVersion == "1.0") {
-                aiProvider = 'ali-Moonshot-Kimi-K2-Instruct';
-            }
-            else if (params.AIVersion == "1.5") {
-                aiProvider = 'doubao-kimi-k2-250711';
-            }
+            
+            // 按照权重概率分配AI提供商：40%概率使用ali-Moonshot-Kimi-K2-Instruct，60%概率使用doubao-kimi-k2-250711
+            aiProvider = stringUtils.weightedRandom({
+                'ali-Moonshot-Kimi-K2-Instruct': 40,
+                'doubao-kimi-k2-250711': 60
+            });
+           
 
             try {
                 //开始时间

src/util/stringClass.js

@@ -1623,5 +1623,37 @@ export const stringUtils = {
         }
         
         return false;
+    },
+
+    /**
+     * 按照权重概率分配返回字符串
+     * @param {Object} options - 包含字符串和对应权重的对象
+     * @example
+     * // 返回 'ali-Moonshot-Kimi-K2-Instruct' 的概率为 40%，返回 'doubao-kimi-k2-250711' 的概率为 60%
+     * stringUtils.weightedRandom({
+     *   'ali-Moonshot-Kimi-K2-Instruct': 40,
+     *   'doubao-kimi-k2-250711': 60
+     * });
+     * @returns {string} - 根据权重随机选择的字符串
+     */
+    weightedRandom(options) {
+        // 计算权重总和
+        const weights = Object.values(options);
+        const totalWeight = weights.reduce((sum, weight) => sum + weight, 0);
+        
+        // 生成一个随机数，范围是 [0, totalWeight)
+        const random = Math.random() * totalWeight;
+        
+        // 根据随机数和权重分布选择结果
+        let cumulativeWeight = 0;
+        for (const [item, weight] of Object.entries(options)) {
+            cumulativeWeight += weight;
+            if (random < cumulativeWeight) {
+                return item;
+            }
+        }
+        
+        // 如果出现意外情况，返回第一个选项
+        return Object.keys(options)[0];
     }
 }