Add state to runnable (#17)

Author: PerezIgnacio

lib/mars/agent.rb

@@ -4,7 +4,9 @@ module Mars
   class Agent < Runnable
     attr_reader :name
 
-    def initialize(name:, options: {}, tools: [], schema: nil, instructions: nil)
+    def initialize(name:, options: {}, tools: [], schema: nil, instructions: nil, **kwargs)
+      super(**kwargs)
+
       @name = name
       @tools = Array(tools)
       @schema = schema

lib/mars/aggregator.rb

@@ -4,7 +4,9 @@ module Mars
   class Aggregator < Runnable
     attr_reader :name, :operation
 
-    def initialize(name = "Aggregator", operation: nil)
+    def initialize(name = "Aggregator", operation: nil, **kwargs)
+      super(**kwargs)
+
       @name = name
       @operation = operation || ->(inputs) { inputs.join("\n") }
     end

lib/mars/gate.rb

@@ -4,7 +4,9 @@ module Mars
   class Gate < Runnable
     attr_reader :name
 
-    def initialize(name:, condition:, branches:)
+    def initialize(name:, condition:, branches:, **kwargs)
+      super(**kwargs)
+
       @name = name
       @condition = condition
       @branches = branches

lib/mars/runnable.rb

@@ -2,6 +2,12 @@
 
 module Mars
   class Runnable
+    attr_accessor :state
+
+    def initialize(state: {})
+      @state = state
+    end
+
     def run(input)
       raise NotImplementedError
     end

lib/mars/workflows/parallel.rb

@@ -5,7 +5,9 @@ module Workflows
     class Parallel < Runnable
       attr_reader :name
 
-      def initialize(name, steps:, aggregator: nil)
+      def initialize(name, steps:, aggregator: nil, **kwargs)
+        super(**kwargs)
+
         @name = name
         @steps = steps
         @aggregator = aggregator || Aggregator.new("#{name} Aggregator")

lib/mars/workflows/sequential.rb

@@ -5,7 +5,9 @@ module Workflows
     class Sequential < Runnable
       attr_reader :name
 
-      def initialize(name, steps:)
+      def initialize(name, steps:, **kwargs)
+        super(**kwargs)
+
         @name = name
         @steps = steps
       end