@comate/zulu 0.0.12 → 0.0.13

package/comate-engine/plugins/devaux/dist/index.js

@@ -49336,6 +49336,207 @@ class mrd2CodeProvider extends SkillProvider {
     }
 }
 
+class tangoProvider extends SkillProvider {
+    static{
+        this.skillName = 'tango';
+    }
+    static{
+        this.displayName = 'tango';
+    }
+    static{
+        this.description = 'tango';
+    }
+    static{
+        this.parameters = {
+            type: 'object',
+            properties: {}
+        };
+    }
+    constructor(init){
+        super(init);
+        this.git = new GitUtil(init.cwd);
+        this.cwd = init.cwd;
+    }
+    // eslint-disable-next-line
+    async *execute() {
+        const stream = new ElementChunkStream();
+        const user = await this.currentUser.requestDetail();
+        const userName = user ? user.name : 'unknown';
+        let isSuccess = false;
+        // wbox平台注册本次扫描任务
+        const wboxTaskId = await wboxRegister(user, 'tango', this.currentContext.query, this.git);
+        try {
+            const moduleName = await this.git.getModuleName(this.cwd);
+            const kirinTaskID = await startKirinAsyncStrategy(kirinCodeModificationAsynURL, {
+                parameter: JSON.stringify({
+                    requirement: 'tango_gen_main\n' + this.currentContext.query,
+                    // eslint-disable-next-line
+                    conf_file: '',
+                    conf: '',
+                    scene: 'tango_gen_main',
+                    // eslint-disable-next-line
+                    rd: userName,
+                    moduleName: await this.git.getModuleName()
+                }),
+                name: kirinUserName,
+                token: kirinToken
+            });
+            if (!kirinTaskID) {
+                const failMessage = renderTemplate(taskExecuteErrorInfo, {
+                    groupID: userServiceHiGroupNumber,
+                    taskID: kirinTaskID
+                });
+                yield stream.flushReplaceLast(/*#__PURE__*/ jsx("p", {
+                    children: failMessage + 'error: kirinTaskID is null'
+                }));
+                return;
+            }
+            const { status, kirinResponse, message } = await getKirinAsyncStrategyResult(kirinTaskID, 600 * 1000);
+            if (status === FeedStrategyStatus.ERROR_STRATEGY_FAIL) {
+                const failMessage = renderTemplate(taskExecuteErrorInfo, {
+                    groupID: userServiceHiGroupNumber,
+                    taskID: kirinTaskID
+                });
+                yield stream.flush(/*#__PURE__*/ jsx("p", {
+                    children: failMessage + 'fail reason:' + message
+                }));
+                return;
+            } else if (status === FeedStrategyStatus.ERROR_TIMEOUT) {
+                const failMessage = renderTemplate(taskExecuteTimeoutErroInfo, {
+                    groupID: userServiceHiGroupNumber,
+                    taskID: kirinTaskID
+                });
+                yield stream.flush(/*#__PURE__*/ jsx("p", {
+                    children: failMessage + 'timeout'
+                }));
+                return;
+            }
+            if (kirinResponse.length === 0) {
+                const failMessage = renderTemplate(taskExecuteErrorInfo, {
+                    groupID: userServiceHiGroupNumber,
+                    taskID: kirinTaskID
+                });
+                yield stream.flushReplaceLast(/*#__PURE__*/ jsx("p", {
+                    children: failMessage + 'error: kirinResponse is empty'
+                }));
+                return;
+            }
+            let chunkStr = '';
+            let result = [];
+            // eslint-disable-next-line
+            for (const [index, kirinRes] of kirinResponse.entries()){
+                let singleChunkStr = '';
+                let modelReader = null;
+                const models = kirinRes.models;
+                if (!models || !models.url) {
+                    if (kirinRes.add_before && kirinRes.add_before.length !== 0) {
+                        let { elemResult } = await parsePathLists(kirinRes.add_before, this.cwd, kirinRes.file_path);
+                        result = result.concat(elemResult);
+                        yield stream.flush(/*#__PURE__*/ jsx("markdown", {
+                            children: kirinRes.add_before
+                        }));
+                    }
+                    continue;
+                } else if (models && (models.url.includes('deepseek') || models.url.includes('10.55.119.216') || models.url.includes('volces.com') || models.url.includes('aliyun') || models.url.includes('https://qianfan.baidubce.com/v2/chat/completions'))) {
+                    modelReader = await fetchStreamUrl(models.url, models.headers, models.data, models.method);
+                } else if (models && models.url) {
+                    modelReader = await fetchStreamUrl(`${models.url}?access_token=${models.params.access_token}`, commonHeadersConfig, models.data, models.method);
+                }
+                if (!modelReader) {
+                    const failMessage = renderTemplate(taskExecuteErrorInfo, {
+                        groupID: userServiceHiGroupNumber,
+                        taskID: kirinTaskID
+                    });
+                    yield stream.flushReplaceLast(/*#__PURE__*/ jsx("p", {
+                        children: failMessage + 'error: modelReader is null'
+                    }));
+                    return;
+                }
+                if (kirinRes.add_before && kirinRes.add_before.length !== 0) {
+                    result.push(/*#__PURE__*/ jsx("markdown", {
+                        children: kirinRes.add_before
+                    }));
+                    yield stream.flush(/*#__PURE__*/ jsx("p", {
+                        children: kirinRes.add_before
+                    }));
+                }
+                try {
+                    let modelType = ModelType.WENXIN;
+                    if (models.url.includes('deepseek') || models.url.includes('10.55.119.216') || models.url.includes('volces.com') || models.url.includes('aliyun') || models.url.includes('https://qianfan.baidubce.com/v2/chat/completions')) {
+                        modelType = ModelType.DEEPSEEK;
+                    } else {
+                        modelType = ModelType.WENXIN;
+                    }
+                    let isFirstFlush = true;
+                    for await (const data of getReaderStr(modelReader, modelType)){
+                        if (data.message !== null) {
+                            chunkStr += data.message;
+                            singleChunkStr += data.message;
+                        }
+                        if (isFirstFlush) {
+                            yield stream.flush(/*#__PURE__*/ jsx("markdown", {
+                                children: singleChunkStr
+                            }));
+                            isFirstFlush = false;
+                        } else {
+                            yield stream.flushReplaceLast(/*#__PURE__*/ jsx("markdown", {
+                                children: singleChunkStr
+                            }));
+                        }
+                    }
+                    let { elemResult } = await parsePathLists(singleChunkStr, this.cwd, kirinRes.file_path);
+                    result = result.concat(elemResult);
+                    yield stream.flush(/*#__PURE__*/ jsx("br", {}));
+                } catch (error) {
+                    const failMessage = renderTemplate(taskExecuteErrorInfo, {
+                        groupID: userServiceHiGroupNumber,
+                        taskID: kirinTaskID
+                    });
+                    // eslint-disable-next-line
+                    yield stream.flush(/*#__PURE__*/ jsx("p", {
+                        children: failMessage + ' error:' + error
+                    }));
+                } finally{
+                    modelReader.releaseLock();
+                    await flushRes2Qadc([
+                        {
+                            kirin_id: kirinTaskID,
+                            order_id: `${kirinTaskID}_${index}`,
+                            model: JSON.stringify(kirinRes.models),
+                            add_before: kirinRes.add_before,
+                            add_after: kirinRes.add_after,
+                            result: singleChunkStr,
+                            rd_name: user ? user.name : 'unknown',
+                            module: moduleName
+                        }
+                    ], 'code_gen_record');
+                    if (kirinRes.add_after && kirinRes.add_after.length !== 0) {
+                        result.push(/*#__PURE__*/ jsx("markdown", {
+                            children: kirinRes.add_after
+                        }));
+                        yield stream.flush(/*#__PURE__*/ jsx("markdown", {
+                            children: kirinRes.add_after
+                        }));
+                    }
+                }
+            }
+            yield stream.flushReplaceAll(result);
+            isSuccess = true;
+        } // eslint-disable-next-line
+        catch (e) {
+            yield stream.flush(/*#__PURE__*/ jsx("p", {
+                children: 'exception:' + e
+            }));
+        } finally{
+            await wboxCallBack({
+                taskId: wboxTaskId ? wboxTaskId : 0,
+                status: isSuccess ? 2 : 1,
+                caller: userName
+            });
+        }
+    }
+}
+
 function mapToJson(mapObj) {
     const obj = {};
     for (let [k, v] of mapObj){
@@ -50481,6 +50682,8 @@ async function* getKirinSkillRecogRes(provider, extendsArgs, scene = 'fallback')
         }
         if (kirinResponse.providerOnly) {
             switch(kirinResponse.provider){
+                case 'tangoProvider':
+                    return new tangoProvider(provider);
                 case 'HarmonyGen':
                     return new HarmonyGen(provider);
                 case 'OPGeneratorProvider':