cloving 0.1.0 → 0.1.1-8.1

package/dist/commands/generate/review.js

@@ -0,0 +1,146 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const cli_highlight_1 = __importDefault(require("cli-highlight"));
+const prompts_1 = require("@inquirer/prompts");
+const copy_paste_1 = __importDefault(require("copy-paste"));
+const cloving_gpt_1 = __importDefault(require("../../cloving_gpt"));
+const git_utils_1 = require("../../utils/git_utils");
+const config_utils_1 = require("../../utils/config_utils");
+const string_utils_1 = require("../../utils/string_utils");
+const review = async (options) => {
+    options.silent = (0, config_utils_1.getConfig)(options).globalSilent || false;
+    const gpt = new cloving_gpt_1.default(options);
+    try {
+        // Define the prompt for analysis
+        const gitDiff = await (0, git_utils_1.getGitDiff)();
+        const promptText = `==== begin diff =====
+${gitDiff}
+==== end diff =====
+
+### Example of a well-structured response
+
+# Code Review: Enhanced Code Generation and File Handling
+
+## Changes Overview
+
+These changes primarily focus on improving the code generation process and adding more robust file handling capabilities. The main modifications include:
+
+1. Enhancing the code generation prompt with a structured example response.
+2. Implementing a new function to extract file names and contents from the generated code.
+3. Adding a new option to save generated code directly to files.
+4. Improving the copy-to-clipboard functionality to allow copying individual files.
+5. Removing an unused import in the review.ts file.
+
+## Reason for Changes
+
+The changes aim to enhance the usability and flexibility of the code generation tool by providing clearer prompts, better file management, and more control over the generated code.
+
+## Detailed Description
+
+1. **Enhanced Code Generation Prompt** - The \`generateCodePrompt\` function now includes a structured example of a well-formatted response. This helps guide the AI to produce more consistent and usable output, making it easier to parse and use the generated code.
+2. **File Extraction Function** - A new function \`extractFilesAndContent\` has been added to parse the raw code command and extract individual file names and their contents. This enables more granular control over the generated code, allowing for file-specific operations.
+3. **Save to File Option** - A new 'Save Source Code to Files' option has been added to the user action menu. This feature allows users to save generated code directly to files in the appropriate directory structure.
+4. **Improved Copy to Clipboard** - The 'Copy Source Code to Clipboard' option has been enhanced to allow users to choose specific files to copy, rather than copying all generated code at once.
+5. **Removed Unused Import** - The unused import of \`execFileSync\` from 'child_process' has been removed from the review.ts file, improving code cleanliness.
+
+## Potential Bugs and Recommended Fixes
+
+### 1. Possible undefined access in \`extractFilesAndContent\`
+
+\`\`\`typescript
+const contentMatch = regex.exec(rawCodeCommand)
+if (contentMatch) {
+  files.push(fileName)
+  let content = contentMatch[1]
+  // ...
+}
+\`\`\`
+
+There's a potential for \`contentMatch[1]\` to be undefined if the regex doesn't capture any groups. To fix this, add a null check:
+
+\`\`\`typescript
+const contentMatch = regex.exec(rawCodeCommand)
+if (contentMatch && contentMatch[1]) {
+  files.push(fileName)
+  let content = contentMatch[1]
+  // ...
+}
+\`\`\`
+
+### Request
+
+Do not use any data from the example response structure, only use the structure.
+I would like you to explain why these change are being made and document a description of these changes.
+Also list any bugs in the new code as well as recommended fixes for those bugs with code examples.
+Format the output of this code review in Markdown format.`;
+        // get the analysis
+        const analysis = await gpt.generateText({ prompt: promptText });
+        // Print the analysis to the console
+        (0, string_utils_1.parseMarkdownInstructions)(analysis).map(code => {
+            if (code.trim().startsWith('```')) {
+                const lines = code.split('\n');
+                const language = code.match(/```(\w+)/)?.[1] || 'plaintext';
+                console.log(lines[0]);
+                try {
+                    console.log((0, cli_highlight_1.default)(lines.slice(1, -1).join('\n'), { language }));
+                }
+                catch (error) {
+                    // don't highlight if it fails
+                    console.log(lines.slice(1, -1).join('\n'));
+                }
+                console.log(lines.slice(-1)[0]);
+            }
+            else {
+                console.log((0, cli_highlight_1.default)(code, { language: 'markdown' }));
+            }
+        });
+        // Modified Function to extract specific sections from the analysis
+        const extractSection = (section) => {
+            const regex = new RegExp(`## ${section}[\\s\\S]*?(?=\\n## Potential Bugs and Recommended Fixes|$)`, 'g');
+            const match = analysis.match(regex);
+            return match ? match[0].trim() : '';
+        };
+        // Prompt the user with options for copying to clipboard
+        const clipboardOption = await (0, prompts_1.select)({
+            message: 'What would you like to copy to the clipboard?',
+            choices: [
+                { name: 'Copy only the Changes Overview', value: 'Changes Overview' },
+                { name: 'Copy only the Potential Bugs and Recommended Fixes', value: 'Potential Bugs and Recommended Fixes' },
+                { name: 'Copy everything', value: 'Everything' },
+                { name: 'Done', value: 'Done' }
+            ]
+        });
+        let contentToCopy = '';
+        switch (clipboardOption) {
+            case 'Changes Overview':
+                contentToCopy = extractSection('Changes Overview');
+                break;
+            case 'Potential Bugs and Recommended Fixes':
+                contentToCopy = extractSection('Potential Bugs and Recommended Fixes');
+                break;
+            case 'Everything':
+                contentToCopy = analysis;
+                break;
+            case 'Done':
+                console.log('No content copied to clipboard.');
+                return;
+        }
+        if (contentToCopy) {
+            copy_paste_1.default.copy(contentToCopy, (err) => {
+                if (err) {
+                    console.error('Error: Unable to copy to clipboard.', err);
+                }
+                else {
+                    console.log('Selected content copied to clipboard');
+                }
+            });
+        }
+    }
+    catch (error) {
+        console.error('Error during analysis:', error.message);
+    }
+};
+exports.default = review;

package/dist/commands/generate/sh.js

@@ -0,0 +1,95 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const inquirer_1 = __importDefault(require("inquirer"));
+const cli_highlight_1 = __importDefault(require("cli-highlight"));
+const child_process_1 = require("child_process");
+const string_utils_1 = require("../../utils/string_utils");
+const command_utils_1 = require("../../utils/command_utils");
+const config_utils_1 = require("../../utils/config_utils");
+const cloving_gpt_1 = __importDefault(require("../../cloving_gpt"));
+const generateShellPrompt = (prompt) => {
+    const shell = (0, child_process_1.execSync)('echo $SHELL').toString().trim();
+    const os = (0, child_process_1.execSync)('echo $OSTYPE').toString().trim();
+    return `Generate an executable ${shell} script that works on ${os}. Try to make it a single line if possible and as simple and straightforward as possible.
+
+Do not add any commentary or context to the message other than the commit message itself.
+
+An example of the output for this should look like the following:
+
+\`\`\`sh
+find . -type f -name "*.ts" -exec sed -i '' 's/old/new/g' {} +
+\`\`\`
+
+Don't use that script, it is only an example.
+
+Here is the description of what I want the script to do:
+
+${prompt}`;
+};
+const generateExplainShellPrompt = (prompt) => {
+    return `${prompt}
+  
+Please briefly explain how this shell script works.`;
+};
+const sh = async (options) => {
+    let { prompt } = options;
+    options.silent = (0, config_utils_1.getConfig)(options).globalSilent || false;
+    const gpt = new cloving_gpt_1.default(options);
+    try {
+        if (!prompt) {
+            prompt = await (0, command_utils_1.promptUser)('What would you like to do: ');
+        }
+        // Generate the prompt for commit message
+        const shellPrompt = generateShellPrompt(prompt);
+        // Instantiate ClovingGPT and get the commit message
+        const rawShellCommand = await gpt.generateText({ prompt: shellPrompt });
+        // Clean the commit message using extractMarkdown
+        const generatedShell = (0, string_utils_1.extractMarkdown)(rawShellCommand);
+        // remove #! from the generated shell script
+        const generatedShellWithoutShebang = generatedShell.replace(/^#!.*?\s/, '');
+        console.log((0, cli_highlight_1.default)(generatedShellWithoutShebang));
+        // Inquirer prompt for further actions
+        const { action } = await inquirer_1.default.prompt([
+            {
+                type: 'list',
+                name: 'action',
+                message: 'What would you like to do?',
+                choices: [
+                    { name: 'Execute', value: 'execute' },
+                    { name: 'Revise', value: 'revise' },
+                    { name: 'Explain', value: 'explain' },
+                    { name: 'Copy to Clipboard', value: 'copy' },
+                    { name: 'Cancel', value: 'cancel' },
+                ],
+            },
+        ]);
+        switch (action) {
+            case 'execute':
+                (0, child_process_1.execSync)(generatedShellWithoutShebang, { stdio: 'inherit' });
+                break;
+            case 'revise':
+                // Logic to revise the prompt (not implemented in this example)
+                console.log('Revise option selected. Implement revision logic here.');
+                break;
+            case 'explain':
+                const explainPrompt = generateExplainShellPrompt(generatedShellWithoutShebang);
+                const explainShellCommand = await gpt.generateText({ prompt: explainPrompt });
+                console.log((0, cli_highlight_1.default)(explainShellCommand, { language: 'markdown' }));
+                break;
+            case 'copy':
+                (0, child_process_1.execFileSync)('pbcopy', { input: generatedShellWithoutShebang });
+                console.log('Script copied to clipboard.');
+                break;
+            case 'cancel':
+                console.log('Operation cancelled.');
+                break;
+        }
+    }
+    catch (error) {
+        console.error('Could not generate shell script');
+    }
+};
+exports.default = sh;

package/dist/commands/generate/shell.js

@@ -0,0 +1,111 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const child_process_1 = require("child_process");
+const cli_highlight_1 = __importDefault(require("cli-highlight"));
+const copy_paste_1 = __importDefault(require("copy-paste"));
+const string_utils_1 = require("../../utils/string_utils");
+const config_utils_1 = require("../../utils/config_utils");
+const cloving_gpt_1 = __importDefault(require("../../cloving_gpt"));
+const prompts_1 = require("@inquirer/prompts");
+const generateShellPrompt = (prompt) => {
+    const shell = (0, child_process_1.execSync)('echo $SHELL').toString().trim();
+    const os = (0, child_process_1.execSync)('echo $OSTYPE').toString().trim();
+    return `Generate an executable ${shell} script that works on ${os}. Try to make it a single line if possible and as simple and straightforward as possible.
+
+Do not add any commentary or context to the message other than the commit message itself.
+
+An example of the output for this should look like the following:
+
+\`\`\`sh
+find . -type f -name "*.ts" -exec sed -i '' 's/old/new/g' {} +
+\`\`\`
+
+Don't use that script, it is only an example.
+
+Here is the description of what I want the script to do:
+
+${prompt}`;
+};
+const generateExplainShellPrompt = (prompt) => {
+    return `${prompt}
+
+## Task
+
+Please briefly explain how this shell script works.`;
+};
+const generateShell = async (gpt, prompt) => {
+    const shellPrompt = generateShellPrompt(prompt);
+    return await gpt.generateText({ prompt: shellPrompt });
+};
+const displayGeneratedShell = (rawShellCommand) => {
+    const generatedShell = (0, string_utils_1.extractMarkdown)(rawShellCommand);
+    const generatedShellWithoutShebang = generatedShell.replace(/^#!.*?\s/, '');
+    console.log((0, cli_highlight_1.default)(generatedShellWithoutShebang));
+};
+const handleUserAction = async (gpt, rawShellCommand, prompt) => {
+    const generatedShell = (0, string_utils_1.extractMarkdown)(rawShellCommand);
+    const generatedShellWithoutShebang = generatedShell.replace(/^#!.*?\s/, '');
+    const action = await (0, prompts_1.select)({
+        message: 'What would you like to do?',
+        choices: [
+            { name: 'Execute', value: 'execute' },
+            { name: 'Revise', value: 'revise' },
+            { name: 'Explain', value: 'explain' },
+            { name: 'Copy to Clipboard', value: 'copy' },
+            { name: 'Cancel', value: 'cancel' },
+        ],
+    });
+    switch (action) {
+        case 'execute':
+            (0, child_process_1.execSync)(generatedShellWithoutShebang, { stdio: 'inherit' });
+            break;
+        case 'revise':
+            const newPrompt = await (0, prompts_1.input)({
+                message: 'How would you like to modify the output:',
+            });
+            const newRawShellCommand = await generateShell(gpt, newPrompt);
+            displayGeneratedShell(newRawShellCommand);
+            await handleUserAction(gpt, newRawShellCommand, newPrompt);
+            break;
+        case 'explain':
+            const explainPrompt = generateExplainShellPrompt(generatedShellWithoutShebang);
+            const explainShellCommand = await gpt.generateText({ prompt: explainPrompt });
+            console.log((0, cli_highlight_1.default)(explainShellCommand, { language: 'markdown' }));
+            break;
+        case 'copy':
+            copy_paste_1.default.copy(generatedShellWithoutShebang, (err) => {
+                if (err) {
+                    console.error('Error: Unable to copy to clipboard.', err);
+                }
+                else {
+                    console.log('Script copied to clipboard.');
+                }
+            });
+            break;
+        case 'cancel':
+            console.log('Operation cancelled.');
+            break;
+    }
+};
+const shell = async (options) => {
+    let { prompt: userPrompt } = options;
+    options.silent = (0, config_utils_1.getConfig)(options).globalSilent || false;
+    const gpt = new cloving_gpt_1.default(options);
+    try {
+        if (!userPrompt) {
+            userPrompt = await (0, prompts_1.input)({
+                message: 'What would you like to do: '
+            });
+        }
+        const rawShellCommand = await generateShell(gpt, userPrompt);
+        displayGeneratedShell(rawShellCommand);
+        await handleUserAction(gpt, rawShellCommand, userPrompt);
+    }
+    catch (error) {
+        console.error('Could not generate shell script', error);
+    }
+};
+exports.default = shell;

package/dist/commands/generate/unit_tests.js

@@ -0,0 +1,314 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const fs_1 = require("fs");
+const child_process_1 = require("child_process");
+const cli_highlight_1 = __importDefault(require("cli-highlight"));
+const prompts_1 = require("@inquirer/prompts");
+const path_1 = __importDefault(require("path"));
+const git_utils_1 = require("../../utils/git_utils");
+const config_utils_1 = require("../../utils/config_utils");
+const string_utils_1 = require("../../utils/string_utils");
+const cloving_gpt_1 = __importDefault(require("../../cloving_gpt"));
+const extractChangedFiles = (gitDiff) => {
+    const fileRegex = /diff --git a\/(.+?) b\/(.+?)\n/g;
+    const files = new Set();
+    let match;
+    while ((match = fileRegex.exec(gitDiff)) !== null) {
+        files.add(match[1]);
+    }
+    return Array.from(files);
+};
+const generatePrompt = (files, srcFiles, testFiles, gitDiff) => {
+    const filesList = files.length > 0 ? files.join('\n') : 'No files provided';
+    return `Enumerate all the files in the provided list${gitDiff ? ' and git diff' : ''} as well as the file names of anything that these files interact with.
+Also, list any test files that might be relevant to these files.
+
+### List of Code Files
+
+${filesList}
+
+### List of All Source Files
+
+${srcFiles}
+
+### List of all test files
+
+${testFiles}
+
+${gitDiff ? '### Git Diff Content' : ''}
+
+${gitDiff || ''}
+
+## Example Output
+
+Here are the files that are relevant:
+
+\`\`\`plaintext
+## files
+
+app/models/foo.rb
+app/views/baz/bar.html.erb
+
+## relevant test files
+
+test/models/foo_test.rb
+test/controllers/baz_controller_test.rb
+\`\`\`
+
+### Prompt
+
+Enumerate all the files in the provided list${gitDiff ? ' and git diff' : ''} as well as the file names of anything that these files interact with.
+Also, list any test files that might be relevant to these files.`;
+};
+const handleUserAction = async (analysis, autoSave = false) => {
+    const [files, fileContents] = (0, string_utils_1.extractFilesAndContent)(analysis);
+    if (autoSave) {
+        for (const file of files) {
+            if (fileContents[file]) {
+                const filePath = path_1.default.resolve(file);
+                await fs_1.promises.mkdir(path_1.default.dirname(filePath), { recursive: true });
+                await fs_1.promises.writeFile(filePath, fileContents[file]);
+                console.log(`${file} has been saved.`);
+            }
+            else {
+                console.log(`File content not found for ${file}.`);
+            }
+        }
+        console.log('All unit test files have been saved.');
+        return;
+    }
+    const action = await (0, prompts_1.select)({
+        message: 'What would you like to do?',
+        choices: [
+            { name: 'Save a Unit Test File', value: 'save' },
+            { name: 'Save All Unit Test Files', value: 'saveAll' },
+            { name: 'Copy Unit Test to Clipboard', value: 'copyTest' },
+            { name: 'Copy Entire Response to Clipboard', value: 'copyAll' },
+            { name: 'Done', value: 'done' },
+        ],
+    });
+    switch (action) {
+        case 'save':
+            let saveAnother = true;
+            while (saveAnother) {
+                const fileToSave = await (0, prompts_1.select)({
+                    message: 'Which unit test file do you want to save?',
+                    choices: files.map(file => ({ name: file, value: file })),
+                });
+                if (fileContents[fileToSave]) {
+                    const filePath = path_1.default.resolve(fileToSave);
+                    await fs_1.promises.mkdir(path_1.default.dirname(filePath), { recursive: true });
+                    await fs_1.promises.writeFile(filePath, fileContents[fileToSave]);
+                    console.log(`${fileToSave} has been saved.`);
+                }
+                else {
+                    console.log('File content not found.');
+                }
+                saveAnother = await (0, prompts_1.confirm)({
+                    message: 'Do you want to save another file?',
+                    default: false,
+                });
+            }
+            break;
+        case 'saveAll':
+            for (const file of files) {
+                if (fileContents[file]) {
+                    const filePath = path_1.default.resolve(file);
+                    await fs_1.promises.mkdir(path_1.default.dirname(filePath), { recursive: true });
+                    await fs_1.promises.writeFile(filePath, fileContents[file]);
+                    console.log(`${file} has been saved.`);
+                }
+                else {
+                    console.log(`File content not found for ${file}.`);
+                }
+            }
+            console.log('All unit test files have been saved.');
+            break;
+        case 'copyTest':
+            const fileToCopy = await (0, prompts_1.select)({
+                message: 'Which unit test file do you want to copy to the clipboard?',
+                choices: files.map(file => ({ name: file, value: file })),
+            });
+            if (fileContents[fileToCopy]) {
+                process.nextTick(() => {
+                    require('copy-paste').copy(fileContents[fileToCopy], () => {
+                        console.log(`${fileToCopy} copied to clipboard`);
+                    });
+                });
+            }
+            else {
+                console.log('File content not found.');
+            }
+            break;
+        case 'copyAll':
+            process.nextTick(() => {
+                require('copy-paste').copy(analysis, () => {
+                    console.log('Entire response copied to clipboard');
+                });
+            });
+            break;
+        case 'done':
+            break;
+    }
+};
+const unitTests = async (options) => {
+    const { files, save } = options;
+    const gpt = new cloving_gpt_1.default(options);
+    const allSrcFiles = await (0, config_utils_1.getAllFiles)(options, true);
+    const testingDirectory = (0, config_utils_1.getTestingDirectory)();
+    if (allSrcFiles.length === 0 || !testingDirectory) {
+        console.error('Could not find any source files to generate unit tests. Please run: cloving init. Then try again.');
+        process.exit(1);
+    }
+    const testFiles = (0, child_process_1.execFileSync)('find', [testingDirectory, '-type', 'f']).toString().trim();
+    let contextFiles;
+    if (files) {
+        console.log('Generating unit tests for the provided files...');
+        console.log('Gathering a list of files related to the changes...');
+        const prompt = generatePrompt(files, allSrcFiles.join('\n'), testFiles);
+        contextFiles = await gpt.generateText({ prompt });
+    }
+    else {
+        console.log('Generating unit tests for the git diff between this branch and the default branch...');
+        console.log('Gathering a list of files related to the changes...');
+        const gitDiff = await (0, git_utils_1.getGitDiff)();
+        const changedFiles = extractChangedFiles(gitDiff);
+        const prompt = generatePrompt(changedFiles, allSrcFiles.join('\n'), testFiles, gitDiff);
+        contextFiles = await gpt.generateText({ prompt });
+    }
+    contextFiles = (0, string_utils_1.extractMarkdown)(contextFiles);
+    console.log(contextFiles);
+    console.log('\nBuilding the tests for these files...');
+    // Initialize variables
+    const lines = [];
+    const context = [];
+    // Read each line of the context files
+    contextFiles.split('\n').forEach((line) => {
+        if (line.trim()) {
+            lines.push(line.trim());
+        }
+    });
+    for (const codeFile of lines) {
+        if (await fs_1.promises.stat(codeFile).then(stat => stat.isFile()).catch(() => false)) {
+            const fileContents = await fs_1.promises.readFile(codeFile, 'utf-8');
+            context.push(`### Contents of **${codeFile}\n\n${fileContents.split('\n').map(line => `    ${line}`).join('\n')}**\n\n`);
+        }
+    }
+    // Generate the message for unit test creation
+    let analysis = '';
+    if ((files || []).length > 0) {
+        const message = `Create unit tests for the List of Files. Always show filenames for the generated code.
+
+## List of Files
+${(files || []).join('\n')}
+
+## Example Response Structure
+
+Here are the unit tests for the provided list of files:
+
+1. **tests/utils/model_utils.test.ts**
+
+\`\`\`typescript
+import { getModel } from '../../src/utils/model_utils'
+
+describe('modelUtils', () => {
+  const originalEnv = process.env
+
+  beforeEach(() => {
+    jest.resetModules()
+    process.env = { ...originalEnv }
+  })
+
+  afterAll(() => {
+    process.env = originalEnv
+  })
+
+  test('getModel should return default model when CLOVING_MODEL is not set', () => {
+    delete process.env.CLOVING_MODEL
+    const result = getModel()
+    expect(result).toBe('claude:claude-3-5-sonnet-20240620')
+  })
+
+  test('getModel should return CLOVING_MODEL when it is set', () => {
+    process.env.CLOVING_MODEL = 'openai:gpt-4'
+    const result = getModel()
+    expect(result).toBe('openai:gpt-4')
+  })
+})
+\`\`\`
+
+## Context
+
+${context.join('\n\n')}
+
+## Prompt
+
+Create unit tests for the List of Files. Always show filenames for the generated code.`;
+        // Get the model and analysis using ClovingGPT
+        analysis = await gpt.generateText({ prompt: message });
+    }
+    else {
+        const message = `Create unit tests for the Code Diff. Always show filenames for the generated code.
+
+## Code Diff
+
+${await (0, git_utils_1.getGitDiff)()}
+
+## Context
+
+${context.join('\n\n')}
+
+## Example Response Structure
+
+Here are the unit tests for the code diff:
+
+1. **tests/utils/model_utils.test.ts**
+
+\`\`\`typescript
+import { getModel } from '../../src/utils/model_utils'
+
+describe('modelUtils', () => {
+  const originalEnv = process.env
+
+  beforeEach(() => {
+    jest.resetModules()
+    process.env = { ...originalEnv }
+  })
+
+  afterAll(() => {
+    process.env = originalEnv
+  })
+
+  test('getModel should return default model when CLOVING_MODEL is not set', () => {
+    delete process.env.CLOVING_MODEL
+    const result = getModel()
+    expect(result).toBe('claude:claude-3-5-sonnet-20240620')
+  })
+
+  test('getModel should return CLOVING_MODEL when it is set', () => {
+    process.env.CLOVING_MODEL = 'openai:gpt-4'
+    const result = getModel()
+    expect(result).toBe('openai:gpt-4')
+  })
+})
+\`\`\`
+
+## Prompt
+
+Create unit tests for the Code Diff. Always show filenames for the generated code.`;
+        // Get the model and analysis using ClovingGPT
+        analysis = await gpt.generateText({ prompt: message });
+    }
+    if (analysis) {
+        console.log((0, cli_highlight_1.default)(analysis));
+    }
+    else {
+        console.log('No unit tests generated.');
+    }
+    // Add the new user action handling
+    await handleUserAction(analysis, save);
+};
+exports.default = unitTests;