From 8d4d549a9834bdc6833b49a76140edef7d71a75d Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Thu, 15 Aug 2024 10:34:49 -0700
Subject: [PATCH] catch litellm bug for image size

---
 aider/models.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/aider/models.py b/aider/models.py
index cad99d1df..4e5bf74ca 100644
--- a/aider/models.py
+++ b/aider/models.py
@@ -516,7 +516,11 @@ class Model:
 
     def token_count(self, messages):
         if type(messages) is list:
-            return litellm.token_counter(model=self.name, messages=messages)
+            try:
+                return litellm.token_counter(model=self.name, messages=messages)
+            except Exception as err:
+                print(f"Unable to count tokens: {err}")
+                return 0
 
         if not self.tokenizer:
             return