swarm_memory 2.1.2 → 2.1.4

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (118) hide show
  1. checksums.yaml +4 -4
  2. data/lib/claude_swarm/claude_mcp_server.rb +1 -0
  3. data/lib/claude_swarm/cli.rb +5 -18
  4. data/lib/claude_swarm/configuration.rb +30 -19
  5. data/lib/claude_swarm/mcp_generator.rb +5 -10
  6. data/lib/claude_swarm/openai/chat_completion.rb +4 -12
  7. data/lib/claude_swarm/openai/executor.rb +3 -1
  8. data/lib/claude_swarm/openai/responses.rb +13 -32
  9. data/lib/claude_swarm/version.rb +1 -1
  10. data/lib/swarm_cli/commands/mcp_serve.rb +2 -2
  11. data/lib/swarm_cli/commands/run.rb +2 -2
  12. data/lib/swarm_cli/config_loader.rb +14 -14
  13. data/lib/swarm_cli/formatters/human_formatter.rb +70 -0
  14. data/lib/swarm_cli/interactive_repl.rb +11 -5
  15. data/lib/swarm_cli/ui/icons.rb +0 -23
  16. data/lib/swarm_cli/version.rb +1 -1
  17. data/lib/swarm_memory/adapters/base.rb +4 -4
  18. data/lib/swarm_memory/adapters/filesystem_adapter.rb +11 -34
  19. data/lib/swarm_memory/core/storage_read_tracker.rb +51 -14
  20. data/lib/swarm_memory/integration/cli_registration.rb +3 -2
  21. data/lib/swarm_memory/integration/sdk_plugin.rb +98 -12
  22. data/lib/swarm_memory/tools/memory_edit.rb +2 -2
  23. data/lib/swarm_memory/tools/memory_multi_edit.rb +2 -2
  24. data/lib/swarm_memory/tools/memory_read.rb +3 -3
  25. data/lib/swarm_memory/version.rb +1 -1
  26. data/lib/swarm_memory.rb +6 -1
  27. data/lib/swarm_sdk/agent/builder.rb +91 -0
  28. data/lib/swarm_sdk/agent/chat.rb +540 -925
  29. data/lib/swarm_sdk/agent/{chat → chat_helpers}/context_tracker.rb +33 -79
  30. data/lib/swarm_sdk/agent/chat_helpers/event_emitter.rb +204 -0
  31. data/lib/swarm_sdk/agent/{chat → chat_helpers}/hook_integration.rb +147 -39
  32. data/lib/swarm_sdk/agent/chat_helpers/instrumentation.rb +78 -0
  33. data/lib/swarm_sdk/agent/chat_helpers/llm_configuration.rb +233 -0
  34. data/lib/swarm_sdk/agent/{chat → chat_helpers}/logging_helpers.rb +1 -1
  35. data/lib/swarm_sdk/agent/chat_helpers/serialization.rb +83 -0
  36. data/lib/swarm_sdk/agent/{chat → chat_helpers}/system_reminder_injector.rb +22 -38
  37. data/lib/swarm_sdk/agent/chat_helpers/system_reminders.rb +79 -0
  38. data/lib/swarm_sdk/agent/chat_helpers/token_tracking.rb +98 -0
  39. data/lib/swarm_sdk/agent/context.rb +8 -4
  40. data/lib/swarm_sdk/agent/context_manager.rb +6 -0
  41. data/lib/swarm_sdk/agent/definition.rb +79 -174
  42. data/lib/swarm_sdk/agent/llm_instrumentation_middleware.rb +182 -0
  43. data/lib/swarm_sdk/agent/system_prompt_builder.rb +161 -0
  44. data/lib/swarm_sdk/builders/base_builder.rb +409 -0
  45. data/lib/swarm_sdk/concerns/cleanupable.rb +39 -0
  46. data/lib/swarm_sdk/concerns/snapshotable.rb +67 -0
  47. data/lib/swarm_sdk/concerns/validatable.rb +55 -0
  48. data/lib/swarm_sdk/configuration/parser.rb +353 -0
  49. data/lib/swarm_sdk/configuration/translator.rb +255 -0
  50. data/lib/swarm_sdk/configuration.rb +100 -261
  51. data/lib/swarm_sdk/context_compactor/token_counter.rb +3 -3
  52. data/lib/swarm_sdk/context_compactor.rb +6 -11
  53. data/lib/swarm_sdk/context_management/builder.rb +128 -0
  54. data/lib/swarm_sdk/context_management/context.rb +328 -0
  55. data/lib/swarm_sdk/defaults.rb +196 -0
  56. data/lib/swarm_sdk/events_to_messages.rb +199 -0
  57. data/lib/swarm_sdk/hooks/shell_executor.rb +2 -1
  58. data/lib/swarm_sdk/log_collector.rb +192 -16
  59. data/lib/swarm_sdk/log_stream.rb +66 -8
  60. data/lib/swarm_sdk/model_aliases.json +4 -1
  61. data/lib/swarm_sdk/node_context.rb +1 -1
  62. data/lib/swarm_sdk/observer/builder.rb +81 -0
  63. data/lib/swarm_sdk/observer/config.rb +45 -0
  64. data/lib/swarm_sdk/observer/manager.rb +236 -0
  65. data/lib/swarm_sdk/patterns/agent_observer.rb +160 -0
  66. data/lib/swarm_sdk/plugin.rb +93 -3
  67. data/lib/swarm_sdk/proc_helpers.rb +53 -0
  68. data/lib/swarm_sdk/prompts/base_system_prompt.md.erb +0 -126
  69. data/lib/swarm_sdk/restore_result.rb +65 -0
  70. data/lib/swarm_sdk/snapshot.rb +156 -0
  71. data/lib/swarm_sdk/snapshot_from_events.rb +397 -0
  72. data/lib/swarm_sdk/state_restorer.rb +476 -0
  73. data/lib/swarm_sdk/state_snapshot.rb +334 -0
  74. data/lib/swarm_sdk/swarm/agent_initializer.rb +428 -79
  75. data/lib/swarm_sdk/swarm/all_agents_builder.rb +28 -1
  76. data/lib/swarm_sdk/swarm/builder.rb +69 -407
  77. data/lib/swarm_sdk/swarm/executor.rb +213 -0
  78. data/lib/swarm_sdk/swarm/hook_triggers.rb +150 -0
  79. data/lib/swarm_sdk/swarm/logging_callbacks.rb +340 -0
  80. data/lib/swarm_sdk/swarm/mcp_configurator.rb +7 -4
  81. data/lib/swarm_sdk/swarm/swarm_registry_builder.rb +67 -0
  82. data/lib/swarm_sdk/swarm/tool_configurator.rb +88 -149
  83. data/lib/swarm_sdk/swarm.rb +366 -631
  84. data/lib/swarm_sdk/swarm_loader.rb +145 -0
  85. data/lib/swarm_sdk/swarm_registry.rb +136 -0
  86. data/lib/swarm_sdk/tools/bash.rb +11 -3
  87. data/lib/swarm_sdk/tools/delegate.rb +127 -24
  88. data/lib/swarm_sdk/tools/edit.rb +8 -13
  89. data/lib/swarm_sdk/tools/glob.rb +9 -1
  90. data/lib/swarm_sdk/tools/grep.rb +7 -0
  91. data/lib/swarm_sdk/tools/multi_edit.rb +15 -11
  92. data/lib/swarm_sdk/tools/path_resolver.rb +51 -2
  93. data/lib/swarm_sdk/tools/read.rb +28 -18
  94. data/lib/swarm_sdk/tools/registry.rb +122 -10
  95. data/lib/swarm_sdk/tools/scratchpad/scratchpad_list.rb +23 -2
  96. data/lib/swarm_sdk/tools/scratchpad/scratchpad_read.rb +23 -2
  97. data/lib/swarm_sdk/tools/scratchpad/scratchpad_write.rb +21 -4
  98. data/lib/swarm_sdk/tools/stores/read_tracker.rb +47 -12
  99. data/lib/swarm_sdk/tools/stores/scratchpad_storage.rb +53 -5
  100. data/lib/swarm_sdk/tools/stores/storage.rb +0 -6
  101. data/lib/swarm_sdk/tools/think.rb +4 -1
  102. data/lib/swarm_sdk/tools/todo_write.rb +27 -8
  103. data/lib/swarm_sdk/tools/web_fetch.rb +3 -2
  104. data/lib/swarm_sdk/tools/write.rb +8 -13
  105. data/lib/swarm_sdk/utils.rb +18 -0
  106. data/lib/swarm_sdk/validation_result.rb +33 -0
  107. data/lib/swarm_sdk/version.rb +1 -1
  108. data/lib/swarm_sdk/{node → workflow}/agent_config.rb +34 -9
  109. data/lib/swarm_sdk/workflow/builder.rb +143 -0
  110. data/lib/swarm_sdk/workflow/executor.rb +497 -0
  111. data/lib/swarm_sdk/{node/builder.rb → workflow/node_builder.rb} +42 -21
  112. data/lib/swarm_sdk/{node → workflow}/transformer_executor.rb +3 -2
  113. data/lib/swarm_sdk/workflow.rb +554 -0
  114. data/lib/swarm_sdk.rb +393 -22
  115. metadata +51 -16
  116. data/lib/swarm_memory/chat_extension.rb +0 -34
  117. data/lib/swarm_sdk/node_orchestrator.rb +0 -591
  118. data/lib/swarm_sdk/providers/openai_with_responses.rb +0 -582
