From 0b2d0004b6b7bb20ced942f8f40262400d1eae22 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Mon, 26 Aug 2024 15:40:03 -0700
Subject: [PATCH] feat: add num_cache_warming_pings arg to __init__

---
 aider/coders/base_coder.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py
index 60b2391f5..0056ad360 100755
--- a/aider/coders/base_coder.py
+++ b/aider/coders/base_coder.py
@@ -90,6 +90,7 @@ class Coder:
     message_tokens_received = 0
     add_cache_headers = False
     cache_warming_thread = None
+    num_cache_warming_pings = 5
 
     @classmethod
     def create(
@@ -252,6 +253,7 @@ class Coder:
         total_cost=0.0,
         map_refresh="auto",
         cache_prompts=False,
+        num_cache_warming_pings=5,
     ):
         self.commit_before_message = []
         self.aider_commit_hashes = set()
@@ -992,7 +994,7 @@ class Coder:
             self.cache_warming_thread.cancel()
 
         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
                 try:
                     completion = litellm.completion(