roast-ai 0.4.9 → 0.4.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fc01ff6a90dbe17c735b17935bb75bcfda0ee47c0b9dbee6234292cc4f5f3799
-  data.tar.gz: caa21e581b5476a65c2dc8b0b7a27f1b20baf2f493260d954785724f73a3ff40
+  metadata.gz: ef6e5b6234cde031dbec2498a1edfe20822de1afb336292ff0e55d3abada5748
+  data.tar.gz: 26992714777862482a9935b7b924f7064db3eed24d3fb25b18b21ef7b741385c
 SHA512:
-  metadata.gz: 6ea65bb2381627119b5740f9353b064d046e26b2769e0be36baf3f13b9696e30e6bd6332fbbf8c4305fea33d1b2bb12d81a4a56871f3793c3718e2ac8407c6ae
-  data.tar.gz: 23ff60c01817d64bae76bb16c0395d8cc7aa95cd621550cebc5fc6763fcf5ba8a023542bf07b49067fa9210d8f228f538b0e82159d897397d08532deece73bc4
+  metadata.gz: 4dc6d30fa71c300fb83ae5144e6fbd55ab7770b25534daa6dadc24c58484276d590951840d196b074038328db317adda90c885d3f6779cfadc0125348e40cf23
+  data.tar.gz: 0d35f0494140641ec3c2a643e86f4b6b174adbbf44af866d617471c405b0247f51a45f53514488a28b4dbd6b0175ff87521752ffba663ab926e51259b7398eeb

data/Gemfile

@@ -8,20 +8,19 @@ git_source(:github) { |repo_name| "https://github.com/#{repo_name}" }
 gemspec
 
 gem "cgi"
+gem "claude_swarm"
 gem "cli-ui", "2.3.0"
 gem "dotenv"
-gem "guard"
 gem "guard-minitest"
+gem "guard"
+gem "minitest-rg"
 gem "mocha"
 gem "rake", require: false
 gem "rubocop-shopify", require: false
 gem "rubocop-sorbet", require: false
+gem "ruby_llm"
 gem "simplecov", require: false
-gem "minitest-rg"
-gem "vcr", require: false
-gem "webmock", require: false
-
 gem "sorbet", require: false
 gem "tapioca", require: false
-
-gem "claude_swarm"
+gem "vcr", require: false
+gem "webmock", require: false

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    roast-ai (0.4.9)
+    roast-ai (0.4.10)
       activesupport (>= 7.0)
       cli-kit (~> 5.0)
       cli-ui (= 2.3.0)
@@ -10,6 +10,7 @@ PATH
       open_router (~> 0.3)
       raix (~> 1.0.2)
       ruby-graphviz (~> 1.2)
+      ruby_llm
       sqlite3 (~> 2.6)
       thor (~> 1.3)
       zeitwerk (~> 2.6)
@@ -133,6 +134,7 @@ GEM
       rb-inotify (~> 0.9, >= 0.9.10)
     logger (1.7.0)
     lumberjack (1.2.10)
+    marcel (1.1.0)
     method_source (1.1.0)
     mime-types (3.7.0)
       logger
@@ -167,7 +169,7 @@ GEM
       method_source (~> 1.0)
     public_suffix (6.0.2)
     racc (1.8.1)
-    rack (2.2.19)
+    rack (2.2.20)
     rainbow (3.1.1)
     raix (1.0.2)
       activesupport (>= 6.0)
@@ -213,6 +215,15 @@ GEM
       faraday-multipart (>= 1)
     ruby-progressbar (1.13.0)
     ruby2_keywords (0.0.5)
+    ruby_llm (1.8.2)
+      base64
+      event_stream_parser (~> 1)
+      faraday (>= 1.10.0)
+      faraday-multipart (>= 1)
+      faraday-net_http (>= 1)
+      faraday-retry (>= 1)
+      marcel (~> 1.0)
+      zeitwerk (~> 2)
     securerandom (0.4.1)
     shellany (0.0.1)
     simplecov (0.22.0)
@@ -284,6 +295,7 @@ DEPENDENCIES
   roast-ai!
   rubocop-shopify
   rubocop-sorbet
+  ruby_llm
   simplecov
   sorbet
   tapioca

data/dsl/demo/Gemfile

@@ -0,0 +1,4 @@
+# frozen_string_literal: true
+
+gem "roast-ai", path: "../.."
+gem "plugin-gem-example", path: "../plugin-gem-example"

data/dsl/demo/Gemfile.lock

@@ -0,0 +1,120 @@
+PATH
+  remote: ../..
+  specs:
+    roast-ai (0.4.9)
+      activesupport (>= 7.0)
+      cli-kit (~> 5.0)
+      cli-ui (= 2.3.0)
+      diff-lcs (~> 1.5)
+      json-schema
+      open_router (~> 0.3)
+      raix (~> 1.0.2)
+      ruby-graphviz (~> 1.2)
+      ruby_llm
+      sqlite3 (~> 2.6)
+      thor (~> 1.3)
+      zeitwerk (~> 2.6)
+
+PATH
+  remote: ../plugin-gem-example
+  specs:
+    plugin-gem-example (0.1.0)
+      roast-ai
+
+GEM
+  specs:
+    activesupport (8.0.3)
+      base64
+      benchmark (>= 0.3)
+      bigdecimal
+      concurrent-ruby (~> 1.0, >= 1.3.1)
+      connection_pool (>= 2.2.5)
+      drb
+      i18n (>= 1.6, < 2)
+      logger (>= 1.4.2)
+      minitest (>= 5.1)
+      securerandom (>= 0.3)
+      tzinfo (~> 2.0, >= 2.0.5)
+      uri (>= 0.13.1)
+    addressable (2.8.7)
+      public_suffix (>= 2.0.2, < 7.0)
+    base64 (0.3.0)
+    benchmark (0.4.1)
+    bigdecimal (3.3.1)
+    cli-kit (5.0.1)
+      cli-ui (~> 2.0)
+    cli-ui (2.3.0)
+    concurrent-ruby (1.3.5)
+    connection_pool (2.5.4)
+    diff-lcs (1.6.2)
+    dotenv (3.1.8)
+    drb (2.2.3)
+    event_stream_parser (1.0.0)
+    faraday (2.14.0)
+      faraday-net_http (>= 2.0, < 3.5)
+      json
+      logger
+    faraday-multipart (1.1.1)
+      multipart-post (~> 2.0)
+    faraday-net_http (3.4.1)
+      net-http (>= 0.5.0)
+    faraday-retry (2.3.2)
+      faraday (~> 2.0)
+    i18n (1.14.7)
+      concurrent-ruby (~> 1.0)
+    json (2.15.1)
+    json-schema (6.0.0)
+      addressable (~> 2.8)
+      bigdecimal (~> 3.1)
+    logger (1.7.0)
+    marcel (1.1.0)
+    minitest (5.26.0)
+    multipart-post (2.4.1)
+    net-http (0.6.0)
+      uri
+    open_router (0.3.3)
+      activesupport (>= 6.0)
+      dotenv (>= 2)
+      faraday (>= 1)
+      faraday-multipart (>= 1)
+    ostruct (0.6.3)
+    public_suffix (6.0.2)
+    raix (1.0.3)
+      activesupport (>= 6.0)
+      faraday-retry (~> 2.0)
+      open_router (~> 0.2)
+      ostruct
+      ruby-openai (~> 8.1)
+    rexml (3.4.4)
+    ruby-graphviz (1.2.5)
+      rexml
+    ruby-openai (8.3.0)
+      event_stream_parser (>= 0.3.0, < 2.0.0)
+      faraday (>= 1)
+      faraday-multipart (>= 1)
+    ruby_llm (1.8.2)
+      base64
+      event_stream_parser (~> 1)
+      faraday (>= 1.10.0)
+      faraday-multipart (>= 1)
+      faraday-net_http (>= 1)
+      faraday-retry (>= 1)
+      marcel (~> 1.0)
+      zeitwerk (~> 2)
+    securerandom (0.4.1)
+    sqlite3 (2.7.4-arm64-darwin)
+    thor (1.4.0)
+    tzinfo (2.0.6)
+      concurrent-ruby (~> 1.0)
+    uri (1.0.4)
+    zeitwerk (2.7.3)
+
+PLATFORMS
+  arm64-darwin
+
+DEPENDENCIES
+  plugin-gem-example!
+  roast-ai!
+
+BUNDLED WITH
+   2.6.8

data/dsl/demo/cogs/local.rb

@@ -0,0 +1,15 @@
+# typed: true
+# frozen_string_literal: true
+
+class Local < Roast::DSL::Cog
+  class Input < Roast::DSL::Cog::Input
+    def validate!
+      true
+    end
+  end
+
+  #: (Input) -> void
+  def execute(input)
+    puts "I'm a workflow-specific cog!"
+  end
+end

data/dsl/demo/simple_external_cog.rb

@@ -0,0 +1,17 @@
+# typed: true
+# frozen_string_literal: true
+
+#: self as Roast::DSL::Workflow
+
+use "simple", from: "plugin_gem_example"
+use "MyCogNamespace::Other", from: "plugin_gem_example"
+use "local"
+
+# Use multiple cogs
+# use ["simple", "other"], from: "plugin_gem_example"
+
+execute do
+  simple
+  other
+  local
+end

