1 line
11 KiB
JSON
1 line
11 KiB
JSON
{"repo": "adriel007/matrioska", "files": {"src/matrioska/cli/init_wizard.py": {"7b5683c5cd36cb1e791c80e4eff8c1bd266d92ba": {"t": 1785674548.011795, "n": 9943, "h": "8195412eb2da4e0d", "c": "\"\"\"Interactive `.env` scaffolding for `matrioska init`.\n\nPrompts the user for provider, model, and API key, detects existing\n`.env`, and writes a Matrioska-ready environment file. Optionally\nseeds `MATRIOSKA.md` with project conventions.\n\"\"\"\n\nfrom __future__ import annotations\n\nimport os\nimport sys\nfrom pathlib import Path\nfrom typing import Dict, List, Optional, Tuple\n\n# Provider catalog: (id, display name, default base_url, suggested model, key_hint)\n_PROVIDERS: List[Tuple[str, str, str, str, str]] = [\n (\"openai\", \"OpenAI\", \"https://api.openai.com/v1\", \"gpt-4o-mini\", \"sk-...\"),\n (\"anthropic\", \"Anthropic\", \"https://api.anthropic.com\", \"claude-sonnet-4\", \"sk-ant-...\"),\n (\"groq\", \"Groq\", \"https://api.groq.com/openai/v1\", \"llama-3.3-70b-versatile\", \"gsk_...\"),\n (\"openrouter\", \"OpenRouter\", \"https://openrouter.ai/api/v1\", \"anthropic/claude-sonnet-4\", \"sk-or-...\"),\n (\"deepseek\", \"DeepSeek\", \"https://api.deepseek.com/v1\", \"deepseek-chat\", \"ds_...\"),\n (\"xai\", \"xAI (Grok)\", \"https://api.x.ai/v1\", \"grok-2-latest\", \"xai-...\"),\n (\"mistral\", \"Mistral\", \"https://api.mistral.ai/v1\", \"mistral-large-latest\", \"...\"),\n (\"together\", \"Together\", \"https://api.together.xyz/v1\", \"Qwen/Qwen2.5-Coder-32B-Instruct\", \"...\"),\n (\"nvidia\", \"NVIDIA NIM\", \"https://integrate.api.nvidia.com/v1\", \"meta/llama-3.1-70b-instruct\", \"nvapi-...\"),\n (\"ollama\", \"Ollama (local)\", \"http://localhost:11434/v1\", \"llama3.1:8b\", \"ollama\"),\n]\n\n\ndef _ask(prompt: str, default: str = \"\") -> str:\n \"\"\"Prompt the user, returning the input (or default if blank).\"\"\"\n suffix = f\" [{default}]\" if default else \"\"\n try:\n ans = input(f\"{prompt}{suffix}: \").strip()\n except EOFError:\n return default\n return ans or default\n\n\ndef _ask_choice(prompt: str, choices: List[str], default_idx: int = 0) -> int:\n \"\"\"Numeric multiple-choice prompt. Returns the chosen index.\"\"\"\n print(prompt)\n for i, label in enumerate(choices, 1):\n marker = \"*\" if i - 1 == default_idx else \" \"\n print(f\" {marker} {i}) {label}\")\n while True:\n raw = _ask(\"Choose\", default=str(default_idx + 1))\n if not raw.isdigit():\n print(\" → please enter a number\")\n continue\n n = int(raw)\n if 1 <= n <= len(choices):\n return n - 1\n print(f\" → out of range (1–{len(choices)})\")\n\n\ndef _ask_yes_no(prompt: str, default: bool = True) -> bool:\n suffix = \"Y/n\" if default else \"y/N\"\n raw = _ask(f\"{prompt} ({suffix})\", default=\"Y\" if default else \"N\").lower()\n return raw.startswith(\"y\") if raw else default\n\n\ndef _read_existing(env_path: Path) -> Dict[str, str]:\n out: Dict[str, str] = {}\n if not env_path.exists():\n return out\n for raw in env_path.read_text(encoding=\"utf-8\", errors=\"ignore\").splitlines():\n line = raw.strip()\n if not line or line.startswith(\"#\") or \"=\" not in line:\n continue\n k, _, v = line.partition(\"=\")\n out[k.strip()] = v.strip().strip('\"').strip(\"'\")\n return out\n\n\ndef _render_env(values: Dict[str, str]) -> str:\n \"\"\"Render the .env file with grouped sections + comments.\"\"\"\n sections = [\n (\"Provider\", [\"MATRIOSKA_PROVIDER\", \"MATRIOSKA_BASE_URL\",\n \"MATRIOSKA_API_KEY\", \"MATRIOSKA_API_KEYS\", \"MATRIOSKA_EXTRA_ENDPOINTS\"]),\n (\"Models\", [\"MATRIOSKA_MODEL\", \"MATRIOSKA_ARCHITECT_MODEL\",\n \"MATRIOSKA_GENERATOR_MODEL\", \"MATRIOSKA_VALIDATOR_MODEL\",\n \"MATRIOSKA_JUDGE_MODEL\", \"MATRIOSKA_REPAIRER_MODEL\"]),\n (\"Pipeline\", [\"MATRIOSKA_MAX_REPAIRS\", \"MATRIOSKA_MAX_DEPTH\",\n \"MATRIOSKA_PARALLEL\", \"MATRIOSKA_ARCHITECT_CANDIDATES\"]),\n (\"Phases\", [\"MATRIOSKA_ENABLE_TOT\", \"MATRIOSKA_ENABLE_REFLEXION\",\n \"MATRIOSKA_ENABLE_TEST_DESIGN\", \"MATRIOSKA_USE_ACI_REPAIR\",\n \"MATRIOSKA_ENABLE_SANDBOX\"]),\n (\"Vault & misc\", [\"MATRIOSKA_ENABLE_VAULT\", \"MATRIOSKA_VAULT_DIR\",\n \"MATRIOSKA_LOG_LEVEL\"]),\n ]\n lines = [\n \"# Generated by `matrioska init`\",\n \"# Edit freely — every key can also be overridden via CLI flag.\",\n \"\",\n ]\n written = set()\n for title, keys in sections:\n block: List[str] = []\n for k in keys:\n if k in values and values[k] != \"\":\n block.append(f'{k}={values[k]}')\n written.add(k)\n if block:\n lines.append(f\"# ── {title} ─────────────────────────────────────────\")\n lines.extend(block)\n lines.append(\"\")\n # Spill remaining keys\n leftovers = [k for k in values if k not in written and values[k] != \"\"]\n if leftovers:\n lines.append(\"# ── Other ──────────────────────────────────────────────\")\n for k in leftovers:\n lines.append(f'{k}={values[k]}')\n lines.append(\"\")\n return \"\\n\".join(lines).rstrip() + \"\\n\"\n\n\n_MATRIOSKA_MD_TEMPLATE = \"\"\"# {project_name}\n\n## Project conventions\n\nNotes injected into the Architect on every run. Edit freely.\n\n### Stack\n- Language: Python 3.11+\n- Frameworks: <add yours>\n- Storage: <add yours>\n\n### Style\n- Prefer small, focused modules\n- Avoid unnecessary abstractions until 3rd repetition\n- Type hints on public functions; docstrings only when the why is non-obvious\n\n### Don'ts\n- No global mutable state\n- No silent exception swallowing\n\n### Notes for the agent\n- <add domain notes or constraints here>\n\"\"\"\n\n\ndef run_init(cwd: Optional[Path] = None) -> int:\n \"\"\"Interactive `.env` scaffolder. Returns process exit code.\"\"\"\n cwd = cwd or Path.cwd()\n env_path = cwd / \".env\"\n\n print(\"─\" * 60)\n print(\" Matrioska init\")\n print(\"─\" * 60)\n print(f\" Target: {env_path}\")\n print()\n\n existing = _read_existing(env_path)\n if existing:\n print(f\"Found existing .env with {len(existing)} key(s).\")\n action = _ask_choice(\n \"What should I do?\",\n [\"Merge (keep existing values, fill missing)\",\n \"Overwrite (replace with new values)\",\n \"Abort\"],\n default_idx=0,\n )\n if action == 2:\n print(\"Aborted.\")\n return 0\n merge = action == 0\n else:\n merge = False\n\n # Provider selection\n labels = [f\"{disp} — model: {model}\" for _id, disp, _url, model, _hint in _PROVIDERS]\n current_provider = existing.get(\"MATRIOSKA_PROVIDER\", \"\")\n default_idx = next(\n (i for i, (pid, *_r) in enumerate(_PROVIDERS) if pid == current_provider),\n 0,\n )\n idx = _ask_choice(\"\\nPick a provider:\", labels, default_idx=default_idx)\n pid, _disp, default_url, default_model, key_hint = _PROVIDERS[idx]\n\n base_url = _ask(\n \"Base URL\",\n default=existing.get(\"MATRIOSKA_BASE_URL\") or default_url,\n )\n\n api_key = \"\"\n if pid != \"ollama\":\n api_key_default = existing.get(\"MATRIOSKA_API_KEY\", \"\")\n prompt_label = \"API key\" + (f\" (hint: {key_hint})\" if not api_key_default else \" (press Enter to keep current)\")\n api_key = _ask(prompt_label, default=api_key_default)\n else:\n api_key = \"ollama\"\n\n model = _ask(\"Default model\", default=existing.get(\"MATRIOSKA_MODEL\") or default_model)\n\n use_role_models = _ask_yes_no(\"\\nSet per-role model overrides? (architect/generator/validator/judge)\", default=False)\n arch_model = gen_model = val_model = judge_model = rep_model = \"\"\n if use_role_models:\n arch_model = _ask(\"Architect model\", default=existing.get(\"MATRIOSKA_ARCHITECT_MODEL\") or model)\n gen_model = _ask(\"Generator model\", default=existing.get(\"MATRIOSKA_GENERATOR_MODEL\") or model)\n val_model = _ask(\"Validator model\", default=existing.get(\"MATRIOSKA_VALIDATOR_MODEL\") or model)\n judge_model = _ask(\"Judge model\", default=existing.get(\"MATRIOSKA_JUDGE_MODEL\") or model)\n rep_model = _ask(\"Repairer model\", default=existing.get(\"MATRIOSKA_REPAIRER_MODEL\") or model)\n\n multikey = \"\"\n if _ask_yes_no(\"\\nUse multiple API keys for rotation? (comma-separated)\", default=False):\n multikey = _ask(\"Comma-separated API keys\", default=existing.get(\"MATRIOSKA_API_KEYS\", \"\"))\n\n # Build values dict (merge-respecting)\n values: Dict[str, str] = dict(existing) if merge else {}\n values.update({\n \"MATRIOSKA_PROVIDER\": pid,\n \"MATRIOSKA_BASE_URL\": base_url,\n \"MATRIOSKA_API_KEY\": api_key,\n \"MATRIOSKA_MODEL\": model,\n })\n if multikey:\n values[\"MATRIOSKA_API_KEYS\"] = multikey\n if use_role_models:\n values.update({\n \"MATRIOSKA_ARCHITECT_MODEL\": arch_model,\n \"MATRIOSKA_GENERATOR_MODEL\": gen_model,\n \"MATRIOSKA_VALIDATOR_MODEL\": val_model,\n \"MATRIOSKA_JUDGE_MODEL\": judge_model,\n \"MATRIOSKA_REPAIRER_MODEL\": rep_model,\n })\n\n # Write .env\n env_path.write_text(_render_env(values), encoding=\"utf-8\")\n print(f\"\\n✓ Wrote {env_path}\")\n\n # MATRIOSKA.md scaffold\n md_path = cwd / \"MATRIOSKA.md\"\n if not md_path.exists():\n if _ask_yes_no(\"\\nCreate MATRIOSKA.md (project conventions injected into the Architect)?\", default=True):\n project_name = _ask(\"Project name\", default=cwd.name)\n md_path.write_text(\n _MATRIOSKA_MD_TEMPLATE.format(project_name=project_name),\n encoding=\"utf-8\",\n )\n print(f\"✓ Wrote {md_path}\")\n\n print(\"\\nNext steps:\")\n print(f\" matrioska run --task \\\"Create a CLI todo app with SQLite\\\"\")\n print(f\" matrioska run --task '…' --quick # rapid iteration\")\n print(f\" matrioska run --task '…' --mode plan # plan only\")\n return 0\n"}}}} |