roast-ai 0.2.3 → 0.3.1

data/lib/roast/workflow/step_orchestrator.rb

@@ -22,13 +22,15 @@ module Roast
         @workflow_executor = workflow_executor
       end
 
-      def execute_step(name, exit_on_error: true)
+      def execute_step(name, exit_on_error: true, step_key: nil)
         resource_type = @workflow.respond_to?(:resource) ? @workflow.resource&.type : nil
 
         @error_handler.with_error_handling(name, resource_type: resource_type) do
           $stderr.puts "Executing: #{name} (Resource type: #{resource_type || "unknown"})"
 
-          step_object = @step_loader.load(name)
+          # Use step_key for loading if provided, otherwise use name
+          load_key = step_key || name
+          step_object = @step_loader.load(name, step_key: load_key)
           step_result = step_object.call
 
           # Store result in workflow output

data/lib/roast/workflow/workflow_execution_context.rb

@@ -1,7 +1,5 @@
 # frozen_string_literal: true
 
-require "active_support/core_ext/string/inflections"
-
 module Roast
   module Workflow
     # Manages execution context across pre-processing, target workflows, and post-processing phases

data/lib/roast/workflow/workflow_executor.rb

@@ -1,9 +1,7 @@
 # frozen_string_literal: true
 
 require "English"
-require "active_support"
-require "active_support/isolated_execution_state"
-require "active_support/notifications"
+
 require "roast/workflow/command_executor"
 require "roast/workflow/conditional_executor"
 require "roast/workflow/error_handler"

data/lib/roast/workflow/workflow_initializer.rb

@@ -5,6 +5,7 @@ require "roast/initializers"
 require "roast/helpers/function_caching_interceptor"
 require "roast/helpers/logger"
 require "roast/workflow/base_workflow"
+require "roast/workflow/interpolator"
 
 module Roast
   module Workflow
@@ -27,11 +28,57 @@ module Roast
       end
 
       def include_tools
-        return unless @configuration.tools.present?
+        return unless @configuration.tools.present? || @configuration.mcp_tools.present?
 
         BaseWorkflow.include(Raix::FunctionDispatch)
         BaseWorkflow.include(Roast::Helpers::FunctionCachingInterceptor) # Add caching support
-        BaseWorkflow.include(*@configuration.tools.map(&:constantize))
+
+        if @configuration.tools.present?
+          BaseWorkflow.include(*@configuration.tools.map(&:constantize))
+        end
+
+        if @configuration.mcp_tools.present?
+          BaseWorkflow.include(Raix::MCP)
+
+          # Create an interpolator for MCP tool configuration
+          # We use Object.new as the context because this interpolation happens during
+          # initialization, before any workflow instance exists. Since we don't have
+          # a workflow instance yet, we use a minimal object that can still evaluate
+          # Ruby expressions like ENV['HOME'] or any other valid Ruby code.
+          interpolator = Interpolator.new(Object.new)
+
+          @configuration.mcp_tools.each do |tool|
+            # Interpolate the config values
+            config = interpolate_config(tool.config, interpolator)
+
+            # Create the appropriate client based on config
+            client = if config["url"]
+              Raix::MCP::SseClient.new(
+                config["url"],
+                headers: config["env"] || {},
+              )
+            elsif config["command"]
+              args = [config["command"]]
+              args += config["args"] if config["args"]
+              Raix::MCP::StdioClient.new(*args, config["env"] || {})
+            end
+
+            BaseWorkflow.mcp(client: client, only: tool.only, except: tool.except)
+          end
+        end
+
+        post_configure_tools
+      end
+
+      def post_configure_tools
+        @configuration.tools.each do |tool_name|
+          tool_module = tool_name.constantize
+
+          if tool_module.respond_to?(:post_configuration_setup)
+            tool_config = @configuration.tool_config(tool_name)
+            tool_module.post_configuration_setup(BaseWorkflow, tool_config)
+          end
+        end
       end
 
       def configure_api_client
@@ -116,6 +163,23 @@ module Roast
         # Make a lightweight API call to validate the token
         client.models.list if client.respond_to?(:models)
       end
