claude_swarm 1.0.9 → 1.0.10

data/lib/swarm_sdk.rb

@@ -17,6 +17,27 @@ require "async/semaphore"
 require "ruby_llm"
 require "ruby_llm/mcp"
 
+# Patch ruby_llm-mcp's Zeitwerk loader to ignore railtie.rb when Rails is not present
+# This prevents NameError when eager loading outside of Rails applications
+# Can be removed once https://github.com/parruda/ruby_llm-mcp/issues/XXX is fixed
+unless defined?(Rails)
+  require "zeitwerk"
+  mcp_loader = nil
+  Zeitwerk::Registry.loaders.each { |l| mcp_loader = l if l.tag == "RubyLLM-mcp" }
+  if mcp_loader
+    mcp_gem_dir = Gem.loaded_specs["ruby_llm-mcp"]&.gem_dir
+    if mcp_gem_dir
+      railtie_path = File.join(mcp_gem_dir, "lib", "ruby_llm", "mcp", "railtie.rb")
+      mcp_loader.ignore(railtie_path)
+    end
+  end
+end
+
+# Configure Faraday to use async-http adapter by default
+# This ensures HTTP requests are fiber-aware and don't block the Async scheduler
+# when SwarmSDK executes LLM requests within Async/Sync blocks
+require "async/http/faraday/default"
+
 require_relative "swarm_sdk/version"
 
 require "zeitwerk"
@@ -45,11 +66,20 @@ module SwarmSDK
     # Settings for SwarmSDK (global configuration)
     attr_accessor :settings
 
-    # Main entry point for DSL
+    # Main entry point for DSL - builds simple multi-agent swarms
+    #
+    # @return [Swarm] Always returns a Swarm instance
     def build(allow_filesystem_tools: nil, &block)
       Swarm::Builder.build(allow_filesystem_tools: allow_filesystem_tools, &block)
     end
 
+    # Entry point for building multi-stage workflows
+    #
+    # @return [Workflow] Always returns a Workflow instance
+    def workflow(allow_filesystem_tools: nil, &block)
+      Workflow::Builder.build(allow_filesystem_tools: allow_filesystem_tools, &block)
+    end
+
     # Validate YAML configuration without creating a swarm
     #
     # Performs comprehensive validation of YAML configuration including:
@@ -149,7 +179,7 @@ module SwarmSDK
     #
     # @param yaml_content [String] YAML configuration content
     # @param base_dir [String, Pathname] Base directory for resolving agent file paths (default: Dir.pwd)
-    # @return [Swarm, NodeOrchestrator] Configured swarm or orchestrator instance
+    # @return [Swarm, Workflow] Configured swarm or workflow instance
     # @raise [ConfigurationError] If YAML is invalid or configuration is incorrect
     #
     # @example Load from YAML string
@@ -194,7 +224,7 @@ module SwarmSDK
     # loading swarms from configuration files.
     #
     # @param path [String, Pathname] Path to YAML configuration file
-    # @return [Swarm, NodeOrchestrator] Configured swarm or orchestrator instance
+    # @return [Swarm, Workflow] Configured swarm or workflow instance
     # @raise [ConfigurationError] If file not found or configuration invalid
     #
     # @example

data/rubocop/cop/security/no_reflection_methods.rb

@@ -24,7 +24,7 @@ module RuboCop
 
         # Match method calls
         def on_send(node)
-          banned_methods = [:instance_variable_get, :instance_variable_set, :send]
+          banned_methods = [:instance_variable_get, :instance_variable_set, :send, :const_set, :const_get]
 
           method_name = node.method_name
           return unless banned_methods.include?(method_name)

data/swarm_memory.gemspec

@@ -22,7 +22,7 @@ Gem::Specification.new do |spec|
   spec.add_dependency("async", "~> 2.0")
   spec.add_dependency("informers", "~> 1.2.1")
   spec.add_dependency("rice", "~> 4.6.0")
