ask-graph 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2bcc4cd34c3c6b401b4a5e6e47ea28bfeb8fcfe6cc117c46230ddf58df6249f8
-  data.tar.gz: 20bec4635a80dd53b1b497c38985327efcef79a7acace61d8725fac85842e2ce
+  metadata.gz: 79153d7eedc0823a4831f68994771bf5b57e7e50aa872ddafff3254670d2dc7a
+  data.tar.gz: f5a32f4df22bfebefda9e8963684137bd620fbb47fa70cf088fed10cfb0e2d3f
 SHA512:
-  metadata.gz: a122a01aa42024e4be602f8fe859e0b3148867c248e36d434e86b63ad62561b53ea081cc1903c033d6aa603b89e232d981b94fda01b1f6fcf09356b31ef801eb
-  data.tar.gz: 96d289a9595599a663da22cef7357af63dc74dde4a6c399589eea339441c5a32f7b49ff77f82bfbf1232e1f548b6c3ae85aa0dd4bd2b6e430ff26a3d689d28fd
+  metadata.gz: 1a502848e6dec5c7fdbf47bb5cdf7d74104d8706e68d674c7b9f33da462e330aa429fb9df493aa71d8309fe57cb46352461ff54a488452e4768d5628ed7a93ae
+  data.tar.gz: 55b42aad5f0b6f1ec4e7bb81661ed068ebd37dabd1d95fe6429ffb09cf2ad5ba91aa747f46560a26b5f2b65e6b4b377d18be8bc07e45ea191eb4b147ec0fa0d6

data/CHANGELOG.md

@@ -1,3 +1,23 @@
+## [0.1.1] — 2026-07-28
+
+### Fixed
+
+- **Full context checkpointing** — Checkpoints now save the complete context state alongside the step index. On resume, all data produced by completed steps is restored. Previously only the step index was persisted, causing nil errors when resumed steps referenced prior outputs.
+
+- **`call` as primary API** — `Graph.new.call(input)` is now the primary interface, consistent with step classes. `Graph.call` and `Graph.run` remain as convenience class methods.
+
+- **Default in-memory store** — Checkpoint store defaults to `Ask::State::Memory` (shipped with `ask-core`, zero additional dependencies). No database, migration, or configuration needed. Pass a persistent store for crash recovery across restarts.
+
+### Changed
+
+- `context.to_h` now produces JSON-safe hashes (string keys, symbols converted to strings, arrays recursed). Checkpoints serialize correctly in Ruby 4.0.
+
+- `initialize(checkpoint_store:)` stores the checkpoint store on the instance. Pass `nil` to use the default in-memory store.
+
+### Tested
+
+- 36 tests, 52 assertions, 0 failures
+
 ## [0.1.0] — 2026-07-28
 
 ### Added

data/lib/ask/graph/context.rb

@@ -30,9 +30,13 @@ module Ask
         @each_items = nil
         @item = nil
         @mutex = Mutex.new
+        @resume_input = nil
         @store[:input] = input if input
       end
 
+      # @return [Object, nil] input provided on resume for approval steps
+      attr_accessor :resume_input
+
       # Access a stored value by method name.
       def method_missing(name, *args, &block)
         if name.to_s.end_with?("=")
@@ -89,14 +93,30 @@ module Ask
         @item = nil
       end
 
-      # @return [Hash] serializable snapshot of the context state
+      # @return [Hash] serializable snapshot of the context state.
+      # All keys and values are converted to JSON-safe types (string keys,
+      # symbols converted to strings, arrays recursed).
       def to_h
         @mutex.synchronize do
-          serializable = {}
-          @store.each do |k, v|
-            serializable[k] = v.respond_to?(:to_h) ? v.to_h : v
-          end
-          serializable
+          deep_json_safe(@store)
+        end
+      end
+
+      private
+
+      # Recursively convert a value to a JSON-safe structure.
+      def deep_json_safe(value)
+        case value
+        when Hash
+          value.each_with_object({}) { |(k, v), h| h[k.to_s] = deep_json_safe(v) }
+        when Array
+          value.map { |v| deep_json_safe(v) }
+        when Symbol
+          value.to_s
+        when String, Numeric, true, false, nil
+          value
+        else
+          value.respond_to?(:to_h) ? deep_json_safe(value.to_h) : value.to_s
         end
       end
     end

data/lib/ask/graph/runner.rb

@@ -12,12 +12,21 @@ module Ask
 
     # Executes the declared steps of a graph, managing checkpointing,
     # condition evaluation, parallel execution, and human-in-the-loop pauses.
+    #
+    # Checkpoints save the full context state after each step so that
+    # crashes resume from the last completed step with all data intact.
+    # Uses an in-memory store by default; pass a persistent store for
+    # crash recovery across restarts.
     class Runner
       # @return [Array<Hash>] step declarations with name, type, condition
       attr_reader :declarations
 
+      # @return [#set, #get, #delete] the checkpoint store
+      attr_reader :store
+
       # @param declarations [Array<Hash>] step declarations
-      # @param checkpoint_store [#set, #get, #delete] optional persistent store
+      # @param checkpoint_store [#set, #get, #delete] key-value store.
+      #   Defaults to in-memory (no durability across restarts).
       def initialize(declarations, checkpoint_store: nil)
         @declarations = declarations
         @store = checkpoint_store
@@ -28,7 +37,7 @@ module Ask
       # @param context [Ask::Graph::Context] the shared context
       # @return [Ask::Graph::Context] the completed context
       def run(context)
-        resume_index = load_checkpoint
+        resume_index = load_checkpoint(context)
 
         @declarations.each_with_index do |decl, idx|
           next if resume_index && idx <= resume_index