data/dsl/plugin-gem-example/.gitignore

@@ -0,0 +1,8 @@
+/.bundle/
+/.yardoc
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/

data/dsl/plugin-gem-example/Gemfile

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+source "https://rubygems.org"
+
+# Specify your gem's dependencies in plugin-gem-example.gemspec
+gemspec
+
+gem "irb"
+gem "rake", "~> 13.0"
+
+gem "minitest", "~> 5.16"
+
+gem "rubocop", "~> 1.21"

data/dsl/plugin-gem-example/Gemfile.lock

@@ -0,0 +1,178 @@
+PATH
+  remote: .
+  specs:
+    plugin-gem-example (0.1.0)
+      roast-ai
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    activesupport (8.0.3)
+      base64
+      benchmark (>= 0.3)
+      bigdecimal
+      concurrent-ruby (~> 1.0, >= 1.3.1)
+      connection_pool (>= 2.2.5)
+      drb
+      i18n (>= 1.6, < 2)
+      logger (>= 1.4.2)
+      minitest (>= 5.1)
+      securerandom (>= 0.3)
+      tzinfo (~> 2.0, >= 2.0.5)
+      uri (>= 0.13.1)
+    addressable (2.8.7)
+      public_suffix (>= 2.0.2, < 7.0)
+    ast (2.4.3)
+    base64 (0.3.0)
+    benchmark (0.4.1)
+    bigdecimal (3.3.1)
+    cli-kit (5.0.1)
+      cli-ui (~> 2.0)
+    cli-ui (2.3.0)
+    concurrent-ruby (1.3.5)
+    connection_pool (2.5.4)
+    date (3.4.1)
+    diff-lcs (1.6.2)
+    dotenv (3.1.8)
+    drb (2.2.3)
+    erb (5.1.1)
+    event_stream_parser (1.0.0)
+    faraday (2.14.0)
+      faraday-net_http (>= 2.0, < 3.5)
+      json
+      logger
+    faraday-multipart (1.1.1)
+      multipart-post (~> 2.0)
+    faraday-net_http (3.4.1)
+      net-http (>= 0.5.0)
+    faraday-retry (2.3.2)
+      faraday (~> 2.0)
+    i18n (1.14.7)
+      concurrent-ruby (~> 1.0)
+    io-console (0.8.1)
+    irb (1.15.2)
+      pp (>= 0.6.0)
+      rdoc (>= 4.0.0)
+      reline (>= 0.4.2)
+    json (2.15.1)
+    json-schema (6.0.0)
+      addressable (~> 2.8)
+      bigdecimal (~> 3.1)
+    language_server-protocol (3.17.0.5)
+    lint_roller (1.1.0)
+    logger (1.7.0)
+    minitest (5.26.0)
+    multipart-post (2.4.1)
+    net-http (0.6.0)
+      uri
+    open_router (0.3.3)
+      activesupport (>= 6.0)
+      dotenv (>= 2)
+      faraday (>= 1)
+      faraday-multipart (>= 1)
+    ostruct (0.6.3)
+    parallel (1.27.0)
+    parser (3.3.9.0)
+      ast (~> 2.4.1)
+      racc
+    pp (0.6.3)
+      prettyprint
+    prettyprint (0.2.0)
+    prism (1.6.0)
+    psych (5.2.6)
+      date
+      stringio
+    public_suffix (6.0.2)
+    racc (1.8.1)
+    rainbow (3.1.1)
+    raix (1.0.3)
+      activesupport (>= 6.0)
+      faraday-retry (~> 2.0)
+      open_router (~> 0.2)
+      ostruct
+      ruby-openai (~> 8.1)
+    rake (13.3.0)
+    rdoc (6.15.0)
+      erb
+      psych (>= 4.0.0)
+      tsort
+    regexp_parser (2.11.3)
+    reline (0.6.2)
+      io-console (~> 0.5)
+    rexml (3.4.4)
+    roast-ai (0.4.9)
+      activesupport (>= 7.0)
+      cli-kit (~> 5.0)
+      cli-ui (= 2.3.0)
+      diff-lcs (~> 1.5)
+      json-schema
+      open_router (~> 0.3)
+      raix (~> 1.0.2)
+      ruby-graphviz (~> 1.2)
+      sqlite3 (~> 2.6)
+      thor (~> 1.3)
+      zeitwerk (~> 2.6)
+    rubocop (1.81.1)
+      json (~> 2.3)
+      language_server-protocol (~> 3.17.0.2)
+      lint_roller (~> 1.1.0)
+      parallel (~> 1.10)
+      parser (>= 3.3.0.2)
+      rainbow (>= 2.2.2, < 4.0)
+      regexp_parser (>= 2.9.3, < 3.0)
+      rubocop-ast (>= 1.47.1, < 2.0)
+      ruby-progressbar (~> 1.7)
+      unicode-display_width (>= 2.4.0, < 4.0)
+    rubocop-ast (1.47.1)
+      parser (>= 3.3.7.2)
+      prism (~> 1.4)
+    ruby-graphviz (1.2.5)
+      rexml
+    ruby-openai (8.3.0)
+      event_stream_parser (>= 0.3.0, < 2.0.0)
+      faraday (>= 1)
+      faraday-multipart (>= 1)
+    ruby-progressbar (1.13.0)
+    securerandom (0.4.1)
+    sqlite3 (2.7.4-aarch64-linux-gnu)
+    sqlite3 (2.7.4-aarch64-linux-musl)
+    sqlite3 (2.7.4-arm-linux-gnu)
+    sqlite3 (2.7.4-arm-linux-musl)
+    sqlite3 (2.7.4-arm64-darwin)
+    sqlite3 (2.7.4-x86-linux-gnu)
+    sqlite3 (2.7.4-x86-linux-musl)
+    sqlite3 (2.7.4-x86_64-darwin)
+    sqlite3 (2.7.4-x86_64-linux-gnu)
+    sqlite3 (2.7.4-x86_64-linux-musl)
+    stringio (3.1.7)
+    thor (1.4.0)
+    tsort (0.2.0)
+    tzinfo (2.0.6)
+      concurrent-ruby (~> 1.0)
+    unicode-display_width (3.2.0)
+      unicode-emoji (~> 4.1)
+    unicode-emoji (4.1.0)
+    uri (1.0.4)
+    zeitwerk (2.7.3)
+
+PLATFORMS
+  aarch64-linux-gnu
+  aarch64-linux-musl
+  arm-linux-gnu
+  arm-linux-musl
+  arm64-darwin
+  x86-linux-gnu
+  x86-linux-musl
+  x86_64-darwin
+  x86_64-linux-gnu
+  x86_64-linux-musl
+
+DEPENDENCIES
+  irb
+  minitest (~> 5.16)
+  plugin-gem-example!
+  rake (~> 13.0)
+  rubocop (~> 1.21)
+
+BUNDLED WITH
+   2.6.8

