langgraph_rb 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9c2d260c430ce9be20fea65d6dff4d8013a535a6db46cdd89e15b19200c8b6dd
-  data.tar.gz: a5213444bbf90954158836c6e73237752ecc2d689e0e4de4807621edb2460f44
+  metadata.gz: b3189c15397a79e8ae6eb273adc89aa9148cdd82a6dfbaab22bd6ee0367c02cd
+  data.tar.gz: ef3b7472a7b8aea84d2dc8a80e4647a4fda9d1346169e304c16357d507b02b84
 SHA512:
-  metadata.gz: ce7b663fb858e5c2adea131d21682df32eddc0cbb72b324773fa0b1cf932338c105d155e7b7b046556748e9f5d60342b9171537a21e1f41c7f7e4933499968cb
-  data.tar.gz: e30e5ad223be3a455361f753b44ba928054eff285429ca0a8c73474eb851eb9c6c59c627357010032d5133d8ba9d891669225e3640aea5873017910724766536
+  metadata.gz: 8084cb63ceacdf4d207daa068a458efa596016232457d863f261533ba3a29de7c88bfc3bdca73a7fcac3312521091cccc61ecd1e99b2df72bfc8ae9b23c0d769
+  data.tar.gz: 1b2315e962fa8204b4eaa487304492435b866ff20102444f9e4ff76dc8675461131352554e311e4b32aa5106331d55d5cba3d38ff9081021e95196a8c71a2858

data/lib/langgraph_rb/observers/base.rb

@@ -117,11 +117,11 @@ module LangGraphRB
 
     class NodeEvent
       attr_reader :type, :node_name, :node_class, :state_before, :state_after, 
-                  :context, :thread_id, :step_number, :duration, :error, :result, :timestamp
+                  :context, :thread_id, :step_number, :duration, :error, :result, :timestamp, :from_node
 
       def initialize(type:, node_name:, node_class: nil, state_before: nil, state_after: nil,
                      context: nil, thread_id: nil, step_number: nil, duration: nil, 
-                     error: nil, result: nil)
+                     error: nil, result: nil, from_node: nil)
         @type = type
         @node_name = node_name
         @node_class = node_class
@@ -133,6 +133,7 @@ module LangGraphRB
         @duration = duration
         @error = error
         @result = result
+        @from_node = from_node
         @timestamp = Time.now.utc
       end
 
@@ -141,6 +142,7 @@ module LangGraphRB
           type: @type,
           node_name: @node_name,
           node_class: @node_class&.name,
+          from_node: @from_node,
           state_before: @state_before,
           state_after: @state_after,
           context: @context,

data/lib/langgraph_rb/runner.rb

@@ -64,7 +64,7 @@ module LangGraphRB
               if dest_name == Graph::FINISH
                 final_state = dest_state
               else
-                next_active << ExecutionFrame.new(dest_name, dest_state, @step_number)
+                next_active << ExecutionFrame.new(dest_name, dest_state, @step_number, from_node: result[:node_name])
               end
             else
               # Use normal edge routing
@@ -78,7 +78,7 @@ module LangGraphRB
                 if dest_name == Graph::FINISH
                   final_state = dest_state
                 else
-                  next_active << ExecutionFrame.new(dest_name, dest_state, @step_number)
+                  next_active << ExecutionFrame.new(dest_name, dest_state, @step_number, from_node: result[:node_name])
                 end
               end
             end
@@ -87,7 +87,7 @@ module LangGraphRB
             # Handle Send commands (map-reduce)
             result[:sends].each do |send_cmd|
               payload_state = result[:state].merge_delta(send_cmd.payload)
-              next_active << ExecutionFrame.new(send_cmd.to, payload_state, @step_number)
+              next_active << ExecutionFrame.new(send_cmd.to, payload_state, @step_number, from_node: result[:node_name])
             end
             
           when :interrupt
@@ -96,7 +96,7 @@ module LangGraphRB
               user_input = @interrupt_handler.call(result[:interrupt])
               # Continue with user input merged into state
               updated_state = result[:state].merge_delta(user_input || {})
