@@ -703,6 +703,12 @@ func (a *sessionAgent) Summarize(ctx context.Context, sessionID string, opts fan
703 deleteErr := a.messages.Delete(ctx, summaryMessage.ID)
703 deleteErr := a.messages.Delete(ctx, summaryMessage.ID)
704 return deleteErr
704 return deleteErr
706+ // Mark the summary message as finished with an error so the UI
708+ summaryMessage.AddFinish(message.FinishReasonError, "Summarization Error", err.Error())
709+ if updateErr := a.messages.Update(ctx, summaryMessage); updateErr != nil {
706 return err
712 return err