Module: SwarmSDK::Concerns::Cleanupable

Included in:
Swarm
Defined in:
lib/swarm_sdk/concerns/cleanupable.rb

Overview

Shared cleanup functionality for Swarm and Workflow

Both classes must have:

  • mcp_clients: Hash of MCP client arrays
  • delegation_instances_hash: Hash of delegation instances (via Snapshotable)

Instance Method Summary collapse

Instance Method Details

#cleanup ⇒ void

This method returns an undefined value.

Cleanup all MCP clients

Stops all MCP client connections gracefully. Should be called when the swarm/workflow is no longer needed.



18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
 
# File 'lib/swarm_sdk/concerns/cleanupable.rb', line 18

def cleanup
  # Check if there's anything to clean up
  return if @mcp_clients.empty? && (!delegation_instances_hash || delegation_instances_hash.empty?)

  # Stop MCP clients for all agents
  @mcp_clients.each do |agent_name, clients|
    clients.each do |client|
      client.stop
      RubyLLM.logger.debug("SwarmSDK: Stopped MCP client '#{client.name}' for agent #{agent_name}")
    rescue StandardError => e
      RubyLLM.logger.debug("SwarmSDK: Error stopping MCP client '#{client.name}': #{e.message}")
    end
  end

  @mcp_clients.clear

  # Shutdown sub-swarms if registry exists
  @swarm_registry&.shutdown_all

  # Clear delegation instances
  delegation_instances_hash&.clear
end