-              next_active << ExecutionFrame.new(result[:node_name], updated_state, @step_number)
+              next_active << ExecutionFrame.new(result[:node_name], updated_state, @step_number, from_node: result[:node_name])
             else
               # No interrupt handler, treat as completion
               final_state = result[:state]
@@ -193,7 +193,7 @@ module LangGraphRB
       notify_observers(:on_graph_end, event)
     end
 
-    def notify_node_start(node, state, context)
+    def notify_node_start(node, state, context, from_node: nil)
       event = Observers::NodeEvent.new(
         type: :start,
         node_name: node.name,
@@ -201,12 +201,13 @@ module LangGraphRB
         state_before: state,
         context: context,
         thread_id: @thread_id,
-        step_number: @step_number
+        step_number: @step_number,
+        from_node: from_node
       )
       notify_observers(:on_node_start, event)
     end
 
-    def notify_node_end(node, state_before, state_after, result, duration)
+    def notify_node_end(node, state_before, state_after, result, duration, from_node: nil)
       event = Observers::NodeEvent.new(
         type: :end,
         node_name: node.name,
@@ -216,12 +217,13 @@ module LangGraphRB
         result: result,
         duration: duration,
         thread_id: @thread_id,
-        step_number: @step_number
+        step_number: @step_number,
+        from_node: from_node
       )
       notify_observers(:on_node_end, event)
     end
 
-    def notify_node_error(node, state, error)
+    def notify_node_error(node, state, error, from_node: nil)
       event = Observers::NodeEvent.new(
         type: :error,
         node_name: node.name,
@@ -229,7 +231,8 @@ module LangGraphRB
         state_before: state,
         error: error,
         thread_id: @thread_id,
-        step_number: @step_number
+        step_number: @step_number,
+        from_node: from_node
       )
       notify_observers(:on_node_error, event)
     end
@@ -251,7 +254,7 @@ module LangGraphRB
         # Execute each frame for this node
         executions.each do |frame|
           thread = Thread.new do
-            execute_node_safely(node, frame.state, context, frame.step)
+            execute_node_safely(node, frame.state, context, frame.step, from_node: frame.from_node)
           end
           threads << thread
         end
@@ -267,8 +270,8 @@ module LangGraphRB
     end
 
     # Safely execute a single node
-    def execute_node_safely(node, state, context, step)
-      notify_node_start(node, state, context)
+    def execute_node_safely(node, state, context, step, from_node: nil)
+      notify_node_start(node, state, context, from_node: from_node)
       
       start_time = Time.now
       begin
@@ -285,11 +288,11 @@ module LangGraphRB
                        state
                      end
         
-        notify_node_end(node, state, final_state, result, duration)
+        notify_node_end(node, state, final_state, result, duration, from_node: from_node)
         processed_result
       rescue => error
         duration = Time.now - start_time
-        notify_node_error(node, state, error)
+        notify_node_error(node, state, error, from_node: from_node)
         
         {
           type: :error,
@@ -421,16 +424,17 @@ module LangGraphRB
 
     # Execution frame for tracking active node executions
     class ExecutionFrame
-      attr_reader :node_name, :state, :step
+      attr_reader :node_name, :state, :step, :from_node
 
-      def initialize(node_name, state, step)
+      def initialize(node_name, state, step, from_node: nil)
         @node_name = node_name.to_sym
         @state = state
         @step = step
+        @from_node = from_node
       end
 
       def to_s
-        "#<ExecutionFrame node: #{@node_name}, step: #{@step}>"
+        "#<ExecutionFrame node: #{@node_name}, step: #{@step}, from: #{@from_node}>"
       end
     end
   end

data/lib/langgraph_rb/version.rb

@@ -1,3 +1,3 @@
 module LangGraphRB
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: langgraph_rb
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Julian Toro
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-08-05 00:00:00.000000000 Z
+date: 2025-08-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: json