+
+      def interpolate_config(config, interpolator)
+        interpolated = {}
+        config.each do |key, value|
+          interpolated[key] = case value
+          when String
+            interpolator.interpolate(value)
+          when Array
+            value.map { |v| v.is_a?(String) ? interpolator.interpolate(v) : v }
+          when Hash
+            interpolate_config(value, interpolator)
+          else
+            value
+          end
+        end
+        interpolated
+      end
     end
   end
 end

data/lib/roast/workflow/workflow_runner.rb

@@ -1,6 +1,5 @@
 # frozen_string_literal: true
 
-require "active_support/notifications"
 require "erb"
 require "roast/workflow/replay_handler"
 require "roast/workflow/workflow_executor"
@@ -168,7 +167,7 @@ module Roast
           context_path: @configuration.context_path,
           resource: @configuration.resource,
           session_name: @configuration.name,
-          configuration: @configuration,
+          workflow_configuration: @configuration,
           pre_processing_data:,
         ).tap do |workflow|
           workflow.output_file = @options[:output] if @options[:output].present?

data/lib/roast.rb

@@ -1,5 +1,13 @@
 # frozen_string_literal: true
 
+require "active_support"
+require "active_support/cache"
+require "active_support/notifications"
+require "active_support/core_ext/hash/indifferent_access"
+require "active_support/core_ext/string"
+require "active_support/core_ext/string/inflections"
+require "active_support/core_ext/module/delegation"
+require "active_support/isolated_execution_state"
 require "fileutils"
 require "cli/ui"
 require "raix"

data/package-lock.json

@@ -0,0 +1,6 @@
+{
+  "name": "roast",
+  "lockfileVersion": 3,
+  "requires": true,
+  "packages": {}
+}

data/roast.gemspec

@@ -41,6 +41,7 @@ Gem::Specification.new do |spec|
   spec.add_dependency("diff-lcs", "~> 1.5")
   spec.add_dependency("faraday-retry")
   spec.add_dependency("json-schema")
-  spec.add_dependency("raix", "~> 0.8.6")
+  spec.add_dependency("open_router", "~> 0.3")
+  spec.add_dependency("raix", "~> 1.0")
   spec.add_dependency("thor", "~> 1.3")
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: roast-ai
 version: !ruby/object:Gem::Version
-  version: 0.2.3
+  version: 0.3.1
 platform: ruby
 authors:
 - Shopify
@@ -79,20 +79,34 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: open_router
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.3'
 - !ruby/object:Gem::Dependency
   name: raix
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.8.6
+        version: '1.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.8.6
+        version: '1.0'
 - !ruby/object:Gem::Dependency
   name: thor
   requirement: !ruby/object:Gem::Requirement
@@ -142,9 +156,28 @@ files:
 - examples/api_workflow/prompt.md
 - examples/api_workflow/transform_data/prompt.md
 - examples/api_workflow/workflow.yml
+- examples/bash_prototyping/README.md
+- examples/bash_prototyping/analyze_network/prompt.md
+- examples/bash_prototyping/analyze_system/prompt.md
+- examples/bash_prototyping/api_testing.yml
+- examples/bash_prototyping/check_processes/prompt.md
+- examples/bash_prototyping/generate_report/prompt.md
+- examples/bash_prototyping/process_json_response/prompt.md
+- examples/bash_prototyping/system_analysis.yml
+- examples/bash_prototyping/test_public_api/prompt.md
 - examples/case_when/README.md
 - examples/case_when/detect_language/prompt.md
 - examples/case_when/workflow.yml
+- examples/cmd/README.md
+- examples/cmd/analyze_project/prompt.md
+- examples/cmd/basic_demo/prompt.md
+- examples/cmd/basic_workflow.yml
+- examples/cmd/check_repository/prompt.md
+- examples/cmd/create_and_verify/prompt.md
+- examples/cmd/dev_workflow.yml
+- examples/cmd/explore_project/prompt.md
+- examples/cmd/explorer_workflow.yml
+- examples/cmd/smart_tool_selection/prompt.md
 - examples/conditional/README.md
 - examples/conditional/check_condition/prompt.md
 - examples/conditional/simple_workflow.yml
