dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-28 16:25:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refactor: Modify cache warming mechanism with ok_to_warm_cache flag

Browse source
This commit is contained in:
Paul Gauthier 2025-02-07 16:39:35 -08:00 • committed by Paul Gauthier (aider)
parent 35f30bde04
commit f7dd0fc582
2 changed files with 10 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

15
aider/coders/base_coder.py
View file

@ -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:

3
aider/main.py
View file

@ -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: