feat(acp): streaming WebSocket route for prompt chunks

Adds GET /api/workspaces/:id/acp/stream. Client opens a WebSocket,
sends {"type":"prompt","content":"..."}, and receives a stream of
{"type":"chunk","messageId","text"} events followed by exactly
one {"type":"complete","stopReason"} or {"type":"error","error"}.
Closing the WS early triggers session/cancel.

- internal/acp/messages.go: StreamEvent wire shape.
- internal/acp/client.go:
  - streamChs []chan StreamEvent set; AddStream / RemoveStream.
  - sendStream non-blocking fanout.
  - Client.Stream(ctx, content, out) registers out, sends prompt,
    emits complete/error after the prompt response, unregisters.
  - handleNotification fans chunk events to all stream consumers.
  - notifyWG ensures chunk ordering vs the terminal event.
- internal/acp/service.go: Service.Stream(workspaceID, content, out)
  mirrors Prompt (per-workspace lock, 5-min timeout, EnsureReady).
- internal/service/acp_service.go: thin AcpService.Stream wrapper
  that maps acp.StreamEvent -> model.AcpStreamEvent.
- internal/model/acp.go: AcpStreamRequest, AcpStreamEvent DTOs.
- internal/api/acp_handler.go: stream WS handler (upgrade, read
  prompt, run Stream in a goroutine, write events, ping/pong, Cancel
  on client close).
- internal/api/router.go: register the new route.
- internal/acp/transport.go: dispatch notifications synchronously
  (vs. goroutine per notification) so chunks preserve order before
  the session/prompt response.
- internal/acp/client_test.go: TestClientStreamEmitsChunkAndComplete
  with a fake transport that drives a known sequence.
- internal/api/acp_handler_test.go: TestAcpStreamHandlerRoutes
  smoke test using a fake opencode acp script.

Existing POST /api/workspaces/:id/acp/prompt is unchanged.

E2E: prompt 'say hi in exactly 3 words' -> 3 chunk events
('Hi',' there','!') + 1 complete {stopReason: 'end_turn'}.

Conversation: 019f3680-200f-79b0-860b-43302e60d0ea
This commit is contained in:
tao.chen
2026-07-06 16:43:17 +08:00
parent 3d34de96bf
commit 031451fe3e
10 changed files with 473 additions and 14 deletions
+98 -10
View File
@@ -31,6 +31,8 @@ type Client struct {
sessionID string
transport *transport
history *History
streamChs []chan<- StreamEvent
notifyWG sync.WaitGroup
}
// NewClient creates an ACP client for the given workspace.
@@ -159,6 +161,73 @@ func (c *Client) Prompt(ctx context.Context, content string) (PromptResult, erro
return result, nil
}
// sendStream writes ev to out without blocking. It is used for both chunk
// notifications and terminal events so that a slow consumer cannot stall the
// transport goroutine.
func (c *Client) sendStream(out chan<- StreamEvent, ev StreamEvent) {
select {
case out <- ev:
default:
}
}
// AddStream registers a new streaming consumer. It must be paired with RemoveStream.
func (c *Client) AddStream(ch chan<- StreamEvent) int {
c.mu.Lock()
defer c.mu.Unlock()
c.streamChs = append(c.streamChs, ch)
return len(c.streamChs)
}
// RemoveStream unregisters a streaming consumer previously added with AddStream.
func (c *Client) RemoveStream(ch chan<- StreamEvent) {
c.mu.Lock()
defer c.mu.Unlock()
for i, existing := range c.streamChs {
if existing == ch {
c.streamChs[i] = c.streamChs[len(c.streamChs)-1]
c.streamChs = c.streamChs[:len(c.streamChs)-1]
return
}
}
}
// Stream sends a user prompt and forwards streamed chunks to out as they arrive.
// It emits exactly one terminal event, either complete or error, before returning.
func (c *Client) Stream(ctx context.Context, content string, out chan<- StreamEvent) error {
c.mu.RLock()
if !c.ready {
c.mu.RUnlock()
c.sendStream(out, StreamEvent{Type: "error", Error: "client not ready"})
return util.New(util.CodeConflict, "client not ready")
}
sessionID := c.sessionID
tr := c.transport
c.mu.RUnlock()
c.history.Add("user", content)
c.AddStream(out)
defer c.RemoveStream(out)
params := SessionPromptParams{
SessionID: sessionID,
Prompt: []PromptMessage{{Type: "text", Text: content}},
}
var result PromptResult
if err := tr.call(ctx, "session/prompt", params, &result); err != nil {
c.sendStream(out, StreamEvent{Type: "error", Error: err.Error()})
return util.Wrap(util.CodeInternal, "session/prompt failed", err)
}
// Wait for any chunk notifications that were already in flight to be
// delivered before we emit the terminal complete event.
c.notifyWG.Wait()
c.sendStream(out, StreamEvent{Type: "complete", StopReason: result.StopReason})
return nil
}
// Cancel sends a best-effort session/cancel notification.
func (c *Client) Cancel() error {
c.mu.RLock()
@@ -230,26 +299,45 @@ func (c *Client) handleNotification(method string, params json.RawMessage) {
c.lg.Debug("acp drop notification", "workspace_id", c.workspaceID, "method", method)
return
}
c.notifyWG.Add(1)
defer c.notifyWG.Done()
var up SessionUpdateParams
if err := json.Unmarshal(params, &up); err != nil {
c.lg.Error("acp invalid session/update", "workspace_id", c.workspaceID, "error", err)
return
}
var role string
switch up.Update.SessionUpdate {
case "agent_message_chunk":
if up.Update.Content.Type == "text" {
c.history.AppendOrCreate("agent", up.Update.MessageID, up.Update.Content.Text)
} else {
c.lg.Debug("acp drop non-text agent chunk", "workspace_id", c.workspaceID, "type", up.Update.Content.Type)
}
role = "agent"
case "user_message_chunk":
if up.Update.Content.Type == "text" {
c.history.AppendOrCreate("user", up.Update.MessageID, up.Update.Content.Text)
} else {
c.lg.Debug("acp drop non-text user chunk", "workspace_id", c.workspaceID, "type", up.Update.Content.Type)
}
role = "user"
default:
c.lg.Debug("acp drop update", "workspace_id", c.workspaceID, "type", up.Update.SessionUpdate)
return
}
if up.Update.Content.Type != "text" {
c.lg.Debug("acp drop non-text chunk", "workspace_id", c.workspaceID, "type", up.Update.Content.Type)
return
}
c.history.AppendOrCreate(role, up.Update.MessageID, up.Update.Content.Text)
ev := StreamEvent{
Type: "chunk",
MessageID: up.Update.MessageID,
Text: up.Update.Content.Text,
}
c.mu.RLock()
streams := make([]chan<- StreamEvent, len(c.streamChs))
copy(streams, c.streamChs)
c.mu.RUnlock()
for _, ch := range streams {
c.sendStream(ch, ev)
}
}