UNPKG

gaunt-sloth-assistant

Version:

[![Tests and Lint](https://github.com/Galvanized-Pukeko/gaunt-sloth-assistant/actions/workflows/unit-tests.yml/badge.svg)](https://github.com/Galvanized-Pukeko/gaunt-sloth-assistant/actions/workflows/unit-tests.yml) [![Integration Tests](https://github.co

62 lines 3.47 kB
import { Option } from 'commander'; import { readBackstory, readGuidelines, readReviewInstructions, readSystemPrompt, wrapContent, } from '#src/prompt.js'; import { readMultipleFilesFromProjectDir } from '#src/utils.js'; import { getStringFromStdin } from '#src/systemUtils.js'; import { REQUIREMENTS_PROVIDERS, CONTENT_PROVIDERS, getRequirementsFromProvider, getContentFromProvider, } from '#src/commands/commandUtils.js'; export function reviewCommand(program, cliConfigOverrides) { program .command('review') .description('Review provided diff or other content') .argument('[contentId]', 'Optional content ID argument to retrieve content with content provider') .alias('r') // TODO add provider to get results of git --no-pager diff .option('-f, --file [files...]', 'Input files. Content of these files will be added BEFORE the diff, but after requirements') // TODO figure out what to do with this (we probably want to merge it with requirementsId)? .option('-r, --requirements <requirements>', 'Requirements for this review.') .addOption(new Option('-p, --requirements-provider <requirementsProvider>', 'Requirements provider for this review.').choices(Object.keys(REQUIREMENTS_PROVIDERS))) .addOption(new Option('--content-provider <contentProvider>', 'Content provider').choices(Object.keys(CONTENT_PROVIDERS))) .option('-m, --message <message>', 'Extra message to provide just before the content') .action(async (contentId, options) => { const { initConfig } = await import('#src/config.js'); const config = await initConfig(cliConfigOverrides); // Initialize and get config const systemPrompt = readSystemPrompt(); const systemMessage = [ readBackstory(), readGuidelines(config.projectGuidelines), readReviewInstructions(config.projectReviewInstructions), ]; if (systemPrompt) { systemMessage.push(systemPrompt); } const content = []; const requirementsId = options.requirements; const requirementsProvider = options.requirementsProvider ?? config?.commands?.review?.requirementsProvider ?? config?.requirementsProvider; const contentProvider = options.contentProvider ?? config?.commands?.review?.contentProvider ?? config?.contentProvider; // TODO consider calling these in parallel const requirements = await getRequirementsFromProvider(requirementsProvider, requirementsId, config); if (requirements) { content.push(requirements); } const providedContent = await getContentFromProvider(contentProvider, contentId, config); if (providedContent) { content.push(providedContent); } if (options.file) { content.push(readMultipleFilesFromProjectDir(options.file)); } const stringFromStdin = getStringFromStdin(); if (stringFromStdin) { content.push(wrapContent(stringFromStdin, 'stdin-content')); } if (options.message) { content.push(wrapContent(options.message, 'message', 'user message')); } const { review } = await import('#src/modules/reviewModule.js'); await review('REVIEW', systemMessage.join('\n'), content.join('\n'), config); }); } //# sourceMappingURL=reviewCommand.js.map