From e10fe50c6f00f1cc800a7594dd468a10bb735bd4 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <paul@aider.chat>
Date: Tue, 11 Mar 2025 11:29:38 -0700
Subject: [PATCH] feat: Add flexible token parsing for set_thinking_tokens
 method

---
 aider/models.py | 43 +++++++++++++++++++++++++++++++++++++++----
 1 file changed, 39 insertions(+), 4 deletions(-)

diff --git a/aider/models.py b/aider/models.py
index 7eb022681..9d8da8cd1 100644
--- a/aider/models.py
+++ b/aider/models.py
@@ -602,13 +602,48 @@ class Model(ModelSettings):
                 self.extra_params["extra_body"] = {}
             self.extra_params["extra_body"]["reasoning_effort"] = effort
 
-    def set_thinking_tokens(self, num):
-        """Set the thinking token budget for models that support it"""
-        if num is not None:
+    def parse_token_value(self, value):
+        """
+        Parse a token value string into an integer.
+        Accepts formats: 8096, "8k", "10.5k", "0.5M", "10K", etc.
+        
+        Args:
+            value: String or int token value
+            
+        Returns:
+            Integer token value
+        """
+        if isinstance(value, int):
+            return value
+            
+        if not isinstance(value, str):
+            return int(value)  # Try to convert to int
+            
+        value = value.strip().upper()
+        
+        if value.endswith('K'):
+            multiplier = 1024
+            value = value[:-1]
+        elif value.endswith('M'):
+            multiplier = 1024 * 1024
+            value = value[:-1]
+        else:
+            multiplier = 1
+            
+        # Convert to float first to handle decimal values like "10.5k"
+        return int(float(value) * multiplier)
+
+    def set_thinking_tokens(self, value):
+        """
+        Set the thinking token budget for models that support it.
+        Accepts formats: 8096, "8k", "10.5k", "0.5M", "10K", etc.
+        """
+        if value is not None:
+            num_tokens = self.parse_token_value(value)
             self.use_temperature = False
             if not self.extra_params:
                 self.extra_params = {}
-            self.extra_params["thinking"] = {"type": "enabled", "budget_tokens": num}
+            self.extra_params["thinking"] = {"type": "enabled", "budget_tokens": num_tokens}
 
     def is_deepseek_r1(self):
         name = self.name.lower()