dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-05 20:25:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Add fallback to second model when first model fails in ChatSummary

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-07-30 12:05:54 -03:00 • committed by Paul Gauthier
parent 963420d16f
commit 0980985338
2 changed files with 110 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

8
aider/history.py
View file

@ -11,12 +11,7 @@ class ChatSummary:
raise ValueError("At least one model must be provided")
self.models = models if isinstance(models, list) else [models]
self.max_tokens = max_tokens
self.current_model = None
self.set_current_model(self.models[0])
def set_current_model(self, model):
self.current_model = model
self.token_count = model.token_count
self.token_count = self.models[0].token_count
def too_big(self, messages):
sized = self.tokenize(messages)
@ -109,7 +104,6 @@ class ChatSummary:
]
for model in self.models:
self.set_current_model(model)
try:
summary = simple_send_with_retries(model.name, summarize_messages)
if summary is not None: