dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-06 12:45:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

feat: add num_cache_warming_pings arg to __init__

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-26 15:40:03 -07:00
parent 8cec00eb9c
commit 0b2d0004b6
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
aider/coders/base_coder.py
View file

@ -90,6 +90,7 @@ class Coder:
message_tokens_received = 0 message_tokens_received = 0
add_cache_headers = False add_cache_headers = False
cache_warming_thread = None cache_warming_thread = None
num_cache_warming_pings = 5
@classmethod @classmethod
def create( def create(
@ -252,6 +253,7 @@ class Coder:
total_cost=0.0, total_cost=0.0,
map_refresh="auto", map_refresh="auto",
cache_prompts=False, cache_prompts=False,
num_cache_warming_pings=5,
): ):
self.commit_before_message = [] self.commit_before_message = []
self.aider_commit_hashes = set() self.aider_commit_hashes = set()
@ -992,7 +994,7 @@ class Coder:
self.cache_warming_thread.cancel() self.cache_warming_thread.cancel()
def warm_cache_worker(): def warm_cache_worker():
for _ in range(5): for _ in range(self.num_cache_warming_pings):
time.sleep(10) # 290 == 4 minutes and 50 seconds time.sleep(10) # 290 == 4 minutes and 50 seconds
try: try:
completion = litellm.completion( completion = litellm.completion(