Install Now

IDE Agents

Agents that understand your entire codebase

Complete features, fix bugs, and ship code in VS Code and JetBrains. Powered by Augment's Context Engine, agents reason across your entire codebase to deliver production-ready changes.

VS Code - Augment Agent
U

Add JWT refresh token rotation to the auth flow

A

I'll implement JWT refresh token rotation. Let me analyze the auth code.

Context Engine: Codebasejwt refresh auth
Read src/api/auth.ts
A

Found it. Adding token rotation with automatic invalidation.

Editing src/api/auth.ts+47 -12
Editing src/middleware/session.ts+23 -5
Auth Module
auth.ts
Auto

THEY SHIP PRODUCTION CODE WITH AUGMENT

How it works

From prompt to pull request

Describe what you want in natural language. Agent breaks it down into steps, implements each one, and keeps you in control the whole time.

01

Plan

Agent analyzes your request against your codebase and creates an actionable plan. Review it before any code changes.

1. Read existing auth middleware
2. Create refresh token handler
3. Update session storage
4. Add token rotation logic
02

Implement

Watch Agent create, edit, and delete files. Run terminal commands. Call external tools. Every action is visible and reversible.

src/api/auth.tsModified
14 const token = jwt.sign(payload, secret);
15+ const refreshToken = generateRefreshToken();
16+ await storeRefreshToken(userId, refreshToken);
17 return { token, refreshToken };
03

Review

Checkpoints save your work at every step. Accept changes, revert to any point, or redirect the agent mid-task.

Checkpoint 4

Added refresh token rotation

3 files changed, 47 insertions

Capabilities

Built for how engineers actually work

001

Plan, then execute

Most AI coding tools dive straight into implementation. They start writing code, and you discover halfway through they misunderstood the scope.

Augment plans first. Ask it to “implement OAuth refresh token flow” and the Tasklist maps the full sequence before touching code: analyze current auth, create token handler, update middleware, add rotation logic, write tests. Add steps when scope grows. Skip what's not relevant. Come back tomorrow and pick up exactly where you left off.

Fig. 1

VS Code - Augment Agent
Task Progress
2/5

Task planning interface in Augment

002

Remember what matters

AI tools that store context automatically sound helpful until you realize you have no idea what they've saved. Outdated patterns persist. You're debugging why the Agent keeps suggesting a library you stopped using months ago.

Augment surfaces every memory before it's saved. A project goal, a debugging decision, a pattern you corrected. Approve, edit, or discard. Nothing gets stored without your sign-off. Promote the good stuff to workspace Rules and share it with your team.

Fig. 2

VS Code - Augment Agent
2 memories pending review

Use TAB to navigate through the list, and save, reject or edit memories as you see fit.

Memory

User roles follow RBAC pattern: Admin, Manager, Member. Always check permissions via hasPermission() helper, never raw role strings.

Created: 12/17/2025Source: Agent
Memory

Tenant isolation is enforced at the database layer. All user queries must include organizationId from the auth context.

Created: 12/16/2025Source: Correction

Memory review interface in Augment

003

Prompts, enhanced

Writing detailed prompts is tedious. Spelling out every file path, naming convention, and edge case takes longer than just doing the work yourself.

Type “fix the login bug” and let Augment expand it. The prompt enhancer pulls in your auth implementation paths, your error handling patterns, your test conventions. Review the enhanced prompt before sending, edit if something's off, or just hit enter and let it run.

Fig. 3

VS Code - Augment Agent
Auth Module
signup.ts
Auto

Prompt enhancement with codebase context

004

Context Engine

The only agent that actually knows your codebase

Most agents see a few files at a time. Augment's Context Engine indexes your entire codebase in real time, including commit history, cross-repo dependencies, and architectural patterns.

Commit history

Why changes were made, not just what changed

Codebase patterns

How your team actually builds, not generic best practices

External sources

Docs, tickets, and design decisions via native integrations and MCP

Tribal knowledge

Edge cases and team conventions discovered through deep codebase analysis

005

Everything else, built in

Execution

Parallel Tool Calls

Execute multiple tools simultaneously for 2x faster turns. Agent automatically parallelizes file reads, searches, and API calls instead of running them sequentially. Complex tasks that touch dozens of files complete in seconds, not minutes.

VS Code
Reading src/api/auth.ts
Reading src/middleware/session.ts
Searching codebase...

Auto Mode

Let Agent work independently. It plans, implements, and iterates without pausing for approval. File edits, terminal commands, and integration calls all execute automatically. Review everything when it's done, or interrupt anytime to redirect.

