From f7dd0fc58201711c4e483fa4340e3cb1fbd224c3 Mon Sep 17 00:00:00 2001
From: Paul Gauthier <paul@aider.chat>
Date: Fri, 7 Feb 2025 16:39:35 -0800
Subject: [PATCH] refactor: Modify cache warming mechanism with
 ok_to_warm_cache flag

---
 aider/coders/base_coder.py | 15 +++++++--------
 aider/main.py              |  3 +++
 2 files changed, 10 insertions(+), 8 deletions(-)

diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py
index 1c2c59b55..7e228dd5f 100755
--- a/aider/coders/base_coder.py
+++ b/aider/coders/base_coder.py
@@ -168,6 +168,7 @@ class Coder:
             use_kwargs.update(kwargs)  # override passed kwargs
 
             kwargs = use_kwargs
+            from_coder.ok_to_warm_cache = False
 
         for coder in coders.__all__:
             if hasattr(coder, "edit_format") and coder.edit_format == edit_format:
@@ -264,7 +265,7 @@ class Coder:
 
         return lines
 
-    _cache_warming_stop = False
+    ok_to_warm_cache = False
 
     def __init__(
         self,
@@ -1198,25 +1199,23 @@ class Coder:
         return chunks
 
     def warm_cache(self, chunks):
-        dump(self.add_cache_headers)
-        dump(self.num_cache_warming_pings)
         if not self.add_cache_headers:
             return
         if not self.num_cache_warming_pings:
             return
+        if not self.ok_to_warm_cache:
+            return
 
-        delay = 5  # * 60 - 5
+        delay = 5 * 60 - 5
         self.next_cache_warm = time.time() + delay
         self.warming_pings_left = self.num_cache_warming_pings
         self.cache_warming_chunks = chunks
-        self._cache_warming_stop = False
 
         if self.cache_warming_thread:
             return
 
         def warm_cache_worker():
-            while not self._cache_warming_stop:
-                dump(self.warming_pings_left)
+            while self.ok_to_warm_cache:
                 time.sleep(1)
                 if self.warming_pings_left <= 0:
                     continue
@@ -1543,7 +1542,7 @@ class Coder:
 
     def __del__(self):
         """Cleanup when the Coder object is destroyed."""
-        self._cache_warming_stop = True
+        self.ok_to_warm_cache = False
 
     def add_assistant_reply_to_cur_messages(self):
         if self.partial_response_content:
diff --git a/aider/main.py b/aider/main.py
index f88b8268b..398737079 100644
--- a/aider/main.py
+++ b/aider/main.py
@@ -1060,10 +1060,13 @@ def main(argv=None, input=None, output=None, force_git_root=None, return_coder=F
 
     while True:
         try:
+            coder.ok_to_warm_cache = True
             coder.run()
             analytics.event("exit", reason="Completed main CLI coder.run")
             return
         except SwitchCoder as switch:
+            coder.ok_to_warm_cache = False
+
             kwargs = dict(io=io, from_coder=coder)
             kwargs.update(switch.kwargs)
             if "show_announcements" in kwargs: