From ea97d0844947d0f6ec1472dd8f7b7f6455680476 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Tue, 25 Jun 2024 20:36:57 -0700 Subject: [PATCH] Handle null model.info max_input_tokens --- aider/models.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/aider/models.py b/aider/models.py index 09c250da5..b9ae02590 100644 --- a/aider/models.py +++ b/aider/models.py @@ -328,7 +328,10 @@ class Model: self.missing_keys = res.get("missing_keys") self.keys_in_environment = res.get("keys_in_environment") - if self.info.get("max_input_tokens", 0) < 32 * 1024: + max_input_tokens = self.info.get("max_input_tokens") + if not max_input_tokens: + max_input_tokens = 0 + if max_input_tokens < 32 * 1024: self.max_chat_history_tokens = 1024 else: self.max_chat_history_tokens = 2 * 1024