swarm_sdk 2.0.2 → 2.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f2cedd1197d0fea1e52b40531ecb6d6f030524436c3164c109bce37da1db883b
-  data.tar.gz: ac77a97bd54374a9049de888e0dfc4a1b1c99d5a4fdc2c01fabe20e54a7cd83b
+  metadata.gz: 9aba64a75c41f9957bb3e11afbe0dd5a5fc6422793fda5d3fb829ac0042cbe18
+  data.tar.gz: 4f403fb7d107fa712a8b847775fea1140a03904cf066b8e848b914f4b3c573e3
 SHA512:
-  metadata.gz: 9d1dfdd64d1247e15a3fea9ce56b0d28657e85a5f9d706d1a2ce9c72700226cf443dcdc691596710d2e948816a3fb5d93fe1520f3bc798df78a2b6a91e9a2085
-  data.tar.gz: b54ae2918f3cb04eb2a49a80faecc2aac1a878cd14828cc3287705eae22ba4da09780c8e38a54c5d048cf4eb1fec75e523dc609192758f6ad60294467d68df01
+  metadata.gz: 204b9d889968b2d582557b06169d02828abd4c365b00e2a6551cc7f4c44a2ec7ae452a55d1e25b59fdc8a0b10c486c365cce2b2332d02f0d54b502ec6667af75
+  data.tar.gz: ae46da01755ef0c98b162c7c2be3c51ba8831dbbdd10301f937e0eed0260f9694b7514658d437e9d65d9cc58e06d7a877600ec29fd48eb773eec8514d74440b1

data/lib/swarm_sdk/agent/builder.rb

@@ -44,7 +44,7 @@ module SwarmSDK
         @headers = {}
         @timeout = nil
         @mcp_servers = []
-        @include_default_tools = true
+        @disable_default_tools = nil # nil = include all default tools
         @bypass_permissions = false
         @coding_agent = nil # nil = not set (will default to false in Definition)
         @assume_model_exists = nil
@@ -124,9 +124,19 @@ module SwarmSDK
         @mcp_servers << server_config
       end
 
-      # Set include_default_tools flag (deprecated - use tools method with include_default parameter)
-      def include_default_tools(enabled)
-        @include_default_tools = enabled
+      # Disable default tools
+      #
+      # @param value [Boolean, Array<Symbol>]
+      #   - true: Disable ALL default tools
+      #   - Array of symbols: Disable specific tools (e.g., [:Think, :TodoWrite])
+      #
+      # @example Disable all default tools
+      #   disable_default_tools true
+      #
+      # @example Disable specific tools
+      #   disable_default_tools [:Think, :TodoWrite]
+      def disable_default_tools(value)
+        @disable_default_tools = value
       end
 
       # Set bypass_permissions flag
@@ -188,7 +198,8 @@ module SwarmSDK
       def tools(*tool_names, include_default: true, replace: false)
         @tools = Set.new if replace
         @tools.merge(tool_names.map(&:to_sym))
-        @include_default_tools = include_default
+        # When include_default is false, disable all default tools
+        @disable_default_tools = true unless include_default
       end
 
       # Add tools from all_agents configuration
@@ -342,7 +353,7 @@ module SwarmSDK
         agent_config[:headers] = @headers if @headers.any?
         agent_config[:timeout] = @timeout if @timeout
         agent_config[:mcp_servers] = @mcp_servers if @mcp_servers.any?
-        agent_config[:include_default_tools] = @include_default_tools
+        agent_config[:disable_default_tools] = @disable_default_tools unless @disable_default_tools.nil?
         agent_config[:bypass_permissions] = @bypass_permissions
         agent_config[:coding_agent] = @coding_agent
         agent_config[:assume_model_exists] = @assume_model_exists unless @assume_model_exists.nil?

data/lib/swarm_sdk/agent/definition.rb

@@ -38,7 +38,7 @@ module SwarmSDK
         :parameters,
         :headers,
         :timeout,
-        :include_default_tools,
+        :disable_default_tools,
         :coding_agent,
         :default_permissions,
         :agent_permissions,
@@ -77,8 +77,11 @@ module SwarmSDK
         # This prevents RubyLLM from trying to validate models in its registry
         @assume_model_exists = true
 