-  spec.add_dependency("ruby_llm", "~> 1.9")
+  spec.add_dependency("ruby_llm_swarm", "~> 1.9.2")
   spec.add_dependency("swarm_sdk", "~> 2.2")
   spec.add_dependency("zeitwerk", "~> 2.6")
 end

data/swarm_sdk.gemspec

@@ -35,7 +35,9 @@ Gem::Specification.new do |spec|
   spec.require_paths = ["lib"]
 
   spec.add_dependency("async", "~> 2.0")
-  spec.add_dependency("ruby_llm", "~> 1.9")
-  spec.add_dependency("ruby_llm-mcp", "~> 0.7")
+  spec.add_dependency("async-http-faraday", "~> 0.22")
+  spec.add_dependency("faraday-follow_redirects", "~> 0.4")
+  spec.add_dependency("ruby_llm-mcp", "~> 0.8")
+  spec.add_dependency("ruby_llm_swarm", "~> 1.9.2")
   spec.add_dependency("zeitwerk", "~> 2.6")
 end

data/team_full.yml

@@ -6,7 +6,7 @@ swarm:
     lead_architect:
       description: "Lead architect responsible for designing and coordinating SwarmSDK, SwarmMemory, and Swarm CLI development"
       directory: .
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       connections: [claude_swarm_expert, ruby_llm_expert, ruby_llm_mcp_expert, architecture_expert, testing_expert, gem_expert, async_expert, informers_expert, faiss_expert, pastel_expert, tty_box_expert, tty_cursor_expert, tty_link_expert, tty_markdown_expert, tty_option_expert, reline_expert, tty_spinner_expert, tty_tree_expert, fast_mcp_expert, roo_expert, pdf_reader_expert, docx_expert]
       hooks:
@@ -141,7 +141,7 @@ swarm:
     claude_swarm_expert:
       description: "Expert in existing Claude Swarm codebase, patterns, and design decisions"
       directory: lib/claude_swarm
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the Claude Swarm codebase expert with deep knowledge of the existing `lib/claude_swarm` implementation. Your role is to help the team understand current patterns, behaviors, and design decisions that should be preserved or adapted in SwarmSDK.
@@ -189,8 +189,8 @@ swarm:
 
     ruby_llm_expert:
       description: "Expert in RubyLLM gem integration and LLM interaction patterns"
