From 0932d0c535ab6efcdd67553fb93296fe04c4403b Mon Sep 17 00:00:00 2001
From: ggurdin <ggurdin@gmail.com>
Date: Fri, 16 Jan 2026 11:59:48 -0500
Subject: [PATCH] feat: backoff after failed igc and tokens requests in message
 sending flow

---
 .../choreographer/choreo_constants.dart       |  1 +
 lib/pangea/choreographer/choreographer.dart   | 43 +++++++++++++++++--
 .../choreographer/igc/igc_controller.dart     |  5 ++-
 3 files changed, 45 insertions(+), 4 deletions(-)

diff --git a/lib/pangea/choreographer/choreo_constants.dart b/lib/pangea/choreographer/choreo_constants.dart
index 85b9e6186..a303266ce 100644
--- a/lib/pangea/choreographer/choreo_constants.dart
+++ b/lib/pangea/choreographer/choreo_constants.dart
@@ -11,4 +11,5 @@ class ChoreoConstants {
   static const int msBeforeIGCStart = 10000;
   static const int maxLength = 1000;
   static const String inputTransformTargetKey = 'input_text_field';
+  static const int defaultErrorBackoffSeconds = 5;
 }
diff --git a/lib/pangea/choreographer/choreographer.dart b/lib/pangea/choreographer/choreographer.dart
index 52348dea6..a04c1e738 100644
--- a/lib/pangea/choreographer/choreographer.dart
+++ b/lib/pangea/choreographer/choreographer.dart
@@ -2,6 +2,8 @@ import 'dart:async';
 
 import 'package:flutter/material.dart';
 
+import 'package:async/async.dart';
+
 import 'package:fluffychat/pangea/choreographer/assistance_state_enum.dart';
 import 'package:fluffychat/pangea/choreographer/choreo_constants.dart';
 import 'package:fluffychat/pangea/choreographer/choreo_mode_enum.dart';
@@ -45,6 +47,11 @@ class Choreographer extends ChangeNotifier {
   String? _lastChecked;
   ChoreoModeEnum _choreoMode = ChoreoModeEnum.igc;
 
+  DateTime? _lastIgcError;
+  DateTime? _lastTokensError;
+  int _igcErrorBackoff = ChoreoConstants.defaultErrorBackoffSeconds;
+  int _tokenErrorBackoff = ChoreoConstants.defaultErrorBackoffSeconds;
+
   StreamSubscription? _languageSub;
   StreamSubscription? _settingsUpdateSub;
   StreamSubscription? _acceptedContinuanceSub;
@@ -68,6 +75,12 @@ class Choreographer extends ChangeNotifier {
         openMatches: [],
       );
 
+  bool _backoffRequest(DateTime? error, int backoffSeconds) {
+    if (error == null) return false;
+    final secondsSinceError = DateTime.now().difference(error).inSeconds;
+    return secondsSinceError <= backoffSeconds;
+  }
+
   void _initialize() {
     textController = PangeaTextController(choreographer: this);
     textController.addListener(_onChange);
@@ -82,7 +95,14 @@ class Choreographer extends ChangeNotifier {
     itController.editing.addListener(_onSubmitSourceTextEdits);
 
     igcController = IgcController(
-      (e) => errorService.setErrorAndLock(ChoreoError(raw: e)),
+      (e) {
+        errorService.setErrorAndLock(ChoreoError(raw: e));
+        _lastIgcError = DateTime.now();
+        _igcErrorBackoff *= 2;
+      },
+      () {
+        _igcErrorBackoff = ChoreoConstants.defaultErrorBackoffSeconds;
+      },
     );
 
     _languageSub ??= MatrixState
@@ -233,7 +253,8 @@ class Choreographer extends ChangeNotifier {
             !ToolSetting.interactiveTranslator.enabled) ||
         (!ToolSetting.autoIGC.enabled &&
             !manual &&
-            _choreoMode != ChoreoModeEnum.it)) {
+            _choreoMode != ChoreoModeEnum.it) ||
+        _backoffRequest(_lastIgcError, _igcErrorBackoff)) {
       return;
     }
 
@@ -275,7 +296,9 @@ class Choreographer extends ChangeNotifier {
         MatrixState.pangeaController.userController.userL2?.langCode;
     final l1LangCode =
         MatrixState.pangeaController.userController.userL1?.langCode;
-    if (l1LangCode != null && l2LangCode != null) {
+    if (l1LangCode != null &&
+        l2LangCode != null &&
+        !_backoffRequest(_lastTokensError, _tokenErrorBackoff)) {
       final res = await TokensRepo.get(
         MatrixState.pangeaController.userController.accessToken,
         TokensRequestModel(
@@ -283,7 +306,21 @@ class Choreographer extends ChangeNotifier {
           senderL1: l1LangCode,
           senderL2: l2LangCode,
         ),
+      ).timeout(
+        const Duration(seconds: 10),
+        onTimeout: () {
+          return Result.error("Token request timed out");
+        },
       );
+
+      if (res.isError) {
+        _lastTokensError = DateTime.now();
+        _tokenErrorBackoff *= 2;
+      } else {
+        // reset backoff on success
+        _tokenErrorBackoff = ChoreoConstants.defaultErrorBackoffSeconds;
+      }
+
       tokensResp = res.isValue ? res.result : null;
     }
 
diff --git a/lib/pangea/choreographer/igc/igc_controller.dart b/lib/pangea/choreographer/igc/igc_controller.dart
index afe661ff9..9b78cb439 100644
--- a/lib/pangea/choreographer/igc/igc_controller.dart
+++ b/lib/pangea/choreographer/igc/igc_controller.dart
@@ -18,7 +18,8 @@ import 'package:fluffychat/widgets/matrix.dart';
 
 class IgcController {
   final Function(Object) onError;
-  IgcController(this.onError);
+  final VoidCallback onFetch;
+  IgcController(this.onError, this.onFetch);
 
   bool _isFetching = false;
   String? _currentText;
@@ -321,6 +322,8 @@ class IgcController {
       onError(res.asError!);
       clear();
       return;
+    } else {
+      onFetch();
     }
 
     if (!_isFetching) return;