VS Code
AUTO MODE
Created auth/refresh.ts
Updated middleware/session.ts
Running npm test...

Ask Mode

Read-only mode for exploration. Ask questions about your codebase, trace dependencies, or understand unfamiliar code without any file modifications. Agent searches and analyzes but never writes. Safe for production repos and shared environments.

VS Code
U

How does auth work in this codebase?

A

Authentication uses JWT tokens stored in httpOnly cookies. The flow starts in...

Terminal Execution

Run commands directly in your terminal. npm install, dev servers, test suites, git operations. Agent sees the output and reacts to errors. Build fails? It reads the logs and fixes the issue without you copy-pasting anything.

VS Code
$ npm run test
Running 47 tests...
✓ auth.test.ts (12 passed)
✓ api.test.ts (23 passed)
✓ utils.test.ts (12 passed)

Checkpoints

Every change creates a snapshot. Roll back to any step without git archaeology. Agent keeps working while you review, and you can rewind to any point if something goes wrong. Undo with one click, not git reflog.

VS Code
1
2
3
4
5
Checkpoint 3: Added auth middleware

Multi Models

Switch between Claude, GPT, and other frontier models based on the task. Some excel at reasoning, others at speed. Pick the right tool for the job without changing your workflow or leaving Augment.

VS Code
Sonnet 4.5Selected
Opus 4.5
GPT 5.1
Context & Integrations

Multi Repos

Work across multiple repositories seamlessly with coordinated changes. Agent understands cross-repo dependencies and keeps related changes in sync. Refactor a shared library and update all downstream consumers in one session.

VS Code
Workspaces
frontend
backend
shared-lib

Image Understanding

Drop in screenshots, mockups, or Figma files. Agent interprets visual context and translates designs into code. Paste an error dialog and ask for a fix. Share a wireframe and get a working component.

VS Code
Analyzing image...
Found: Login form with email/password fields and OAuth buttons

Web Search

Agent pulls in documentation, Stack Overflow, and API references in real time. No stale training data. Always current, always cited. Ask about a library you've never used and get answers from the latest docs.

VS Code
Searching for React 19 docs...
react.dev/blog/react-19
github.com/facebook/react

Rules & Guidelines

Define coding standards and conventions that Agent follows consistently. Preferred libraries, naming patterns, architectural decisions. Store them in your repo so the whole team gets the same behavior.

VS Code
# .augment-guidelines
- Use TypeScript strict mode
- Prefer async/await over callbacks
- Use Zod for validation
- Follow REST naming conventions

Native Integrations

GitHub, Linear, Jira, Confluence, Notion, Sentry, Stripe. One-click OAuth setup, no MCP config required. Pull in an issue, implement the fix, open a PR, and update the ticket without leaving your editor.

VS Code
GH
Issue #1842 imported
Li
Linear connected
Ji
Jira connected

MCP Integrations

Connect to 100+ tools in your stack through the Model Context Protocol. Databases, cloud providers, internal APIs. Community servers or build your own. Agent calls them like native tools.

VS Code
Connected Tools
Figma MCP
Stripe MCP
MongoDB MCP
AWS MCP

Use Cases

Get your agents to do real work

Implement features from tickets

Point Agent at a Linear or Jira ticket. It reads the requirements, explores the codebase, and starts implementation.

Fix bugs with stack traces

Paste an error. Agent traces the issue through your code and proposes a fix that actually fits your architecture.

Write tests for existing code

Generate unit tests that follow your testing patterns, use your mocking conventions, and cover the right edge cases.

Refactor without breaking things

Move functions between files, rename modules, update API signatures. Agent handles the ripple effects.

Generate documentation

Create READMEs, API docs, and inline comments that reflect what the code actually does.

Upgrade dependencies

Agent updates package versions, fixes breaking changes, and runs your test suite to verify.

Onboard to unfamiliar code

Ask Agent to explain how a feature works end-to-end. It traces the request flow, maps dependencies, and explains the architecture in context of your specific codebase.

Resolve merge conflicts

Point Agent at a conflicted branch. It understands both sides of the change, resolves conflicts intelligently, and preserves the intent of each commit.

Debug performance issues

Track down slow endpoints, N+1 queries, and memory leaks. Agent profiles the code path, identifies bottlenecks, and implements fixes with before/after benchmarks.

IDE Support

Works where you work

Native extensions for VS Code and JetBrains IDEs. Same agent, same features, same Context Engine.

Get Started

Give your codebase the agents it deserves

Install Augment to get started. Works with codebases of any size, from side projects to enterprise monorepos.