From fb0691914c7cb061d29c411c75193a84308de436 Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Tue, 6 Aug 2024 09:11:21 -0300
Subject: [PATCH] fix: Use repo map only when map_tokens is greater than 0

---
 aider/coders/base_coder.py | 2 +-
 tests/basic/test_main.py   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py
index 7fe2451a6..f2a04f4a0 100755
--- a/aider/coders/base_coder.py
+++ b/aider/coders/base_coder.py
@@ -317,7 +317,7 @@ class Coder:
             use_repo_map = main_model.use_repo_map
             map_tokens = 1024
         else:
-            use_repo_map = True
+            use_repo_map = map_tokens > 0
 
         max_inp_tokens = self.main_model.info.get("max_input_tokens") or 0
 
diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py
index 8252907b4..0525fa8df 100644
--- a/tests/basic/test_main.py
+++ b/tests/basic/test_main.py
@@ -384,7 +384,7 @@ class TestMain(TestCase):
                     input=DummyInput(),
                     output=DummyOutput(),
                 )
-                MockRepoMap.assert_called_once()
+                MockRepoMap.assert_not_called()
 
     def test_map_tokens_option_with_non_zero_value(self):
         with GitTemporaryDirectory():