data/dsl/plugin-gem-example/lib/other.rb

@@ -0,0 +1,17 @@
+# typed: true
+# frozen_string_literal: true
+
+module MyCogNamespace
+  class Other < Roast::DSL::Cog
+    class Input < Roast::DSL::Cog::Input
+      def validate!
+        true
+      end
+    end
+
+    #: (Input) -> void
+    def execute(input)
+      puts "I'm a different cog!"
+    end
+  end
+end

data/dsl/plugin-gem-example/lib/plugin_gem_example.rb

@@ -0,0 +1,5 @@
+# typed: false
+# frozen_string_literal: true
+
+require "simple"
+require "other"

data/dsl/plugin-gem-example/lib/simple.rb

@@ -0,0 +1,15 @@
+# typed: true
+# frozen_string_literal: true
+
+class Simple < Roast::DSL::Cog
+  class Input < Roast::DSL::Cog::Input
+    def validate!
+      true
+    end
+  end
+
+  #: (Input) -> void
+  def execute(input)
+    puts "I'm a cog!"
+  end
+end

data/dsl/plugin-gem-example/lib/version.rb

@@ -0,0 +1,10 @@
+# typed: false
+# frozen_string_literal: true
+
+module Plugin
+  module Gem
+    module Example
+      VERSION = "0.1.0"
+    end
+  end
+end

data/dsl/plugin-gem-example/plugin-gem-example.gemspec

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+require_relative "lib/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "plugin-gem-example"
+  spec.version = Plugin::Gem::Example::VERSION
+  spec.authors = ["Sam Schmidt"]
+  spec.email = ["sam.schmidt@shopify.com"]
+
+  spec.summary = "TODO: Write a short summary, because RubyGems requires one."
+  spec.description = "TODO: Write a longer description or delete this line."
+  spec.homepage = "TODO: Put your gem's website or public repo URL here."
+  spec.license = "MIT"
+  spec.required_ruby_version = ">= 3.1.0"
+
+  spec.metadata["allowed_push_host"] = "TODO: Set to your gem server 'https://example.com'"
+
+  spec.metadata["homepage_uri"] = spec.homepage
+  spec.metadata["source_code_uri"] = "TODO: Put your gem's public repo URL here."
+  spec.metadata["changelog_uri"] = "TODO: Put your gem's CHANGELOG.md URL here."
+
+  # Uncomment to register a new dependency of your gem
+  spec.add_dependency("roast-ai")
+
+  # For more information and examples about making a new gem, check out our
+  # guide at: https://bundler.io/guides/creating_gem.html
+end