data/lib/swarm_sdk.rb CHANGED
@@ -17,6 +17,27 @@ require "async/semaphore"
17
17
  require "ruby_llm"
18
18
  require "ruby_llm/mcp"
19
19
 
20
+ # Patch ruby_llm-mcp's Zeitwerk loader to ignore railtie.rb when Rails is not present
21
+ # This prevents NameError when eager loading outside of Rails applications
22
+ # Can be removed once https://github.com/parruda/ruby_llm-mcp/issues/XXX is fixed
23
+ unless defined?(Rails)
24
+ require "zeitwerk"
25
+ mcp_loader = nil
26
+ Zeitwerk::Registry.loaders.each { |l| mcp_loader = l if l.tag == "RubyLLM-mcp" }
27
+ if mcp_loader
28
+ mcp_gem_dir = Gem.loaded_specs["ruby_llm-mcp"]&.gem_dir
29
+ if mcp_gem_dir
30
+ railtie_path = File.join(mcp_gem_dir, "lib", "ruby_llm", "mcp", "railtie.rb")
31
+ mcp_loader.ignore(railtie_path)
32
+ end
33
+ end
34
+ end
35
+
36
+ # Configure Faraday to use async-http adapter by default
37
+ # This ensures HTTP requests are fiber-aware and don't block the Async scheduler
38
+ # when SwarmSDK executes LLM requests within Async/Sync blocks
39
+ require "async/http/faraday/default"
40
+
20
41
  require_relative "swarm_sdk/version"
