ai-worktool 1.0.0

package/LICENSE.txt

@@ -0,0 +1 @@
+NO LICENSE

package/README.md

@@ -0,0 +1,67 @@
+# 🎮 吃豆豆:单测智能体
+
+<p align="center">
+  <img src="https://aicoder.jianguoke.cn/assets/images/logo.png" alt="吃豆豆logo" width="120" height="120">
+</p>
+
+<p align="center">
+  <strong>自动生成高质量单元测试，让代码覆盖率轻松达到100% 🚀</strong>
+</p>
+
+<p align="center">
+  <em>你负责编写源代码，我负责编写测试代码，解放双手，专注核心业务逻辑开发 💻</em>
+</p>
+
+---
+
+## 目录
+- [🎮 吃豆豆:单测智能体](#-吃豆豆单测智能体)
+  - [目录](#目录)
+  - [🌟 特性亮点](#-特性亮点)
+  - [🚀 核心功能](#-核心功能)
+  - [💎 核心价值](#-核心价值)
+  - [📝 使用指南](#-使用指南)
+  - [🔧 适用范围](#-适用范围)
+  - [🚀 开始使用](#-开始使用)
+  - [💬 联系我们](#-联系我们)
+
+## 🌟 特性亮点
+- 🔍 **智能代码分析**：深入理解代码结构和逻辑关系
+- 📝 **精准测试生成**：针对不同代码场景生成最佳测试策略
+- 🔄 **持续优化**：根据测试结果不断完善测试用例
+- 💡 **AI驱动**：利用先进的大语言模型提升测试质量
+
+## 🚀 核心功能
+- 🤖 **自动化测试生成**：自动识别未覆盖代码，生成完整测试用例，直到100%代码覆盖通过
+- 🛠️ **智能错误修复**：检测并修复单元测试中的语法错误、逻辑错误及断言失败
+- 📊 **覆盖率可视化**：实时展示行/语句/函数/分支覆盖率，通过颜色编码直观呈现测试状态
+- 🔄 **全流程自动化**：从环境初始化、依赖安装到持续测试优化，全程无需人工干预
+
+## 💎 核心价值
+- ⚡ **提升开发效率**：再也**不用手写**单测代码，专注核心业务逻辑开发
+- 🛡️ **保障代码质量**：确保测试覆盖率达标，降低线上bug风险
+- 🧩 **简化测试流程**：无需深入掌握测试框架细节，一键启动全自动化测试流程
+- 👁️ **可视化反馈**：通过直观的覆盖率报表和进度展示，让测试状态一目了然
+
+## 📝 使用指南
+1. 📂 打开项目文件夹
+2. ▶️ 启动吃豆豆智能体
+3. 📈 查看实时测试进度和覆盖率报告
+4. 🧠 智能体自动编写代码
+
+## 🔧 适用范围
+- 🔌 支持多种调用方式：VSCode插件、CLI命令
+- 💻 支持多种编程语言：TypeScript、JavaScript等
+- 🧪 支持主流测试框架：Jest、Vitest等
+
+---
+
+## 🚀 开始使用
+准备好提升你的开发效率了吗？立即安装吃豆豆单测智能体，让测试工作变得轻松简单！
+
+## 💬 联系我们
+如有任何问题或建议，可以给我发邮件 9160294@qq.com。
+
+<p align="center">
+  <strong>© 2025 吃豆豆单测智能体 - 让单元测试自动化、智能化</strong>
+</p>

package/dist/agents/chanjet.js

@@ -0,0 +1,382 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.startAgent = exports.stopAgent = exports.setConversationId = exports.addMessage = void 0;
+const tools = __importStar(require("../tools"));
+const os = __importStar(require("os"));
+const toolCall_1 = require("./toolCall");
+// API配置 - 方便修改和更新
+const API_CONFIG = {
+    // 聊天API地址
+    // eslint-disable-next-line @typescript-eslint/naming-convention
+    CHAT_ENDPOINT: 'https://inte-oneai.chanjet.com.cn/assistant/v1/chat',
+    // 函数调用结果上报地址前缀
+    // eslint-disable-next-line @typescript-eslint/naming-convention
+    FUNCTION_RESULT_ENDPOINT_PREFIX: 'https://inte-oneai.chanjet.com.cn/assistant/v1/function_call/update_result/',
+    // 认证信息
+    // eslint-disable-next-line @typescript-eslint/naming-convention
+    AUTH_TOKEN: 'Bearer DF#app-35EgnCA8VOlHFaRkR39pXvOw',
+    // 函数调用结果上报API密钥
+    // eslint-disable-next-line @typescript-eslint/naming-convention
+    API_KEY: 'your_api_key',
+    // 请求配置
+    // eslint-disable-next-line @typescript-eslint/naming-convention
+    REQUEST_CONFIG: {
+        channel: 'VSCode插件',
+        // eslint-disable-next-line @typescript-eslint/naming-convention
+        user_id: os.userInfo().username,
+        // eslint-disable-next-line @typescript-eslint/naming-convention
+        tenant_id: 'none'
+    }
+};
+let conversationId = ''; // 会话ID
+// 添加日志
+const addLog = (text, type = '') => {
+    console.log(text);
+};
+let lateLines = 0;
+function wrapText(text, width = process.stdout.columns) {
+    // 按现有换行符分割文本
+    const paragraphs = text.split('\n');
+    let totalLines = 0;
+    let wrappedText = '';
+    paragraphs.forEach((paragraph, index) => {
+        let currentLine = '';
+        let currentLineWidth = 0;
+        let paragraphLines = 1; // 每个段落至少有一行
+        for (let i = 0; i < paragraph.length; i++) {
+            const char = paragraph[i];
+            const charWidth = getCharWidth(char);
+            if (currentLineWidth + charWidth > width) {
+                currentLine += '\n';
+                currentLine += char;
+                currentLineWidth = charWidth;
+                paragraphLines++;
+            }
+            else {
+                currentLine += char;
+                currentLineWidth += charWidth;
+            }
+        }
+        // 添加处理后的段落（最后一个段落不添加额外换行）
+        wrappedText += currentLine + (index < paragraphs.length - 1 ? '\n' : '');
+        totalLines += paragraphLines;
+    });
+    return {
+        wrappedText,
+        lines: totalLines
+    };
+}
+function getCharWidth(char) {
+    const code = char.codePointAt(0) ?? 0;
+    // 全角字符范围判断
+    return (code >= 0x1100 && code <= 0x11ff) || // Hangul Jamo
+        (code >= 0x2e80 && code <= 0xa4cf) || // CJK ... Yi
+        (code >= 0xac00 && code <= 0xd7af) || // Hangul Syllables
+        (code >= 0xf900 && code <= 0xfaff) || // CJK Compatibility Ideographs
+        (code >= 0xfe30 && code <= 0xfe4f) || // CJK Compatibility Forms
+        (code >= 0xff00 && code <= 0xffef) // Halfwidth and Fullwidth Forms
+        ? 2
+        : 1;
+}
+const addMessage = (text, type = 'bot') => {
+    if (!process.stdout.moveCursor) {
+        console.log(text);
+        return;
+    }
+    const { lines } = wrapText(text);
+    process.stdout.moveCursor(0, -lateLines);
+    process.stdout.write(text + '\n');
+    lateLines = lines;
+};
+exports.addMessage = addMessage;
+const setConversationId = (id) => {
+    conversationId = id;
+};
+exports.setConversationId = setConversationId;
+const invokeTool = async (toolName, params) => {
+    let data = null;
+    try {
+        const tool = tools[toolName];
+        if (!tool) {
+            throw new Error(`Tool ${toolName} not found`);
+        }
+        data = await tool.apply(null, params);
+    }
+    catch (err) {
+        data = err.message || '执行失败';
+    }
+    return data || '执行完成';
+};
+// 处理事件消息
+const handleEventMessage = (contentObj) => {
+    switch (contentObj.type) {
+        case 'workflow_started':
+            addLog('工作流开始', 'workflow');
+            break;
+        case 'workflow_finished':
+            addLog('工作流结束', 'workflow');
+            break;
+        case 'node_started':
+            const startNodeInfo = contentObj.text || {};
+            addLog(`节点开始: ${startNodeInfo.title || ''} (${startNodeInfo.node_type || '未知类型'})`, 'node');
+            break;
+        case 'node_finished':
+            const finishNodeInfo = contentObj.text || {};
+            addLog(`节点结束: ${finishNodeInfo.title || ''} (${finishNodeInfo.node_type || '未知类型'})`, 'node');
+            break;
+        case 'message_end':
+            addLog('消息结束', 'workflow');
+            break;
+        default:
+            addLog(`事件: ${contentObj.type}`, 'workflow');
+    }
+};
+const reportFunctionResult = async (taskId, result) => {
+    addLog(`上报函数调用结果: ${taskId} ${result.success ? 'success' : 'failed'} ${typeof result.data === 'string' ? result.data : JSON.stringify(result.data, null, 2)}`);
+    try {
+        const response = await fetch(API_CONFIG.FUNCTION_RESULT_ENDPOINT_PREFIX + taskId, {
+            method: 'POST',
+            headers: {
+                // eslint-disable-next-line @typescript-eslint/naming-convention
+                'Content-Type': 'application/json'
+            },
+            body: JSON.stringify({
+                result
+            })
+        });
+        // if (response.ok) {
+        //   addLog('函数调用结果上报成功');
+        // } else {
+        //   addLog(`函数调用结果上报失败: ${response.status}`);
+        // }
+    }
+    catch (error) {
+        console.error('上报函数调用结果错误:', error);
+        addLog(`上报函数调用结果错误: ${error.message}`, 'error');
+    }
+};
+const stopAgent = async (taskId) => {
+    try {
+        await fetch(`${API_CONFIG.CHAT_ENDPOINT}/${taskId}/stop`, {
+            method: 'POST',
+            headers: {
+                // eslint-disable-next-line @typescript-eslint/naming-convention
+                Authorization: API_CONFIG.AUTH_TOKEN,
+                // eslint-disable-next-line @typescript-eslint/naming-convention
+                'Content-Type': 'application/json'
+            },
+            body: JSON.stringify(API_CONFIG.REQUEST_CONFIG)
+        });
+    }
+    catch (err) {
+        console.log(err);
+    }
+};
+exports.stopAgent = stopAgent;
+// 获取机器人响应
+const startAgent = async (name, userMessage, vars, withConversation, ctrl, mode, onMessage, onToolCall) => {
+    if (withConversation !== true) {
+        (0, exports.setConversationId)(''); // 每次重来
+    }
+    // console.log('conversationId:', conversationId);
+    const requestData = {
+        ...API_CONFIG.REQUEST_CONFIG,
+        // eslint-disable-next-line @typescript-eslint/naming-convention
+        conversation_id: conversationId,
+        variables: {
+            agent: name,
+            ...vars,
+            testFilePatterns: vars?.testFilePatterns?.join(','),
+            languages: vars?.languages?.join(',')
+        },
+        // eslint-disable-next-line @typescript-eslint/naming-convention
+        completion_option: {
+            stream: true,
+            append: false
+        },
+        messages: [
+            {
+                role: 'user',
+                // eslint-disable-next-line @typescript-eslint/naming-convention
+                content_type: 'text',
+                content: JSON.stringify({ type: 'text', text: userMessage }),
+                attachments: []
+            }
+        ]
+    };
+    console.log(userMessage);
+    let response;
+    try {
+        response = await fetch(API_CONFIG.CHAT_ENDPOINT, {
+            method: 'POST',
+            signal: ctrl?.signal,
+            headers: {
+                // eslint-disable-next-line @typescript-eslint/naming-convention
+                Authorization: API_CONFIG.AUTH_TOKEN,
+                // eslint-disable-next-line @typescript-eslint/naming-convention
+                'Content-Type': 'application/json'
+            },
+            body: JSON.stringify(requestData)
+        });
+    }
+    catch (err) {
+        throw new Error('大模型调用失败,' + err.message);
+    }
+    if (!response.ok) {
+        throw new Error(`HTTP错误! 状态码: ${response.status}`);
+    }
+    const reader = response.body.getReader();
+    const decoder = new TextDecoder();
+    let botResponse = '';
+    let botId = Date.now();
+    const clearMessage = () => {
+        if (botResponse) {
+            addLog(botResponse, 'bot');
+            botResponse = '';
+        }
+        lateLines = 0;
+        botId = Date.now();
+    };
+    // 处理函数调用
+    const handleFunctionCall = async (contentObj) => {
+        if (contentObj.type !== 'function_call') {
+            return;
+        }
+        try {
+            const functionCallData = JSON.parse(contentObj.text);
+            if (!functionCallData) {
+                return;
+            }
+            const { function_name: functionName, params, task_id: taskId } = functionCallData;
+            // console.log(JSON.stringify(params))
+            addLog(`函数调用: ${functionName}(\n${params
+                .map((it) => (typeof it === 'string' ? it.replace(/\\n/g, '\n') : JSON.stringify(it, null, 2)))
+                .join(',\n')})`, 'function');
+            await onMessage?.((0, toolCall_1.formatToolCallInfo)(functionCallData, vars.projectRoot), taskId);
+            let result = null;
+            try {
+                const ret = await invokeTool(functionName, params);
+                result = { success: true, data: ret };
+                await onToolCall?.({
+                    functionName,
+                    params,
+                    result
+                });
+            }
+            catch (e) {
+                addLog(`函数调用失败: ${functionName}`, 'error');
+                result = { success: false, message: e instanceof Error ? e.message : '未知错误' };
+                await onToolCall?.({
+                    functionName,
+                    params,
+                    result
+                });
+            }
+            await reportFunctionResult(taskId, result);
+            await onMessage?.((0, toolCall_1.formatToolCallInfo)(functionCallData, vars.projectRoot, result), taskId);
+        }
+        catch (error) {
+            console.error('函数调用错误:', error);
+            addLog(`函数调用错误: ${error instanceof Error ? error.message : '未知错误'}`, 'error');
+        }
+    };
+    let taskId = '';
+    while (true) {
+        const { value, done } = await reader.read();
+        if (done) {
+            break;
+        }
+        const chunk = decoder.decode(value, { stream: true });
+        const sseChunks = chunk.split('\n\n').filter((line) => line.trim().startsWith('data: '));
+        for (const sseChunk of sseChunks) {
+            const jsonStr = sseChunk.substring(6);
+            const data = JSON.parse(jsonStr);
+            if (!data?.answer?.length) {
+                continue;
+            }
+            if (data.conversation_id && !conversationId) {
+                (0, exports.setConversationId)(data.conversation_id);
+                // addLog(`会话ID: ${data.conversation_id}`);
+            }
+            for (const item of data.answer) {
+                if (!item.message) {
+                    continue;
+                }
+                const { role, content_type: contentType, content } = item.message;
+                if (role !== 'assistant') {
+                    continue;
+                }
+                const contentObj = JSON.parse(content);
+                if (!contentObj) {
+                    continue;
+                }
+                switch (contentType) {
+                    case 'event':
+                        clearMessage();
+                        handleEventMessage(contentObj);
+                        if (contentObj.type === 'workflow_started') {
+                            taskId = contentObj.task_id;
+                        }
+                        break;
+                    case 'text':
+                        if (contentObj.type === 'text' && contentObj.text) {
+                            if (mode === 'full') {
+                                botResponse += contentObj.text;
+                            }
+                            else {
+                                botResponse = contentObj.text;
+                            }
+                            await onMessage?.(botResponse, name + botId);
+                            // addMessage(botResponse, 'bot');
+                        }
+                        break;
+                    case 'function_call':
+                        clearMessage();
+                        await handleFunctionCall(contentObj);
+                        break;
+                    default:
+                        clearMessage();
+                        addLog(`未知的内容类型: ${contentType}`);
+                }
+            }
+        }
+    }
+    if (ctrl?.signal.aborted && taskId) {
+        await (0, exports.stopAgent)(taskId);
+    }
+};
+exports.startAgent = startAgent;
+//# sourceMappingURL=chanjet.js.map

package/dist/agents/index.js

@@ -0,0 +1,73 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.setConversationId = void 0;
+exports.getPlatform = getPlatform;
+exports.setPlatform = setPlatform;
+exports.startAgent = startAgent;
+const jianguoke = __importStar(require("./jianguoke"));
+const chanjet = __importStar(require("./chanjet"));
+__exportStar(require("./prompt"), exports);
+const platforms = {
+    chanjet,
+    jianguoke
+};
+let currentPlatform;
+function getPlatform() {
+    return platforms[currentPlatform || 'jianguoke'];
+}
+function setPlatform(target) {
+    currentPlatform = target;
+}
+const setConversationId = (id) => {
+    const platform = getPlatform();
+    if (!platform) {
+        throw new Error(`智能体 "${currentPlatform}" 不支持`);
+    }
+    platform.setConversationId(id);
+};
+exports.setConversationId = setConversationId;
+async function startAgent(currentPlatform, name, question, vars, withConversation, ctrl, mode, onMessage, onToolCall) {
+    const platform = platforms[currentPlatform] || getPlatform();
+    if (!platform) {
+        throw new Error(`智能体 "${currentPlatform}" 不支持`);
+    }
+    await platform.startAgent.apply(null, [name, question, vars, withConversation, ctrl, mode, onMessage, onToolCall]);
+    console.log(`智能体 "${currentPlatform}" 调用完成`);
+}
+//# sourceMappingURL=index.js.map