data/dsl/prototype.rb

@@ -1,17 +1,25 @@
 # typed: true
 # frozen_string_literal: true
 
-#: self as Roast::DSL::Executor
+#: self as Roast::DSL::Workflow
 
 config do
+  # TODO: make a way to do print_all on all and turn off printing on one
   cmd(:echo) { print_all! }
+  cmd(:date) { print_stdout! }
+  chat { "" }
+  # TODO: add a way to apply config to a subset of named cog by pattern
+  # cmd(/^date.*/) { print_all! }
 end
 
 execute do
   # Anonymous cog. Added to the stack directly and given an autogenerated key in cog storage
   # Use for actions you do once and forget about, don't need configuration
-  cmd { "touch tmp/#{SecureRandom.uuid} " }
+  cmd { |my| my.command = "touch tmp/#{SecureRandom.uuid}" }
 
   # Named cog. Configuration for this specific instance will be looked up from config block
-  cmd(:echo) { "echo 'Hello World!'" }
+  cmd(:echo) { |my| my.command = "echo 'Hello World!'" }
+
+  # Cogs can implement input coercion for simple return values
+  cmd(:date) { "date" }
 end

data/dsl/scoped_executors.rb

@@ -0,0 +1,28 @@
+# typed: true
+# frozen_string_literal: true
+
+#: self as Roast::DSL::Workflow
+
+config do
+  cmd do
+    print_all!
+  end
+end
+
+execute(:capitalize_a_random_word) do
+  cmd(:word) { "shuf /usr/share/dict/words -n 1" }
+  cmd(:capitalize) do |my|
+    word = cmd(:word).out.strip
+    my.command = "sh"
+    my.args << "-c"
+    my.args << "echo '#{word}' | tr '[:lower:]' '[:upper:]'"
+  end
+end
+
+execute do
+  cmd(:before) { "echo '--> before'" }
+  execute { :capitalize_a_random_word }
+  execute { :capitalize_a_random_word }
+  execute { :capitalize_a_random_word }
+  cmd(:after) { "echo '--> after'" }
+end

data/dsl/simple_chat.rb

@@ -0,0 +1,12 @@
+# typed: true
+# frozen_string_literal: true
+
+#: self as Roast::DSL::Workflow
+
+config do
+  # ...
+end
+
+execute do
+  chat(:lake) { "What is the deepest lake?" }
+end

data/dsl/step_communication.rb

@@ -1,6 +1,8 @@
-# typed: false
+# typed: true
 # frozen_string_literal: true
 
+#: self as Roast::DSL::Workflow
+
 # How do we pass information between steps?
 # Demonstrate by passing result of a command output to another step
 
@@ -10,9 +12,13 @@ end
 
 execute do
   cmd(:ls) { "ls -al" }
-  cmd(:echo) do
+  cmd(:echo) do |my|
+    my.command = "echo"
     # TODO: this is a bespoke output object for cmd, is there a generic one we can offer
-    first_line = cmd(:ls).command_output.split("\n").second
-    "echo '#{first_line}'"
+    first_line = cmd(:ls).out.split("\n").second
+    last_line = cmd(:ls).out.split("\n").last
+    my.args << first_line unless first_line.blank?
+    my.args << "\n---\n"
+    my.args << last_line if last_line != first_line && last_line.present?
   end
 end

data/lib/roast/dsl/cog/config.rb

@@ -1,16 +1,19 @@
-# typed: false
+# typed: true
 # frozen_string_literal: true
 
 module Roast
   module DSL
     class Cog
       class Config
+        #: Hash[Symbol, untyped]
         attr_reader :values
 
+        #: (?Hash[Symbol, untyped]) -> void
         def initialize(initial = {})
           @values = initial
         end
 
+        #: (Cog::Config) -> Cog::Config
         def merge(config_object)
           self.class.new(values.merge(config_object.values))
         end
@@ -18,10 +21,12 @@ module Roast
         # It is recommended to implement a custom config object for a nicer interface,
         # but for simple cases where it would just be a key value store we provide one by default.
 
+        #: (Symbol, untyped) -> void
         def []=(key, value)
           @values[key] = value
         end
 
+        #: (Symbol) -> untyped
         def [](key)
           @values[key]
         end