21
42
 
22
43
  require "zeitwerk"
@@ -26,6 +47,8 @@ loader.push_dir("#{__dir__}/swarm_sdk", namespace: SwarmSDK)
26
47
  loader.inflector = Zeitwerk::GemInflector.new(__FILE__)
27
48
  loader.inflector.inflect(
28
49
  "cli" => "CLI",
50
+ "llm_instrumentation_middleware" => "LLMInstrumentationMiddleware",
51
+ "mcp" => "MCP",
29
52
  "openai_with_responses" => "OpenAIWithResponses",
30
53
  )
31
54
  loader.setup
@@ -43,9 +66,186 @@ module SwarmSDK
43
66
  # Settings for SwarmSDK (global configuration)
44
67
  attr_accessor :settings
45
68
 
46
- # Main entry point for DSL
47
- def build(&block)
48
- Swarm::Builder.build(&block)
69
+ # Main entry point for DSL - builds simple multi-agent swarms
70
+ #
71
+ # @return [Swarm] Always returns a Swarm instance
72
+ def build(allow_filesystem_tools: nil, &block)
73
+ Swarm::Builder.build(allow_filesystem_tools: allow_filesystem_tools, &block)
74
+ end
75
+
76
+ # Entry point for building multi-stage workflows
77
+ #
78
+ # @return [Workflow] Always returns a Workflow instance
79
+ def workflow(allow_filesystem_tools: nil, &block)
80
+ Workflow::Builder.build(allow_filesystem_tools: allow_filesystem_tools, &block)
81
+ end
82
+
83
+ # Validate YAML configuration without creating a swarm
84
+ #
85
+ # Performs comprehensive validation of YAML configuration including:
86
+ # - YAML syntax
87
+ # - Required fields (version, swarm name, lead, agents)
88
+ # - Agent configurations (description, directory existence)
89
+ # - Circular dependencies
90
+ # - File references (agent_file paths)
91
+ # - Hook configurations
92
+ #
93
+ # @param yaml_content [String] YAML configuration content
94
+ # @param base_dir [String, Pathname] Base directory for resolving agent file paths (default: Dir.pwd)
95
+ # @return [Array<Hash>] Array of error hashes (empty if valid)
96
+ #
97
+ # @example Validate YAML string
98
+ # errors = SwarmSDK.validate(yaml_content)
99
+ # if errors.empty?
100
+ # puts "Configuration is valid!"
101
+ # else
102
+ # errors.each do |error|
103
+ # puts "#{error[:field]}: #{error[:message]}"
104
+ # end
105
+ # end
106
+ #
107
+ # @example Error hash structure
108
+ # {
109
+ # type: :missing_field, # Error type
110
+ # field: "swarm.agents.backend.description", # JSON-style path to field
111
+ # message: "Agent 'backend' missing required 'description' field",
112
+ # agent: "backend" # Optional, present if error is agent-specific
113
+ # }
114
+ def validate(yaml_content, base_dir: Dir.pwd)
115
+ errors = []
116
+
117
+ begin
118
+ config = Configuration.new(yaml_content, base_dir: base_dir)
119
+ config.load_and_validate
120
+
121
+ # Build swarm to trigger DSL validation
122
+ # This catches errors from Agent::Definition, Builder, etc.
123
+ config.to_swarm
124
+ rescue ConfigurationError, CircularDependencyError => e
125
+ errors << parse_configuration_error(e)
126
+ rescue StandardError => e
127
+ errors << {
128
+ type: :unknown_error,
129
+ field: nil,
130
+ message: e.message,
131
+ }
132
+ end
133
+
134
+ errors
135
+ end
136
+
137
+ # Validate YAML configuration file
138
+ #
139
+ # Convenience method that reads the file and validates the content.
140
+ #
141
+ # @param path [String, Pathname] Path to YAML configuration file
142
+ # @return [Array<Hash>] Array of error hashes (empty if valid)
143
+ #
144
+ # @example
145
+ # errors = SwarmSDK.validate_file("config.yml")
146
+ # if errors.empty?
147
+ # puts "Valid configuration!"
148
+ # swarm = SwarmSDK.load_file("config.yml")
149
+ # else
150
+ # errors.each { |e| puts "Error: #{e[:message]}" }
151
+ # end
152
+ def validate_file(path)
153
+ path = Pathname.new(path).expand_path
154
+
155
+ unless path.exist?
156
+ return [{
157
+ type: :file_not_found,
158
+ field: nil,
159
+ message: "Configuration file not found: #{path}",
160
+ }]
161
+ end
162
+
163
+ yaml_content = File.read(path)
164
+ base_dir = path.dirname
165
+
166
+ validate(yaml_content, base_dir: base_dir)
167
+ rescue StandardError => e
168
+ [{
169
+ type: :file_read_error,
170
+ field: nil,
171
+ message: "Error reading file: #{e.message}",
172
+ }]
173
+ end
174
+
175
+ # Load swarm from YAML string
176
+ #
177
+ # This is the primary programmatic API for loading YAML configurations.
178
+ # For file-based loading, use SwarmSDK.load_file for convenience.
179
+ #
180
+ # @param yaml_content [String] YAML configuration content
181
+ # @param base_dir [String, Pathname] Base directory for resolving agent file paths (default: Dir.pwd)
182
+ # @return [Swarm, Workflow] Configured swarm or workflow instance
183
+ # @raise [ConfigurationError] If YAML is invalid or configuration is incorrect
184
+ #
185
+ # @example Load from YAML string
186
+ # yaml = <<~YAML
187
+ # version: 2
188
+ # swarm:
189
+ # name: "Dev Team"
190
+ # lead: backend
191
+ # agents:
192
+ # backend:
193
+ # description: "Backend developer"
194
+ # model: "gpt-4"
195
+ # agent_file: "agents/backend.md" # Resolved relative to base_dir
196
+ # YAML
197
+ #
198
+ # swarm = SwarmSDK.load(yaml, base_dir: "/path/to/project")
199
+ # result = swarm.execute("Build authentication")
200
+ #
201
+ # @example Load with default base_dir (Dir.pwd)
202
+ # yaml = File.read("config.yml")
203
+ # swarm = SwarmSDK.load(yaml) # base_dir defaults to Dir.pwd
204
+ def load(yaml_content, base_dir: Dir.pwd, allow_filesystem_tools: nil)
205
+ config = Configuration.new(yaml_content, base_dir: base_dir)
206
+ config.load_and_validate
207
+ swarm = config.to_swarm(allow_filesystem_tools: allow_filesystem_tools)
208
+
209
+ # Apply hooks if any are configured (YAML-only feature)
210
+ if hooks_configured?(config)
211
+ Hooks::Adapter.apply_hooks(swarm, config)
212
+ end
213
+
214
+ # Store config reference for agent hooks (applied during initialize_agents)
215
+ swarm.config_for_hooks = config
216
+
217
+ swarm
218
+ end
219
+
220
+ # Load swarm from YAML file (convenience method)
221
+ #
222
+ # Reads the YAML file and uses the file's directory as the base directory
223
+ # for resolving agent file paths. This is the recommended method for
224
+ # loading swarms from configuration files.
225
+ #
226
+ # @param path [String, Pathname] Path to YAML configuration file
227
+ # @return [Swarm, Workflow] Configured swarm or workflow instance
228
+ # @raise [ConfigurationError] If file not found or configuration invalid
229
+ #
230
+ # @example
231
+ # swarm = SwarmSDK.load_file("config.yml")
232
+ # result = swarm.execute("Build authentication")
233
+ #
234
+ # @example With absolute path
235
+ # swarm = SwarmSDK.load_file("/absolute/path/config.yml")
236
+ def load_file(path, allow_filesystem_tools: nil)
237
+ config = Configuration.load_file(path)
238
+ swarm = config.to_swarm(allow_filesystem_tools: allow_filesystem_tools)
239
+
240
+ # Apply hooks if any are configured (YAML-only feature)
241
+ if hooks_configured?(config)
242
+ Hooks::Adapter.apply_hooks(swarm, config)
243
+ end
244
+
245
+ # Store config reference for agent hooks (applied during initialize_agents)
246
+ swarm.config_for_hooks = config
247
+
248
+ swarm
49
249
  end
