fix(auth): stop draining stream pool goroutines after context cancellation

This commit is contained in:
chujian
2026-03-07 18:30:33 +08:00
parent 522a68a4ea
commit a52da26b5d

View File

@@ -471,7 +471,10 @@ func (m *Manager) wrapStreamResult(ctx context.Context, auth *Auth, provider, ro
}
}
for chunk := range remaining {
_ = emit(chunk)
if ok := emit(chunk); !ok {
discardStreamChunks(remaining)
return
}
}
if !failed {
m.MarkResult(ctx, Result{AuthID: auth.ID, Provider: provider, Model: routeModel, Success: true})