WrightAI — AI Code Documentation for VS Code

WrightAI brings AI-powered documentation directly into your editor. Generate docstrings, track coverage, detect drift, and chat with your codebase — all without leaving VS Code.

Powered by Claude (Anthropic). No local server or Python installation required — the backend is fully hosted.

Getting Started

1. Install this extension from the VS Code Marketplace
2. Sign in at wrightai-web.fly.dev with GitHub or Google
3. Copy your personal API key (starts with wai_)
4. Open VS Code Settings (Cmd+, / Ctrl+,), search for WrightAI, and paste the key into Wright: Api Key
5. Open any supported file — a Generate Docs CodeLens button appears above each function

Features

Generate Docstrings

Click the Generate Docs CodeLens button above any function, or press Cmd+Shift+D / Ctrl+Shift+D with your cursor inside a function.

A side-by-side diff preview shows the generated docstring before anything is written — accept or discard with one click. After your first injection, Wright reminds you that saving the file will automatically check for documentation drift.

You can also:

• Right-click anywhere in a file → Wright: Generate docs for this function
• Hover over any function name to see the existing docstring and a Regenerate link inline

Generate Docs for an Entire File

Open the Command Palette (Cmd+Shift+P / Ctrl+Shift+P) and run:

Wright: Generate docs for entire file

Wright documents every undocumented function in the file with a live progress bar showing per-function status.

Gutter Icons

Every function line shows a real-time status icon in the editor gutter:

• ✓ (green) — documented and up to date
• ○ (grey) — no documentation yet
• ⚠ (amber) — documentation is out of date (drifted)

Hover over a gutter icon to see the function name and a quick-action link.

Chat with Your Codebase

Run Wright: Chat with codebase from the Command Palette to open an interactive chat panel. Ask anything:

• "How does authentication work?"
• "Where is the payment logic?"
• "What does the process_batch function do?"

Answers stream in real time, include clickable file:line citations that jump straight to the source, and suggest follow-up questions. Full conversation history is maintained across turns.

Coverage Dashboard

The Wright Coverage panel in the Explorer sidebar shows live documentation coverage:

• Overall percentage and function counts
• Workspace path and file count
• Color-coded icons (green ≥ 80%, yellow ≥ 50%, red < 50%)

Run Wright: Show coverage from the Command Palette to refresh it manually.

Drift Detection

Run Wright: Check for doc drift to find functions whose implementation has changed since their docstring was written. Stale functions are highlighted with an ⚠ Docs outdated — regenerate CodeLens.

Drift is also checked automatically every time you save a file.

Keyboard Shortcuts

Extension Settings

Note: JavaScript, TypeScript, and Java always use JSDoc. Go always uses // line comments. Rust always uses /// doc comments. The style setting only applies to Python.

Supported Languages

Python · JavaScript · TypeScript · Java · Go · Rust

Support

• Bug reports and feature requests: GitHub Issues
• Questions and feedback: GitHub Discussions