50
250
 
51
251
  # Configure SwarmSDK global settings
@@ -62,6 +262,180 @@ module SwarmSDK
62
262
  # Alias for backward compatibility
63
263
  alias_method :configuration, :settings
64
264
  alias_method :reset_configuration!, :reset_settings!
265
+
266
+ private
267
+
268
+ # Check if hooks are configured in the configuration
269
+ #
270
+ # @param config [Configuration] Configuration instance
271
+ # @return [Boolean] true if any hooks are configured
272
+ def hooks_configured?(config)
273
+ config.swarm_hooks.any? ||
274
+ config.all_agents_hooks.any? ||
275
+ config.agents.any? { |_, agent_config| agent_config[:hooks]&.any? }
276
+ end
277
+
278
+ # Parse configuration error and extract structured information
279
+ #
280
+ # Attempts to extract field path and agent name from error messages.
281
+ # Returns a structured error hash with type, field, message, and optional agent.
282
+ #
283
+ # @param error [StandardError] The caught error
284
+ # @return [Hash] Structured error hash
285
+ def parse_configuration_error(error)
286
+ message = error.message
287
+ error_hash = { message: message }
288
+
289
+ # Detect error type and extract field information
290
+ case message
291
+ # YAML syntax errors
292
+ when /Invalid YAML syntax/i
293
+ error_hash.merge!(
294
+ type: :syntax_error,
295
+ field: nil,
296
+ )
297
+
298
+ # Missing version field
299
+ when /Missing 'version' field/i
300
+ error_hash.merge!(
301
+ type: :missing_field,
302
+ field: "version",
303
+ )
304
+
305
+ # Invalid version
306
+ when /SwarmSDK requires version: (\d+)/i
307
+ error_hash.merge!(
308
+ type: :invalid_value,
309
+ field: "version",
310
+ )
311
+
312
+ # Missing swarm fields
313
+ when /Missing '(\w+)' field in swarm configuration/i
314
+ field_name = Regexp.last_match(1)
315
+ error_hash.merge!(
316
+ type: :missing_field,
317
+ field: "swarm.#{field_name}",
318
+ )
319
+
320
+ # Agent missing required field
321
+ when /Agent '([^']+)' missing required '([^']+)' field/i
322
+ agent_name = Regexp.last_match(1)
323
+ field_name = Regexp.last_match(2)
324
+ error_hash.merge!(
325
+ type: :missing_field,
326
+ field: "swarm.agents.#{agent_name}.#{field_name}",
327
+ agent: agent_name,
328
+ )
329
+
330
+ # Directory does not exist
331
+ when /Directory '([^']+)' for agent '([^']+)' does not exist/i
332
+ agent_name = Regexp.last_match(2)
333
+ error_hash.merge!(
334
+ type: :directory_not_found,
335
+ field: "swarm.agents.#{agent_name}.directory",
336
+ agent: agent_name,
337
+ )
338
+
339
+ # Error loading agent from file (must come before "Agent file not found")
340
+ when /Error loading agent '([^']+)' from file/i
341
+ agent_name = Regexp.last_match(1)
342
+ error_hash.merge!(
343
+ type: :file_load_error,
344
+ field: "swarm.agents.#{agent_name}.agent_file",
345
+ agent: agent_name,
346
+ )
347
+
348
+ # Agent file not found
349
+ when /Agent file not found: (.+)/i
350
+ # Try to extract agent name from the error context if available
351
+ error_hash.merge!(
352
+ type: :file_not_found,
353
+ field: nil, # We don't know which agent without more context
354
+ )
355
+
356
+ # Lead agent not found
357
+ when /Lead agent '([^']+)' not found in agents/i
358
+ error_hash.merge!(
359
+ type: :invalid_reference,
360
+ field: "swarm.lead",
361
+ )
362
+
363
+ # Unknown agent in connections (old format)
364
+ when /Agent '([^']+)' has connection to unknown agent '([^']+)'/i
365
+ agent_name = Regexp.last_match(1)
366
+ error_hash.merge!(
367
+ type: :invalid_reference,
368
+ field: "swarm.agents.#{agent_name}.delegates_to",
369
+ agent: agent_name,
370
+ )
371
+
372
+ # Unknown agent in connections (new format with composable swarms)
373
+ when /Agent '([^']+)' delegates to unknown target '([^']+)'/i
374
+ agent_name = Regexp.last_match(1)
375
+ error_hash.merge!(
376
+ type: :invalid_reference,
377
+ field: "swarm.agents.#{agent_name}.delegates_to",
378
+ agent: agent_name,
379
+ )
380
+
381
+ # Circular dependency
382
+ when /Circular dependency detected/i
383
+ error_hash.merge!(
384
+ type: :circular_dependency,
385
+ field: nil,
386
+ )
387
+
388
+ # Configuration file not found
389
+ when /Configuration file not found/i
390
+ error_hash.merge!(
391
+ type: :file_not_found,
392
+ field: nil,
393
+ )
394
+
395
+ # Invalid hook event
396
+ when /Invalid hook event '([^']+)' for agent '([^']+)'/i
397
+ agent_name = Regexp.last_match(2)
398
+ error_hash.merge!(
399
+ type: :invalid_value,
400
+ field: "swarm.agents.#{agent_name}.hooks",
401
+ agent: agent_name,
402
+ )
403
+
404
+ # api_version validation error
405
+ when /Agent '([^']+)' has api_version set, but provider is/i
406
+ agent_name = Regexp.last_match(1)
407
+ error_hash.merge!(
408
+ type: :invalid_value,
409
+ field: "swarm.agents.#{agent_name}.api_version",
410
+ agent: agent_name,
411
+ )
412
+
413
+ # api_version invalid value
414
+ when /Agent '([^']+)' has invalid api_version/i
415
+ agent_name = Regexp.last_match(1)
416
+ error_hash.merge!(
417
+ type: :invalid_value,
418
+ field: "swarm.agents.#{agent_name}.api_version",
419
+ agent: agent_name,
420
+ )
421
+
422
+ # No agents defined
423
+ when /No agents defined/i
424
+ error_hash.merge!(
425
+ type: :missing_field,
426
+ field: "swarm.agents",
427
+ )
428
+
429
+ # Default: unknown error
430
+ else
431
+ error_hash.merge!(
432
+ type: :validation_error,
433
+ field: nil,
434
+ )
435
+ end
436
+
437
+ error_hash.compact
438
+ end
65
439
  end
