@comate/zulu 0.7.1 → 0.7.3

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

@@ -1,1046 +0,0 @@
-import { SkillProvider, StringChunkStream, TextModel } from '@comate/plugin-host';
-import { readFileSync } from 'fs';
-import { fileURLToPath } from 'node:url';
-import { dirname } from 'path';
-
-class HelpSkillProvider extends SkillProvider {
-    static{
-        this.skillName = 'help';
-    }
-    static{
-        this.description = '插件介绍';
-    }
-    static{
-        this.parameters = {
-            type: 'object',
-            properties: {}
-        };
-    }
-    async *execute() {
-        const stream = new StringChunkStream();
-        const filename = fileURLToPath(import.meta.url);
-        const content = readFileSync(`${dirname(filename)}/help.md`, 'utf8');
-        yield stream.flush(content);
-    }
-}
-
-const API_SERVER_HOST = 'http://youfang-new.baidu-int.com';
-const apiRequest$1 = async (method, path, options = {}, timeout = 5000)=>{
-    try {
-        const response = await fetch(`${API_SERVER_HOST}${path}`, {
-            method,
-            ...options,
-            signal: AbortSignal.timeout(timeout)
-        });
-        if (!response.ok) {
-            throw new Error(`[${response.status}] 请求错误。请求参数：${JSON.stringify({
-                method,
-                path,
-                ...options
-            })}`);
-        }
-        const jsonRes = await response.json();
-        // console.log('------ api response: ', jsonRes);
-        return jsonRes;
-    } catch (err) {
-        // console.log('------ api err: ', err);
-        throw err;
-    }
-};
-const queryKnowledgeConfig = (username, icode, source)=>{
-    return apiRequest$1('GET', `/api/v1/products/centrino/comate/knowledge?icode=${icode}&source=${source}`, {
-        headers: {
-            'X-COMATE-USER': username
-        }
-    });
-};
-const setKnowledgeConfig = (username, body)=>{
-    return apiRequest$1('POST', `/api/v1/products/centrino/comate/knowledge`, {
-        headers: {
-            'X-COMATE-USER': username,
-            'Content-Type': 'application/json'
-        },
-        body: JSON.stringify(body)
-    });
-};
-const createTask = (username, body)=>{
-    return apiRequest$1('POST', `/api/v1/products/centrino/comate/task`, {
-        headers: {
-            'X-COMATE-USER': username,
-            'Content-Type': 'application/json'
-        },
-        body: JSON.stringify(body)
-    });
-};
-const updateTaskStatus = (username, body)=>{
-    return apiRequest$1('POST', `/api/v1/products/centrino/comate/task`, {
-        headers: {
-            'X-COMATE-USER': username,
-            'Content-Type': 'application/json'
-        },
-        body: JSON.stringify(body)
-    });
-};
-const updateTaskKnowledge = (username, body)=>{
-    return apiRequest$1('POST', `/api/v1/products/centrino/comate/task`, {
-        headers: {
-            'X-COMATE-USER': username,
-            'Content-Type': 'application/json'
-        },
-        body: JSON.stringify(body)
-    });
-};
-const getTaskDetail = (username, taskId)=>{
-    return apiRequest$1('GET', `/api/v1/products/centrino/comate/task?task_id=${taskId}`, {
-        headers: {
-            'X-COMATE-USER': username
-        }
-    });
-};
-const addKnowledgeAB = (username, body)=>{
-    return apiRequest$1('POST', `/api/v1/products/centrino/comate/ab/knowledge`, {
-        headers: {
-            'X-COMATE-USER': username,
-            'Content-Type': 'application/json'
-        },
-        body: JSON.stringify(body)
-    }, 600000);
-};
-
-class knowledgeConfigSkillProvider extends SkillProvider {
-    static{
-        this.skillName = 'konwledgeConfig';
-    }
-    static{
-        this.displayName = '知识库配置';
-    }
-    static{
-        this.description = '为当前项目绑定私域知识库ID，或查询绑定状态';
-    }
-    static{
-        this.parameters = {
-            type: 'object',
-            properties: {}
-        };
-    }
-    async *execute() {
-        const userDetail = await this.currentUser.requestDetail() || {
-            name: "",
-            email: 'xxx@baidu.com'
-        };
-        const { activeFilePath, selectedCode, activeFileLanguage, query } = this.currentContext;
-        const stream = new StringChunkStream();
-        // 获取仓库信息
-        const repository = await this.retriever.repositoryInfo();
-        let repositoryUrl;
-        if (repository.versionControl === 'git') {
-            repositoryUrl = repository.repositoryUrl;
-            const matched = repositoryUrl?.match(/baidu\/.*\/.*/);
-            if (matched?.length) {
-                repositoryUrl = matched[0];
-            }
-        }
-        if (!repositoryUrl) {
-            yield stream.flushReplaceLast('无法获取当前项目的仓库地址，可能不是一个iCode仓库或者缺少.git配置目录');
-            return;
-        }
-        try {
-            // 绑定(或更新)私域知识库
-            if (this.isValidUUID(query)) {
-                const response = await setKnowledgeConfig(userDetail.name, {
-                    icode: repositoryUrl,
-                    source: 'comate',
-                    knowledgeId: query,
-                    createUser: userDetail.name
-                });
-                if (!response || !response.knowledgeId) {
-                    yield stream.flushReplaceLast('私域知识绑定失败，请联系管理员查看');
-                } else {
-                    yield stream.flushReplaceLast('私域知识绑定成功。\n\n' + '可输入用例描述信息，进行知识检索结果验证；\n\n' + '可通过指令 **/用例生成** 尝试生成测试用例。');
-                }
-                return;
-            }
-            // 查询当前代码库绑定的知识集ID
-            const knowledgeConfig = await queryKnowledgeConfig(userDetail.name, repositoryUrl, 'comate');
-            if (!knowledgeConfig || !knowledgeConfig.knowledgeId) {
-                yield stream.flushReplaceLast('当前代码库未绑定私域知识，请输入知识集ID进行绑定\n\n若无知识集，可前往 [Comate知识中心]' + '(https://comate.baidu-int.com/cop/knowledge/list)创建并添加知识');
-                return;
-            }
-            if (query.length === 0) {
-                // 查询知识库
-                yield stream.flushReplaceLast(`当前代码库已绑定私域知识：[${knowledgeConfig.knowledgeId}](https://comate.baidu-int.com/cop/knowledge/` + `detail/NORMAL/${knowledgeConfig.knowledgeId})\n\n若无访问权限，请联系业务负责人或管理员`);
-                return;
-            } else {
-                yield stream.flushReplaceLast('私域知识检索中...');
-                //检索知识测试
-                let queryWithCode = query;
-                // 拼接query与选中代码
-                if (selectedCode) {
-                    queryWithCode = `${query}\n\`\`\`${activeFileLanguage}\n${selectedCode}\n\`\`\``;
-                }
-                let queryCodePrompt = `请根据下列用例步骤描述检索出相关的函数或引用\n${query}`;
-                const [knowledge, code] = await Promise.all([
-                    this.retriever.knowledgeFromQuery(queryWithCode, {
-                        knowledgeSets: [
-                            {
-                                uuid: knowledgeConfig.knowledgeId,
-                                type: 'NORMAL'
-                            }
-                        ],
-                        queryRewrite: true
-                    }),
-                    this.retriever.codeFromQuery(queryCodePrompt, {
-                        knowledgeSets: [
-                            {
-                                uuid: knowledgeConfig.knowledgeId,
-                                type: 'NORMAL'
-                            }
-                        ],
-                        queryRewrite: false
-                    })
-                ]);
-                yield stream.flushReplaceLast('只返回top6个知识点\n\n');
-                let index = 1;
-                for(let i = 0; i < knowledge.length; i++){
-                    if (i >= 6) {
-                        break;
-                    }
-                    index = i + 1;
-                    yield stream.flush(`# 知识${index}\n\n`);
-                    yield stream.flush(knowledge[i]);
-                    yield stream.flush('\n\n');
-                }
-            }
-        } catch (error) {
-            yield stream.flushReplaceLast('出现错误，请稍后再试');
-        }
-    }
-    trimKnowledge(list, maxCount = 6, maxLength = 10000) {
-        const state = {
-            length: 0
-        };
-        const output = [];
-        for (const item of list){
-            if (state.length + item.length > maxLength || output.length >= maxCount) {
-                break;
-            }
-            output.push(item);
-            state.length += item.length;
-        }
-        return output;
-    }
-    isValidUUID(uuid) {
-        // UUID v4 regex pattern
-        const uuidPattern = /^[0-9a-fA-F]{8}-[0-9a-fA-F]{4}-4[0-9a-fA-F]{3}-[89aAbB][0-9a-fA-F]{3}-[0-9a-fA-F]{12}$/;
-        // Use .match to check if the UUID matches the pattern
-        return uuid.match(uuidPattern) !== null;
-    }
-}
-
-var promptTemplate = "你是一位资深的{{frame}}自动化测试技术专家。你的核心职责是理解输出代码示例、并根据需解答的问题及回答要求生成一个自动化测试用例代码。在解答问题前，你会收到多个背景知识和一个具体的问题。\n* 在提供的背景知识中，将通过『##』明确标注需要参考的信息类型。例如，`## 编程语言`表示提问者当前使用的编程语言，`## 参考文档片段`表示引用特定的文档内容，`## 输出代码示例`表示需要输出的代码样式示例。准确识别这些信息对于解答问题至关重要。\n\n# 背景知识\n## 编程语言\n{{activeFileLanguage}}\n\n## 参考文档片段\n{{trim_knowledge}}\n\n## 输出代码示例\n{{templateCaseCode}}\n\n# 回答要求\n1. 请参考所提供的背景知识来回答问题，但不要提及你参考了这些背景知识，因为背景知识对提问者是不可见的。\n2. 保持简洁，直接、明确地回应问题，避免复述问题本身。\n3. 回答中如包含代码，应优先使用背景知识中的封装函数来实现。若使用了非背景知识中的函数，请输出相应代码。\n4. 请保证引用包和代码逻辑的语法正确性，并应减少文字解释，必要的解释以代码注释形式给出。\n4. 回答中如包含代码，则请参考示例代码的样式进行输出，且单行不超过120字符。代码应作为完整块展示，避免分割解释。\n5. 回答中如有时间信息，请更新为当前时间；若提及用户名信息，请更新为{{email}}；若用例名中有编号，请递增编号。\n6. 使用Markdown格式编排回答，确保逻辑清晰。对于代码块，请采用标准的Markdown代码块格式，并指定编程语言，例如：\n```python\n# 这是一个Python代码块示例\n```\n\n# 需解答的问题\n请根据如下用例步骤信息生成一个测试用例：\n{{query}}";
-
-class CaseByCaseSkillProvider extends SkillProvider {
-    static{
-        this.skillName = 'genCaseByCase';
-    }
-    static{
-        this.displayName = '用例生成';
-    }
-    static{
-        this.description = '基于用例功能和步骤描述生成用例';
-    }
-    static{
-        this.parameters = {
-            type: 'object',
-            properties: {}
-        };
-    }
-    async *execute() {
-        const userDetail = await this.currentUser.requestDetail() || {
-            name: "",
-            email: 'xxx@baidu.com'
-        };
-        const email = userDetail.email;
-        let { activeFilePath, selectedCode, activeFileLanguage, query } = this.currentContext;
-        query = query.trim();
-        selectedCode = selectedCode.trim();
-        const stream = new StringChunkStream();
-        // 入参校验
-        if (query.length === 0) {
-            yield stream.flush("用例步骤描述不可为空，请输入并重试");
-            return;
-        }
-        if (selectedCode === '') {
-            yield stream.flush('（选中一段用例代码作为示例，有助于统一代码风格）\n\n');
-        // return;
-        }
-        // 获取仓库信息
-        const repository = await this.retriever.repositoryInfo();
-        let repositoryUrl;
-        if (repository.versionControl === 'git') {
-            repositoryUrl = repository.repositoryUrl;
-            const matched = repositoryUrl?.match(/baidu\/.*\/.*/);
-            if (matched?.length) {
-                repositoryUrl = matched[0];
-            }
-        }
-        if (!repositoryUrl) {
-            yield stream.flushReplaceLast('无法获取当前项目的仓库地址，可能不是一个iCode仓库或者缺少.git配置目录');
-            return;
-        }
-        // 打点：初始化
-        let taskInfo = await createTask(userDetail.name, {
-            conversationId: '',
-            icode: repositoryUrl,
-            query: query,
-            selected: selectedCode,
-            knowledge: '',
-            frame: '',
-            language: activeFileLanguage,
-            status: 'init',
-            createUser: userDetail.name
-        });
-        if (!taskInfo || !taskInfo.taskId) {
-            yield stream.flush('打点失败，不影响生成\n\n');
-        }
-        // 查询当前代码库绑定的知识集ID
-        const knowledgeConfig = await queryKnowledgeConfig(userDetail.name, repositoryUrl, 'comate');
-        if (!knowledgeConfig || !knowledgeConfig.knowledgeId) {
-            yield stream.flushReplaceLast('请通过指令 **/知识库** 为当前代码仓库绑定知识集ID。去创建知识：[Comate知识中心](https://comate.baidu-int.com/cop/knowledge/list)');
-            return;
-        }
-        // 框架
-        let frame = knowledgeConfig.frame;
-        if (frame !== '') {
-            frame = frame + '框架';
-        } else {
-            frame = 'pytest框架';
-        }
-        // yield stream.flush('若超时未响应，请尝试重新生成\n\n');
-        // 打点：running
-        if (taskInfo && taskInfo.taskId) {
-            await updateTaskStatus(userDetail.name, {
-                taskId: taskInfo.taskId,
-                status: 'running'
-            });
-        }
-        // 拆解步骤
-        const [queryFunc, querySteps] = this.parseQueryStep(query);
-        let knowledgeList = [];
-        const maxCount = Math.max(Math.floor(10 / querySteps.length), 1);
-        // 记录开始时间
-        const startTime = performance.now();
-        try {
-            yield stream.flush(`检索知识`);
-            for(let i = 0; i < querySteps.length; i++){
-                const retreiveTemplate = '请根据下列描述信息检索出相关的函数或引用，并根据相关性排序，描述如下：';
-                let queryCodePrompt = `${retreiveTemplate}\n${querySteps[i]}`;
-                if (queryFunc.length > 0) {
-                    queryCodePrompt = `${retreiveTemplate}\n${queryFunc}：${querySteps[i]}`;
-                }
-                yield stream.flush(`.`);
-                const [knowledge] = await Promise.all([
-                    this.retriever.knowledgeFromQuery(queryCodePrompt, {
-                        knowledgeSets: [
-                            {
-                                uuid: knowledgeConfig.knowledgeId,
-                                type: 'NORMAL'
-                            }
-                        ],
-                        queryRewrite: true
-                    })
-                ]);
-                knowledgeList = this.updateKnowledgeList(knowledgeList, knowledge, maxCount);
-            }
-            yield stream.flush(`\n\n检索完成\n\n`);
-            // 记录结束时间
-            const retriveTime = performance.now();
-            // 计算执行时长
-            let duration = retriveTime - startTime;
-            const trim_knowledge = this.trimKnowledge(knowledgeList, 9);
-            // const trim_code = this.trimKnowledge(code, 5, 3000);
-            const trim_code = '';
-            let templateCaseCode = `\`\`\`${activeFileLanguage}\n${selectedCode}\n\`\`\``;
-            const prompt = this.llm.createPrompt(promptTemplate, {
-                frame,
-                activeFileLanguage,
-                trim_knowledge: trim_knowledge.join('\n\n'),
-                templateCaseCode,
-                email,
-                query
-            });
-            const chunks = this.llm.askForTextStreaming(prompt, {
-                model: TextModel.Default
-            });
-            yield stream.flushReplaceLast('\n\n');
-            for await (const chunk of chunks){
-                yield stream.flush(chunk);
-            }
-            // 记录结束时间
-            const endTime = performance.now();
-            // 计算执行时长
-            duration = endTime - retriveTime;
-            // 打点：success
-            if (taskInfo && taskInfo.taskId) {
-                let response = await updateTaskStatus(userDetail.name, {
-                    taskId: taskInfo.taskId,
-                    status: 'success'
-                });
-            }
-        } catch (error) {
-            yield stream.flushReplaceLast('出现错误，请稍后再试');
-            // 打点：fail
-            if (taskInfo && taskInfo.taskId) {
-                await updateTaskStatus(userDetail.name, {
-                    taskId: taskInfo.taskId,
-                    status: 'fail'
-                });
-            }
-        }
-    }
-    trimKnowledge(list, maxCount = 6, maxLength = 10000) {
-        const state = {
-            length: 0
-        };
-        const output = [];
-        for (const item of list){
-            if (state.length + item.length > maxLength || output.length >= maxCount) {
-                break;
-            }
-            output.push(item);
-            state.length += item.length;
-        }
-        return output;
-    }
-    parseQueryStep(queryRaw) {
-        if (queryRaw === '') {
-            return [
-                '',
-                []
-            ];
-        }
-        let queryFuncDesc = '';
-        const list = queryRaw.split('\n');
-        if (!queryRaw.startsWith('1')) {
-            queryFuncDesc = list[0];
-            list.splice(0, 1);
-        // if (!queryFuncDesc.includes('步骤')) {
-        //     queryFuncDesc = queryFuncDesc + '，步骤描述如下：';
-        // }
-        }
-        for(let i = 0; i < list.length; i++){
-            list[i] = list[i].trim();
-        }
-        return [
-            queryFuncDesc,
-            list
-        ];
-    }
-    updateKnowledgeList(knowledgeList, knowledge, maxCount) {
-        // 使用 Set 去除 knowledgeList 中的重复项并方便快速查找
-        const knowledgeSet = new Set(knowledgeList);
-        // 新知识去重
-        const newKnowledgeSet = new Set(knowledge);
-        const newKnowledge = [
-            ...newKnowledgeSet
-        ];
-        // 过滤掉已经在 knowledgeList 中的新知识，并限制最大数量为 maxCount
-        const newKnowledgeList = newKnowledge.filter((k)=>!knowledgeSet.has(k)).slice(0, maxCount);
-        // 将新知识合并到旧知识列表中
-        return [
-            ...knowledgeList,
-            ...newKnowledgeList
-        ];
-    }
-}
-
-const AB_AUTH = 'Bearer bce-v3/ALTAK-wPXaAeVTpgBy9uLMFHtII/6860c1ecfe6aa6c6366e374057f51bf5b5400dd4';
-const AB_KNOWLEDGE_APP_ID = '8c0358d9-8854-4e53-96a2-5bf8bf966913';
-const AB_CASE_APP_ID = 'c3d7d0ad-39cc-4ca1-8912-8ad2f3b1cf49';
-
-function createParser(onParse) {
-  let isFirstChunk, buffer, startingPosition, startingFieldLength, eventId, eventName, data;
-  return reset(), { feed, reset };
-  function reset() {
-    isFirstChunk = !0, buffer = "", startingPosition = 0, startingFieldLength = -1, eventId = void 0, eventName = void 0, data = "";
-  }
-  function feed(chunk) {
-    buffer = buffer ? buffer + chunk : chunk, isFirstChunk && hasBom(buffer) && (buffer = buffer.slice(BOM.length)), isFirstChunk = !1;
-    const length = buffer.length;
-    let position = 0, discardTrailingNewline = !1;
-    for (; position < length; ) {
-      discardTrailingNewline && (buffer[position] === `
-` && ++position, discardTrailingNewline = !1);
-      let lineLength = -1, fieldLength = startingFieldLength, character;
-      for (let index = startingPosition; lineLength < 0 && index < length; ++index)
-        character = buffer[index], character === ":" && fieldLength < 0 ? fieldLength = index - position : character === "\r" ? (discardTrailingNewline = !0, lineLength = index - position) : character === `
-` && (lineLength = index - position);
-      if (lineLength < 0) {
-        startingPosition = length - position, startingFieldLength = fieldLength;
-        break;
-      } else
-        startingPosition = 0, startingFieldLength = -1;
-      parseEventStreamLine(buffer, position, fieldLength, lineLength), position += lineLength + 1;
-    }
-    position === length ? buffer = "" : position > 0 && (buffer = buffer.slice(position));
-  }
-  function parseEventStreamLine(lineBuffer, index, fieldLength, lineLength) {
-    if (lineLength === 0) {
-      data.length > 0 && (onParse({
-        type: "event",
-        id: eventId,
-        event: eventName || void 0,
-        data: data.slice(0, -1)
-        // remove trailing newline
-      }), data = "", eventId = void 0), eventName = void 0;
-      return;
-    }
-    const noValue = fieldLength < 0, field = lineBuffer.slice(index, index + (noValue ? lineLength : fieldLength));
-    let step = 0;
-    noValue ? step = lineLength : lineBuffer[index + fieldLength + 1] === " " ? step = fieldLength + 2 : step = fieldLength + 1;
-    const position = index + step, valueLength = lineLength - step, value = lineBuffer.slice(position, position + valueLength).toString();
-    if (field === "data")
-      data += value ? `${value}
-` : `
-`;
-    else if (field === "event")
-      eventName = value;
-    else if (field === "id" && !value.includes("\0"))
-      eventId = value;
-    else if (field === "retry") {
-      const retry = parseInt(value, 10);
-      Number.isNaN(retry) || onParse({ type: "reconnect-interval", value: retry });
-    }
-  }
-}
-const BOM = [239, 187, 191];
-function hasBom(buffer) {
-  return BOM.every((charCode, index) => buffer.charCodeAt(index) === charCode);
-}
-
-class EventSourceParserStream extends TransformStream {
-  constructor() {
-    let parser;
-    super({
-      start(controller) {
-        parser = createParser((event) => {
-          event.type === "event" && controller.enqueue(event);
-        });
-      },
-      transform(chunk) {
-        parser.feed(chunk);
-      }
-    });
-  }
-}
-
-const AB_SERVER_HOST = 'https://qianfan.baidubce.com';
-const apiRequest = async (method, path, options = {}, timeout = 20000)=>{
-    try {
-        const response = await fetch(`${AB_SERVER_HOST}${path}`, {
-            method,
-            ...options,
-            signal: AbortSignal.timeout(timeout)
-        });
-        if (!response.ok) {
-            throw new Error(`[${response.status}] 请求错误。请求参数：${JSON.stringify({
-                method,
-                path,
-                ...options
-            })}`);
-        }
-        const jsonRes = await response.json();
-        // console.log('------ api response: ', jsonRes);
-        return jsonRes;
-    } catch (err) {
-        // console.error('------ api error: ', err);
-        throw err;
-    }
-};
-async function* abSteamRequest(method, path, options = {}) {
-    try {
-        const response = await fetch(`${AB_SERVER_HOST}${path}`, {
-            method,
-            headers: {
-                'Content-Type': 'application/json'
-            },
-            ...options
-        });
-        if (!response.ok) {
-            throw new Error(`[${response.status}] 请求错误。请求参数：${JSON.stringify({
-                method,
-                path,
-                ...options
-            })}`);
-        }
-        if (!response.body) {
-            throw new Error(`[${response.status}] 响应体错误。请求参数：${JSON.stringify({
-                method,
-                path,
-                ...options
-            })}`);
-        }
-        // 兼容非流式返回的处理逻辑
-        const contentType = response.headers.get("Content-Type");
-        if (contentType?.indexOf("application/json") !== -1) {
-            const jsonRes = response.json();
-            yield jsonRes.then((res)=>{
-                if (res.code !== 'OK') {
-                    throw new Error(`[${res.code}] ${res.message}`);
-                } else {
-                    return res.data;
-                }
-            });
-        }
-        const resBody = response.body.pipeThrough(new TextDecoderStream()).pipeThrough(new EventSourceParserStream());
-        const reader = resBody.getReader();
-        try {
-            while(true){
-                const { value, done } = await reader.read();
-                if (done) {
-                    break;
-                }
-                const text = value?.type === 'event' ? JSON.parse(value.data).answer : undefined;
-                if (!text) {
-                    continue;
-                }
-                yield text;
-            }
-        } catch (e) {
-            throw e;
-        } finally{
-            reader.releaseLock();
-        }
-    } catch (err) {
-        throw err;
-    }
-}
-const initABChat = (auth, body)=>{
-    return apiRequest('POST', `/v2/app/conversation`, {
-        headers: {
-            'Content-Type': 'application/json',
-            'X-Appbuilder-Authorization': auth
-        },
-        body: JSON.stringify(body)
-    });
-};
-const runABChat = (auth, body)=>{
-    body.query = JSON.stringify(body.query);
-    return apiRequest('POST', `/v2/app/conversation/runs`, {
-        headers: {
-            'Content-Type': 'application/json',
-            'X-Appbuilder-Authorization': auth
-        },
-        body: JSON.stringify(body)
-    });
-};
-const runABChatStream = (auth, body)=>{
-    console.log('runABChatStream body:', body);
-    const res = abSteamRequest('POST', `/v2/app/conversation/runs`, {
-        headers: {
-            'Content-Type': 'application/json',
-            'X-Appbuilder-Authorization': auth
-        },
-        body: JSON.stringify(body)
-    });
-    return res;
-};
-
-class CaseABSkillProvider extends SkillProvider {
-    static{
-        this.skillName = 'genCaseAB';
-    }
-    static{
-        this.displayName = '用例生成New';
-    }
-    static{
-        this.description = '基于用例功能和步骤描述生成用例';
-    }
-    static{
-        this.parameters = {
-            type: 'object',
-            properties: {}
-        };
-    }
-    async *execute() {
-        const userDetail = await this.currentUser.requestDetail() || {
-            name: "",
-            email: 'xxx@baidu.com'
-        };
-        userDetail.email;
-        let { selectedCode, activeFileLanguage, query } = this.currentContext;
-        query = query.trim();
-        selectedCode = selectedCode.trim();
-        const stream = new StringChunkStream();
-        // 入参校验
-        if (query.length === 0) {
-            yield stream.flush("用例步骤描述不可为空，请输入并重试");
-            return;
-        }
-        if (selectedCode === '') {
-            yield stream.flush('（选中一段用例代码作为示例，有助于统一代码风格）\n\n');
-        // return;
-        }
-        // 回放模式
-        try {
-            if (Number(query) > 0) {
-                yield stream.flushReplaceLast(' **进入了回放模式（debug）** \n\n');
-                let taskDetail = await getTaskDetail(userDetail.name, query);
-                if (!taskDetail || !taskDetail.conversationId) {
-                    yield stream.flush(`获取任务记录失败，任务ID:${query}\n\n`);
-                    return;
-                }
-                // 生成对话
-                const caseChunks = runABChatStream(AB_AUTH, {
-                    app_id: AB_CASE_APP_ID,
-                    conversation_id: taskDetail?.conversationId,
-                    query: query,
-                    stream: true
-                });
-                yield stream.flush('将基于历史任务记录，直接进入LLM用例生成节点\n\n');
-                for await (const chunk of caseChunks){
-                    yield stream.flush(chunk);
-                }
-                yield stream.flush('\n\n');
-                return;
-            }
-        } catch (error) {
-            yield stream.flush('回放失败，请确认任务ID是否存在');
-            return;
-        }
-        yield stream.flush('知识库配置检查•••');
-        // 获取仓库信息
-        const repository = await this.retriever.repositoryInfo();
-        let repositoryUrl;
-        if (repository.versionControl === 'git') {
-            repositoryUrl = repository.repositoryUrl;
-            const matched = repositoryUrl?.match(/baidu\/.*\/.*/);
-            if (matched?.length) {
-                repositoryUrl = matched[0];
-            }
-        }
-        if (!repositoryUrl) {
-            yield stream.flushReplaceLast('无法获取当前项目的仓库地址，可能不是一个iCode仓库或者缺少.git配置目录');
-            return;
-        }
-        // 打点：init
-        let taskInfo = await createTask(userDetail.name, {
-            conversationId: '',
-            icode: repositoryUrl,
-            query: query,
-            selected: selectedCode,
-            knowledge: '',
-            frame: '',
-            language: activeFileLanguage,
-            status: 'init',
-            createUser: userDetail.name
-        });
-        if (!taskInfo || !taskInfo.taskId) {
-            yield stream.flush('创建任务失败\n\n');
-            return;
-        }
-        // 查询当前代码库绑定的知识集ID
-        const knowledgeConfig = await queryKnowledgeConfig(userDetail.name, repositoryUrl, 'ab');
-        if (!knowledgeConfig || !knowledgeConfig.knowledgeId) {
-            yield stream.flushReplaceLast('\n\n ** 请联系管理员为当前库添加知识库配置，用于添加私域知识 ** \n\n');
-            return;
-        } else {
-            yield stream.flush('done\n\n');
-        }
-        // 框架
-        let frame = knowledgeConfig.frame;
-        if (frame !== '' && frame !== 'undefined') {
-            frame = frame + '框架';
-        } else {
-            frame = 'pytest框架';
-        }
-        // 打点：running
-        await updateTaskStatus(userDetail.name, {
-            taskId: taskInfo.taskId,
-            status: 'running'
-        });
-        // 拆解步骤
-        const [queryFunc, querySteps] = this.parseQueryStep(query);
-        let knowledgeList = [];
-        const maxKnowledgeCount = 20;
-        const maxKnowledgeLength = 20000;
-        const maxCount = Math.max(Math.floor(maxKnowledgeCount / querySteps.length), 1);
-        // 记录开始时间
-        const startTime = performance.now();
-        try {
-            // 创建检索会话
-            const chatInfo = await initABChat(AB_AUTH, {
-                app_id: AB_KNOWLEDGE_APP_ID
-            });
-            if (!chatInfo || !chatInfo.conversation_id) {
-                yield stream.flushReplaceLast('创建会话失败，请联系管理员\n\n');
-                return;
-            }
-            yield stream.flush(`检索知识`);
-            for(let i = 0; i < querySteps.length; i++){
-                const retreiveTemplate = '请检索函数或步骤：';
-                let queryCodePrompt = `${retreiveTemplate}\n${querySteps[i]}`;
-                if (queryFunc.length > 0) {
-                    queryCodePrompt = `${retreiveTemplate}：${querySteps[i]}\n${queryFunc}`;
-                }
-                yield stream.flush(`${i + 1}`);
-                // 知识检索对话
-                const chatResp = await runABChat(AB_AUTH, {
-                    app_id: AB_KNOWLEDGE_APP_ID,
-                    conversation_id: chatInfo.conversation_id,
-                    query: {
-                        icode: repositoryUrl,
-                        query: queryCodePrompt,
-                        selected: selectedCode,
-                        username: userDetail.name
-                    },
-                    stream: false
-                });
-                if (chatResp?.answer === undefined || chatResp?.answer === null) {
-                    yield stream.flushReplaceLast('检索知识失败\n\n');
-                    return;
-                }
-                const knowledge = this.parseKnowledge(chatResp.answer);
-                knowledgeList = this.updateKnowledgeList(knowledgeList, knowledge, maxCount);
-                yield stream.flushReplaceLast(`•`);
-            }
-            yield stream.flush(`done\n\n`);
-            // 记录结束时间
-            const retriveTime = performance.now();
-            // 计算执行时长
-            let duration = retriveTime - startTime;
-            if (knowledgeList.length === 0) {
-                yield stream.flushReplaceLast(' **未检索到任何相关知识，无法生成用例** \n\n请确认是否已配置知识库且添加了知识，否则请联系管理员\n\n');
-                return;
-            }
-            const trim_knowledge = this.trimKnowledge(knowledgeList, maxKnowledgeCount, maxKnowledgeLength);
-            // 创建生成会话
-            const caseChatInfo = await initABChat(AB_AUTH, {
-                app_id: AB_CASE_APP_ID
-            });
-            if (!caseChatInfo || !caseChatInfo.conversation_id) {
-                yield stream.flushReplaceLast('创建用例生成会话失败，请联系管理员\n\n');
-                return;
-            }
-            // 打点：记录数据
-            let taskResponse = await updateTaskKnowledge(userDetail.name, {
-                taskId: taskInfo.taskId,
-                conversation_id: caseChatInfo.conversation_id,
-                knowledge: trim_knowledge.join('\n\n'),
-                frame: frame
-            });
-            // 用例对话会通过api获取知识等信息，sleep 0.5s打一个时间差
-            // await this.sleep(500);
-            yield stream.flush('生成中•••\n\n');
-            // 开始用例对话
-            const caseChunks = runABChatStream(AB_AUTH, {
-                app_id: AB_CASE_APP_ID,
-                conversation_id: caseChatInfo.conversation_id,
-                query: String(taskInfo.taskId),
-                stream: true
-            });
-            // 输出stream内容
-            yield stream.flushReplaceLast(`\n\n`);
-            for await (const chunk of caseChunks){
-                yield stream.flush(chunk);
-            }
-            yield stream.flush('\n\n');
-            // 记录结束时间
-            const endTime = performance.now();
-            // 计算执行时长
-            duration = endTime - retriveTime;
-            // 打点：success
-            let response = await updateTaskStatus(userDetail.name, {
-                taskId: taskInfo.taskId,
-                status: 'success'
-            });
-        } catch (error) {
-            yield stream.flushReplaceLast('出现错误，请稍后再试');
-            // 打点：fail
-            await updateTaskStatus(userDetail.name, {
-                taskId: taskInfo.taskId,
-                status: 'fail'
-            });
-        }
-    }
-    trimKnowledge(list, maxCount = 6, maxLength = 20000) {
-        const state = {
-            length: 0
-        };
-        const output = [];
-        for (const item of list){
-            if (state.length + item.length > maxLength || output.length >= maxCount) {
-                break;
-            }
-            output.push(item);
-            state.length += item.length;
-        }
-        return output;
-    }
-    parseQueryStep(queryRaw) {
-        if (queryRaw === '') {
-            return [
-                '',
-                []
-            ];
-        }
-        let queryFuncDesc = '';
-        const list = queryRaw.split('\n');
-        if (!queryRaw.startsWith('1')) {
-            queryFuncDesc = list[0];
-            list.splice(0, 1);
-        // if (!queryFuncDesc.includes('步骤')) {
-        //     queryFuncDesc = queryFuncDesc + '，步骤描述如下：';
-        // }
-        }
-        for(let i = 0; i < list.length; i++){
-            list[i] = list[i].trim();
-        }
-        return [
-            queryFuncDesc,
-            list
-        ];
-    }
-    updateKnowledgeList(knowledgeList, knowledge, maxCount) {
-        // 过滤掉不是函数名和步骤名的知识
-        let rawKnowledge = knowledge.map((str)=>str.trim()).filter((str)=>str.includes("函数名") || str.includes("步骤名"));
-        // 使用 Set 去除 knowledgeList 中的重复项并方便快速查找
-        const knowledgeSet = new Set(knowledgeList);
-        // 新知识去重
-        const newKnowledgeSet = new Set(rawKnowledge);
-        const newKnowledge = [
-            ...newKnowledgeSet
-        ];
-        // 过滤掉已经在 knowledgeList 中的新知识，并限制最大数量为 maxCount
-        const newKnowledgeList = newKnowledge.filter((k)=>!knowledgeSet.has(k)).slice(0, maxCount);
-        // 将新知识合并到旧知识列表中
-        return [
-            ...knowledgeList,
-            ...newKnowledgeList
-        ];
-    }
-    parseKnowledge(answer) {
-        if (answer === '') {
-            return [];
-        }
-        const jsonObject = JSON.parse(answer);
-        if (jsonObject && jsonObject.knowledge) {
-            return jsonObject.knowledge;
-        }
-        return [];
-    }
-    sleep(ms) {
-        return new Promise((resolve)=>setTimeout(resolve, ms));
-    }
-}
-
-class addKnowledgeSkillProvider extends SkillProvider {
-    static{
-        this.skillName = 'addKonwledge';
-    }
-    static{
-        this.displayName = '添加知识';
-    }
-    static{
-        this.description = '将函数或用例代码转化为知识文本，添加到知识库中';
-    }
-    static{
-        this.parameters = {
-            type: 'object',
-            properties: {}
-        };
-    }
-    async *execute() {
-        const userDetail = await this.currentUser.requestDetail() || {
-            name: "",
-            email: 'xxx@baidu.com'
-        };
-        const { activeFilePath, selectedCode, activeFileLanguage, query } = this.currentContext;
-        const stream = new StringChunkStream();
-        if (selectedCode === "" && query === '') {
-            yield stream.flushReplaceLast('请选中或输入一段代码，再重试\n\n代码片段类型：完整的函数片段或一个用例的完整代码');
-            return;
-        }
-        // 获取仓库信息
-        const repository = await this.retriever.repositoryInfo();
-        let repositoryUrl;
-        if (repository.versionControl === 'git') {
-            repositoryUrl = repository.repositoryUrl;
-            const matched = repositoryUrl?.match(/baidu\/.*\/.*/);
-            if (matched?.length) {
-                repositoryUrl = matched[0];
-            }
-        }
-        if (!repositoryUrl) {
-            yield stream.flushReplaceLast('无法获取当前项目的仓库地址，可能不是一个iCode仓库或者缺少.git配置目录');
-            return;
-        }
-        try {
-            // 查询当前代码库绑定的知识集ID
-            const knowledgeConfig = await queryKnowledgeConfig(userDetail.name, repositoryUrl, 'ab');
-            if (!knowledgeConfig || !knowledgeConfig.knowledgeId) {
-                yield stream.flushReplaceLast('当前代码库不支持使用，请联系管理员添加配置');
-                return;
-            }
-            // 文本长度限制
-            const contentLines = selectedCode.split('\n');
-            if (contentLines.length > 500) {
-                yield stream.flushReplaceLast('代码行数过长，请控制在500行以内，再重试');
-            }
-            yield stream.flushReplaceLast('执行时间与代码行数正相关，请耐心等待\n\n');
-            // TODO 调用service接口，添加知识
-            let taskResponse = await addKnowledgeAB(userDetail.name, {
-                knowledgeBaseId: knowledgeConfig.knowledgeId,
-                icode: repositoryUrl,
-                path: activeFilePath,
-                query: query,
-                selected: selectedCode,
-                language: activeFileLanguage,
-                createUser: userDetail.name
-            });
-            yield stream.flushReplaceLast('\n\n');
-            if (taskResponse && taskResponse?.message.length > 0) {
-                yield stream.flush(`${taskResponse?.message}`);
-            }
-            if (taskResponse && taskResponse?.succeedList.length > 0) {
-                yield stream.flush(`添加成功清单：\n\n- ${taskResponse?.succeedList.join('\n\n- ')}`);
-            }
-            yield stream.flush('\n\n');
-            if (taskResponse && taskResponse?.failedList.length > 0) {
-                yield stream.flush(`添加失败清单：\n\n- ${taskResponse?.failedList.join('\n\n- ')}`);
-            }
-        } catch (error) {
-            yield stream.flushReplaceLast('出现错误，请稍后再试');
-        }
-    }
-    trimKnowledge(list, maxCount = 6, maxLength = 10000) {
-        const state = {
-            length: 0
-        };
-        const output = [];
-        for (const item of list){
-            if (state.length + item.length > maxLength || output.length >= maxCount) {
-                break;
-            }
-            output.push(item);
-            state.length += item.length;
-        }
-        return output;
-    }
-    isValidUUID(uuid) {
-        // UUID v4 regex pattern
-        const uuidPattern = /^[0-9a-fA-F]{8}-[0-9a-fA-F]{4}-4[0-9a-fA-F]{3}-[89aAbB][0-9a-fA-F]{3}-[0-9a-fA-F]{12}$/;
-        // Use .match to check if the UUID matches the pattern
-        return uuid.match(uuidPattern) !== null;
-    }
-}
-
-function setup({ registry }) {
-    registry.registerSkillProvider('help', HelpSkillProvider);
-    registry.registerSkillProvider('knowledgeConfig', knowledgeConfigSkillProvider);
-    registry.registerSkillProvider('genCaseByCase', CaseByCaseSkillProvider);
-    registry.registerSkillProvider('genCaseAB', CaseABSkillProvider);
-    registry.registerSkillProvider('addKnowledge', addKnowledgeSkillProvider);
-}
-
-export { setup };