@@ -157,6 +190,7 @@ files:
 - examples/exit_on_error/analyze_lint_output/prompt.md
 - examples/exit_on_error/apply_fixes/prompt.md
 - examples/exit_on_error/workflow.yml
+- examples/grading/README.md
 - examples/grading/analyze_coverage/prompt.md
 - examples/grading/calculate_final_grade.rb
 - examples/grading/format_result.rb
@@ -203,6 +237,33 @@ files:
 - examples/iteration/workflow.yml
 - examples/json_handling/README.md
 - examples/json_handling/workflow.yml
+- examples/mcp/README.md
+- examples/mcp/analyze_changes/prompt.md
+- examples/mcp/analyze_issues/prompt.md
+- examples/mcp/analyze_schema/prompt.md
+- examples/mcp/check_data_quality/prompt.md
+- examples/mcp/check_documentation/prompt.md
+- examples/mcp/create_recommendations/prompt.md
+- examples/mcp/database_workflow.yml
+- examples/mcp/env_demo/workflow.yml
+- examples/mcp/fetch_pr_context/prompt.md
+- examples/mcp/filesystem_demo/create_test_file/prompt.md
+- examples/mcp/filesystem_demo/list_files/prompt.md
+- examples/mcp/filesystem_demo/read_with_mcp/prompt.md
+- examples/mcp/filesystem_demo/workflow.yml
+- examples/mcp/generate_insights/prompt.md
+- examples/mcp/generate_report/prompt.md
+- examples/mcp/generate_review/prompt.md
+- examples/mcp/github_workflow.yml
+- examples/mcp/multi_mcp_workflow.yml
+- examples/mcp/post_review/prompt.md
+- examples/mcp/save_report/prompt.md
+- examples/mcp/search_issues/prompt.md
+- examples/mcp/summarize/prompt.md
+- examples/mcp/test_filesystem/prompt.md
+- examples/mcp/test_github/prompt.md
+- examples/mcp/test_read/prompt.md
+- examples/mcp/workflow.yml
 - examples/openrouter_example/README.md
 - examples/openrouter_example/analyze_input/prompt.md
 - examples/openrouter_example/generate_response/prompt.md
@@ -225,6 +286,9 @@ files:
 - examples/rspec_to_minitest/run_and_improve/prompt.md
 - examples/rspec_to_minitest/workflow.md
 - examples/rspec_to_minitest/workflow.yml
+- examples/shared_config/README.md
+- examples/shared_config/example_with_shared_config/workflow.yml
+- examples/shared_config/shared.yml
 - examples/single_target_prepost/README.md
 - examples/single_target_prepost/post_processing/output.txt
 - examples/single_target_prepost/pre_processing/gather_dependencies/prompt.md
@@ -233,6 +297,9 @@ files:
 - examples/smart_coercion_defaults/workflow.yml
 - examples/step_configuration/README.md
 - examples/step_configuration/workflow.yml
+- examples/tool_config_example/README.md
+- examples/tool_config_example/example_step/prompt.md
+- examples/tool_config_example/workflow.yml
 - examples/workflow_generator/README.md
 - examples/workflow_generator/analyze_user_request/prompt.md
 - examples/workflow_generator/create_workflow_files/prompt.md
@@ -261,9 +328,11 @@ files:
 - lib/roast/resources/url_resource.rb
 - lib/roast/tools.rb
 - lib/roast/tools/ask_user.rb
+- lib/roast/tools/bash.rb
 - lib/roast/tools/cmd.rb
 - lib/roast/tools/coding_agent.rb
 - lib/roast/tools/grep.rb
+- lib/roast/tools/helpers/coding_agent_message_formatter.rb
 - lib/roast/tools/read_file.rb
 - lib/roast/tools/search_file.rb
 - lib/roast/tools/update_files.rb
@@ -325,6 +394,7 @@ files:
 - lib/roast/workflow/workflow_executor.rb
 - lib/roast/workflow/workflow_initializer.rb
 - lib/roast/workflow/workflow_runner.rb
+- package-lock.json
 - roast.gemspec
 - schema/workflow.json
 - shipit.rubygems.yml