feat: Add error handling for ImportError in RepoMap.find_definitions

aider/models.py

@@ -658,7 +658,13 @@ def validate_environment(self):
         # https://github.com/BerriAI/litellm/issues/3190
 
         model = self.name
-        res = litellm.validate_environment(model)
+
+        try:
+            res = litellm.validate_environment(model)
+        except ImportError as err:
+            self.io.tool_error("Error while loading necessary module: {err}")
+            sys.exit(1)
+
         if res["keys_in_environment"]:
             return res
         if res["missing_keys"]:

aider/repomap.py

@@ -389,6 +389,9 @@ def get_ranked_tags(
             ranked = nx.pagerank(G, weight="weight", **pers_args)
         except ZeroDivisionError:
             return []
+        except ImportError as err:
+            self.io.tool_error("Error while loading necessary module: {err}")
+            sys.exit(1)
 
         # distribute the rank from each source node, across all of its out edges
         ranked_definitions = defaultdict(float)

aider/sendchat.py

@@ -1,5 +1,6 @@
 import hashlib
 import json
+import sys
 
 import backoff
 
@@ -83,7 +84,11 @@ def send_completion(
 
     # del kwargs['stream']
 
-    res = litellm.completion(**kwargs)
+    try:
+        res = litellm.completion(**kwargs)
+    except ImportError as err:
+        print("Error while loading necessary module: {err}")
+        sys.exit(1)
 
     if not stream and CACHE is not None:
         CACHE[key] = res