roast-ai 0.4.9 → 0.4.10

data/sorbet/rbi/shims/lib/roast/dsl/cog_input_context.rbi

@@ -0,0 +1,17 @@
+# typed: true
+# frozen_string_literal: true
+
+module Roast
+  module DSL
+    class CogInputContext
+      #: (Symbol) -> Roast::DSL::Cogs::Cmd::Output
+      def cmd(name); end
+
+      #: (Symbol) -> Roast::DSL::Cogs::Chat::Output
+      def chat(name); end
+
+      #: (Symbol) -> Roast::DSL::Cog::Output
+      def execute(name); end
+    end
+  end
+end

data/sorbet/rbi/shims/lib/roast/dsl/config_context.rbi

@@ -6,6 +6,12 @@ module Roast
     class ConfigContext
       #: (?Symbol?) {() [self: Roast::DSL::Cogs::Cmd::Config] -> void} -> void
       def cmd(name = nil, &block); end
+
+      #: (?Symbol?) {() [self: Roast::DSL::Cogs::Chat::Config] -> void} -> void
+      def chat(name = nil, &block); end
+
+      #: (?Symbol?) {() [self: Roast::DSL::Cog::Config] -> void} -> void
+      def execute(name = nil, &block); end
     end
   end
 end

data/sorbet/rbi/shims/lib/roast/dsl/execution_context.rbi

@@ -0,0 +1,17 @@
+# typed: true
+# frozen_string_literal: true
+
+module Roast
+  module DSL
+    class ExecutionContext
+      #: (?Symbol?) {(Roast::DSL::Cogs::Cmd::Input) [self: Roast::DSL::CogInputContext] -> (String | Array[String] | nil)} -> void
+      def cmd(name = nil, &block); end
+
+      #: (?Symbol?) {(Roast::DSL::Cogs::Chat::Input) [self: Roast::DSL::CogInputContext] -> String?} -> void
+      def chat(name = nil, &block); end
+
+      #: (?Symbol?) {(Roast::DSL::Cogs::Execute::Input) [self: Roast::DSL::CogInputContext] -> (Symbol | nil)} -> void
+      def execute(name = nil, &block); end
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: roast-ai
 version: !ruby/object:Gem::Version
-  version: 0.4.9
+  version: 0.4.10
 platform: ruby
 authors:
 - Shopify
@@ -121,6 +121,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1.2'
+- !ruby/object:Gem::Dependency
+  name: ruby_llm
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: sqlite3
   requirement: !ruby/object:Gem::Requirement
@@ -201,9 +215,23 @@ files:
 - docs/INSTRUMENTATION.md
 - docs/ITERATION_SYNTAX.md
 - docs/VALIDATION.md
+- dsl/demo/Gemfile
+- dsl/demo/Gemfile.lock
+- dsl/demo/cogs/local.rb
+- dsl/demo/simple_external_cog.rb
 - dsl/less_simple.rb
+- dsl/plugin-gem-example/.gitignore
+- dsl/plugin-gem-example/Gemfile
+- dsl/plugin-gem-example/Gemfile.lock
+- dsl/plugin-gem-example/lib/other.rb
+- dsl/plugin-gem-example/lib/plugin_gem_example.rb
+- dsl/plugin-gem-example/lib/simple.rb
+- dsl/plugin-gem-example/lib/version.rb
+- dsl/plugin-gem-example/plugin-gem-example.gemspec
 - dsl/prototype.rb
+- dsl/scoped_executors.rb
 - dsl/simple.rb
+- dsl/simple_chat.rb
 - dsl/step_communication.rb
 - examples/README.md
 - examples/available_tools_demo/README.md
@@ -232,15 +260,22 @@ files:
 - lib/roast/dsl.rb
 - lib/roast/dsl/cog.rb
 - lib/roast/dsl/cog/config.rb
+- lib/roast/dsl/cog/input.rb
+- lib/roast/dsl/cog/output.rb
+- lib/roast/dsl/cog/registry.rb
 - lib/roast/dsl/cog/stack.rb
 - lib/roast/dsl/cog/store.rb
-- lib/roast/dsl/cog_execution_context.rb
-- lib/roast/dsl/cogs.rb
+- lib/roast/dsl/cog_input_context.rb
+- lib/roast/dsl/cog_input_manager.rb
+- lib/roast/dsl/cogs/chat.rb
 - lib/roast/dsl/cogs/cmd.rb
+- lib/roast/dsl/cogs/execute.rb
 - lib/roast/dsl/cogs/graph.rb
 - lib/roast/dsl/config_context.rb
-- lib/roast/dsl/executor.rb
-- lib/roast/dsl/workflow_execution_context.rb
+- lib/roast/dsl/config_manager.rb
+- lib/roast/dsl/execution_context.rb
+- lib/roast/dsl/execution_manager.rb
+- lib/roast/dsl/workflow.rb
 - lib/roast/error.rb
 - lib/roast/errors.rb
 - lib/roast/factories/api_provider_factory.rb
