smith-agents 0.4.5 → 0.6.0

data/lib/smith/workflow/split_step_persistence.rb

@@ -4,17 +4,46 @@ require "digest"
 require "json"
 require "securerandom"
 
+require_relative "prepared_step"
+require_relative "prepared_step_dispatch"
+require_relative "prepared_step_execution_authorization"
+require_relative "prepared_step_execution_result"
+require_relative "prepared_step_recovery"
 require_relative "split_step_persistence/subclass_boundary"
 require_relative "split_step_persistence/inheritance"
+require_relative "split_step_persistence/recovery_class_methods"
 require_relative "split_step_persistence/transition_contract_structured_values"
 require_relative "split_step_persistence/transition_contract_signature"
 require_relative "split_step_persistence/transition_contract_freezer"
 require_relative "split_step_persistence/transition_contract"
+require_relative "split_step_persistence/transaction_identity"
+require_relative "split_step_persistence/canonical_payload_digest"
+require_relative "split_step_persistence/definition_boundary"
 require_relative "split_step_persistence/boundary"
+require_relative "split_step_persistence/boundary_reset"
 require_relative "split_step_persistence/state_snapshot"
 require_relative "split_step_persistence/preparation_claim"
 require_relative "split_step_persistence/preparation_recovery"
 require_relative "split_step_persistence/preparation"
+require_relative "split_step_persistence/restart_safe_adapter"
+require_relative "split_step_persistence/recovery"
+require_relative "split_step_persistence/recovery_boundary"
+require_relative "split_step_persistence/preparation_payload"
+require_relative "split_step_persistence/dispatch_boundary"
+require_relative "split_step_persistence/dispatch_claim"
+require_relative "split_step_persistence/dispatch_verification"
+require_relative "split_step_persistence/dispatch_confirmation"
+require_relative "split_step_persistence/execution_verification"
+require_relative "split_step_persistence/execution_binding_snapshot"
+require_relative "split_step_persistence/execution_authorization"
+require_relative "split_step_persistence/execution_authorization_issuance"
+require_relative "split_step_persistence/execution_result_capture"
+require_relative "split_step_persistence/execution_lifecycle"
+require_relative "split_step_persistence/composite_preparation"
+require_relative "split_step_persistence/composite_branch_authorization"
+require_relative "split_step_persistence/composite_branch_execution"
+require_relative "split_step_persistence/composite_reduction_execution"
+require_relative "split_step_persistence/composite_execution"
 require_relative "split_step_persistence/execution"
 require_relative "split_step_persistence/checkpoint_state"
 require_relative "split_step_persistence/checkpoint"
@@ -27,13 +56,34 @@ module Smith
 
       def self.prepended(base)
         base.singleton_class.prepend(Inheritance)
+        base.extend(RecoveryClassMethods)
       end
 
       include Boundary
+      include BoundaryReset
+      include DefinitionBoundary
       include StateSnapshot
       include PreparationClaim
       include PreparationRecovery
       include Preparation
+      include RestartSafeAdapter
+      include Recovery
+      include RecoveryBoundary
+      include PreparationPayload
+      include DispatchBoundary
+      include DispatchClaim
+      include DispatchVerification
+      include DispatchConfirmation
+      include ExecutionVerification
+      include ExecutionAuthorization
+      include ExecutionAuthorizationIssuance
+      include ExecutionResultCapture
+      include ExecutionLifecycle
+      include CompositePreparation
+      include CompositeBranchAuthorization
+      include CompositeBranchExecution
+      include CompositeReductionExecution
+      include CompositeExecution
       include Execution
       include CheckpointState
       include Checkpoint

data/lib/smith/workflow/step_completion.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module Smith
+  class Workflow
+    module StepCompletion
+      private
+
+      def complete_step(transition, output)
+        step = { transition: transition.name, from: transition.from, to: transition.to, output: output }
+        result = SplitStepPersistence
+                 .instance_method(:prepare_split_step_execution_result)
+                 .bind_call(self, step)
+        StepCompletion.instance_method(:append_accepted_output).bind_call(self, output)
+        SplitStepPersistence
+          .instance_method(:commit_split_step_execution_result!)
+          .bind_call(self, result)
+        @state = transition.to
+        @next_transition_name = @router_next_transition || transition.success_transition
+        @router_next_transition = nil
+        emit_step_completed(transition, output)
+        step
+      end
+
+      def append_accepted_output(output)
+        return unless @session_messages && !output.nil?
+
+        @session_messages << { role: :assistant, content: output }
+      end
+    end
+  end
+end

