Add can_reason to model as suggested

Author: bashtoni

lua/codecompanion/adapters/anthropic.lua

@@ -21,7 +21,7 @@ return {
   opts = {
     stream = true,
     cache_breakpoints = 4, -- Cache up to this many messages
-    cache_over = 300, -- Cache any message which has this many tokens or more
+    cache_over = 300,      -- Cache any message which has this many tokens or more
   },
   url = "https://api.anthropic.com/v1/messages",
   env = {
@@ -81,18 +81,18 @@ return {
     form_messages = function(self, messages)
       -- Extract and format system messages
       local system = vim
-        .iter(messages)
-        :filter(function(msg)
-          return msg.role == "system"
-        end)
-        :map(function(msg)
-          return {
-            type = "text",
-            text = msg.content,
-            cache_control = nil, -- To be set later if needed
-          }
-        end)
-        :totable()
+          .iter(messages)
+          :filter(function(msg)
+            return msg.role == "system"
+          end)
+          :map(function(msg)
+            return {
+              type = "text",
+              text = msg.content,
+              cache_control = nil, -- To be set later if needed
+            }
+          end)
+          :totable()
       system = next(system) and system or nil
 
       -- Remove system messages and merge user/assistant messages
@@ -114,9 +114,9 @@ return {
       for i = #messages, 1, -1 do
         local message = messages[i]
         if
-          message.role == self.roles.user
-          and tokens.calculate(message.content) >= self.opts.cache_over
-          and breakpoints_used < self.opts.cache_breakpoints
+            message.role == self.roles.user
+            and tokens.calculate(message.content) >= self.opts.cache_over
+            and breakpoints_used < self.opts.cache_breakpoints
         then
           message.content = {
             {
@@ -158,7 +158,7 @@ return {
         if ok then
           if json.type == "message_start" then
             input_tokens = (json.message.usage.input_tokens or 0)
-              + (json.message.usage.cache_creation_input_tokens or 0)
+                + (json.message.usage.cache_creation_input_tokens or 0)
 
             output_tokens = json.message.usage.output_tokens or 0
           elseif json.type == "message_delta" then
@@ -260,10 +260,11 @@ return {
       order = 1,
       mapping = "parameters",
       type = "enum",
-      desc = "The model that will complete your prompt. See https://docs.anthropic.com/claude/docs/models-overview for additional details and options.",
+      desc =
+      "The model that will complete your prompt. See https://docs.anthropic.com/claude/docs/models-overview for additional details and options.",
       default = "claude-3-7-sonnet-20250219",
       choices = {
-        "claude-3-7-sonnet-20250219",
+        ["claude-3-7-sonnet-20250219"] = { opts = { can_reason = true } },
         "claude-3-5-sonnet-20241022",
         "claude-3-5-haiku-20241022",
         "claude-3-opus-20240229",
@@ -285,25 +286,39 @@ return {
       optional = true,
       default = false,
       desc = "Enable extended thinking for more thorough reasoning. Requires thinking_budget to be set.",
+      condition = function(schema)
+        local model = schema.model.default
+        if schema.model.choices[model] and schema.model.choices[model].opts then
+          return schema.model.choices[model].opts.can_reason
+        end
+      end,
     },
     thinking_budget = {
       order = 4,
       mapping = "parameters",
       type = "number",
       optional = true,
       default = 16000,
-      desc = "The maximum number of tokens to use for thinking when extended_thinking is enabled. Must be less than max_tokens.",
+      desc =
+      "The maximum number of tokens to use for thinking when extended_thinking is enabled. Must be less than max_tokens.",
       validate = function(n)
         return n > 0, "Must be greater than 0"
       end,
+      condition = function(schema)
+        local model = schema.model.default
+        if schema.model.choices[model] and schema.model.choices[model].opts then
+          return schema.model.choices[model].opts.can_reason
+        end
+      end,
     },
     max_tokens = {
       order = 5,
       mapping = "parameters",
       type = "number",
       optional = true,
       default = 4096,
-      desc = "The maximum number of tokens to generate before stopping. This parameter only specifies the absolute maximum number of tokens to generate. Different models have different maximum values for this parameter.",
+      desc =
+      "The maximum number of tokens to generate before stopping. This parameter only specifies the absolute maximum number of tokens to generate. Different models have different maximum values for this parameter.",
       validate = function(n)
         return n > 0 and n <= 32768, "Must be between 0 and 32768"
       end,
@@ -314,7 +329,8 @@ return {
       type = "number",
       optional = true,
       default = 0,
-      desc = "Amount of randomness injected into the response. Ranges from 0.0 to 1.0. Use temperature closer to 0.0 for analytical / multiple choice, and closer to 1.0 for creative and generative tasks. Note that even with temperature of 0.0, the results will not be fully deterministic.",
+      desc =
+      "Amount of randomness injected into the response. Ranges from 0.0 to 1.0. Use temperature closer to 0.0 for analytical / multiple choice, and closer to 1.0 for creative and generative tasks. Note that even with temperature of 0.0, the results will not be fully deterministic.",
       validate = function(n)
         return n >= 0 and n <= 1, "Must be between 0 and 1.0"
       end,
@@ -325,7 +341,8 @@ return {
       type = "number",
       optional = true,
       default = nil,
-      desc = "Computes the cumulative distribution over all the options for each subsequent token in decreasing probability order and cuts it off once it reaches a particular probability specified by top_p",
+      desc =
+      "Computes the cumulative distribution over all the options for each subsequent token in decreasing probability order and cuts it off once it reaches a particular probability specified by top_p",
       validate = function(n)
         return n >= 0 and n <= 1, "Must be between 0 and 1"
       end,
@@ -336,7 +353,8 @@ return {
       type = "number",
       optional = true,
       default = nil,
-      desc = "Only sample from the top K options for each subsequent token. Use top_k to remove long tail low probability responses",
+      desc =
+      "Only sample from the top K options for each subsequent token. Use top_k to remove long tail low probability responses",
       validate = function(n)
         return n >= 0, "Must be greater than 0"
       end,