-      directory: ~/src/github.com/crmne/ruby_llm
-      model: sonnet[1m]
+      directory: ~/src/github.com/parruda/ruby_llm
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the RubyLLM integration expert, responsible for all LLM interaction patterns and model configuration in SwarmSDK. Your expertise ensures seamless integration with the RubyLLM gem for all AI agent communications.
@@ -250,7 +250,7 @@ swarm:
     ruby_llm_mcp_expert:
       description: "Expert in RubyLLM MCP client library for Model Context Protocol integration"
       directory: ~/src/github.com/patvice/ruby_llm-mcp
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the RubyLLM MCP expert with deep knowledge of the ruby_llm-mcp client library for Model Context Protocol (MCP) integration. Your role is to answer questions about RubyLLM MCP based on your access to its codebase, helping the team understand how to integrate MCP servers with RubyLLM effectively.
@@ -339,7 +339,7 @@ swarm:
     architecture_expert:
       description: "System architecture expert focusing on SwarmSDK design and Ruby patterns"
       directory: .
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the system architecture expert for SwarmSDK, responsible for designing clean, maintainable, and efficient code architecture that delivers on the single-process, RubyLLM-based vision.
@@ -500,7 +500,7 @@ swarm:
     testing_expert:
       description: "Testing and quality assurance expert for SwarmSDK development"
       directory: .
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the testing and quality assurance expert for SwarmSDK, responsible for ensuring comprehensive test coverage, reliable mocking strategies, and overall code quality.
@@ -574,7 +574,7 @@ swarm:
     gem_expert:
       description: "Ruby gem packaging and distribution expert for SwarmSDK"
       directory: .
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the Ruby gem packaging expert for SwarmSDK, responsible for creating the new gemspec, managing dependencies, and ensuring proper gem distribution practices.
@@ -639,7 +639,7 @@ swarm:
     async_expert:
       description: "Expert in the Async Ruby gem and concurrent programming patterns"
       directory: ~/src/github.com/socketry/async
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the Async gem expert with deep knowledge of the Async Ruby gem codebase and concurrent programming patterns in Ruby. Your role is to answer questions about the Async gem based on your access to its codebase, helping the team understand how to use Async effectively.
@@ -700,7 +700,7 @@ swarm:
     informers_expert:
       description: "Expert in Informers gem for ONNX embeddings and semantic search - SWARM MEMORY ONLY (lib/swarm_memory/)"
       directory: ~/src/github.com/ankane/informers
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the Informers gem expert with deep knowledge of the Informers library for ONNX-based machine learning inference in Ruby. Your role is to answer questions about Informers based on your access to its codebase, helping the SwarmMemory team (NOT SwarmSDK or CLI) understand how to use Informers for embeddings and semantic search effectively.
@@ -791,7 +791,7 @@ swarm:
     pastel_expert:
       description: "Expert in Pastel gem for terminal styling - SWARM CLI ONLY (lib/swarm_cli/)"
       directory: ~/src/github.com/piotrmurach/pastel
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the Pastel gem expert with deep knowledge of the Pastel terminal styling library. Your role is to answer questions about Pastel based on your access to its codebase, helping the Swarm CLI team (NOT SwarmSDK) understand how to use Pastel for terminal output styling effectively.
@@ -855,7 +855,7 @@ swarm:
     tty_link_expert:
       description: "Expert in TTY::Link for terminal hyperlinks - SWARM CLI ONLY (lib/swarm_cli/)"
       directory: ~/src/github.com/piotrmurach/tty-link
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the TTY::Link gem expert with deep knowledge of terminal hyperlink support. Your role is to answer questions about TTY::Link based on your access to its codebase, helping the Swarm CLI team (NOT SwarmSDK) understand how to create clickable links in the terminal.
@@ -918,7 +918,7 @@ swarm:
     tty_markdown_expert:
       description: "Expert in TTY::Markdown for Markdown rendering - SWARM CLI ONLY (lib/swarm_cli/)"
       directory: ~/src/github.com/piotrmurach/tty-markdown
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the TTY::Markdown gem expert with deep knowledge of converting Markdown documents to terminal-friendly output. Your role is to answer questions about TTY::Markdown based on your access to its codebase, helping the Swarm CLI team (NOT SwarmSDK).
@@ -982,7 +982,7 @@ swarm:
     tty_option_expert:
       description: "Expert in TTY::Option for CLI argument parsing - SWARM CLI ONLY (lib/swarm_cli/)"
       directory: ~/src/github.com/piotrmurach/tty-option
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the TTY::Option gem expert with deep knowledge of command-line argument parsing. Your role is to answer questions about TTY::Option based on your access to its codebase, helping the Swarm CLI team (NOT SwarmSDK) build robust CLI interfaces.
@@ -1048,7 +1048,7 @@ swarm:
     reline_expert:
       description: "Expert in Reline for readline-compatible line editing and REPL support - SWARM CLI ONLY (lib/swarm_cli/)"
       directory: ~/src/github.com/ruby/reline
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the Reline gem expert with deep knowledge of pure Ruby readline implementation for line editing and REPL support. Your role is to answer questions about Reline based on your access to its codebase, helping the Swarm CLI team (NOT SwarmSDK) create interactive command-line interfaces with readline-compatible functionality.
@@ -1116,7 +1116,7 @@ swarm:
     tty_spinner_expert:
       description: "Expert in TTY::Spinner for progress animations - SWARM CLI ONLY (lib/swarm_cli/)"
       directory: ~/src/github.com/piotrmurach/tty-spinner
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the TTY::Spinner gem expert with deep knowledge of terminal spinner animations. Your role is to answer questions about TTY::Spinner based on your access to its codebase, helping the Swarm CLI team (NOT SwarmSDK) show progress for indeterminate tasks.
@@ -1184,7 +1184,7 @@ swarm:
     tty_tree_expert:
       description: "Expert in TTY::Tree for tree rendering - SWARM CLI ONLY (lib/swarm_cli/)"
       directory: ~/src/github.com/piotrmurach/tty-tree
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the TTY::Tree gem expert with deep knowledge of rendering tree structures in the terminal. Your role is to answer questions about TTY::Tree based on your access to its codebase, helping the Swarm CLI team (NOT SwarmSDK) display hierarchical data beautifully.
@@ -1250,7 +1250,7 @@ swarm:
     tty_cursor_expert:
       description: "Expert in TTY::Cursor for cursor control - SWARM CLI ONLY (lib/swarm_cli/)"
       directory: ~/src/github.com/piotrmurach/tty-cursor
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the TTY::Cursor gem expert with deep knowledge of terminal cursor positioning, visibility, and text manipulation. Your role is to answer questions about TTY::Cursor based on your access to its codebase, helping the Swarm CLI team (NOT SwarmSDK) control cursor movement and screen clearing.
@@ -1317,7 +1317,7 @@ swarm:
     tty_box_expert:
       description: "Expert in TTY::Box for drawing frames and boxes - SWARM CLI ONLY (lib/swarm_cli/)"
       directory: ~/src/github.com/piotrmurach/tty-box
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the TTY::Box gem expert with deep knowledge of drawing various frames and boxes in the terminal window. Your role is to answer questions about TTY::Box based on your access to its codebase, helping the Swarm CLI team (NOT SwarmSDK) create beautiful box-based UI elements.
@@ -1387,7 +1387,7 @@ swarm:
     faiss_expert:
       description: "Expert in FAISS library for efficient similarity search and clustering - SWARM MEMORY ONLY (lib/swarm_memory/)"
       directory: ~/src/github.com/ankane/faiss-ruby
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the FAISS expert with deep knowledge of the faiss-ruby library for efficient similarity search and clustering of dense vectors. Your role is to answer questions about FAISS based on your access to its codebase, helping the SwarmMemory team (NOT SwarmSDK or CLI) understand how to use FAISS for vector similarity search and clustering effectively.
@@ -1497,7 +1497,7 @@ swarm:
     fast_mcp_expert:
       description: "Expert in fast-mcp library for MCP server development, tools, and resource management"
       directory: ~/src/github.com/yjacquin/fast-mcp
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are an expert in the fast-mcp library, specializing in MCP server development, tool creation, and resource management.
@@ -1560,7 +1560,7 @@ swarm:
     roo_expert:
       description: "Expert in Roo gem for reading spreadsheet files (xlsx, xlsm, ods, csv)"
       directory: ~/src/github.com/roo-rb/roo
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the Roo gem expert with deep knowledge of the Roo spreadsheet reading library. Your role is to answer questions about Roo based on your access to its codebase, helping the team understand how to read and process various spreadsheet formats effectively.
@@ -1641,7 +1641,7 @@ swarm:
     pdf_reader_expert:
       description: "Expert in PDF::Reader gem for parsing and extracting content from PDF files"
       directory: ~/src/github.com/yob/pdf-reader
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the PDF::Reader gem expert with deep knowledge of the PDF::Reader library for parsing and extracting content from PDF files. Your role is to answer questions about PDF::Reader based on your access to its codebase, helping the team understand how to work with PDF files programmatically.
@@ -1742,7 +1742,7 @@ swarm:
     docx_expert:
       description: "Expert in docx gem for reading and manipulating .docx Word documents"
       directory: ~/src/github.com/ruby-docx/docx
