@@ -567,7 +567,7 @@ loop:
568 // Reset the timeout timer since we received a chunk
568 // Reset the timeout timer since we received a chunk
569 timer.Reset(streamChunkTimeout)
569 timer.Reset(streamChunkTimeout)
571 if processErr := a.processEvent(ctx, sessionID, &assistantMsg, event); processErr != nil {
571 if processErr := a.processEvent(ctx, sessionID, &assistantMsg, event); processErr != nil {
572 if errors.Is(processErr, context.Canceled) {
572 if errors.Is(processErr, context.Canceled) {
573 a.finishMessage(context.Background(), &assistantMsg, message.FinishReasonCanceled, "Request cancelled", "")
573 a.finishMessage(context.Background(), &assistantMsg, message.FinishReasonCanceled, "Request cancelled", "")