data/lib/smith/workflow/step_context.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+require_relative "thread_context_snapshot"
+
+module Smith
+  class Workflow
+    module StepContext
+      private
+
+      def with_step_context(transition, &block)
+        ThreadContextSnapshot.new.around do
+          setup_step_context
+          Thread.handle_interrupt(Object => :immediate, &block)
+        rescue StandardError => e
+          @outcome = nil
+          GuardrailIntegration.instance_method(:handle_step_failure).bind_call(self, transition, e)
+        ensure
+          teardown_step_context
+        end
+      end
+
+      def within_raw_step_context(&block)
+        ThreadContextSnapshot.new.around do
+          setup_step_context
+          Thread.handle_interrupt(Object => :immediate, &block)
+        ensure
+          teardown_step_context
+        end
+      end
+
+      def setup_step_context
+        Tool.current_deadline = wall_clock_deadline
+        Tool.current_ledger = @ledger
+        Tool.current_tool_result_collector = tool_result_collector
+      end
+
+      def teardown_step_context
+        Tool.current_guardrails = nil
+        Tool.current_deadline = nil
+        Tool.current_ledger = nil
+        Tool.current_tool_result_collector = nil
+        Smith.scoped_artifacts = nil
+      end
+    end
+  end
+end

data/lib/smith/workflow/string_snapshot.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module Smith
+  class Workflow
+    class StringSnapshot
+      BYTESIZE = String.instance_method(:bytesize)
+      FREEZE = String.instance_method(:freeze)
+      INITIALIZE_COPY = String.instance_method(:initialize_copy)
+      private_constant :BYTESIZE, :FREEZE, :INITIALIZE_COPY
+
+      class << self
+        def bytesize(value)
+          BYTESIZE.bind_call(value)
+        end
+
+        def copy(value, freeze: false)
+          String.allocate.tap do |copy|
+            INITIALIZE_COPY.bind_call(copy, value)
+            FREEZE.bind_call(copy) if freeze
+          end
+        end
+      end
+    end
+  end
+end

data/lib/smith/workflow/thread_context_snapshot.rb

@@ -0,0 +1,103 @@
+# frozen_string_literal: true
+
+require "dry-initializer"
+
+require_relative "process_local"
+
+module Smith
+  class Workflow
+    class ThreadContextSnapshot
+      extend Dry::Initializer
+      include ProcessLocal
+
+      TOOL_ATTRIBUTES = %i[
+        current_guardrails
+        current_deadline
+        current_ledger
+        current_tool_call_allowance
+        current_tool_result_collector
+      ].freeze
+      THREAD_KEYS = %i[
+        smith_call_deadline
+        smith_call_ledger
+        smith_failed_agent_results
+        smith_last_agent_result
+        smith_parallel_agent_binding
+      ].freeze
+
+      option :tool_attributes, default: proc { TOOL_ATTRIBUTES }
+      option :thread_keys, default: proc { THREAD_KEYS }
+      option :scoped_artifacts, default: proc { true }
+
+      def initialize(...)
+        super
+        @tool_attributes = tool_attributes.dup.freeze
+        @thread_keys = thread_keys.dup.freeze
+        validate_selection!
+        @tool_values = capture_tool_values
+        @thread_values = capture_thread_values
+        @artifact_store = Smith.scoped_artifacts if scoped_artifacts
+        @process_id = Process.pid
+        @thread = Thread.current
+        @fiber = Fiber.current
+        @restored = false
+      end
+
+      def restore!
+        ensure_owner!
+        Thread.handle_interrupt(Object => :never) { restore_context! }
+        self
+      end
+
+      def around
+        ensure_owner!
+        Thread.handle_interrupt(Object => :never) do
+          raise WorkflowError, "thread context snapshot has already been restored" if @restored
+
+          begin
+            yield
+          ensure
+            restore_context!
+          end
+        end
+      end
+
+      private
+
+      def restore_context!
+        raise WorkflowError, "thread context snapshot has already been restored" if @restored
+
+        @tool_values.each { |attribute, value| Tool.public_send("#{attribute}=", value) }
+        @thread_values.each { |key, value| Thread.current[key] = value }
+        Smith.scoped_artifacts = @artifact_store if scoped_artifacts
+        @restored = true
+      end
+
+      def capture_tool_values
+        tool_attributes.to_h { |attribute| [attribute, Tool.public_send(attribute)] }.freeze
+      end
+
+      def capture_thread_values
+        thread_keys.to_h { |key| [key, Thread.current[key]] }.freeze
+      end
+
+      def ensure_owner!
+        owner = @process_id == Process.pid && @thread.equal?(Thread.current) && @fiber.equal?(Fiber.current)
+        return if owner
+
+        raise WorkflowError, "thread context snapshot belongs to another process, thread, or fiber"
+      end
+
+      def validate_selection!
+        unless (tool_attributes - TOOL_ATTRIBUTES).empty?
+          raise ArgumentError, "thread context snapshot contains unsupported tool attributes"
+        end
+        return if (thread_keys - THREAD_KEYS).empty?
+
+        raise ArgumentError, "thread context snapshot contains unsupported thread keys"
+      end
+    end
+
+    private_constant :ThreadContextSnapshot
+  end
+end