DRI Assist — VS Code Extension + CLI

AI-powered incident triage assistant for on-call engineers. Works with ICM, ADO, Kusto, and EngHub.

Install from Marketplace

VS Code → Extensions → Search "DRI Assist" → Install

Or install the CLI

npm install -g driassist-agent

VS Code Usage

After installing, use Copilot Chat:

@driassist What are my active incidents?
@driassist /triage
@driassist /investigate 12345678
@driassist /checklist
@driassist /trends
@driassist /setup

First-time setup

1. Install the extension
2. Run DRI Assist: Setup Team Config from the command palette (Ctrl+Shift+P)
3. Fill in your team details in the generated TeamConfig.yaml
4. Start chatting with @driassist

CLI Usage

# Interactive chat
driassist chat

# Run DRI startup scan
driassist triage

# Investigate a specific incident
driassist investigate 12345678

# Generate TeamConfig.yaml
driassist setup

CLI Prerequisites

• agency CLI installed
• az login completed (for ICM/ADO/Kusto auth)
• GITHUB_TOKEN env var set (for LLM via GitHub Models)

# Set your token
export GITHUB_TOKEN=$(gh auth token)

# Or on Windows
$env:GITHUB_TOKEN = $(gh auth token)

How It Works

┌─────────────────────────────────────────────────────────────┐
│  VS Code Copilot Chat          OR         CLI Terminal      │
│  @driassist /triage                     driassist chat   │
└──────────────────────┬──────────────────────────────────────┘
                       │
                       ▼
┌─────────────────────────────────────────────────────────────┐
│  Agent Core                                                 │
│  ├─ System prompt from Dri.agent.md + TeamConfig.yaml       │
│  ├─ LLM (Copilot in VS Code, GitHub Models in CLI)         │
│  ├─ MCP Bridge → spawns agency MCP servers                  │
│  │   ├─ agency mcp icm      → ICM incidents                │
│  │   ├─ agency mcp es-chat  → EngHub docs, TSGs            │
│  │   ├─ agency mcp ado      → ADO work items               │
│  │   └─ kusto-mcp-server    → KQL queries                  │
│  └─ Tool calling loop (multi-turn)                          │
└─────────────────────────────────────────────────────────────┘

What Ships in the Package

driassist-agent/
├─ out/              Extension TypeScript → JS compiled output
│   ├─ extension.js  VS Code entry point (chat participant)
│   ├─ mcp-bridge.js MCP server manager
│   └─ config-loader.js  TeamConfig + agent.md → system prompt
├─ cli/
│   └─ index.js      CLI entry point (oncall-dri command)
├─ agents/           Bundled agent definitions
│   ├─ Dri.agent.md           DRI agent instructions
│   ├─ ConfigHelper.agent.md  Setup wizard instructions
│   ├─ TeamConfigTemplate.yaml  Template for teams to fill in
│   ├─ mcp-config.json        MCP server definitions
│   └─ DriAgentRef/
│       └─ IcmKqlReference.md KQL query reference
└─ assets/
    └─ icon.png      Extension icon

Building from Source

cd vscode-extension

# Copy latest agent files from main repo
pwsh copy-agents.ps1

# Install deps and compile
npm install
npm run compile

# Package as .vsix
npm run package

# Publish to marketplace
npm run publish

Configuration

Settings available in VS Code (Settings → On-Call DRI Agent):