66
440
 
67
441
  # Settings class for SwarmSDK global settings (not to be confused with Configuration for YAML loading)
@@ -69,17 +443,33 @@ module SwarmSDK
69
443
  # WebFetch tool LLM processing configuration
70
444
  attr_accessor :webfetch_provider, :webfetch_model, :webfetch_base_url, :webfetch_max_tokens
71
445
 
446
+ # Filesystem tools control
447
+ attr_accessor :allow_filesystem_tools
448
+
72
449
  def initialize
73
450
  @webfetch_provider = nil
74
451
  @webfetch_model = nil
75
452
  @webfetch_base_url = nil
76
453
  @webfetch_max_tokens = 4096
454
+ @allow_filesystem_tools = parse_env_bool("SWARM_SDK_ALLOW_FILESYSTEM_TOOLS", default: true)
77
455
  end
78
456
 
79
457
  # Check if WebFetch LLM processing is enabled
80
458
  def webfetch_llm_enabled?
81
459
  !@webfetch_provider.nil? && !@webfetch_model.nil?
82
460
  end
461
+
462
+ private
463
+
464
+ def parse_env_bool(key, default:)
465
+ return default unless ENV.key?(key)
466
+
467
+ value = ENV[key].to_s.downcase
468
+ return true if ["true", "yes", "1", "on", "enabled"].include?(value)
469
+ return false if ["false", "no", "0", "off", "disabled"].include?(value)
470
+
471
+ default
472
+ end
83
473
  end