-        # include_default_tools defaults to true if not specified
-        @include_default_tools = config.key?(:include_default_tools) ? config[:include_default_tools] : true
+        # disable_default_tools can be:
+        # - nil/not set: include all default tools (default behavior)
+        # - true: disable ALL default tools
+        # - Array of symbols: disable specific tools (e.g., [:Think, :TodoWrite])
+        @disable_default_tools = config[:disable_default_tools]
 
         # coding_agent defaults to false if not specified
         # When true, includes the base system prompt for coding tasks
@@ -117,7 +120,7 @@ module SwarmSDK
         {
           name: @name,
           description: @description,
-          model: @model,
+          model: SwarmSDK::Models.resolve_alias(@model), # Resolve model aliases
           directory: @directory,
           tools: @tools,
           delegates_to: @delegates_to,
@@ -130,7 +133,7 @@ module SwarmSDK
           headers: @headers,
           timeout: @timeout,
           bypass_permissions: @bypass_permissions,
-          include_default_tools: @include_default_tools,
+          disable_default_tools: @disable_default_tools,
           coding_agent: @coding_agent,
           assume_model_exists: @assume_model_exists,
           max_concurrent_tools: @max_concurrent_tools,
@@ -224,7 +227,7 @@ module SwarmSDK
           else
             rendered_base
           end
-        elsif @include_default_tools
+        elsif default_tools_enabled?
           # Non-coding agent: optionally include TODO/Scratchpad sections if default tools available
           non_coding_base = render_non_coding_base_prompt
 
@@ -242,6 +245,13 @@ module SwarmSDK
         end
       end
 
+      # Check if default tools are enabled (i.e., not disabled)
+      #
+      # @return [Boolean] True if default tools should be included
+      def default_tools_enabled?
+        @disable_default_tools != true
+      end
+
       def render_base_system_prompt
         cwd = @directory || Dir.pwd
         platform = RUBY_PLATFORM

data/lib/swarm_sdk/configuration.rb

@@ -116,6 +116,11 @@ module SwarmSDK
       return unless @config[:swarm]
 
       @all_agents_config = @config[:swarm][:all_agents] || {}
+
+      # Convert disable_default_tools array elements to symbols
+      if @all_agents_config[:disable_default_tools].is_a?(Array)
+        @all_agents_config[:disable_default_tools] = @all_agents_config[:disable_default_tools].map(&:to_sym)
+      end
     end
 
     def load_hooks_config
@@ -212,6 +217,9 @@ module SwarmSDK
           # Merge headers: all_agents.headers + agent.headers
           # Agent values override all_agents values for same keys
           merged[:headers] = (merged[:headers] || {}).merge(value || {})
+        when :disable_default_tools
+          # Convert array elements to symbols if it's an array
+          merged[key] = value.is_a?(Array) ? value.map(&:to_sym) : value
         else
           # For everything else (model, provider, base_url, timeout, coding_agent, etc.),
           # agent value overrides all_agents value

data/lib/swarm_sdk/swarm/builder.rb

@@ -278,7 +278,7 @@ module SwarmSDK
         # Don't apply assume_model_exists from markdown - let DSL overrides or auto-enable handle it
         # builder.assume_model_exists(config[:assume_model_exists]) unless config[:assume_model_exists].nil?
         builder.bypass_permissions(config[:bypass_permissions]) if config[:bypass_permissions]
-        builder.include_default_tools(config[:include_default_tools]) unless config[:include_default_tools].nil?
+        builder.disable_default_tools(config[:disable_default_tools]) unless config[:disable_default_tools].nil?
 
         # Add tools from markdown
         if config[:tools]&.any?

data/lib/swarm_sdk/swarm/tool_configurator.rb

@@ -12,7 +12,7 @@ module SwarmSDK
     #
     # This encapsulates all tool-related logic that was previously in Swarm.
     class ToolConfigurator
-      # Default tools available to all agents (unless include_default_tools: false)
+      # Default tools available to all agents (unless disable_default_tools is set)
       DEFAULT_TOOLS = [
         :Read,
         :Grep,
@@ -21,6 +21,7 @@ module SwarmSDK
         :ScratchpadWrite,
         :ScratchpadRead,
         :ScratchpadList,
+        :Think,
       ].freeze
 
       def initialize(swarm, scratchpad)
@@ -75,6 +76,8 @@ module SwarmSDK
           Tools::ScratchpadRead.create_for_scratchpad(@scratchpad)
         when :ScratchpadList
           Tools::ScratchpadList.create_for_scratchpad(@scratchpad)
+        when :Think
+          Tools::Think.new
         else
           # Regular tools - get class from registry and instantiate
           tool_class = Tools::Registry.get(tool_name_sym)
@@ -133,13 +136,14 @@ module SwarmSDK
         end
       end
 
-      # Register default tools for agents that have include_default_tools enabled
+      # Register default tools for agents (unless disabled)
       #
       # @param chat [AgentChat] The chat instance
       # @param agent_name [Symbol] Agent name
       # @param agent_definition [AgentDefinition] Agent definition
       def register_default_tools(chat, agent_name:, agent_definition:)
-        return unless agent_definition.include_default_tools
+        # If disable_default_tools is true, skip all default tools
+        return if agent_definition.disable_default_tools == true
 
         # Get explicit tool names to avoid duplicates
         explicit_tool_names = agent_definition.tools.map { |t| t[:name] }.to_set
@@ -148,6 +152,9 @@ module SwarmSDK
           # Skip if already registered explicitly
           next if explicit_tool_names.include?(tool_name)
 
+          # Skip if tool is in the disable list
+          next if tool_disabled?(tool_name, agent_definition.disable_default_tools)
+
           tool_instance = create_tool_instance(tool_name, agent_name, agent_definition.directory)
 
           # Resolve permissions for default tool (same logic as AgentDefinition)
@@ -166,6 +173,25 @@ module SwarmSDK
         end
       end
 
+      # Check if a tool should be disabled based on disable_default_tools config
+      #
+      # @param tool_name [Symbol] Tool name to check
+      # @param disable_config [nil, Boolean, Array<Symbol>] Disable configuration
+      # @return [Boolean] True if tool should be disabled
+      def tool_disabled?(tool_name, disable_config)
+        return false if disable_config.nil?
+
+        if disable_config == true
+          # Disable all default tools
+          true
+        elsif disable_config.is_a?(Array)
+          # Disable only tools in the array
+          disable_config.include?(tool_name)
+        else
+          false
+        end
+      end
+
       # Register agent delegation tools
       #
       # Creates delegation tools that allow one agent to call another.

data/lib/swarm_sdk/tools/registry.rb

@@ -20,6 +20,7 @@ module SwarmSDK
         ScratchpadWrite: :special, # Requires scratchpad instance
         ScratchpadRead: :special, # Requires scratchpad instance
         ScratchpadList: :special, # Requires scratchpad instance
+        Think: SwarmSDK::Tools::Think,
       }.freeze
 
       class << self

