diff --git a/opc/scripts/setup/wizard.py b/opc/scripts/setup/wizard.py index 142ddc78..6255876a 100644 --- a/opc/scripts/setup/wizard.py +++ b/opc/scripts/setup/wizard.py @@ -407,7 +407,7 @@ async def prompt_api_keys() -> dict[str, str]: """Prompt user for optional API keys. Returns: - dict with keys: perplexity, nia, braintrust + dict with keys: perplexity, nia, braintrust, firecrawl """ console.print("\n[bold]API Keys (optional)[/bold]") console.print("Press Enter to skip any key you don't have.\n") @@ -415,11 +415,13 @@ async def prompt_api_keys() -> dict[str, str]: perplexity = Prompt.ask("Perplexity API key (web search)", default="") nia = Prompt.ask("Nia API key (documentation search)", default="") braintrust = Prompt.ask("Braintrust API key (observability)", default="") + firecrawl = Prompt.ask("Firecrawl API key (web scraping)", default="") return { "perplexity": perplexity, "nia": nia, "braintrust": braintrust, + "firecrawl": firecrawl, } @@ -500,6 +502,8 @@ def generate_env_file(config: dict[str, Any], env_path: Path) -> None: lines.append(f"NIA_API_KEY={api_keys['nia']}") if api_keys.get("braintrust"): lines.append(f"BRAINTRUST_API_KEY={api_keys['braintrust']}") + if api_keys.get("firecrawl"): + lines.append(f"FIRECRAWL_API_KEY={api_keys['firecrawl']}") lines.append("") # Write file @@ -616,7 +620,7 @@ async def run_setup_wizard() -> None: if Confirm.ask("Configure API keys?", default=False): api_keys = await prompt_api_keys() else: - api_keys = {"perplexity": "", "nia": "", "braintrust": ""} + api_keys = {"perplexity": "", "nia": "", "braintrust": "", "firecrawl": ""} # Step 5: Generate .env console.print("\n[bold]Step 5/13: Generating configuration...[/bold]")