84
474
 
85
475
  # Initialize default settings
@@ -132,22 +522,3 @@ RubyLLM.configure do |config|
132
522
  config.gpustack_api_base ||= ENV["GPUSTACK_API_BASE"]
133
523
  config.gpustack_api_key ||= ENV["GPUSTACK_API_KEY"]
134
524
  end
135
-
136
- # monkey patches
137
- # ruby_llm/mcp
138
- # - add `id` when sending "notifications/initialized" message: https://github.com/patvice/ruby_llm-mcp/issues/65
139
- # - remove `to_sym` on MCP parameter type: https://github.com/patvice/ruby_llm-mcp/issues/62#issuecomment-3421488406
140
- require "ruby_llm/mcp/notifications/initialize"
141
- require "ruby_llm/mcp/parameter"
142
-
143
- module RubyLLM
144
- module MCP
145
- module Notifications
146
- class Initialize
147
- def call
148
- @coordinator.request(notification_body, add_id: true, wait_for_response: false)
149
- end
150
- end
151
- end
152
- end
153
- end
metadata CHANGED
@@ -1,7 +1,7 @@
1
1
  --- !ruby/object:Gem::Specification
2
2
  name: swarm_memory
3
3
  version: !ruby/object:Gem::Version
4
- version: 2.1.2
4
+ version: 2.1.4
5
5
  platform: ruby
6
6
  authors:
7
7
  - Paulo Arruda
@@ -52,33 +52,33 @@ dependencies:
52
52
  - !ruby/object:Gem::Version
53
53
  version: 4.6.0
54
54
  - !ruby/object:Gem::Dependency
55
- name: ruby_llm
55
+ name: ruby_llm_swarm
56
56
  requirement: !ruby/object:Gem::Requirement
57
57
  requirements:
58
58
  - - "~>"
59
59
  - !ruby/object:Gem::Version
60
- version: '1.8'
60
+ version: 1.9.2
61
61
  type: :runtime
62
62
  prerelease: false
