cloving 0.1.15 → 0.1.17

package/dist/commands/generate/review.js

@@ -4,7 +4,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 const cli_highlight_1 = __importDefault(require("cli-highlight"));
-const inquirer_1 = __importDefault(require("inquirer"));
+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");
@@ -16,7 +16,7 @@ const review = async (options) => {
     try {
         // Define the prompt for analysis
         const gitDiff = await (0, git_utils_1.getGitDiff)();
-        const prompt = `==== begin diff =====
+        const promptText = `==== begin diff =====
 ${gitDiff}
 ==== end diff =====
 
@@ -77,7 +77,7 @@ I would like you to explain why these change are being made and document a descr
 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 });
+        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('```')) {
@@ -104,19 +104,15 @@ Format the output of this code review in Markdown format.`;
             return match ? match[0].trim() : '';
         };
         // Prompt the user with options for copying to clipboard
-        const { clipboardOption } = await inquirer_1.default.prompt([
-            {
-                type: 'list',
-                name: 'clipboardOption',
-                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' }
-                ]
-            }
-        ]);
+        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':

package/dist/commands/generate/shell.js

@@ -3,13 +3,13 @@ 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 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();
@@ -48,32 +48,24 @@ const displayGeneratedShell = (rawShellCommand) => {
 const handleUserAction = async (gpt, rawShellCommand, prompt) => {
     const generatedShell = (0, string_utils_1.extractMarkdown)(rawShellCommand);
     const generatedShellWithoutShebang = generatedShell.replace(/^#!.*?\s/, '');
-    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' },
-            ],
-        },
-    ]);
+    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 inquirer_1.default.prompt([
-                {
-                    type: 'input',
-                    name: 'newPrompt',
-                    message: 'How would you like to modify the output:',
-                },
-            ]);
+            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);
@@ -99,23 +91,18 @@ const handleUserAction = async (gpt, rawShellCommand, prompt) => {
     }
 };
 const shell = async (options) => {
-    let { prompt } = 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 (!prompt) {
-            const { userPrompt } = await inquirer_1.default.prompt([
-                {
-                    type: 'input',
-                    name: 'userPrompt',
-                    message: 'What would you like to do: '
-                }
-            ]);
-            prompt = userPrompt;
+        if (!userPrompt) {
+            userPrompt = await (0, prompts_1.input)({
+                message: 'What would you like to do: '
+            });
         }
-        const rawShellCommand = await generateShell(gpt, prompt);
+        const rawShellCommand = await generateShell(gpt, userPrompt);
         displayGeneratedShell(rawShellCommand);
-        await handleUserAction(gpt, rawShellCommand, prompt);
+        await handleUserAction(gpt, rawShellCommand, userPrompt);
     }
     catch (error) {
         console.error('Could not generate shell script', error);

package/dist/commands/generate/unit_tests.js

@@ -6,7 +6,7 @@ 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 inquirer_1 = __importDefault(require("inquirer"));
+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");
@@ -23,7 +23,10 @@ const extractChangedFiles = (gitDiff) => {
 };
 const generatePrompt = (files, srcFiles, testFiles, gitDiff) => {
     const filesList = files.length > 0 ? files.join('\n') : 'No files provided';
-    return `### List of Code Files
+    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}
 
@@ -35,13 +38,13 @@ ${srcFiles}
 
 ${testFiles}
 
-### Git Diff Content
+${gitDiff ? '### Git Diff Content' : ''}
 
-${gitDiff}
+${gitDiff || ''}
 
 ## Example Output
 
