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