From cf2a48b21f56df0779f42390f143df254ce0cf08 Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Thu, 18 Apr 2024 13:55:43 -0700
Subject: [PATCH] get_weak_model

---
 aider/coders/base_coder.py |  2 +-
 aider/history.py           |  2 +-
 aider/models.py            | 17 ++++++++---------
 3 files changed, 10 insertions(+), 11 deletions(-)

diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py
index d503c3221..8384b2545 100755
--- a/aider/coders/base_coder.py
+++ b/aider/coders/base_coder.py
@@ -212,7 +212,7 @@ class Coder:
             self.io.tool_output(f"Added {fname} to the chat.")
 
         self.summarizer = ChatSummary(
-            self.main_model.weak_model(),
+            self.main_model.get_weak_model(),
             self.main_model.max_chat_history_tokens,
         )
 
diff --git a/aider/history.py b/aider/history.py
index 3f12103c2..9ed866554 100644
--- a/aider/history.py
+++ b/aider/history.py
@@ -124,7 +124,7 @@ def main():
 
         assistant.append(line)
 
-    summarizer = ChatSummary(models.Model.weak_model())
+    summarizer = ChatSummary(models.Model.get_weak_model())
     summary = summarizer.summarize(messages[-40:])
     dump(summary)
 
diff --git a/aider/models.py b/aider/models.py
index bd7ab3e76..8e52ceb92 100644
--- a/aider/models.py
+++ b/aider/models.py
@@ -6,7 +6,7 @@ from typing import Optional
 import litellm
 from PIL import Image
 
-from aider.dump import dump
+from aider.dump import dump  # noqa: F401
 
 DEFAULT_MODEL_NAME = "gpt-4-1106-preview"
 
@@ -114,11 +114,13 @@ class Model:
     name = None
 
     weak_model_name = "gpt-3.5-turbo-0125"
+
     edit_format = "whole"
     use_repo_map = False
     send_undo_reply = False
 
     max_chat_history_tokens = 1024
+    weak_model = None
 
     def __init__(self, model):
         self.name = model
@@ -128,8 +130,6 @@ class Model:
         except Exception as err:
             raise NoModelInfo(str(err))
 
-        dump(model, self.info)
-
         if self.info.get("max_input_tokens", 0) < 32 * 1024:
             self.max_chat_history_tokens = 1024
         else:
@@ -159,14 +159,13 @@ class Model:
     def __str__(self):
         return self.name
 
-    def weak_model(self):
-        if self.name == self.weak_model_name:
-            return self
-
-        return Model(self.weak_model_name)
+    def get_weak_model(self):
+        if not self.weak_model:
+            self.weak_model = Model(self.weak_model_name)
+        return self.weak_model
 
     def commit_message_models(self):
-        return [self.weak_model()]
+        return [self.get_weak_model()]
 
     def tokenizer(self, text):
         return litellm.encode(model=self.name, text=text)