robot_lab-ractor 0.1.0 → 0.2.6

data/lib/robot_lab/ractor_network_scheduler.rb

@@ -50,21 +50,10 @@ module RobotLab
       remaining = specs_with_deps.dup
 
       until remaining.empty?
-        ready, remaining = remaining.partition do |entry|
-          deps = entry[:depends_on]
-          deps == :none || deps == :optional ||
-            Array(deps).all? { |d| completed.key?(d) }
-        end
-
-        raise RobotLab::Error, "Circular dependency or unresolvable deps in RactorNetworkScheduler" if ready.empty?
+        ready, remaining = partition_ready(remaining, completed)
+        raise RobotLab::Error, 'Circular dependency or unresolvable deps in RactorNetworkScheduler' if ready.empty?
 
-        threads = ready.map do |entry|
-          spec = entry[:spec]
-          msg  = completed.values.last || message
-          Thread.new { [spec.name, execute_spec(spec, msg)] }.tap { |t| t.report_on_exception = false }
-        end
-
-        threads.each do |t|
+        dispatch_ready(ready, completed, message).each do |t|
           name, result = t.value
           completed[name] = result
         end
@@ -79,7 +68,39 @@ module RobotLab
 
       @closed = true
       @size.times { @work_q.push(nil) }
-      @workers.each { |w| w.join rescue nil }
+      @workers.each do |w|
+        w.join
+      rescue StandardError
+        nil
+      end
+    end
+
+    # Called inside Ractor worker blocks — must be a class method.
+    def self.process_job(job)
+      spec    = job.payload[:spec]
+      message = job.payload[:message]
+      job.reply_queue.push(build_and_run_robot(spec, message))
+    rescue StandardError => e
+      job.reply_queue.push(wrap_error(e))
+    end
+
+    def self.build_and_run_robot(spec, message)
+      config = spec.config_hash.empty? ? nil : RobotLab::RunConfig.new(**spec.config_hash.transform_keys(&:to_sym))
+      robot  = RobotLab::Robot.new(
+        name: spec.name,
+        template: spec.template&.to_sym,
+        system_prompt: spec.system_prompt,
+        config: config
+      )
+      spec.hook_classes.each { |handler| robot.on(handler) }
+      robot.run(message).last_text_content.to_s.freeze
+    end
+
+    def self.wrap_error(err)
+      RobotLab::RactorJobError.new(
+        message: err.message.freeze,
+        backtrace: (err.backtrace || []).map(&:freeze).freeze
+      )
     end
 
     private
@@ -90,18 +111,16 @@ module RobotLab
       reply_q        = RactorQueue.new(capacity: 1)
 
       job = RactorJob.new(
-        id:          SecureRandom.uuid.freeze,
-        type:        :robot,
-        payload:     RactorBoundary.freeze_deep({ spec: frozen_spec, message: frozen_message }),
+        id: SecureRandom.uuid.freeze,
+        type: :robot,
+        payload: RactorBoundary.freeze_deep({ spec: frozen_spec, message: frozen_message }),
         reply_queue: reply_q
       )
 
       @work_q.push(job)
       result = reply_q.pop
 
-      if result.is_a?(RactorJobError)
-        raise RobotLab::Error, "Robot '#{spec.name}' failed in Ractor: #{result.message}"
-      end
+      raise RobotLab::Error, "Robot '#{spec.name}' failed in Ractor: #{result.message}" if result.is_a?(RactorJobError)
 
       result
     end
@@ -112,28 +131,24 @@ module RobotLab
           job = q.pop
           break if job.nil?
 
-          begin
-            spec   = job.payload[:spec]
-            message = job.payload[:message]
-
-            robot = RobotLab::Robot.new(
-              name:          spec.name,
-              template:      spec.template ? spec.template.to_sym : nil,
-              system_prompt: spec.system_prompt,
-              config:        spec.config_hash.empty? ? nil : RobotLab::RunConfig.new(**spec.config_hash.transform_keys(&:to_sym))
-            )
-
-            robot_result = robot.run(message)
-            job.reply_queue.push(robot_result.last_text_content.to_s.freeze)
-          rescue => e
-            err = RobotLab::RactorJobError.new(
-              message:   e.message.freeze,
-              backtrace: (e.backtrace || []).map(&:freeze).freeze
-            )
-            job.reply_queue.push(err)
-          end
+          RobotLab::RactorNetworkScheduler.process_job(job)
         end
       end
     end
+
+    def partition_ready(remaining, completed)
+      remaining.partition do |entry|
+        deps = entry[:depends_on]
+        deps == :none || deps == :optional || Array(deps).all? { |d| completed.key?(d) }
+      end
+    end
+
+    def dispatch_ready(ready, completed, message)
+      ready.map do |entry|
+        spec = entry[:spec]
+        msg  = completed.values.last || message
+        Thread.new { [spec.name, execute_spec(spec, msg)] }.tap { |t| t.report_on_exception = false }
+      end
+    end
   end
 end

