feat: add network calls column to AI sessions table by SasSwart · Pull Request #27269 · coder/coder

SasSwart · 2026-07-15T16:45:59Z

Add a "Total/blocked network calls" column to the AIBridge sessions table.

Update ListAIBridgeSessions query to calculate network called made and blocked per session. See query plan here.

linear-code · 2026-07-15T16:46:03Z

AIGOV-460

github-actions · 2026-07-15T16:46:13Z

Docs preview

📖 View docs preview for docs/reference/api/aigateway.md

SasSwart · 2026-07-20T11:01:53Z

@johnstcn Would you mind having a look at the backend and API? I'll ask @EhabY for a look at the FE. I have self reviewed the backend. I am still self reviewing the frontend.

coder-tasks · 2026-07-20T11:02:30Z

Documentation Check

Updates Needed

~~docs/ai-coder/ai-gateway/audit.md - The Sessions list section enumerates the sessions-table columns and should mention the new Total/blocked network calls column.~~ (No longer needed in this PR: the frontend/UI changes that added the column were removed in 43a0595 "chore: remove frontend changes from backend PR". This is now a backend/API-only PR, so the user-facing sessions-list docs are unaffected. The column docs should be handled by the follow-up frontend PR.)

Note: docs/reference/api/aigateway.md and docs/reference/api/schemas.md are auto-generated and are already correctly updated in this PR; no manual action needed there. The network_calls API field is covered by these generated schema docs.

Automated review via Coder Agents

johnstcn

Was the query plan you shared on the current database state or on a synthetic dataset that includes the new indexes proposed here?

johnstcn · 2026-07-20T11:09:44Z

Would it be simpler to inline two nil-able int64s instead of making an embedded struct?

NetworkCallsTotal *int64 `json:"network_calls_count,omitempty"` NetworkCallsBlocked *int64 `json:"network_calls_blocked,omitempty"`

johnstcn · 2026-07-20T11:11:37Z

+		// seq 3, so A's window is (0,3) and B's is (3, +inf). The logs at seq 0
+		// and 3 are the interceptions' own LLM-provider calls and must be
+		// excluded by the exclusive lower bound.


Restates the inline comments below.

Suggested change

// seq 3, so A's window is (0,3) and B's is (3, +inf). The logs at seq 0

// and 3 are the interceptions' own LLM-provider calls and must be

// excluded by the exclusive lower bound.

// seq 3, so A's window is (0,3) and B's is (3, +inf).

Add a "Total/blocked network calls" column to the AIBridge sessions table, rendering per-session total, blocked, and errored call counts as badges plus "No activity" and "Disabled" states. Define the NetworkCalls contract on the codersdk AIBridgeSession type; the DB aggregation and handler population are left as marked TODO touch points. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

The AI Gateway sessions table shipped with the network-calls column scaffolded but unpopulated, so every session rendered as "Disabled". Aggregate Agent Firewall egress per session in ListAIBridgeSessions by correlating boundary_logs to each interception's firewall session and sequence window (seq, next_seq). The exclusive lower bound drops the interception's own LLM-provider call; matched_rule IS NULL counts blocked requests. Sessions that never passed through Agent Firewall report a nil summary so the UI keeps rendering "Disabled". Drop the errored counter for now; boundary_logs has no error signal yet. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>

Move the non-generated frontend changes for network call badges to a separate branch so this PR is backend only. The generated typesGenerated.ts additions remain, as they are produced by make gen alongside the backend changes. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>

EhabY

A few small comments on the FE, but it looks solid!

EhabY · 2026-07-20T11:32:24Z

 							<TableHead className="text-nowrap">Provider</TableHead>
 							<TableHead className="text-nowrap">Client</TableHead>
 							<TableHead className="text-nowrap">In/Out Tokens</TableHead>
+							<TableHead className="w-40">


Why is this header styled differently, shouldn't it be text-nowrap like above without the width limit?

We could also shorten this to fit in like the other titles to Network Calls since the content it self includes total/blocked

EhabY · 2026-07-20T12:05:56Z