data/lib/swarm_sdk/tools/think.rb

@@ -0,0 +1,95 @@
+# frozen_string_literal: true
+
+module SwarmSDK
+  module Tools
+    # Think tool for explicit reasoning and planning
+    #
+    # Allows the agent to write down thoughts, plans, strategies, and intermediate
+    # calculations. These thoughts become part of the conversation context, enabling
+    # better attention and reasoning through complex problems.
+    #
+    # This is inspired by research showing that explicitly articulating reasoning steps
+    # (chain-of-thought prompting) leads to significantly better outcomes, especially
+    # for complex tasks requiring multi-step reasoning or arithmetic.
+    class Think < RubyLLM::Tool
+      def name
+        "Think"
+      end
+
+      description <<~DESC
+        **IMPORTANT: You SHOULD use this tool frequently throughout your work. Using this tool leads to significantly
+        better outcomes and more accurate solutions. Make it a habit to think before acting.**
+
+        This tool allows you to write down your thoughts, plans, strategies, and intermediate calculations.
+        Think of it as your working memory - just as humans think before speaking or acting, you should think before
+        using other tools or providing responses.
+
+        **STRONGLY RECOMMENDED to use this tool:**
+        - **ALWAYS** before starting any task (even simple ones)
+        - **ALWAYS** when you need to do any arithmetic or counting
+        - **ALWAYS** after reading files or getting search results to process what you learned
+        - **FREQUENTLY** between steps to track progress and plan next actions
+
+        This is your private thinking space - use it liberally to enhance your problem-solving capabilities. Recording
+        your thoughts helps you maintain context across multiple steps and remember important information throughout your task.
+
+        When and how to use this tool:
+
+        1. **Before starting any complex task**: Write down your understanding of the problem, break it into smaller
+           sub-tasks, and create a step-by-step plan. Example:
+           - "The user wants me to refactor this codebase. Let me first understand the structure..."
+           - "I need to: 1) Analyze current architecture, 2) Identify pain points, 3) Propose changes..."
+
+        2. **For arithmetic and calculations**: Work through math problems step by step. Example:
+           - "If we have 150 requests/second and each takes 20ms, that's 150 * 0.02 = 3 seconds of CPU time..."
+           - "Converting 2GB to bytes: 2 * 1024 * 1024 * 1024 = 2,147,483,648 bytes"
+
+        3. **After completing sub-tasks**: Summarize what you've accomplished and what remains. Example:
+           - "I've successfully implemented the authentication module. Next, I need to integrate it with the API..."
+           - "Fixed 3 out of 5 bugs. Remaining: memory leak in parser, race condition in worker thread"
+
+        4. **When encountering complexity**: Break down complex logic or decisions. Example:
+           - "This function has multiple edge cases. Let me list them: null input, empty array, negative numbers..."
+           - "The user's request is ambiguous. Possible interpretations: A) modify existing code, B) create new module..."
+
+        5. **For remembering context**: Store important information you'll need later. Example:
+           - "Important: The user mentioned they're using Ruby 3.2, so I can use pattern matching"
+           - "File structure: main.rb requires from lib/, config is in config.yml"
+
+        6. **When debugging or analyzing**: Track your investigation process. Example:
+           - "The error occurs in line 42. Let me trace backwards: function called from main(), receives data from..."
+           - "Hypothesis: the bug might be due to timezone differences. Let me check..."
+
+        7. **For creative problem-solving**: Brainstorm multiple approaches before choosing one. Example:
+           - "Approaches to optimize this: 1) Add caching, 2) Use parallel processing, 3) Optimize algorithm..."
+           - "Design patterns that could work here: Factory, Observer, or maybe Strategy pattern..."
+
+        **Remember: The most successful agents use this tool 5-10 times per task on average. If you haven't used this
+        tool in the last 2-3 actions, you probably should. Using this tool is a sign of thoughtful, methodical problem
+        solving and leads to fewer mistakes and better solutions.**
+
+        Your thoughts persist throughout your session as part of the conversation history, so you can refer
+        back to earlier thinking. Use clear formatting and organization to make it easy to reference
+        later. Don't hesitate to think out loud - this tool is designed to augment your cognitive capabilities and help
+        you deliver better solutions.
+      DESC
+
+      param :thoughts,
+        type: "string",
+        desc: "Your thoughts, plans, calculations, or any notes you want to record",
+        required: true
+
+      def execute(thoughts:)
+        return validation_error("thoughts are required") if thoughts.nil? || thoughts.empty?
+
+        "Thought noted."
+      end
+
+      private
+
+      def validation_error(message)
+        "<tool_use_error>InputValidationError: #{message}</tool_use_error>"
+      end
+    end
+  end
+end

data/lib/swarm_sdk/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module SwarmSDK
-  VERSION = "2.0.2"
+  VERSION = "2.0.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: swarm_sdk
 version: !ruby/object:Gem::Version
-  version: 2.0.2
+  version: 2.0.3
 platform: ruby
 authors:
 - Paulo Arruda
@@ -148,6 +148,7 @@ files:
 - lib/swarm_sdk/tools/stores/read_tracker.rb
 - lib/swarm_sdk/tools/stores/scratchpad.rb
 - lib/swarm_sdk/tools/stores/todo_manager.rb
+- lib/swarm_sdk/tools/think.rb
 - lib/swarm_sdk/tools/todo_write.rb
 - lib/swarm_sdk/tools/write.rb
 - lib/swarm_sdk/utils.rb