From 73734efe8946dbfd913226747c7a18c5a80c9939 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Mon, 12 Aug 2024 15:20:05 -0700
Subject: [PATCH] feat: Add infinity symbol to edit format if model supports
 assistant prefill

---
 aider/coders/base_coder.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py
index 49a9c61c7..f8c13b0ef 100755
--- a/aider/coders/base_coder.py
+++ b/aider/coders/base_coder.py
@@ -150,6 +150,8 @@ class Coder:
         weak_model = main_model.weak_model
         prefix = "Model:"
         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:
             prefix = "Models:"
             output += f", weak model {weak_model.name}"