mars_rb 0.1.0 → 1.0.0

data/lib/mars/workflows/parallel.rb

@@ -0,0 +1,61 @@
+# frozen_string_literal: true
+
+module MARS
+  module Workflows
+    class Parallel < Runnable
+      def initialize(name, steps:, aggregator: nil, **kwargs)
+        super(name: name, **kwargs)
+
+        @steps = steps
+        @aggregator = aggregator || Aggregator.new("#{name} Aggregator")
+      end
+
+      def run(context)
+        context = ensure_context(context)
+        child_contexts = []
+        results = execute_steps(context, child_contexts)
+
+        context.merge(child_contexts)
+        context.current_input = results
+        aggregator.run(context)
+      end
+
+      private
+
+      attr_reader :steps, :aggregator
+
+      def execute_steps(context, child_contexts)
+        Sync do |workflow|
+          tasks = steps.map do |step|
+            child_ctx = context.fork(state: step.state)
+            child_contexts << child_ctx
+
+            workflow.async do
+              workflow_step(step, child_ctx)
+            end
+          end
+
+          tasks.map(&:wait)
+        end
+      end
+
+      def workflow_step(step, child_ctx)
+        step.run_before_hooks(child_ctx)
+
+        step_input = step.formatter.format_input(child_ctx)
+        child_ctx.current_input = step_input
+
+        result = step.run(child_ctx)
+
+        formatted = step.formatter.format_output(result)
+        child_ctx.record(step.name, formatted)
+        step.run_after_hooks(child_ctx, formatted)
+        formatted
+      end
+
+      def ensure_context(input)
+        input.is_a?(ExecutionContext) ? input : ExecutionContext.new(input: input)
+      end
+    end
+  end
+end

data/lib/mars/workflows/sequential.rb

@@ -1,21 +1,39 @@
 # frozen_string_literal: true
 
-require_relative "../runnable"
-
-module Mars
+module MARS
   module Workflows
     class Sequential < Runnable
-      def initialize(name, steps:)
-        @name = name
+      def initialize(name, steps:, **kwargs)
+        super(name: name, **kwargs)
+
         @steps = steps
       end
 
-      def run(input)
+      def run(context)
+        context = ensure_context(context)
+
         @steps.each do |step|
-          input = step.run(input)
+          step.run_before_hooks(context)
+
+          step_input = step.formatter.format_input(context)
+          context.current_input = step_input
+
+          result = step.run(context)
+
+          formatted = step.formatter.format_output(result)
+          context.record(step.name, formatted)
+          step.run_after_hooks(context, formatted)
         end
 
-        input
+        context.current_input
+      end
+
+      private
+
+      attr_reader :steps
+
+      def ensure_context(input)
+        input.is_a?(ExecutionContext) ? input : ExecutionContext.new(input: input)
       end
     end
   end

data/lib/mars.rb

@@ -1,8 +1,17 @@
 # frozen_string_literal: true
 
-require_relative "mars/version"
+require "zeitwerk"
+require "async"
+require "ruby_llm"
+require "ruby_llm/schema"
 
-module Mars
+loader = Zeitwerk::Loader.for_gem
+loader.inflector.inflect("mars" => "MARS")
+loader.ignore("#{__dir__}/mars_rb.rb")
+loader.setup
+
+module MARS
   class Error < StandardError; end
-  # Your code goes here...
 end
+
+MARS::Rendering::Graph.include_extensions

data/lib/mars_rb.rb

@@ -0,0 +1,3 @@
+# frozen_string_literal: true
+
+require_relative "mars"

data/sig/mars.rbs

@@ -1,4 +1,4 @@
-module Mars
+module MARS
   VERSION: String
   # See the writing guide of rbs: https://github.com/ruby/rbs#guides
 end

metadata

@@ -1,20 +1,62 @@
 --- !ruby/object:Gem::Specification
 name: mars_rb
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 1.0.0
 platform: ruby
 authors:
 - Santiago Bartesaghi
 - Andres Garcia
 - Ignacio Perez
 - Santiago Diaz
-autorequire: 
+autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-09-26 00:00:00.000000000 Z
-dependencies: []
+date: 2026-07-29 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: async
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.34'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.34'
+- !ruby/object:Gem::Dependency
+  name: ruby_llm
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.12'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.12'
+- !ruby/object:Gem::Dependency
+  name: zeitwerk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.7'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.7'
 description: MARS (Multi-Agent Ruby SDK) provides a comprehensive framework for developers
-  to implement multi-agent solutions using pure Ruby. It offers a simple API for orchestrating
+  to implementmulti-agent solutions using pure Ruby. It offers a simple API for orchestrating
   multiple agents.
 email:
 - sbartesaghi@hotmail.com
@@ -32,10 +74,40 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- examples/complex_llm_workflow/diagram.md
+- examples/complex_llm_workflow/generator.rb
+- examples/complex_workflow/diagram.md
+- examples/complex_workflow/generator.rb
+- examples/parallel_workflow/diagram.md
+- examples/parallel_workflow/generator.rb
+- examples/simple_workflow/diagram.md
+- examples/simple_workflow/generator.rb
 - lib/mars.rb
+- lib/mars/agent_step.rb
+- lib/mars/aggregator.rb
+- lib/mars/execution_context.rb
+- lib/mars/formatter.rb
+- lib/mars/gate.rb
+- lib/mars/hooks.rb
+- lib/mars/rendering/graph.rb
+- lib/mars/rendering/graph/agent_step.rb
+- lib/mars/rendering/graph/aggregator.rb
+- lib/mars/rendering/graph/base.rb
+- lib/mars/rendering/graph/builder.rb
+- lib/mars/rendering/graph/gate.rb
+- lib/mars/rendering/graph/node.rb
+- lib/mars/rendering/graph/parallel_workflow.rb
+- lib/mars/rendering/graph/runnable.rb
+- lib/mars/rendering/graph/sequential_workflow.rb
+- lib/mars/rendering/graph/subgraph.rb
+- lib/mars/rendering/html.rb
+- lib/mars/rendering/mermaid.rb
 - lib/mars/runnable.rb
 - lib/mars/version.rb
+- lib/mars/workflows/aggregate_error.rb
+- lib/mars/workflows/parallel.rb
 - lib/mars/workflows/sequential.rb
+- lib/mars_rb.rb
 - sig/mars.rbs
 homepage: https://github.com/rootstrap/mars
 licenses:
@@ -44,7 +116,8 @@ metadata:
   homepage_uri: https://github.com/rootstrap/mars
   source_code_uri: https://github.com/rootstrap/mars
   changelog_uri: https://github.com/rootstrap/mars/releases
-post_install_message: 
+  rubygems_mfa_required: 'true'
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -59,8 +132,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3.1
-signing_key: 
+rubygems_version: 3.5.23
+signing_key:
 specification_version: 4
 summary: Multi-Agent Ruby SDK - A framework for building multi-agent solutions in
   pure Ruby