feat: Add infinity symbol to edit format if model supports assistant prefill

This commit is contained in:
Paul Gauthier (aider) 2024-08-12 15:20:05 -07:00
parent 888211fd48
commit 73734efe89

View file

@ -150,6 +150,8 @@ class Coder:
weak_model = main_model.weak_model weak_model = main_model.weak_model
prefix = "Model:" prefix = "Model:"
output = f" {main_model.name} with {self.edit_format} edit format" output = f" {main_model.name} with {self.edit_format} edit format"
if main_model.info.get("supports_assistant_prefill"):
output += ""
if weak_model is not main_model: if weak_model is not main_model:
prefix = "Models:" prefix = "Models:"
output += f", weak model {weak_model.name}" output += f", weak model {weak_model.name}"