63
63
  version_requirements: !ruby/object:Gem::Requirement
64
64
  requirements:
65
65
  - - "~>"
66
66
  - !ruby/object:Gem::Version
67
- version: '1.8'
67
+ version: 1.9.2
68
68
  - !ruby/object:Gem::Dependency
69
69
  name: swarm_sdk
70
70
  requirement: !ruby/object:Gem::Requirement
71
71
  requirements:
72
72
  - - "~>"
73
73
  - !ruby/object:Gem::Version
74
- version: '2.1'
74
+ version: '2.2'
75
75
  type: :runtime
76
76
  prerelease: false
77
77
  version_requirements: !ruby/object:Gem::Requirement
78
78
  requirements:
79
79
  - - "~>"
80
80
  - !ruby/object:Gem::Version
81
- version: '2.1'
81
+ version: '2.2'
82
82
  - !ruby/object:Gem::Dependency
83
83
  name: zeitwerk
84
84
  requirement: !ruby/object:Gem::Requirement
@@ -163,7 +163,6 @@ files:
163
163
  - lib/swarm_memory.rb
164
164
  - lib/swarm_memory/adapters/base.rb
165
165
  - lib/swarm_memory/adapters/filesystem_adapter.rb
166
- - lib/swarm_memory/chat_extension.rb
167
166
  - lib/swarm_memory/cli/commands.rb
168
167
  - lib/swarm_memory/core/entry.rb
169
168
  - lib/swarm_memory/core/frontmatter_parser.rb
@@ -207,18 +206,36 @@ files:
207
206
  - lib/swarm_sdk/agent/RETRY_LOGIC.md
208
207
  - lib/swarm_sdk/agent/builder.rb
209
208
  - lib/swarm_sdk/agent/chat.rb
210
- - lib/swarm_sdk/agent/chat/context_tracker.rb
211
- - lib/swarm_sdk/agent/chat/hook_integration.rb
212
- - lib/swarm_sdk/agent/chat/logging_helpers.rb
213
- - lib/swarm_sdk/agent/chat/system_reminder_injector.rb
209
+ - lib/swarm_sdk/agent/chat_helpers/context_tracker.rb
210
+ - lib/swarm_sdk/agent/chat_helpers/event_emitter.rb
211
+ - lib/swarm_sdk/agent/chat_helpers/hook_integration.rb
212
+ - lib/swarm_sdk/agent/chat_helpers/instrumentation.rb
213
+ - lib/swarm_sdk/agent/chat_helpers/llm_configuration.rb
214
+ - lib/swarm_sdk/agent/chat_helpers/logging_helpers.rb
215
+ - lib/swarm_sdk/agent/chat_helpers/serialization.rb
216
+ - lib/swarm_sdk/agent/chat_helpers/system_reminder_injector.rb
217
+ - lib/swarm_sdk/agent/chat_helpers/system_reminders.rb
218
+ - lib/swarm_sdk/agent/chat_helpers/token_tracking.rb
214
219
  - lib/swarm_sdk/agent/context.rb
215
220
  - lib/swarm_sdk/agent/context_manager.rb
216
221
  - lib/swarm_sdk/agent/definition.rb
222
+ - lib/swarm_sdk/agent/llm_instrumentation_middleware.rb
223
+ - lib/swarm_sdk/agent/system_prompt_builder.rb
224
+ - lib/swarm_sdk/builders/base_builder.rb
217
225
  - lib/swarm_sdk/claude_code_agent_adapter.rb
226
+ - lib/swarm_sdk/concerns/cleanupable.rb
227
+ - lib/swarm_sdk/concerns/snapshotable.rb
228
+ - lib/swarm_sdk/concerns/validatable.rb
218
229
  - lib/swarm_sdk/configuration.rb
230
+ - lib/swarm_sdk/configuration/parser.rb
231
+ - lib/swarm_sdk/configuration/translator.rb
219
232
  - lib/swarm_sdk/context_compactor.rb
220
233
  - lib/swarm_sdk/context_compactor/metrics.rb
221
234
  - lib/swarm_sdk/context_compactor/token_counter.rb
235
+ - lib/swarm_sdk/context_management/builder.rb
236
+ - lib/swarm_sdk/context_management/context.rb
237
+ - lib/swarm_sdk/defaults.rb
238
+ - lib/swarm_sdk/events_to_messages.rb
222
239
  - lib/swarm_sdk/hooks/adapter.rb
223
240
  - lib/swarm_sdk/hooks/context.rb
224
241
  - lib/swarm_sdk/hooks/definition.rb
@@ -235,11 +252,11 @@ files:
235
252
  - lib/swarm_sdk/model_aliases.json
236
253
  - lib/swarm_sdk/models.json
237
254
  - lib/swarm_sdk/models.rb