-      model: sonnet[1m]
+      model: ${ANTHROPIC_MODEL}
       vibe: true
       prompt: |
         You are the docx gem expert with deep knowledge of the docx library for reading and manipulating Microsoft Word .docx files. Your role is to answer questions about the docx gem based on your access to its codebase, helping the team understand how to work with Word documents programmatically.

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: claude_swarm
 version: !ruby/object:Gem::Version
-  version: 1.0.9
+  version: 1.0.10
 platform: ruby
 authors:
 - Paulo Arruda
@@ -167,6 +167,7 @@ files:
 - docs/v2/guides/getting-started.md
 - docs/v2/guides/memory-adapters.md
 - docs/v2/guides/memory-defrag-guide.md
+- docs/v2/guides/migrating-to-2.3.md
 - docs/v2/guides/plugins.md
 - docs/v2/guides/quick-start-cli.md
 - docs/v2/guides/rails-integration.md
@@ -304,7 +305,6 @@ files:
 - lib/swarm_memory.rb
 - lib/swarm_memory/adapters/base.rb
 - lib/swarm_memory/adapters/filesystem_adapter.rb
-- lib/swarm_memory/chat_extension.rb
 - lib/swarm_memory/cli/commands.rb
 - lib/swarm_memory/core/entry.rb
 - lib/swarm_memory/core/frontmatter_parser.rb