@@ -417,6 +452,7 @@ files:
 - sorbet/rbi/gems/listen@3.9.0.rbi
 - sorbet/rbi/gems/logger@1.7.0.rbi
 - sorbet/rbi/gems/lumberjack@1.2.10.rbi
+- sorbet/rbi/gems/marcel@1.1.0.rbi
 - sorbet/rbi/gems/method_source@1.1.0.rbi
 - sorbet/rbi/gems/mime-types-data@3.2025.0617.rbi
 - sorbet/rbi/gems/mime-types@3.7.0.rbi
@@ -436,14 +472,13 @@ files:
 - sorbet/rbi/gems/pry@0.15.2.rbi
 - sorbet/rbi/gems/public_suffix@6.0.2.rbi
 - sorbet/rbi/gems/racc@1.8.1.rbi
-- sorbet/rbi/gems/rack@2.2.18.rbi
+- sorbet/rbi/gems/rack@2.2.19.rbi
 - sorbet/rbi/gems/rainbow@3.1.1.rbi
 - sorbet/rbi/gems/raix@1.0.2.rbi
 - sorbet/rbi/gems/rake@13.3.0.rbi
 - sorbet/rbi/gems/rb-fsevent@0.11.2.rbi
 - sorbet/rbi/gems/rb-inotify@0.11.1.rbi
 - sorbet/rbi/gems/rbi@0.3.6.rbi
-- sorbet/rbi/gems/rbs-inline@0.12.0.rbi
 - sorbet/rbi/gems/rbs@3.9.4.rbi
 - sorbet/rbi/gems/regexp_parser@2.10.0.rbi
 - sorbet/rbi/gems/rexml@3.4.2.rbi
@@ -455,6 +490,7 @@ files:
 - sorbet/rbi/gems/ruby-openai@8.1.0.rbi
 - sorbet/rbi/gems/ruby-progressbar@1.13.0.rbi
 - sorbet/rbi/gems/ruby2_keywords@0.0.5.rbi
+- sorbet/rbi/gems/ruby_llm@1.8.2.rbi
 - sorbet/rbi/gems/securerandom@0.4.1.rbi
 - sorbet/rbi/gems/shellany@0.0.1.rbi
 - sorbet/rbi/gems/simplecov-html@0.13.2.rbi
@@ -473,8 +509,9 @@ files:
 - sorbet/rbi/gems/yard-sorbet@0.9.0.rbi
 - sorbet/rbi/gems/yard@0.9.37.rbi
 - sorbet/rbi/gems/zeitwerk@2.7.3.rbi
+- sorbet/rbi/shims/lib/roast/dsl/cog_input_context.rbi
 - sorbet/rbi/shims/lib/roast/dsl/config_context.rbi
-- sorbet/rbi/shims/lib/roast/dsl/workflow_execution_context.rbi
+- sorbet/rbi/shims/lib/roast/dsl/execution_context.rbi
 - sorbet/rbi/todo.rbi
 - sorbet/tapioca/config.yml
 - sorbet/tapioca/require.rb

data/lib/roast/dsl/cog_execution_context.rb

@@ -1,29 +0,0 @@
-# typed: true
-# frozen_string_literal: true
-
-module Roast
-  module DSL
-    # Contains the cogs already executed in this run.
-    class CogExecutionContext
-      # Raises if you access a cog in an execution block that hasn't already been run.
-      class IncompleteCogExecutionAccessError < StandardError; end
-
-      def initialize(cogs, bound_names)
-        @cogs = cogs
-        bind_cog_methods(bound_names)
-      end
-
-      private
-
-      def bind_cog_methods(bound_names)
-        bound_names.map do |name|
-          define_singleton_method(name.to_sym, ->(name) do
-            @cogs[name].tap do |cog|
-              raise IncompleteCogExecutionAccessError unless cog.ran?
-            end.output
-          end)
-        end
-      end
-    end
-  end
-end

data/lib/roast/dsl/cogs.rb