+		return (
+			<TooltipProvider>
+				<Tooltip>
+					<TooltipTrigger asChild>


Maybe copy the pattern in LastConnectionHead.tsx:13-21 by adding tabIndex={0} to the InfoIcon plus an sr-only "More info" label

Currently there is no way to see why something is disabled using the keyboard (without a mouse)

EhabY · 2026-07-20T12:06:13Z

+	return (
+		<TooltipProvider>
+			<Tooltip>
+				<TooltipTrigger asChild>


EhabY · 2026-07-20T12:08:02Z

+} from "#/components/Tooltip/Tooltip";
+
+interface NetworkCallBadgesProps {
+	size?: "xs" | "sm" | "md";


AFAIK this is never used right? If so then let's remove it for now and only add it when there are other consumers

johnstcn · 2026-07-20T12:16:55Z

 	-- Fetch only the most recent user prompt across all interceptions
 	-- in the session.
 	SELECT up.prompt
 	FROM aibridge_user_prompts up
 	WHERE up.interception_id = ANY(sr.interception_ids)
 	ORDER BY up.created_at DESC, up.id DESC
 	LIMIT 1


Out of scope of this PR: the plan you linked shows the sample query being absolutely hammered by this join. We should probably follow up separately on this.

SasSwart · 2026-07-20T13:04:49Z

@johnstcn I've updated the link in the PR description to point to a new query plan. This one is against a seeded DB (2k sessions, 240k log lines). The seeded data is perhaps too uniform, but I believe this is still a good estimate. We are introducing some cost. It's only 25% of the cost that you identified in your comment above though.

I have a fix in mind, but I'm trying to determine whether its worth forcing this now or if this is a rabbit hole of diminishing returns.

johnstcn

Let's ship and observe the perf on dogfood.

github-actions Bot assigned SasSwart Jul 15, 2026

SasSwart changed the title ~~feat: add total/blocked network calls column to AI sessions table~~ feat: add network calls column to AI sessions table Jul 16, 2026

SasSwart requested a review from johnstcn July 20, 2026 10:58

SasSwart marked this pull request as ready for review July 20, 2026 11:00

johnstcn reviewed Jul 20, 2026

View reviewed changes

SasSwart and others added 4 commits July 20, 2026 11:32

Remove overly verbose AI comments

6bbefbf

Renumber migrations

6fc1f78

SasSwart force-pushed the aigov-460 branch from df9ee76 to 6fc1f78 Compare July 20, 2026 11:48

EhabY reviewed Jul 20, 2026

View reviewed changes

johnstcn reviewed Jul 20, 2026

View reviewed changes

SasSwart mentioned this pull request Jul 20, 2026

feat: add network call badges to AI sessions table #27341

Merged

johnstcn approved these changes Jul 21, 2026

View reviewed changes

SasSwart added 2 commits July 21, 2026 10:35

enhance boundary_logs session index and optimize query counts

a0b0fcd

fix migration numbers

c164363

SasSwart merged commit a9a1dcc into main Jul 21, 2026
32 of 33 checks passed

SasSwart deleted the aigov-460 branch July 21, 2026 12:34

github-actions Bot locked and limited conversation to collaborators Jul 21, 2026

Uh oh!

Conversation

SasSwart commented Jul 15, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

linear-code Bot commented Jul 15, 2026

Uh oh!

github-actions Bot commented Jul 15, 2026

Docs preview

Uh oh!

SasSwart commented Jul 20, 2026

Uh oh!

coder-tasks Bot commented Jul 20, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Documentation Check

Updates Needed

Uh oh!

johnstcn left a comment

Choose a reason for hiding this comment

Uh oh!

johnstcn Jul 20, 2026

Choose a reason for hiding this comment

Uh oh!

johnstcn Jul 20, 2026

Choose a reason for hiding this comment

Uh oh!

EhabY left a comment

Choose a reason for hiding this comment

Uh oh!

EhabY Jul 20, 2026

Choose a reason for hiding this comment

Uh oh!

EhabY Jul 20, 2026

Choose a reason for hiding this comment

Uh oh!

EhabY Jul 20, 2026

Choose a reason for hiding this comment

Uh oh!

EhabY Jul 20, 2026

Choose a reason for hiding this comment

Uh oh!

johnstcn Jul 20, 2026

Choose a reason for hiding this comment

Uh oh!

SasSwart commented Jul 20, 2026

Uh oh!

johnstcn left a comment

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants

SasSwart commented Jul 15, 2026 •

edited

Loading

coder-tasks Bot commented Jul 20, 2026 •

edited

Loading