From af6d37a8e27f0573b7d4cc17b1e7dd189feff517 Mon Sep 17 00:00:00 2001 From: Michael Neale Date: Wed, 22 Oct 2025 14:58:35 +1100 Subject: [PATCH] Add --interactive-requirements flag for AI-enhanced requirements mode - Adds new --interactive-requirements CLI flag for autonomous mode - Prompts user for brief requirements input - Uses AI to enhance and structure requirements into proper markdown - Shows enhanced requirements and allows user to approve/edit/cancel - Saves to requirements.md and proceeds with autonomous mode if approved - Includes test script for manual verification --- crates/g3-cli/src/lib.rs | 110 +++++++++++++++++++++++++++++++++++++++ test-ai-requirements.sh | 39 ++++++++++++++ 2 files changed, 149 insertions(+) create mode 100755 test-ai-requirements.sh diff --git a/crates/g3-cli/src/lib.rs b/crates/g3-cli/src/lib.rs index e901f22..be0dc19 100644 --- a/crates/g3-cli/src/lib.rs +++ b/crates/g3-cli/src/lib.rs @@ -216,6 +216,10 @@ pub struct Cli { #[arg(long, value_name = "TEXT")] pub requirements: Option, + /// Interactive mode: prompt for requirements and save to requirements.md before starting autonomous mode + #[arg(long)] + pub interactive_requirements: bool, + /// Use retro terminal UI (inspired by 80s sci-fi) #[arg(long)] pub retro: bool, @@ -303,6 +307,112 @@ pub async fn run() -> Result<()> { // Create project model let project = if cli.autonomous { + // Handle interactive requirements mode with AI enhancement + if cli.interactive_requirements { + println!("\nšŸ“ Interactive Requirements Mode"); + println!("================================\n"); + println!("Describe what you want to build (can be brief):"); + println!("Press Ctrl+D (Unix) or Ctrl+Z (Windows) when done.\n"); + + use std::io::{self, Read, Write}; + let mut requirements_input = String::new(); + io::stdin().read_to_string(&mut requirements_input)?; + + if requirements_input.trim().is_empty() { + anyhow::bail!("No requirements provided. Exiting."); + } + + println!("\nšŸ¤– Enhancing your requirements with AI...\n"); + + // Create a temporary agent to enhance the requirements + let temp_config = Config::load_with_overrides( + cli.config.as_deref(), + cli.provider.clone(), + cli.model.clone(), + )?; + + let ui_writer = ConsoleUiWriter::new(); + let mut temp_agent = Agent::new_with_readme_and_quiet( + temp_config, + ui_writer, + None, + true, // quiet mode + ).await?; + + // Craft the enhancement prompt + let enhancement_prompt = format!( + r#"You are a requirements analyst. Take this brief user input and expand it into a structured requirements document. + +USER INPUT: +{} + +Create a professional requirements document with: +1. A clear project title (# heading) +2. An overview section explaining what will be built +3. Organized requirements (functional, technical, quality) +4. Acceptance criteria +5. Any technical constraints or preferences mentioned + +Format as proper markdown. Be specific and actionable. If the user's input is vague, make reasonable assumptions but keep it focused on what they described. + +Output ONLY the markdown content, no explanations or meta-commentary."#, + requirements_input.trim() + ); + + // Execute enhancement task + let result = temp_agent + .execute_task_with_timing(&enhancement_prompt, None, false, false, false, false) + .await?; + + let enhanced_requirements = result.response.trim().to_string(); + + // Show the enhanced requirements + println!("\nšŸ“‹ Enhanced Requirements Document:"); + println!("{}\n", "=".repeat(60)); + println!("{}", enhanced_requirements); + println!("{}\n", "=".repeat(60)); + + // Ask for confirmation + println!("\nā“ Is this requirements document acceptable?"); + println!(" [y] Yes, proceed with autonomous mode"); + println!(" [e] Edit and save manually"); + println!(" [n] No, cancel\n"); + + print!("Your choice (y/e/n): "); + io::stdout().flush()?; + + let mut choice = String::new(); + io::stdin().read_line(&mut choice)?; + let choice = choice.trim().to_lowercase(); + + let requirements_path = workspace_dir.join("requirements.md"); + + match choice.as_str() { + "y" | "yes" => { + // Save enhanced requirements + std::fs::write(&requirements_path, &enhanced_requirements)?; + println!("\nāœ… Requirements saved to: {}", requirements_path.display()); + println!("šŸš€ Starting autonomous mode...\n"); + } + "e" | "edit" => { + // Save enhanced requirements for manual editing + std::fs::write(&requirements_path, &enhanced_requirements)?; + println!("\nāœ… Requirements saved to: {}", requirements_path.display()); + println!("šŸ“ Please edit the file and run: g3 --autonomous"); + println!(" Exiting for now.\n"); + return Ok(()); + } + "n" | "no" => { + println!("\nāŒ Cancelled. No files were saved.\n"); + return Ok(()); + } + _ => { + println!("\nāŒ Invalid choice. Cancelled.\n"); + return Ok(()); + } + } + } + if let Some(requirements_text) = cli.requirements { // Use requirements text override Project::new_autonomous_with_requirements(workspace_dir.clone(), requirements_text)? diff --git a/test-ai-requirements.sh b/test-ai-requirements.sh new file mode 100755 index 0000000..06c97fc --- /dev/null +++ b/test-ai-requirements.sh @@ -0,0 +1,39 @@ +#!/bin/bash +# Test script for AI-enhanced interactive requirements mode + +echo "Testing AI-enhanced interactive requirements mode..." +echo "" + +# Create a test workspace +TEST_WORKSPACE="/tmp/g3-test-interactive-$(date +%s)" +mkdir -p "$TEST_WORKSPACE" + +echo "Test workspace: $TEST_WORKSPACE" +echo "" + +# Create sample brief input +BRIEF_INPUT="build a calculator cli in rust with basic operations" + +echo "Brief input:" +echo "---" +echo "$BRIEF_INPUT" +echo "---" +echo "" + +echo "This will:" +echo "1. Send brief input to AI" +echo "2. AI generates structured requirements.md" +echo "3. Show enhanced requirements" +echo "4. Prompt for confirmation (y/e/n)" +echo "" + +echo "To test manually, run:" +echo "cargo run -- --autonomous --interactive-requirements --workspace $TEST_WORKSPACE" +echo "" +echo "Then type: $BRIEF_INPUT" +echo "Press Ctrl+D" +echo "Review the AI-generated requirements" +echo "Choose 'y' to proceed, 'e' to edit, or 'n' to cancel" +echo "" + +echo "Test workspace will be at: $TEST_WORKSPACE"