@@ -348,19 +348,35 @@ files:
 - lib/swarm_sdk/agent/RETRY_LOGIC.md
 - lib/swarm_sdk/agent/builder.rb
 - lib/swarm_sdk/agent/chat.rb
-- lib/swarm_sdk/agent/chat/context_tracker.rb
-- lib/swarm_sdk/agent/chat/hook_integration.rb
-- lib/swarm_sdk/agent/chat/logging_helpers.rb
-- lib/swarm_sdk/agent/chat/system_reminder_injector.rb
+- lib/swarm_sdk/agent/chat_helpers/context_tracker.rb
+- lib/swarm_sdk/agent/chat_helpers/event_emitter.rb
+- lib/swarm_sdk/agent/chat_helpers/hook_integration.rb
+- lib/swarm_sdk/agent/chat_helpers/instrumentation.rb
+- lib/swarm_sdk/agent/chat_helpers/llm_configuration.rb
+- lib/swarm_sdk/agent/chat_helpers/logging_helpers.rb
+- lib/swarm_sdk/agent/chat_helpers/serialization.rb
+- lib/swarm_sdk/agent/chat_helpers/system_reminder_injector.rb
+- lib/swarm_sdk/agent/chat_helpers/system_reminders.rb
+- lib/swarm_sdk/agent/chat_helpers/token_tracking.rb
 - lib/swarm_sdk/agent/context.rb
 - lib/swarm_sdk/agent/context_manager.rb
 - lib/swarm_sdk/agent/definition.rb
 - lib/swarm_sdk/agent/llm_instrumentation_middleware.rb
+- lib/swarm_sdk/agent/system_prompt_builder.rb
+- lib/swarm_sdk/builders/base_builder.rb
 - lib/swarm_sdk/claude_code_agent_adapter.rb
+- lib/swarm_sdk/concerns/cleanupable.rb
+- lib/swarm_sdk/concerns/snapshotable.rb
+- lib/swarm_sdk/concerns/validatable.rb
 - lib/swarm_sdk/configuration.rb
+- lib/swarm_sdk/configuration/parser.rb
+- lib/swarm_sdk/configuration/translator.rb
 - lib/swarm_sdk/context_compactor.rb
 - lib/swarm_sdk/context_compactor/metrics.rb
 - lib/swarm_sdk/context_compactor/token_counter.rb
+- lib/swarm_sdk/context_management/builder.rb
+- lib/swarm_sdk/context_management/context.rb
+- lib/swarm_sdk/defaults.rb
 - lib/swarm_sdk/events_to_messages.rb
 - lib/swarm_sdk/hooks/adapter.rb
 - lib/swarm_sdk/hooks/context.rb
@@ -378,11 +394,11 @@ files:
 - lib/swarm_sdk/model_aliases.json
 - lib/swarm_sdk/models.json
 - lib/swarm_sdk/models.rb