-Give me this output format for your answer:
+Here are the files that are relevant:
 
 \`\`\`plaintext
 ## files
@@ -55,69 +58,68 @@ test/models/foo_test.rb
 test/controllers/baz_controller_test.rb
 \`\`\`
 
-### Request
-
-Please 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.
+### 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) => {
+const handleUserAction = async (analysis, autoSave = false) => {
     const [files, fileContents] = (0, string_utils_1.extractFilesAndContent)(analysis);
-    const { action } = await inquirer_1.default.prompt([
-        {
-            type: 'list',
-            name: 'action',
-            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' },
-            ],
-        },
-    ]);
+    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 inquirer_1.default.prompt([
-                    {
-                        type: 'list',
-                        name: 'fileToSave',
-                        message: 'Which unit test file do you want to save?',
-                        choices: files.map(file => ({ name: file, value: file })),
-                    },
-                ]);
+                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);
-                    fs_1.promises.mkdir(path_1.default.dirname(filePath), { recursive: true })
-                        .then(() => fs_1.promises.writeFile(filePath, fileContents[fileToSave]))
-                        .then(() => console.log(`${fileToSave} has been saved.`))
-                        .catch((error) => console.error(`Error saving ${fileToSave}:`, error));
+                    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.');
                 }
-                const { saveMore } = await inquirer_1.default.prompt([
-                    {
-                        type: 'confirm',
-                        name: 'saveMore',
-                        message: 'Do you want to save another file?',
-                        default: false,
-                    },
-                ]);
-                saveAnother = saveMore;
+                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);
-                    fs_1.promises.mkdir(path_1.default.dirname(filePath), { recursive: true })
-                        .then(() => fs_1.promises.writeFile(filePath, fileContents[file]))
-                        .then(() => console.log(`${file} has been saved.`))
-                        .catch((error) => console.error(`Error saving ${file}:`, error));
+                    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}.`);
@@ -126,14 +128,10 @@ const handleUserAction = async (analysis) => {
             console.log('All unit test files have been saved.');
             break;
         case 'copyTest':
-            const { fileToCopy } = await inquirer_1.default.prompt([
-                {
-                    type: 'list',
-                    name: 'fileToCopy',
-                    message: 'Which unit test file do you want to copy to the clipboard?',
-                    choices: files.map(file => ({ name: file, value: file })),
-                },
-            ]);
+            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], () => {
@@ -157,7 +155,7 @@ const handleUserAction = async (analysis) => {
     }
 };
 const unitTests = async (options) => {
-    const { files } = 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)();
@@ -168,15 +166,22 @@ const unitTests = async (options) => {
     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 = [];
@@ -189,38 +194,121 @@ const unitTests = async (options) => {
     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}\n\n### End of ${codeFile}`);
+            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 = `please create unit tests for these files:
+        const message = `Create unit tests for the List of Files. Always show filenames for the generated code.
 
-## Begin list of files
+## List of Files
 ${(files || []).join('\n')}
-## End list of files
+
+## 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')}`;
+
+${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 = `please create unit tests for these changes:
+        const message = `Create unit tests for the Code Diff. Always show filenames for the generated code.
+
+## Code Diff
 
-## Begin diff
 ${await (0, git_utils_1.getGitDiff)()}
-## End diff
 
 ## Context
 
-${context.join('\n\n')}`;
+${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 });
     }
-    console.log((0, cli_highlight_1.default)(analysis));
+    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);
+    await handleUserAction(analysis, save);
 };
 exports.default = unitTests;

package/dist/commands/init.js

@@ -7,7 +7,7 @@ exports.init = void 0;
 const fs_1 = __importDefault(require("fs"));
 const os_1 = __importDefault(require("os"));
 const path_1 = __importDefault(require("path"));
-const inquirer_1 = __importDefault(require("inquirer"));
+const prompts_1 = require("@inquirer/prompts");
 const cloving_gpt_1 = __importDefault(require("../cloving_gpt"));
 const ignore_1 = __importDefault(require("ignore"));
 const string_utils_1 = require("../utils/string_utils");
@@ -42,14 +42,10 @@ technologies used. This will provide better context for future Cloving requests.
         return;
     }
     if (!(0, command_utils_1.checkForSpecialFiles)()) {
-        const { technologies } = await inquirer_1.default.prompt([
-            {
-                type: 'input',
-                name: 'technologies',
-                message: 'No special files detected. Please describe the main technologies used in your project:',
-                validate: (input) => input.trim().length > 0 || 'Please provide a description of the technologies used.',
-            },
-        ]);
+        const technologies = await (0, prompts_1.input)({
+            message: 'No special files detected. Please describe the main technologies used in your project:',
+            validate: (input) => input.trim().length > 0 || 'Please provide a description of the technologies used.',
+        });
         specialFileContents['description'] = technologies;
     }
     const tempFilePath = path_1.default.join(os_1.default.tmpdir(), `describe_${Date.now()}.tmp`);
@@ -207,14 +203,10 @@ Here is an example response:
         console.log('[done] Project data saved to cloving.json');
         // Prompt the user if they want to review the generated cloving.json
         if (!options.silent) {
-            const { reviewAnswer } = await inquirer_1.default.prompt([
-                {
-                    type: 'confirm',
-                    name: 'reviewAnswer',
-                    message: 'Do you want to review the generated data?',
-                    default: true,
-                },
-            ]);
+            const reviewAnswer = await (0, prompts_1.confirm)({
+                message: 'Do you want to review the generated data?',
+                default: true,
+            });
             if (reviewAnswer) {
                 console.log(cleanAiChatResponse);
             }

package/dist/commands/project/build.js

@@ -7,7 +7,7 @@ exports.buildProject = void 0;
 const copy_paste_1 = __importDefault(require("copy-paste"));
 const fs_1 = require("fs");
 const cli_highlight_1 = __importDefault(require("cli-highlight"));
-const inquirer_1 = __importDefault(require("inquirer"));
+const prompts_1 = require("@inquirer/prompts");
 const cloving_gpt_1 = __importDefault(require("../../cloving_gpt"));
 const config_utils_1 = require("../../utils/config_utils");
 const git_utils_1 = require("../../utils/git_utils");
@@ -42,15 +42,11 @@ const buildProject = async (options) => {
         console.log('No plan items found.');
         process.exit(1);
     }
-    // Use inquirer to select a plan item
-    const { selectedPlanItemKey } = await inquirer_1.default.prompt([
-        {
-            type: 'list',
-            name: 'selectedPlanItemKey',
-            message: 'Select a plan item:',
-            choices: planItems.map(([key, item]) => ({ name: key, value: key })),
-        },
-    ]);
+    // Use @inquirer/prompts to select a plan item
+    const selectedPlanItemKey = await (0, prompts_1.select)({
+        message: 'Select a plan item:',
+        choices: planItems.map(([key, item]) => ({ name: key, value: key })),
+    });
     const selectedPlanItem = planItems.find(([key]) => key === selectedPlanItemKey);
     if (!selectedPlanItem) {
         console.log('Selected plan item not found.');
@@ -77,15 +73,11 @@ const buildProject = async (options) => {
             console.log((0, cli_highlight_1.default)(code, { language: 'markdown' }));
         }
     });
-    // Prompt the user to copy the project to the clipboard using inquirer
-    const { copyToClipboard } = await inquirer_1.default.prompt([
-        {
-            type: 'confirm',
-            name: 'copyToClipboard',
-            message: 'Do you want to copy this project to the clipboard?',
-            default: true,
-        },
-    ]);
+    // Prompt the user to copy the project to the clipboard using @inquirer/prompts
+    const copyToClipboard = await (0, prompts_1.confirm)({
+        message: 'Do you want to copy this project to the clipboard?',
+        default: true,
+    });
     if (copyToClipboard) {
         copy_paste_1.default.copy(projectCode, () => {
             console.log('Project copied to clipboard');

package/dist/commands/project/init.js

@@ -5,7 +5,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.initProject = void 0;
 const fs_1 = require("fs");
-const inquirer_1 = __importDefault(require("inquirer"));
+const prompts_1 = require("@inquirer/prompts");
 const cloving_gpt_1 = __importDefault(require("../../cloving_gpt"));
 const config_utils_1 = require("../../utils/config_utils");
 const git_utils_1 = require("../../utils/git_utils");
@@ -52,13 +52,9 @@ const initProject = async (options) => {
         console.log(`You are on the ${defaultBranchName} branch. Please checkout to a git branch and try again. For example: git branch -b my-new-project`);
         process.exit(1);
     }
-    const { projectTask } = await inquirer_1.default.prompt([
-        {
-            type: 'input',
-            name: 'projectTask',
-            message: 'Describe the task you want to accomplish with this project:',
-        },
-    ]);
+    const projectTask = await (0, prompts_1.input)({
+        message: 'Describe the task you want to accomplish with this project:',
+    });
     const prompt = await generatePrompt(projectName, projectTask);
     const contextFiles = await gpt.generateText({ prompt });
     // Initialize variables

package/dist/index.js

@@ -65,6 +65,7 @@ generate
     .option('-s, --silent', 'Run the command without asking for confirmation of submitting prompts')
     .option('-m, --model <model>', 'Select the model to use (e.g., openai, claude, ollama, ollama:llama3, claude:claude-3-5-sonnet-20240620)')
     .option('-p, --prompt <prompt>', 'Specify the prompt to use')
+    .option('-t, --temperature <temperature>', 'Temperature for the model (default 0.2)')
     .action(shell_1.default);
 generate
     .command('code')
@@ -74,12 +75,14 @@ generate
     .option('-i, --interactive', 'Auto-save generated code and then show a new prompt with existing context to revise the code further')
     .option('-s, --silent', 'Run the command without asking for confirmation of submitting prompts')
     .option('-m, --model <model>', 'Select the model to use (e.g., openai, claude, ollama, ollama:llama3, claude:claude-3-5:sonnet-20240620)')
+    .option('-t, --temperature <temperature>', 'Temperature for the model (default 0.2)')
     .option('-p, --prompt <prompt>', 'Specify the prompt to use')
     .option('-f, --files <filenames...>', 'Specify filenames of files with context to use for generating code')
     .action(code_1.default);
 generate
     .command('commit')
     .description('Generate a commit message and commit the changes')
+    .option('-t, --temperature <temperature>', 'Temperature for the model (default 0.2)')
     .option('-s, --silent', 'Run the command without asking for confirmation of submitting prompts')
     .option('-m, --model <model>', 'Select the model to use (e.g., openai, claude, ollama, ollama:llama3, claude:claude-3-5-sonnet-20240620)')
     .action(commit_1.default);
@@ -92,9 +95,11 @@ generate
     .command('unit-tests')
     .alias('u')
     .description('Generate unit tests (if you don\'t specify filenames, it will generate tests for commited changes that differ from the main/master branch)')
+    .option('--save', 'Automatically save all files after generating the unit tests')
     .option('-s, --silent', 'Run the command without asking for confirmation of submitting prompts')
     .option('-f, --files <filenames...>', 'Specify filenames for the unit tests')
     .option('-m, --model <model>', 'Select the model to use (e.g., openai, claude, ollama, ollama:llama3, claude:claude-3-5-sonnet-20240620)')
+    .option('-t, --temperature <temperature>', 'Temperature for the model (default 0.2)')
     .action(unit_tests_1.default);
 generate
     .command('review')
@@ -102,6 +107,7 @@ generate
     .description('Review the code for commited changes that differ from the main/master branch')
     .option('-s, --silent', 'Run the command without asking for confirmation of submitting prompts')
     .option('-m, --model <model>', 'Select the model to use (e.g., openai, claude, ollama, ollama:llama3, claude:claude-3-5-sonnet-20240620)')
+    .option('-t, --temperature <temperature>', 'Temperature for the model (default 0.2)')
     .action(review_1.default);
 // Project commands
 const project = program