🔨 chore: add GatewayStreamNotifier for Agent Gateway WebSocket push by arvinxx · Pull Request #13603 · lobehub/lobehub

arvinxx · 2026-04-07T02:38:29Z

Summary

Add GatewayStreamNotifier decorator that wraps IStreamEventManager to push events to the Agent Gateway via HTTP (fire-and-forget)
Update createStreamEventManager factory to auto-wrap with gateway notifier when AGENT_GATEWAY_SERVICE_TOKEN is configured
Redis SSE remains the primary event channel; the gateway is an additive push layer for WebSocket delivery

How it works

AgentRuntimeService.executeStep()
  → publishStreamEvent() / publishStreamChunk() / etc.
    → GatewayStreamNotifier
      ├→ inner.publish*()  [Redis Streams, unchanged]
      └→ HTTP POST → agent-gateway/api/operations/push-event  [new, fire-and-forget]

Environment Variables

Variable	Description
`AGENT_GATEWAY_SERVICE_TOKEN`	Enables gateway push when set
`AGENT_GATEWAY_URL`	Gateway URL (default: `https://agent-gateway.lobehub.com`)

Test plan

Type check passes
Verified with local dev server + CLI agent execution
Confirmed events arrive at gateway (/api/operations/status returns completed)
Existing Redis SSE path unaffected (CLI still receives events via SSE)

Fixes LOBE-6723

🤖 Generated with Claude Code

Add a decorator that wraps IStreamEventManager to additionally push events to the Agent Gateway via HTTP (fire-and-forget). When AGENT_GATEWAY_SERVICE_TOKEN is configured, the factory automatically wraps the base stream manager with the gateway notifier. Redis SSE remains the primary event channel; the gateway is an additive push layer for WebSocket delivery. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

vercel · 2026-04-07T02:38:34Z

The latest updates on your projects. Learn more about Vercel for GitHub.

Project	Deployment	Actions	Updated (UTC)
lobehub	Ready	Preview, Comment	Apr 7, 2026 3:46am

sourcery-ai

We've reviewed this pull request using the Sourcery rules engine

lobehubbot · 2026-04-07T02:39:32Z

@nekomeowww - This PR adds a GatewayStreamNotifier for Agent Gateway WebSocket push, touching backend server modules in src/server/modules/AgentRuntime/. Please take a look.

codecov · 2026-04-07T02:40:10Z

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 66.61%. Comparing base (5b03f00) to head (cbc892f).
⚠️ Report is 10 commits behind head on canary.

Additional details and impacted files

@@            Coverage Diff             @@
##           canary   #13603      +/-   ##
==========================================
+ Coverage   66.58%   66.61%   +0.03%     
==========================================
  Files        1999     2000       +1     
  Lines      165955   166107     +152     
  Branches    19193    19215      +22     
==========================================
+ Hits       110503   110659     +156     
+ Misses      55328    55324       -4     
  Partials      124      124

Flag	Coverage Δ
app	`58.48% <100.00%> (+0.05%)`	⬆️
database	`92.60% <ø> (ø)`
packages/agent-runtime	`88.98% <ø> (ø)`
packages/context-engine	`85.51% <ø> (ø)`
packages/conversation-flow	`92.36% <ø> (ø)`
packages/file-loaders	`87.02% <ø> (ø)`
packages/memory-user-memory	`66.68% <ø> (ø)`
packages/model-bank	`99.85% <ø> (ø)`
packages/model-runtime	`84.66% <ø> (ø)`
packages/prompts	`65.80% <ø> (ø)`
packages/python-interpreter	`92.90% <ø> (ø)`
packages/ssrf-safe-fetch	`0.00% <ø> (ø)`
packages/utils	`90.02% <ø> (ø)`
packages/web-crawler	`88.82% <ø> (ø)`

Flags with carried forward coverage won't be shown. Click here to find out more.

Components	Coverage Δ
Store	`67.10% <ø> (ø)`
Services	`51.55% <ø> (ø)`
Server	`66.08% <100.00%> (+0.12%)`	⬆️
Libs	`51.03% <ø> (ø)`
Utils	`91.01% <ø> (ø)`

🚀 New features to boost your workflow:

❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.
📦 JS Bundle Analysis: Save yourself from yourself by tracking and limiting bundle sizes in JS merges.

chatgpt-codex-connector

💡 Codex Review

Here are some automated review suggestions for this pull request.

Reviewed commit: 388fae1ee2

ℹ️ About Codex in GitHub

Your team has set up Codex to review pull requests in this repo. Reviews are triggered when you

Open a pull request for review
Mark a draft as ready
Comment "@codex review".

If Codex has suggestions, it will comment; otherwise it will react with 👍.

Codex can also answer questions or update the PR. Try commenting "@codex address that feedback".

src/server/modules/AgentRuntime/GatewayStreamNotifier.ts

Ensure the decorator always delegates to the inner stream event manager first, gateway failure never drops Redis events, and the factory correctly wraps/skips based on AGENT_GATEWAY_SERVICE_TOKEN. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

- 5s AbortSignal timeout on every gateway POST to prevent hanging sockets - Max 20 inflight requests; excess silently dropped with a debug log - Use url-join for URL construction instead of string concatenation Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Read AGENT_GATEWAY_SERVICE_TOKEN and AGENT_GATEWAY_URL from the validated appEnv config instead of raw process.env. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

sourcery-ai bot reviewed Apr 7, 2026

View reviewed changes

vercel bot deployed to Preview April 7, 2026 02:39 View deployment

chatgpt-codex-connector bot reviewed Apr 7, 2026

View reviewed changes

src/server/modules/AgentRuntime/GatewayStreamNotifier.ts Outdated Show resolved Hide resolved

vercel bot deployed to Preview April 7, 2026 02:54 View deployment

vercel bot deployed to Preview April 7, 2026 03:08 View deployment

🐛 fix: resolve TS18048 possibly undefined in test

c611ef3

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

vercel bot deployed to Preview April 7, 2026 03:26 View deployment

arvinxx and others added 2 commits April 7, 2026 11:35

♻️ refactor: move gateway env vars to appEnv

d699b15

Read AGENT_GATEWAY_SERVICE_TOKEN and AGENT_GATEWAY_URL from the validated appEnv config instead of raw process.env. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

♻️ refactor: move gateway URL default into appEnv

cbc892f

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

vercel bot deployed to Preview April 7, 2026 03:46 View deployment

arvinxx changed the title ~~✨ feat: add GatewayStreamNotifier for Agent Gateway WebSocket push~~ 🔨 chore: add GatewayStreamNotifier for Agent Gateway WebSocket push Apr 7, 2026

arvinxx merged commit 6c9cbb0 into canary Apr 7, 2026
34 checks passed

arvinxx deleted the feat/agent-gateway-notifier branch April 7, 2026 04:16

arvinxx mentioned this pull request Apr 7, 2026

♻️ refactor: add WebSocket gateway support to CLI agent run #13608

Merged

4 tasks

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

🔨 chore: add GatewayStreamNotifier for Agent Gateway WebSocket push#13603

🔨 chore: add GatewayStreamNotifier for Agent Gateway WebSocket push#13603
arvinxx merged 6 commits intocanaryfrom
feat/agent-gateway-notifier

arvinxx commented Apr 7, 2026

Uh oh!

vercel bot commented Apr 7, 2026 •

edited

Loading

Uh oh!

sourcery-ai bot left a comment

Uh oh!

lobehubbot commented Apr 7, 2026

Uh oh!

codecov bot commented Apr 7, 2026 •

edited

Loading

Uh oh!

chatgpt-codex-connector bot left a comment

Uh oh!

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

2 participants

Uh oh!

Conversation

arvinxx commented Apr 7, 2026

Summary

How it works

Environment Variables

Test plan

Uh oh!

vercel bot commented Apr 7, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

sourcery-ai bot left a comment

Choose a reason for hiding this comment

Uh oh!

lobehubbot commented Apr 7, 2026

Uh oh!

codecov bot commented Apr 7, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Codecov Report

Uh oh!

chatgpt-codex-connector bot left a comment

Choose a reason for hiding this comment

💡 Codex Review

Uh oh!

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

2 participants

vercel bot commented Apr 7, 2026 •

edited

Loading

codecov bot commented Apr 7, 2026 •

edited

Loading