on main help sign in
chore: cancel requests when user closes
commitfc8ec7f200393bd5b8f870971c851dca23634fa1 authorKujtim Hoxha <kujtimii.h@gmail.com> dateFri 20 Jun 2025, 11:21 (1 year ago) parentb1cd3f1
2 files changed+9
internal/app/app.go+1 internal/llm/agent/agent.go+8
internal/app/app.go+1
@@ -163,4 +163,5 @@ func (app *App) Shutdown() {
163163 }
164164 cancel()
165165 }
166+ app.CoderAgent.CancelAll()
166167}
internal/llm/agent/agent.go+8
@@ -50,6 +50,7 @@ type Service interface {
5050 Model() models.Model
5151 Run(ctx context.Context, sessionID string, content string, attachments ...message.Attachment) (<-chan AgentEvent, error)
5252 Cancel(sessionID string)
53+ CancelAll()
5354 IsSessionBusy(sessionID string) bool
5455 IsBusy() bool
5556 Update(agentName config.AgentName, modelID models.ModelID) (models.Model, error)
@@ -698,6 +699,13 @@ func (a *agent) Summarize(ctx context.Context, sessionID string) error {
698699 return nil
699700}
700701
702+func (a *agent) CancelAll() {
703+ a.activeRequests.Range(func(key, value any) bool {
704+ a.Cancel(key.(string)) // key is sessionID
705+ return true
706+ })
707+}
708+
701709func createAgentProvider(agentName config.AgentName) (provider.Provider, error) {
702710 cfg := config.Get()
703711 agentConfig, ok := cfg.Agents[agentName]