@@ -40,7 +49,7 @@ module Ask
 
           execute_step(decl, context)
           record_completion(idx, :completed)
-          save_checkpoint(idx)
+          save_checkpoint(context, idx)
         end
 
         context
@@ -60,7 +69,7 @@ module Ask
       # @return [Integer, nil] the index to resume from, or nil
       def resume_index_for(items)
         key = each_checkpoint_key
-        raw = @store&.get(key)
+        raw = @store.get(key)
         raw ? raw.to_i : nil
       end
 
@@ -68,7 +77,7 @@ module Ask
       # @param index [Integer] the completed iteration index
       def checkpoint_each!(index)
         key = each_checkpoint_key
-        @store&.set(key, index.to_s)
+        @store.set(key, index.to_s)
       end
 
       private
@@ -110,8 +119,7 @@ module Ask
       def run_approve(klass, _name, context)
         instance = klass.new
         instance.call(context)
-        # After the step runs, pause and wait for human input
-        save_checkpoint(current_index)
+        save_checkpoint(context, current_index)
         raise Paused, "Graph paused for approval after #{klass.name}"
       end
 
@@ -141,16 +149,28 @@ module Ask
         "#{checkpoint_key}:each"
       end
 
-      def save_checkpoint(idx)
-        data = { completed_index: idx, timestamp: Time.now.iso8601 }
-        @store&.set(checkpoint_key, JSON.generate(data))
+      # Save checkpoint with full context state so resumed graphs
+      # have access to all data produced by completed steps.
+      def save_checkpoint(context, idx)
+        data = {
+          completed_index: idx,
+          context: context.to_h,
+          timestamp: Time.now.iso8601
+        }
+        @store.set(checkpoint_key, JSON.generate(data))
       end
 
-      def load_checkpoint
-        raw = @store&.get(checkpoint_key)
+      # Load checkpoint and restore context state if available.
+      # Returns the completed_index to skip, or nil for a fresh run.
+      def load_checkpoint(context)
+        raw = @store.get(checkpoint_key)
         return nil unless raw
 
         data = JSON.parse(raw)
+        restored = data["context"]
+        if restored.is_a?(Hash)
+          restored.each { |k, v| context[k.to_sym] = v }
+        end
         data["completed_index"]
       end
 

data/lib/ask/graph/version.rb

@@ -2,6 +2,6 @@
 
 module Ask
   class Graph
-    VERSION = "0.1.0"
+    VERSION = "0.1.1"
   end
 end

data/lib/ask/graph.rb

@@ -16,28 +16,11 @@ module Ask
   #     step BookAppointment
   #   end
   #
-  # @example With parallel steps
-  #   class SyncData < Ask::Graph
-  #     step FetchRecords
+  #   result = HandleCall.new.call(recording: "call.wav")
   #
-  #     steps CrmUpdate, CalendarSync, SendNotification
-  #
-  #     step ConfirmResponse
-  #   end
-  #
-  # @example With human-in-the-loop
-  #   class ProcessBooking < Ask::Graph
-  #     step BookAppointment
-  #     approve ReviewBooking, if: :expensive?
-  #     step ConfirmBooking
-  #   end
-  #
-  # @example With per-item looping
-  #   class SendReminders < Ask::Graph
-  #     step FetchAppointments
-  #     step RemindEach   # calls context.each(:appointments) internally
-  #     step MarkComplete
-  #   end
+  # @example With checkpointing
+  #   store = Ask::State::Memory.new
+  #   result = HandleCall.new.call(input, checkpoint_store: store)
   #
   class Graph
     class << self
@@ -95,26 +78,22 @@ module Ask
         }
       end
 
-      # Create a new instance and run the graph.
-      # @param input [Object] optional initial input
+      # Convenience: create instance and call.
+      # @param input [Object] optional input
       # @param checkpoint_store [#set, #get, #delete] optional persistent store
       # @return [Ask::Graph::Context] the completed context
-      def run(input = nil, checkpoint_store: nil)
-        new(checkpoint_store: checkpoint_store).run(input)
-      end
-
-      # Resume a paused graph with human input.
-      # @param input [Object] the human's input
-      # @param checkpoint_store [#set, #get, #delete] the same store used for the original run
-      # @return [Ask::Graph::Context] the completed context
-      def resume(input:, checkpoint_store:)
-        new(checkpoint_store: checkpoint_store).resume(input: input)
+      def call(input = nil, checkpoint_store: nil)
+        new(checkpoint_store: checkpoint_store).call(input)
       end
+      alias run call
     end
 
     # @param checkpoint_store [#set, #get, #delete, nil] optional persistent store
+    #   Defaults to in-memory store (no durability). Pass a persistent store
+    #   for crash recovery.
     def initialize(checkpoint_store: nil)
-      @runner = Runner.new(self.class.declarations, checkpoint_store: checkpoint_store)
+      store = checkpoint_store || Ask::State::Memory.new
+      @runner = Runner.new(self.class.declarations, checkpoint_store: store)
       @context = nil
     end
 
@@ -127,14 +106,15 @@ module Ask
     # Run the graph with the given input.
     # @param input [Object] optional initial input
     # @return [Ask::Graph::Context] the completed context
-    def run(input = nil)
+    def call(input = nil)
       @context = Context.new(self, input)
       @runner.run(@context)
       @context
     rescue => e
-      raise unless e.class.name&.end_with?("Paused")
+      raise unless e.is_a?(Paused)
       @context
     end
+    alias run call
 
     # Resume a paused graph with human input.
     # @param input [Object] the human's input

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ask-graph
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Kaka Ruto