@@ -1,65 +0,0 @@
-# typed: false
-# frozen_string_literal: true
-
-module Roast
-  module DSL
-    module Cogs
-      class << self
-        #: (String) -> void
-        def load_all_for(rube_rb_fpath)
-          rube_rb_fpath = File.expand_path(rube_rb_fpath)
-          all_cog_files(rube_rb_fpath).each do |cog_fpath|
-            require cog_fpath
-          end
-          bind_all_cog_invokers
-        end
-
-        #: () -> Array[Class]
-        def all_cog_classes
-          # rubocop:disable Sorbet/ConstantsFromStrings
-          Roast::DSL::Cogs.constants.map do |cog_class_name|
-            Roast::DSL::Cogs.const_get(cog_class_name)
-          end
-          # rubocop:enable Sorbet/ConstantsFromStrings
-        end
-
-        private
-
-        #: () -> void
-        def bind_all_cog_invokers
-          all_cog_classes.each do |cog_class|
-            # At some point we may want to tuck this all into a Roast::DSL::Binding/Scope/Context to avoid polluting toplevel.
-            TOPLEVEL_BINDING.eval(binding_string_for(cog_class))
-          end
-        end
-
-        #: (Class) -> String
-        def binding_string_for(cog_class)
-          <<~RUBY
-            def #{cog_class.method_name}(*args, **kwargs, &block)
-              #{cog_class.name}.invoke(*args, **kwargs, &block)
-            end
-          RUBY
-        end
-
-        #: (String) -> Array[String]
-        def all_cog_files(rube_rb_fpath)
-          dirs = [project_cogs_dir(rube_rb_fpath), internal_cogs_dir]
-          dirs.map do |dir|
-            Dir.glob(File.join(dir, "*.rb")) # Just toplevel .rb files
-          end.flatten
-        end
-
-        #: (String) -> String
-        def project_cogs_dir(rube_rb_fpath)
-          File.join(File.dirname(rube_rb_fpath), "cogs")
-        end
-
-        #: () -> String
-        def internal_cogs_dir
-          File.join(File.dirname(__FILE__), "cogs")
-        end
-      end
-    end
-  end
-end

data/lib/roast/dsl/executor.rb

@@ -1,82 +0,0 @@
-# typed: true
-# frozen_string_literal: true
-
-module Roast
-  module DSL
-    class Executor
-      class ExecutorError < Roast::Error; end
-      class ExecutorAlreadyPreparedError < ExecutorError; end
-      class ExecutorAlreadyCompletedError < ExecutorError; end
-
-      class << self
-        def from_file(workflow_path)
-          run!(File.read(workflow_path))
-        end
-
-        private
-
-        def run!(workflow_definition)
-          executor = new
-          executor.prepare!(workflow_definition)
-          executor.start!
-        end
-      end
-
-      def prepare!(input)
-        # You can only initialize an executor once.
-        raise ExecutorAlreadyPreparedError if @prepared
-
-        extract_dsl_procs(input)
-        @cogs = Cog::Store.new
-        @cog_stack = Cog::Stack.new
-
-        @config_context = ConfigContext.new(@cogs, @config_proc)
-        @config_context.prepare!
-        @execution_context = WorkflowExecutionContext.new(@cogs, @cog_stack, @execution_proc)
-        @execution_context.prepare!
-
-        @prepared = true
-      end
-
-      def start!
-        # Now we run the cogs!
-        # You can only do this once, executors are not reusable to avoid state pollution
-        raise ExecutorAlreadyCompletedError if @completed
-
-        @cog_stack.map do |name, cog|
-          cog.run!(
-            @config_context.fetch_merged_config(cog.class, name.to_sym),
-            @execution_context.cog_execution_context,
-          )
-        end
-
-        @completed = true
-      end
-
-      def prepared?
-        @prepared ||= false
-      end
-
-      def completed?
-        @completed ||= false
-      end
-
-      #: { () [self: ConfigContext] -> void} -> void
-      def config(&block)
-        @config_proc = block
-      end
-
-      #: { () [self: WorkflowExecutionContext] -> void} -> void
-      def execute(&block)
-        @execution_proc = block
-      end
-
-      # Separating the instance evals ensures that we can reuse the same cog method
-      # names between config and execute, while have the backing objects be completely
-      # different. This means we have an enforced separation between configuring and running.
-      def extract_dsl_procs(input)
-        instance_eval(input)
-      end
-    end
-  end
-end

data/lib/roast/dsl/workflow_execution_context.rb

@@ -1,47 +0,0 @@
-# typed: true
-# frozen_string_literal: true
-
-module Roast
-  module DSL
-    class WorkflowExecutionContext
-      def initialize(cogs, cog_stack, execution_proc)
-        @cogs = cogs
-        @cog_stack = cog_stack
-        @execution_proc = execution_proc
-        @bound_names = []
-      end
-
-      def prepare!
-        bind_default_cogs
-        instance_eval(&@execution_proc)
-      end
-
-      def cog_execution_context
-        @cog_execution_context ||= CogExecutionContext.new(@cogs, @bound_names)
-      end
-
-      private
-
-      def add_cog_instance(name, cog)
-        @cogs.insert(name, cog)
-        @cog_stack.push([name, cog])
-      end
-
-      def output(name)
-        @cogs[name].output
-      end
-
-      #: () -> void
-      def bind_default_cogs
-        bind_cog(Cogs::Cmd, :cmd)
-      end
-
-      def bind_cog(cog_class, name)
-        @bound_names << name
-        instance_eval do
-          define_singleton_method(name, &cog_class.on_create)
-        end
-      end
-    end
-  end
-end