data/lib/robot_lab/ractor_worker_pool.rb

@@ -39,27 +39,22 @@ module RobotLab
     # @return [Object] the tool's return value
     # @raise [ToolError] if the tool raises inside the Ractor
     def submit(tool_class_name, args)
-      raise ToolError, "Pool is shut down" if @closed
+      raise ToolError, 'Pool is shut down' if @closed
 
       reply_q = RactorQueue.new(capacity: 1)
-      payload = RactorBoundary.freeze_deep({
-        tool_class: tool_class_name.to_s,
-        args:       args
-      })
+      payload = RactorBoundary.freeze_deep({ tool_class: tool_class_name.to_s, args: args })
 
       job = RactorJob.new(
-        id:          SecureRandom.uuid.freeze,
-        type:        :tool,
-        payload:     payload,
+        id: SecureRandom.uuid.freeze,
+        type: :tool,
+        payload: payload,
         reply_queue: reply_q
       )
 
       @work_q.push(job)
       result = reply_q.pop
 
-      if result.is_a?(RactorJobError)
-        raise ToolError, "Tool '#{tool_class_name}' failed in Ractor: #{result.message}"
-      end
+      raise ToolError, "Tool '#{tool_class_name}' failed in Ractor: #{result.message}" if result.is_a?(RactorJobError)
 
       result
     end
@@ -71,7 +66,28 @@ module RobotLab
 
       @closed = true
       @size.times { @work_q.push(nil) }
-      @workers.each { |w| w.join rescue nil }
+      @workers.each do |w|
+        w.join
+      rescue StandardError
+        nil
+      end
+    end
+
+    # Called inside Ractor worker blocks — must be a class method.
+    def self.process_job(job)
+      tool_class    = Object.const_get(job.payload[:tool_class])
+      result        = tool_class.new.execute(**job.payload[:args].transform_keys(&:to_sym))
+      frozen_result = ::Ractor.make_shareable(result.frozen? ? result : result.dup.freeze)
+      job.reply_queue.push(frozen_result)
+    rescue StandardError => e
+      job.reply_queue.push(wrap_error(e))
+    end
+
+    def self.wrap_error(err)
+      RobotLab::RactorJobError.new(
+        message: err.message.freeze,
+        backtrace: (err.backtrace || []).map(&:freeze).freeze
+      )
     end
 
     private
@@ -82,19 +98,7 @@ module RobotLab
           job = q.pop
           break if job.nil?
 
-          begin
-            tool_class    = Object.const_get(job.payload[:tool_class])
-            tool          = tool_class.new
-            result        = tool.execute(**job.payload[:args].transform_keys(&:to_sym))
-            frozen_result = ::Ractor.make_shareable(result.frozen? ? result : result.dup.freeze)
-            job.reply_queue.push(frozen_result)
-          rescue => e
-            err = RobotLab::RactorJobError.new(
-              message:   e.message.freeze,
-              backtrace: (e.backtrace || []).map(&:freeze).freeze
-            )
-            job.reply_queue.push(err)
-          end
+          RobotLab::RactorWorkerPool.process_job(job)
         end
       end
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: robot_lab-ractor
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.6
 platform: ruby
 authors:
 - Dewayne VanHoozer
@@ -10,7 +10,7 @@ cert_chain: []
 date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: robot_lab
+  name: ractor_queue
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -24,7 +24,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: ractor_queue
+  name: ractor-wrapper
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -38,19 +38,19 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: ractor-wrapper
+  name: robot_lab
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.2.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.2.0
 description: Provides RobotLab::RactorWorkerPool — a pool of Ruby Ractor workers for
   executing CPU-bound, Ractor-safe tools in parallel. Includes RactorBoundary (deep-freeze
   utility for crossing Ractor boundaries), RactorJob/RactorJobError/RobotSpec (frozen
@@ -64,7 +64,11 @@ extra_rdoc_files: []
 files:
 - ".envrc"
 - ".github/workflows/deploy-github-pages.yml"
+- ".loki"
+- ".quality/reek_baseline.txt"
+- ".rubocop.yml"
 - CHANGELOG.md
+- CLAUDE.md
 - LICENSE.txt
 - README.md
 - Rakefile
@@ -72,8 +76,11 @@ files:
 - docs/index.md
 - docs/superpowers/plans/2026-04-14-ractor-integration.md
 - docs/superpowers/specs/2026-04-14-ractor-integration-design.md
-- examples/29_ractor_tools.rb
-- examples/30_ractor_network.rb
+- examples/01_ractor_tools.rb
+- examples/02_ractor_network.rb
+- examples/03_ractor_hooks.rb
+- examples/common.rb
+- examples/xyzzy.rb
 - lib/robot_lab/ractor.rb
 - lib/robot_lab/ractor/version.rb
 - lib/robot_lab/ractor_boundary.rb
@@ -104,7 +111,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 4.0.11
+rubygems_version: 4.0.17
 specification_version: 4
 summary: Ractor-based parallel tool execution for RobotLab agents
 test_files: []