238
- - lib/swarm_sdk/node/agent_config.rb
239
- - lib/swarm_sdk/node/builder.rb
240
- - lib/swarm_sdk/node/transformer_executor.rb
241
255
  - lib/swarm_sdk/node_context.rb
242
- - lib/swarm_sdk/node_orchestrator.rb
256
+ - lib/swarm_sdk/observer/builder.rb
257
+ - lib/swarm_sdk/observer/config.rb
258
+ - lib/swarm_sdk/observer/manager.rb
259
+ - lib/swarm_sdk/patterns/agent_observer.rb
243
260
  - lib/swarm_sdk/permissions/config.rb
244
261
  - lib/swarm_sdk/permissions/error_formatter.rb
245
262
  - lib/swarm_sdk/permissions/path_matcher.rb
@@ -247,15 +264,26 @@ files:
247
264
  - lib/swarm_sdk/permissions_builder.rb
248
265
  - lib/swarm_sdk/plugin.rb
249
266
  - lib/swarm_sdk/plugin_registry.rb
267
+ - lib/swarm_sdk/proc_helpers.rb
250
268
  - lib/swarm_sdk/prompts/base_system_prompt.md.erb
251
- - lib/swarm_sdk/providers/openai_with_responses.rb
269
+ - lib/swarm_sdk/restore_result.rb
252
270
  - lib/swarm_sdk/result.rb
271
+ - lib/swarm_sdk/snapshot.rb
272
+ - lib/swarm_sdk/snapshot_from_events.rb
273
+ - lib/swarm_sdk/state_restorer.rb
274
+ - lib/swarm_sdk/state_snapshot.rb
253
275
  - lib/swarm_sdk/swarm.rb
254
276
  - lib/swarm_sdk/swarm/agent_initializer.rb
255
277
  - lib/swarm_sdk/swarm/all_agents_builder.rb
256
278
  - lib/swarm_sdk/swarm/builder.rb
279
+ - lib/swarm_sdk/swarm/executor.rb
280
+ - lib/swarm_sdk/swarm/hook_triggers.rb
281
+ - lib/swarm_sdk/swarm/logging_callbacks.rb
257
282
  - lib/swarm_sdk/swarm/mcp_configurator.rb
283
+ - lib/swarm_sdk/swarm/swarm_registry_builder.rb
258
284
  - lib/swarm_sdk/swarm/tool_configurator.rb
285
+ - lib/swarm_sdk/swarm_loader.rb
286
+ - lib/swarm_sdk/swarm_registry.rb
259
287
  - lib/swarm_sdk/tools/bash.rb
260
288
  - lib/swarm_sdk/tools/clock.rb
261
289
  - lib/swarm_sdk/tools/delegate.rb
@@ -286,7 +314,14 @@ files:
286
314
  - lib/swarm_sdk/tools/web_fetch.rb
287
315
  - lib/swarm_sdk/tools/write.rb
288
316
  - lib/swarm_sdk/utils.rb
317
+ - lib/swarm_sdk/validation_result.rb
289
318
  - lib/swarm_sdk/version.rb
319
+ - lib/swarm_sdk/workflow.rb
320
+ - lib/swarm_sdk/workflow/agent_config.rb
321
+ - lib/swarm_sdk/workflow/builder.rb
322
+ - lib/swarm_sdk/workflow/executor.rb
323
+ - lib/swarm_sdk/workflow/node_builder.rb
324
+ - lib/swarm_sdk/workflow/transformer_executor.rb
290
325
  homepage: https://github.com/parruda/claude-swarm
291
326
  licenses:
292
327
  - MIT
@@ -1,34 +0,0 @@
1
- # frozen_string_literal: true
2
-
3
- module SwarmMemory
4
- # Extension module for SwarmSDK::Agent::Chat
5
- #
6
- # Adds individual tool removal capability needed for:
7
- # 1. Mode-based tool filtering (retrieval/interactive/researcher)
8
- # 2. LoadSkill's fine-grained tool swapping
9
- #
10
- # This is injected into SwarmSDK::Agent::Chat when SwarmMemory is loaded.
11
- module ChatExtension
12
- # Remove a specific tool by name
13
- #
14
- # Used by SwarmMemory to filter tools based on memory mode.
15
- # Unlike remove_mutable_tools (which removes ALL mutable tools),
16
- # this removes a single tool by name.
17
- #
18
- # @param tool_name [String, Symbol] Tool name to remove
19
- # @return [void]
20
- def remove_tool(tool_name)
21
- tool_sym = tool_name.to_sym
22
- tool_str = tool_name.to_s
23
-
24
- # Remove from @tools hash (tools are keyed by symbol)
25
- @tools.delete(tool_sym)
26
- @tools.delete(tool_str)
27
- end
28
- end
29
- end
30
-
31
- # Inject into SwarmSDK when both gems are loaded
32
- if defined?(SwarmSDK::Agent::Chat)
33
- SwarmSDK::Agent::Chat.include(SwarmMemory::ChatExtension)
34
- end