PromptGuard VS Code Extension

Scan, detect, and secure LLM SDK usage in your codebase - directly in VS Code.

Features

• LLM SDK Detection: Automatically detects OpenAI, Anthropic, Cohere, HuggingFace, Gemini, Groq, and AWS Bedrock SDKs in your code
• Status Indicator: See PromptGuard status at a glance in the status bar
• One-Click Setup: Initialize PromptGuard with a single command
• Real-time Diagnostics: Get warnings when LLM SDKs are detected without PromptGuard protection
• Security Scanning: Scan selected text or files for prompt injection and PII
• PII Redaction: Redact sensitive data from selected text
• CLI Integration: Seamlessly wraps the PromptGuard CLI

Tip: For new projects, the PromptGuard SDK with auto-instrumentation (promptguard.init()) is the recommended integration - one line of code secures all LLM calls. The CLI and this extension are best for scanning existing codebases and applying proxy-mode transforms. (Python: install promptguard-sdk, import as promptguard.)

Screenshots

Inline diagnostic squiggle on an unprotected LLM SDK call, with the PromptGuard quick-fix.

Status bar indicator showing Active / Disabled / Not initialized states.

PromptGuard Activity Bar view: status, managed files, and quick actions.

Requirements

• VS Code 1.80.0 or higher
• PromptGuard CLI installed (see Installation)

Installation

1. Install the PromptGuard CLI

curl -fsSL https://raw.githubusercontent.com/acebot712/promptguard-cli/main/install.sh | sh

Or install manually from GitHub Releases.

2. Install the Extension

1. Open VS Code
2. Go to Extensions (Cmd+Shift+X / Ctrl+Shift+X)
3. Search for "PromptGuard"
4. Click Install

Or install from the command line:

code --install-extension promptguard.promptguard-vscode

Usage

Initialize PromptGuard

1. Open Command Palette (Cmd+Shift+P / Ctrl+Shift+P)
2. Run PromptGuard: Initialize
3. Enter your API key when prompted
4. Select providers (or press Escape to use all)

Detect LLM SDKs

1. Run PromptGuard: Detect LLM SDKs from Command Palette
2. View results in the Output channel

Check Status

• Click the status bar indicator (bottom right)
• Or run PromptGuard: Show Status from Command Palette

Apply Protection

1. Run PromptGuard: Apply Protection from Command Palette
2. Confirm the action
3. Your LLM SDK calls are rewritten to route through the PromptGuard proxy (originals are backed up; PromptGuard: Disable Protection restores them)

Configuration

CLI Path

If the CLI is not in your PATH, you can set a custom path:

1. Open Settings (Cmd+, / Ctrl+,)
2. Search for "PromptGuard"
3. Set PromptGuard: Cli Path to the full path to the binary

Example: /usr/local/bin/promptguard or C:\Program Files\PromptGuard\promptguard.exe

Settings

Commands

• PromptGuard: Initialize - Set up PromptGuard in your project
• PromptGuard: Detect LLM SDKs - Detect LLM SDK usage across the project
• PromptGuard: Show Status - Display current configuration
• PromptGuard: Apply Protection - Rewrite detected LLM SDK calls to route through PromptGuard
• PromptGuard: Disable Protection - Temporarily disable PromptGuard
• PromptGuard: Enable Protection - Re-enable PromptGuard
• PromptGuard: Set API Key - Store your PromptGuard API key securely
• PromptGuard: Select Project - Switch the active PromptGuard project
• PromptGuard: Scan Selection for Threats - Scan the selected text for prompt injection and PII
• PromptGuard: Redact PII from Selection - Redact sensitive data from the selected text
• PromptGuard: Scan File for Threats - Scan an entire file for security threats
• PromptGuard: Install/Update CLI - Install or update the PromptGuard CLI
• PromptGuard: Refresh - Reload the PromptGuard views after external changes

Right-click in the editor to Scan/Redact a selection; right-click a file in the Explorer to scan it.

Status Bar

The status bar shows:

• Active - PromptGuard is active and protecting your app
• Disabled - PromptGuard is temporarily disabled
• Not initialized - PromptGuard hasn't been set up yet

Click the status bar to view detailed status.

Sidebar

Open the shield icon in the Activity Bar for the PromptGuard view. Before setup it shows a welcome panel that either offers to Initialize PromptGuard / Detect LLM SDKs, or — if the CLI isn't found — points you at Install/Update CLI. Once initialized it shows current status, managed files, and an Actions group with quick buttons to Detect / Enable Protection / Disable Protection, plus Set API Key and Select Project so you can configure a key or switch projects without opening the Command Palette. Use the refresh icon in the view title bar to update both the tree and the status bar.

Supported Languages

• TypeScript (.ts, .tsx)
• JavaScript (.js, .jsx)
• Python (.py)

Supported Providers

• OpenAI
• Anthropic
• Cohere
• HuggingFace
• Gemini (Google)
• Groq
• AWS Bedrock

Troubleshooting

CLI Not Found

If you see "CLI not found" errors:

1. Verify the CLI is installed: promptguard --version
2. Check it's in your PATH: which promptguard (macOS/Linux) or where promptguard (Windows)
3. Set the promptguard.cliPath setting to the full path

Extension Not Working

1. Check the Output channel: View → Output → Select "PromptGuard"
2. Reload the window: Cmd+Shift+P → "Developer: Reload Window"
3. Check VS Code version: Must be 1.80.0 or higher

Links

• PromptGuard Website
• Documentation
• CLI Repository
• Report Issues

License

Apache 2.0