@comate/zulu 0.7.1 → 0.7.2

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

@@ -1,837 +0,0 @@
-import { SkillProvider, StringChunkStream } from '@comate/plugin-host';
-import { readFileSync } from 'fs';
-import { fileURLToPath } from 'node:url';
-import path, { 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();
-        try {
-            const filename = fileURLToPath(import.meta.url);
-            const content = readFileSync(`${dirname(filename)}/help.md`, 'utf8');
-            yield stream.flush(content);
-        } catch (error) {
-            this.logger.error(`Plugin Error: ${error}`);
-            yield stream.flushReplaceLast(`抱歉\n\n插件发生出错，${error}。`);
-        }
-    }
-}
-
-const API_SERVER_HOST = 'http://10.169.27.242:8089/comate';
-// export const API_SERVER_HOST = 'http://localhost:8080/comate';
-const WEB_LINK = 'https://casebuilder.n.baidu-int.com';
-const MAX_DEMO_CONTENT_LENGTH = 5000;
-
-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 apiRequest = 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 = response.json();
-        return jsonRes.then((res)=>{
-            if (res.code !== 'OK') {
-                throw new Error(`[${res.code}] ${res.message}`);
-            } else {
-                return res.data;
-            }
-        });
-    } catch (err) {
-        throw err;
-    }
-};
-async function* sseRequest(method, path, options = {}) {
-    try {
-        const response = await fetch(`${API_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 = await 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 { event, data } = value;
-                if (event && [
-                    'FatalError',
-                    'BadRequest'
-                ].includes(event)) {
-                    throw new Error(data);
-                }
-                if (event !== 'Msg' || !data.length) {
-                    continue;
-                }
-                yield data.replaceAll('\\x0A', '\n');
-            }
-        } catch (e) {
-            throw e;
-        } finally{
-            reader.releaseLock();
-        }
-    } catch (err) {
-        throw err;
-    }
-}
-const queryRepo = (username, repo)=>{
-    return apiRequest('GET', `/query/repo?repo=${repo}`, {
-        headers: {
-            'X-COMATE-USER': username
-        }
-    });
-};
-const setRepoProject = (username, repo, projectId)=>{
-    return apiRequest('POST', `/set/repo/project?repo=${repo}`, {
-        headers: {
-            'X-COMATE-USER': username,
-            'Content-Type': 'application/json'
-        },
-        body: JSON.stringify({
-            projectId
-        })
-    });
-};
-const setProjectCaseDemoContent = (username, repo, content)=>{
-    return apiRequest('POST', `/set/project/case-demo-content?repo=${repo}`, {
-        headers: {
-            'X-COMATE-USER': username,
-            'Content-Type': 'application/json'
-        },
-        body: JSON.stringify({
-            caseDemoContent: content
-        })
-    });
-};
-const analyzeRepo = (username, repo, fileRelPath, content)=>{
-    return apiRequest('POST', `/analyze/repo?repo=${repo}`, {
-        headers: {
-            'X-COMATE-USER': username,
-            'Content-Type': 'application/json'
-        },
-        body: JSON.stringify({
-            fileRelPath,
-            fileContent: content
-        })
-    }, 10000);
-};
-// 创建AppBuilder会话id
-const newChat = (username, repo, sessionId, ide)=>{
-    return apiRequest('POST', `/chat/new?repo=${repo}`, {
-        headers: {
-            'X-COMATE-USER': username,
-            'Content-Type': 'application/json'
-        },
-        body: JSON.stringify({
-            sessionId,
-            ide
-        })
-    });
-};
-// 函数检索测试
-const ragRepoFunc = (username, repo, query)=>{
-    return apiRequest('POST', `/rag/repo-func?repo=${repo}`, {
-        headers: {
-            'X-COMATE-USER': username,
-            'Content-Type': 'application/json'
-        },
-        body: JSON.stringify({
-            query
-        })
-    }, 20000);
-};
-// AppBuilder多轮对话生成代码
-async function genCaseCodeByAB(username, repo, cid, query, caseDemoContent, isFirstRound) {
-    const res = await sseRequest('POST', `/gen/case-code-by-ab?repo=${repo}`, {
-        body: JSON.stringify({
-            cid,
-            query,
-            caseDemoContent,
-            isFirstRound
-        }),
-        headers: {
-            'X-COMATE-USER': username,
-            'Content-Type': 'application/json'
-        }
-    });
-    return res;
-}
-
-const pickICodeNameRepository = (repository)=>{
-    if (repository.versionControl === 'git') {
-        const repositoryUrl = repository.repositoryUrl;
-        const matched = repositoryUrl?.match(/ssh:\/\/[^/]+\/(.+)/);
-        if (matched && matched.length > 1) {
-            return matched[1];
-        }
-    }
-    return null;
-};
-
-class InfoSkillProvider extends SkillProvider {
-    static{
-        this.skillName = 'info';
-    }
-    static{
-        this.displayName = '查看当前代码库信息';
-    }
-    static{
-        this.description = '查看当前代码库在平台的信息和iAPI绑定关系';
-    }
-    static{
-        this.parameters = {
-            type: 'object',
-            properties: {}
-        };
-    }
-    async *execute() {
-        const stream = new StringChunkStream();
-        try {
-            const userDetail = await this.currentUser.requestDetail();
-            if (!userDetail) {
-                yield stream.flushReplaceLast('请先完成授权，CaseBuilder 插件需在用户授权后才能使用。');
-                return;
-            }
-            const repository = await this.retriever.repositoryInfo();
-            let repositoryUrl = pickICodeNameRepository(repository);
-            if (!repositoryUrl) {
-                yield stream.flushReplaceLast('无法获取当前项目的仓库地址，可能不是一个icode仓库或者缺少.git配置目录。');
-                return;
-            }
-            yield stream.flush('-> 正在获取当前仓库在平台的配置...\n\n');
-            const teamRepo = await queryRepo(userDetail.name, repositoryUrl);
-            yield stream.flush('-> 获取仓库配置成功，信息如下：\n\n');
-            const activeProject = teamRepo?.repoProjects.find((item)=>item.projectId === teamRepo.activeProjectId);
-            stream.append(`仓库名称：${teamRepo.name}\n\n`);
-            stream.append(`默认iAPI项目：${activeProject ? activeProject.name : '未设置'}\n\n`);
-            stream.append(`描述：${teamRepo.description}\n\n`);
-            stream.append('关联的iAPI项目：\n\n');
-            const projectTable = this.render.table(teamRepo?.repoProjects || [], {
-                columns: [
-                    {
-                        field: 'projectId',
-                        title: '项目ID'
-                    },
-                    {
-                        field: 'name',
-                        title: '名称'
-                    }
-                ]
-            });
-            yield stream.flush(projectTable);
-            yield stream.flush('\n\n');
-        } catch (error) {
-            this.logger.error(`Plugin Error: ${error}`);
-            yield stream.flushReplaceLast(`${error}`);
-        }
-    }
-}
-
-class SetRepoProjectSkillProvider extends SkillProvider {
-    static{
-        this.skillName = 'setRepoProject';
-    }
-    static{
-        this.displayName = '设置默认IAPI项目';
-    }
-    static{
-        this.description = '代码库必须要和IAPI项目做绑定，同时只能使用一个IAPI项目做接口生成';
-    }
-    static{
-        this.parameters = {
-            type: 'object',
-            properties: {}
-        };
-    }
-    async *execute() {
-        const stream = new StringChunkStream();
-        try {
-            const { query } = this.currentContext;
-            const projectId = query.trim();
-            const userDetail = await this.currentUser.requestDetail();
-            if (!userDetail) {
-                yield stream.flushReplaceLast('请先完成授权，CaseBuilder 插件需在用户授权后才能使用。');
-                return;
-            }
-            const fs = await this.requestWorkspaceFileSystem();
-            if (!fs) {
-                yield stream.flush('请先完成授权，CaseBuilder 插件需在用户授权工作区的文件读取权限后才能使用。');
-                return;
-            }
-            const repository = await this.retriever.repositoryInfo();
-            let repositoryUrl = pickICodeNameRepository(repository);
-            if (!repositoryUrl) {
-                yield stream.flushReplaceLast('无法获取当前项目的仓库地址，可能不是一个icode仓库或者缺少.git配置目录。');
-                return;
-            }
-            if (projectId == '') {
-                yield stream.flushReplaceLast('请输入需要设置的默认iAPI项目ID。');
-                return;
-            }
-            const repoProject = await setRepoProject(userDetail.name, repositoryUrl, projectId);
-            if (repoProject.name == '') {
-                yield stream.flushReplaceLast('设置默认iAPI项目失败，iAPI项目不存在，请检查输入的iAPI项目ID是否正确。');
-                return;
-            }
-            yield stream.flush('-> 设置默认iAPI项目成功。\n\n');
-            yield stream.flush('-> 当前默认iAPI项目：\n\n');
-            const projectTable = this.render.table([
-                repoProject
-            ], {
-                columns: [
-                    {
-                        field: 'projectId',
-                        title: '项目ID'
-                    },
-                    {
-                        field: 'name',
-                        title: '名称'
-                    }
-                ]
-            });
-            yield stream.flush(projectTable);
-            yield stream.flush('\n\n');
-        } catch (error) {
-            this.logger.error(`Plugin Error: ${error}`);
-            yield stream.flushReplaceLast(`${error}`);
-        }
-    }
-}
-
-class SetCaseDemoContentSkillProvider extends SkillProvider {
-    static{
-        this.skillName = 'setCaseDemoContent';
-    }
-    static{
-        this.displayName = '设置【用例样例】代码';
-    }
-    static{
-        this.description = '用例样例代码可用于用例生成的参考代码';
-    }
-    static{
-        this.parameters = {
-            type: 'object',
-            properties: {}
-        };
-    }
-    async *execute() {
-        const stream = new StringChunkStream();
-        try {
-            const { activeFilePath, activeFileContent, selectedCode } = this.currentContext;
-            const userDetail = await this.currentUser.requestDetail();
-            if (!userDetail) {
-                yield stream.flushReplaceLast('请先完成授权，CaseBuilder 插件需在用户授权后才能使用。');
-                return;
-            }
-            const fs = await this.requestWorkspaceFileSystem();
-            if (!fs) {
-                yield stream.flush('请先完成授权，CaseBuilder 插件需在用户授权工作区的文件读取权限后才能使用。');
-                return;
-            }
-            const repository = await this.retriever.repositoryInfo();
-            let repositoryUrl = pickICodeNameRepository(repository);
-            if (!repositoryUrl) {
-                yield stream.flushReplaceLast('无法获取当前项目的仓库地址，可能不是一个icode仓库或者缺少.git配置目录。');
-                return;
-            }
-            if (activeFilePath == '') {
-                yield stream.flushReplaceLast('未打开任何文件，请先打开需要设置的用例样例文件。');
-                return;
-            }
-            if (activeFileContent == '') {
-                yield stream.flushReplaceLast('当前文件内容为空，请先编辑需要设置的用例样例文件。');
-                return;
-            }
-            let fileContent = activeFileContent;
-            yield stream.flush('-> 使用当前编辑器中打开的文件作为用例样例文件。\n\n');
-            yield stream.flush(`-> 用例样例文件为：${activeFilePath}\n\n`);
-            if (selectedCode.length) {
-                fileContent = selectedCode;
-                yield stream.flush('-> 使用当前编辑器中选中的代码作为用例样例文件。\n\n');
-            }
-            const fileContentLength = fileContent.length;
-            yield stream.flush(`-> 当前样例文件长度: ${fileContentLength}。\n\n`);
-            if (fileContentLength > MAX_DEMO_CONTENT_LENGTH) {
-                yield stream.flush(`-> 样例文件超过 ${MAX_DEMO_CONTENT_LENGTH} 字符限制，请编辑更短的样例文件。\n\n`);
-                return;
-            }
-            await setProjectCaseDemoContent(userDetail.name, repositoryUrl, fileContent);
-            yield stream.flush('-> 设置用例样例文件成功。\n\n');
-        } catch (error) {
-            this.logger.error(`Plugin Error: ${error}`);
-            yield stream.flushReplaceLast(`${error}`);
-        }
-    }
-}
-
-const blackFileList = [
-    '__init__.py',
-    '.DS_Store'
-];
-const getFilesFromDirectory = async (files, rootDir, fs)=>{
-    rootDir = rootDir.replace(/\/$/, '');
-    const dirs = await fs.readDirectory(rootDir);
-    for (const item of dirs){
-        if (item.type === 'directory') {
-            await getFilesFromDirectory(files, path.join(rootDir, item.name), fs);
-        }
-        if (item.type === 'file') {
-            if (blackFileList.includes(item.name)) {
-                continue;
-            }
-            files.push({
-                name: item.name,
-                path: path.normalize(rootDir),
-                status: 'pending'
-            });
-        }
-    }
-    return files;
-};
-
-class RepoAnalyzeSkillProvider extends SkillProvider {
-    static{
-        this.skillName = 'repoAnalyze';
-    }
-    static{
-        this.displayName = '分析代码库';
-    }
-    static{
-        this.description = '指定目录或文件进行分析代码。';
-    }
-    static{
-        this.parameters = {
-            type: 'object',
-            properties: {}
-        };
-    }
-    renderFileTableString(files) {
-        return this.render.table(files, {
-            columns: [
-                {
-                    field: 'path',
-                    title: '目录'
-                },
-                {
-                    field: 'name',
-                    title: '文件名'
-                },
-                {
-                    field: 'status',
-                    title: '状态',
-                    format: (v)=>{
-                        switch(v){
-                            case 'pending':
-                                return '待提交';
-                            case 'processing':
-                                return '提交中';
-                            case 'success':
-                                return '已提交';
-                            case 'failed':
-                                return '提交失败';
-                            case 'ignore':
-                                return '忽略';
-                            case 'empty':
-                                return '空文件';
-                            case 'exists':
-                                return '任务已存在';
-                            default:
-                                return v;
-                        }
-                    }
-                },
-                {
-                    field: 'err',
-                    title: '失败原因',
-                    format: (v)=>v || ''
-                }
-            ]
-        });
-    }
-    async *execute() {
-        const stream = new StringChunkStream();
-        try {
-            const { query, selectedCode, activeFilePath } = this.currentContext;
-            let keyword = path.normalize(query.trim());
-            const userDetail = await this.currentUser.requestDetail();
-            if (!userDetail) {
-                yield stream.flushReplaceLast('请先完成授权，CaseBuilder 插件需在用户授权后才能使用。');
-                return;
-            }
-            const fs = await this.requestWorkspaceFileSystem();
-            if (!fs) {
-                yield stream.flush('请先完成授权，CaseBuilder 插件需在用户授权工作区的文件读取权限后才能使用。');
-                return;
-            }
-            const repository = await this.retriever.repositoryInfo();
-            let repositoryUrl = pickICodeNameRepository(repository);
-            if (!repositoryUrl) {
-                yield stream.flushReplaceLast('无法获取当前项目的仓库地址，可能不是一个icode仓库或者缺少.git配置目录。');
-                return;
-            }
-            const isSelectedCode = activeFilePath.length && selectedCode.length;
-            let fileMode = 'openFile';
-            if ([
-                '',
-                '.',
-                './'
-            ].includes(keyword)) {
-                if ((keyword === '.' || keyword === '') && isSelectedCode) {
-                    keyword = activeFilePath;
-                    fileMode = 'selectedCode';
-                } else {
-                    yield stream.flushReplaceLast('请输入需要分析的目录、文件或者选中代码段，不允许输入根目录。');
-                    return;
-                }
-            }
-            let files = [];
-            let fileOrDir;
-            try {
-                fileOrDir = await fs.stat(keyword);
-            } catch (error) {
-                yield stream.flushReplaceLast('目录或文件不存在。');
-                return;
-            }
-            if (fileOrDir.type === 'file') {
-                files.push({
-                    name: path.basename(keyword),
-                    path: path.dirname(keyword),
-                    status: 'pending'
-                });
-            } else if (fileOrDir.type === 'directory') {
-                // 遍历目录
-                files = await getFilesFromDirectory([], keyword, fs);
-                if (files.length === 0) {
-                    yield stream.flushReplaceLast('目录下没有文件，请选择其他目录');
-                    return;
-                }
-                files.sort((a, b)=>a.path.length - b.path.length);
-            } else {
-                yield stream.flushReplaceLast('请输入正确的目录或文件');
-                return;
-            }
-            yield stream.flushReplaceLast(this.renderFileTableString(files));
-            for(const fileIdx in files){
-                const file = files[fileIdx];
-                const fileRelPath = file.path + '/' + file.name;
-                let fileContent;
-                if (fileMode === 'openFile') {
-                    const buf = await fs.readFile(fileRelPath);
-                    fileContent = buf.toString();
-                }
-                if (fileMode === 'selectedCode') {
-                    fileContent = selectedCode;
-                }
-                if (!fileContent) {
-                    files[fileIdx].status = 'ignore';
-                    yield stream.flushReplaceLast(this.renderFileTableString(files));
-                    continue;
-                }
-                files[fileIdx].status = 'processing';
-                yield stream.flushReplaceLast(this.renderFileTableString(files));
-                try {
-                    const status = await analyzeRepo(userDetail.name, repositoryUrl, fileRelPath, fileContent);
-                    if (status) {
-                        files[fileIdx].status = status;
-                    }
-                } catch (error) {
-                    files[fileIdx].status = 'failed';
-                    files[fileIdx].err = error;
-                }
-                yield stream.flushReplaceLast(this.renderFileTableString(files));
-            }
-            yield stream.flushReplaceLast(this.renderFileTableString(files) + `\n\n分析任务是异步操作，前往 [CaseBuilder-团队-管理函数](${WEB_LINK}/team/repo) 查看任务进度。`);
-        } catch (error) {
-            this.logger.error(`Plugin Error: ${error}`);
-            yield stream.flushReplaceLast(`${error}`);
-        }
-    }
-}
-
-class GenCaseV2SkillProvider extends SkillProvider {
-    static{
-        this.skillName = 'genCaseV2';
-    }
-    static{
-        this.displayName = '用例用例生成V2';
-    }
-    static{
-        this.description = '基于代码库现有函数进行用例生成';
-    }
-    static{
-        this.parameters = {
-            type: 'object',
-            properties: {}
-        };
-    }
-    async *execute() {
-        const stream = new StringChunkStream();
-        try {
-            const { query, selectedCode } = this.currentContext;
-            const trimQuery = query.trim();
-            const userDetail = await this.currentUser.requestDetail();
-            if (!userDetail) {
-                yield stream.flushReplaceLast('请先完成授权，CaseBuilder 插件需在用户授权后才能使用。');
-                return;
-            }
-            const fs = await this.requestWorkspaceFileSystem();
-            if (!fs) {
-                yield stream.flush('请先完成授权，CaseBuilder 插件需在用户授权工作区的文件读取权限后才能使用。');
-                return;
-            }
-            const repository = await this.retriever.repositoryInfo();
-            let repositoryUrl = pickICodeNameRepository(repository);
-            if (!repositoryUrl) {
-                yield stream.flushReplaceLast('无法获取当前项目的仓库地址，可能不是一个icode仓库或者缺少.git配置目录。');
-                return;
-            }
-            if (trimQuery.length == 0) {
-                yield stream.flushReplaceLast('请以自然语言输入用例描述或用例实现步骤。');
-                return;
-            }
-            // 获取当前IDE的sessionId
-            const sessionId = this.sessionUuid;
-            if (!sessionId) {
-                yield stream.flushReplaceLast('IDE sessionId获取失败，请稍后重试。');
-                return;
-            }
-            // 获取当前IDE的名称
-            const host = await this.retriever.hostInfo();
-            const ideName = host.appName.replaceAll(" ", "_").toLowerCase();
-            yield stream.flush('-> 正在获取当前仓库在平台的配置...\n\n');
-            const teamRepo = await queryRepo(userDetail.name, repositoryUrl);
-            if (teamRepo.activeProjectId == '') {
-                yield stream.flush('团队仓库没有激活默认iAPI项目，你可以通过功能【设置默认iAPI项目】快速设置。\n\n');
-                return;
-            }
-            const activeProject = teamRepo?.repoProjects.find((item)=>item.projectId === teamRepo.activeProjectId);
-            if (!activeProject) {
-                yield stream.flush('团队仓库默认iAPI项目不存在。\n\n');
-                return;
-            }
-            yield stream.flush('-> 获取仓库配置成功，信息如下：\n\n');
-            stream.append(`仓库名称：${teamRepo.name}\n\n`);
-            stream.append(`默认iAPI项目：${activeProject.name}\n\n`);
-            stream.append(`描述：${teamRepo.description}\n\n`);
-            stream.append('关联的iAPI项目：\n\n');
-            const projectTable = this.render.table(teamRepo?.repoProjects || [], {
-                columns: [
-                    {
-                        field: 'projectId',
-                        title: '项目ID'
-                    },
-                    {
-                        field: 'name',
-                        title: '名称'
-                    }
-                ]
-            });
-            yield stream.flush(projectTable);
-            yield stream.flush('\n\n');
-            const newChatRes = await newChat(userDetail.name, repositoryUrl, sessionId, ideName);
-            if (!newChatRes || !newChatRes.cid) {
-                yield stream.flush('生成会话失败，请稍后重试。\n\n');
-                return;
-            }
-            const cid = newChatRes.cid;
-            const isFirstRound = newChatRes.isFirstRound;
-            yield stream.flush('-> 根据您的需求，开始为您生成接口用例代码...\n\n');
-            const codeChunks = await genCaseCodeByAB(userDetail.name, repositoryUrl, cid, trimQuery, selectedCode, isFirstRound);
-            for await (const chunk of codeChunks){
-                yield stream.flush(chunk);
-            }
-            yield stream.flush('\n\n');
-        } catch (error) {
-            this.logger.error(`Plugin Error: ${error}`);
-            yield stream.flush(`${error}。`);
-        }
-    }
-}
-
-class RagRepoFuncSkillProvider extends SkillProvider {
-    static{
-        this.skillName = 'ragRepoFunc';
-    }
-    static{
-        this.displayName = '代码库函数命中测试';
-    }
-    static{
-        this.description = '根据关键字检索已经上传AB知识库的函数知识。';
-    }
-    static{
-        this.parameters = {
-            type: 'object',
-            properties: {}
-        };
-    }
-    async *execute() {
-        const stream = new StringChunkStream();
-        try {
-            const { query } = this.currentContext;
-            const keyword = query.trim();
-            const userDetail = await this.currentUser.requestDetail();
-            if (!userDetail) {
-                yield stream.flushReplaceLast('请先完成授权，CaseBuilder 插件需在用户授权后才能使用。');
-                return;
-            }
-            const repository = await this.retriever.repositoryInfo();
-            let repositoryUrl = pickICodeNameRepository(repository);
-            if (!repositoryUrl) {
-                yield stream.flushReplaceLast('无法获取当前项目的仓库地址，可能不是一个icode仓库或者缺少.git配置目录。');
-                return;
-            }
-            if (keyword.length == 0) {
-                yield stream.flushReplaceLast('请输入检索关键字，可以是用例描述的某一个步骤。');
-                return;
-            }
-            yield stream.flush('-> 正在为你进行函数命中测试...\n\n');
-            const res = await ragRepoFunc(userDetail.name, repositoryUrl, keyword);
-            if (res?.length) {
-                yield stream.flushReplaceLast(res);
-            } else {
-                yield stream.flushReplaceLast('没有找到匹配的函数，请调整输入的关键字重新尝试。');
-            }
-        } catch (error) {
-            this.logger.error(`Plugin Error: ${error}`);
-            yield stream.flushReplaceLast(`${error}`);
-        }
-    }
-}
-
-function setup({ registry }) {
-    registry.registerSkillProvider('help', HelpSkillProvider);
-    registry.registerSkillProvider('info', InfoSkillProvider);
-    registry.registerSkillProvider('setRepoProject', SetRepoProjectSkillProvider);
-    registry.registerSkillProvider('setCaseDemoContent', SetCaseDemoContentSkillProvider);
-    registry.registerSkillProvider('repoAnalyze', RepoAnalyzeSkillProvider);
-    registry.registerSkillProvider('genCaseV2', GenCaseV2SkillProvider);
-    registry.registerSkillProvider('ragRepoFunc', RagRepoFuncSkillProvider);
-}
-
-export { setup };