-- lib/swarm_sdk/node/agent_config.rb
-- lib/swarm_sdk/node/builder.rb
-- lib/swarm_sdk/node/transformer_executor.rb
 - lib/swarm_sdk/node_context.rb
-- lib/swarm_sdk/node_orchestrator.rb
+- lib/swarm_sdk/observer/builder.rb
+- lib/swarm_sdk/observer/config.rb
+- lib/swarm_sdk/observer/manager.rb
+- lib/swarm_sdk/patterns/agent_observer.rb
 - lib/swarm_sdk/permissions/config.rb
 - lib/swarm_sdk/permissions/error_formatter.rb
 - lib/swarm_sdk/permissions/path_matcher.rb
@@ -392,7 +408,6 @@ files:
 - lib/swarm_sdk/plugin_registry.rb
 - lib/swarm_sdk/proc_helpers.rb
 - lib/swarm_sdk/prompts/base_system_prompt.md.erb
-- lib/swarm_sdk/providers/openai_with_responses.rb
 - lib/swarm_sdk/restore_result.rb
 - lib/swarm_sdk/result.rb
 - lib/swarm_sdk/snapshot.rb
@@ -403,6 +418,9 @@ files:
 - lib/swarm_sdk/swarm/agent_initializer.rb
 - lib/swarm_sdk/swarm/all_agents_builder.rb
 - lib/swarm_sdk/swarm/builder.rb
+- lib/swarm_sdk/swarm/executor.rb
+- lib/swarm_sdk/swarm/hook_triggers.rb
+- lib/swarm_sdk/swarm/logging_callbacks.rb
 - lib/swarm_sdk/swarm/mcp_configurator.rb
 - lib/swarm_sdk/swarm/swarm_registry_builder.rb
 - lib/swarm_sdk/swarm/tool_configurator.rb
@@ -440,6 +458,12 @@ files:
 - lib/swarm_sdk/utils.rb
 - lib/swarm_sdk/validation_result.rb
 - lib/swarm_sdk/version.rb
+- lib/swarm_sdk/workflow.rb
+- lib/swarm_sdk/workflow/agent_config.rb
+- lib/swarm_sdk/workflow/builder.rb
+- lib/swarm_sdk/workflow/executor.rb
+- lib/swarm_sdk/workflow/node_builder.rb
+- lib/swarm_sdk/workflow/transformer_executor.rb
 - llms.claude-swarm.txt
 - rubocop/cop/security/no_reflection_methods.rb
 - rubocop/cop/security/no_ruby_llm_logger.rb

data/lib/swarm_memory/chat_extension.rb

@@ -1,34 +0,0 @@
-# frozen_string_literal: true
-
-module SwarmMemory
-  # Extension module for SwarmSDK::Agent::Chat
-  #
-  # Adds individual tool removal capability needed for:
-  # 1. Mode-based tool filtering (retrieval/interactive/researcher)
-  # 2. LoadSkill's fine-grained tool swapping
-  #
-  # This is injected into SwarmSDK::Agent::Chat when SwarmMemory is loaded.
-  module ChatExtension
-    # Remove a specific tool by name
-    #
-    # Used by SwarmMemory to filter tools based on memory mode.
-    # Unlike remove_mutable_tools (which removes ALL mutable tools),
-    # this removes a single tool by name.
-    #
-    # @param tool_name [String, Symbol] Tool name to remove
-    # @return [void]
-    def remove_tool(tool_name)
-      tool_sym = tool_name.to_sym
-      tool_str = tool_name.to_s
-
-      # Remove from @tools hash (tools are keyed by symbol)
-      @tools.delete(tool_sym)
-      @tools.delete(tool_str)
-    end
-  end
-end
-
-# Inject into SwarmSDK when both gems are loaded
-if defined?(SwarmSDK::Agent::Chat)
-  SwarmSDK::Agent::Chat.include(SwarmMemory::ChatExtension)
-end