roast-ai 1.0.2 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 309884973e075e4d94e5eb7ea3b3192ebe5cbcf5d08a60312c1054f9567d4eb9
-  data.tar.gz: b79c03de093e577e77b480e7c67ac5dd510572eeb51887ea9147a03474a6f76a
+  metadata.gz: 493182d7cf3f626050e7e580402cda301cccf7da038072fec060e8e142a067a8
+  data.tar.gz: b1a789f8383a9644f0a7d91c7485b14def95574ff5a5e59e7be7ca17a1bf3ff5
 SHA512:
-  metadata.gz: 78d6f38126192dd69853a1bfe8bfe15c5f8ee6ee34f4cf998af534ff215c7a774e0af250409de94027ae36ed0b159a624ad6c4f1c837c40dfc24b72a8b6fe1cd
-  data.tar.gz: f36f8a9ad38f0c4e8d0ee15b88c212a59d831613024d9825f362fceba8619d1ad27289142e5f47c53300ac6036753733b7c4c5d3cc27781efbc21f54c63b69de
+  metadata.gz: 30044c47daf5c6c6856b5d947c2740a0df631f59079332cb15d01c02db18aec6a943b193e3b54718162379d186f9e33ca08b163fe1b2d98d13c2da0c5aa6f358
+  data.tar.gz: 1546e8c8128a4bf80a55faa80b07872bb12faaac9b21b61f75f62a4024f50b033441be0a8f9883b29af15b8d0951121e09c80a68c8c3a4fc69a8df7a298f5704

data/.rubocop.yml

@@ -2,6 +2,8 @@ inherit_from: .rubocop_todo.yml
 
 plugins:
   - rubocop-sorbet
+  - type_toolkit:
+      require_path: rubocop-type_toolkit
 
 inherit_gem:
   rubocop-shopify: rubocop.yml

data/Gemfile

@@ -17,9 +17,9 @@ group :development, :test do
   gem "rubocop-shopify", require: false
   gem "rubocop-sorbet", require: false
   gem "simplecov", require: false
-  gem "sorbet", "~> 0.5.12414", require: false
+  gem "sorbet", ">= 0.6.12698", require: false
   gem "sqlite3", require: false
-  gem "tapioca", "~> 0.16.11", require: false
+  gem "tapioca", ">= 0.17.8", require: false
   gem "vcr", require: false
   gem "webmock", require: false
 end

data/Gemfile.lock

@@ -1,11 +1,12 @@
 PATH
   remote: .
   specs:
-    roast-ai (1.0.2)
+    roast-ai (1.1.0)
       activesupport (~> 8.0)
       async (>= 2.34)
       rainbow (>= 3.0.0)
       ruby_llm (>= 1.8)
+      type_toolkit (>= 0.0.5)
       zeitwerk (>= 2.6)
 
 GEM
@@ -126,12 +127,15 @@ GEM
     rb-fsevent (0.11.2)
     rb-inotify (0.11.1)
       ffi (~> 1.0)
-    rbi (0.3.6)
+    rbi (0.3.9)
       prism (~> 1.0)
       rbs (>= 3.4.4)
-    rbs (3.9.4)
+    rbs (4.0.0.dev.5)
       logger
+      prism (>= 1.3.0)
+      tsort
     regexp_parser (2.10.0)
+    require-hooks (0.2.3)
     rexml (3.4.2)
     rubocop (1.77.0)
       json (~> 2.3)
@@ -171,35 +175,41 @@ GEM
       simplecov_json_formatter (~> 0.1)
     simplecov-html (0.13.2)
     simplecov_json_formatter (0.1.4)
-    sorbet (0.5.12414)
-      sorbet-static (= 0.5.12414)
-    sorbet-runtime (0.5.12414)
-    sorbet-static (0.5.12414-universal-darwin)
-    sorbet-static (0.5.12414-x86_64-linux)
-    sorbet-static-and-runtime (0.5.12414)
-      sorbet (= 0.5.12414)
-      sorbet-runtime (= 0.5.12414)
-    spoom (1.6.3)
+    sorbet (0.6.12984)
+      sorbet-static (= 0.6.12984)
+    sorbet-runtime (0.6.12984)
+    sorbet-static (0.6.12984-universal-darwin)
+    sorbet-static (0.6.12984-x86_64-linux)
+    sorbet-static-and-runtime (0.6.12984)
+      sorbet (= 0.6.12984)
+      sorbet-runtime (= 0.6.12984)
+    spoom (1.7.11)
       erubi (>= 1.10.0)
       prism (>= 0.28.0)
       rbi (>= 0.3.3)
+      rbs (>= 4.0.0.dev.4)
       rexml (>= 3.2.6)
       sorbet-static-and-runtime (>= 0.5.10187)
       thor (>= 0.19.2)
     sqlite3 (2.9.0-arm64-darwin)
     sqlite3 (2.9.0-x86_64-linux-gnu)
-    tapioca (0.16.11)
+    tapioca (0.17.10)
       benchmark
       bundler (>= 2.2.25)
       netrc (>= 0.11.0)
       parallel (>= 1.21.0)
-      rbi (~> 0.2)
+      rbi (>= 0.3.7)
+      require-hooks (>= 0.2.2)
       sorbet-static-and-runtime (>= 0.5.11087)
-      spoom (>= 1.2.0)
+      spoom (>= 1.7.9)
       thor (>= 1.2.0)
       yard-sorbet
     thor (1.4.0)
     traces (0.18.2)
+    tsort (0.2.0)
+    type_toolkit (0.0.5)
+      lint_roller
+      rubocop (>= 1.72.0)
     tzinfo (2.0.6)
       concurrent-ruby (~> 1.0)
     unicode-display_width (3.1.4)
@@ -233,9 +243,9 @@ DEPENDENCIES
   rubocop-shopify
   rubocop-sorbet
   simplecov
-  sorbet (~> 0.5.12414)
+  sorbet (>= 0.6.12698)
   sqlite3
-  tapioca (~> 0.16.11)
+  tapioca (>= 0.17.8)
   vcr
   webmock
 

data/examples/agent_with_multiple_prompts.rb

@@ -0,0 +1,27 @@
+# typed: true
+# frozen_string_literal: true
+
+#: self as Roast::Workflow
+
+config do
+  agent do
+    provider :claude
+    model "haiku"
+    show_prompt!
+    dump_raw_agent_messages_to "tmp/claude-messages2.log"
+  end
+end
+
+execute do
+  agent(:multi_step) do
+    [
+      "What is 2+2?",
+      "Now multiply that by 3",
+      "Now subtract 5"
+    ]
+  end
+  ruby do
+    answer = agent!(:multi_step).integer!
+    puts "((2 + 2) * 3) - 5 = #{answer}"
+  end
+end

data/examples/demo/Gemfile.lock

@@ -1,11 +1,12 @@
 PATH
   remote: ../..
   specs:
-    roast-ai (1.0.2)
+    roast-ai (1.1.0)
       activesupport (~> 8.0)
       async (>= 2.34)
       rainbow (>= 3.0.0)
       ruby_llm (>= 1.8)
+      type_toolkit (>= 0.0.5)
       zeitwerk (>= 2.6)
 
 PATH
@@ -30,6 +31,7 @@ GEM
       securerandom (>= 0.3)
       tzinfo (~> 2.0, >= 2.0.5)
       uri (>= 0.13.1)
+    ast (2.4.3)
     async (2.34.0)
       console (~> 1.29)
       fiber-annotation
@@ -65,6 +67,8 @@ GEM
       concurrent-ruby (~> 1.0)
     io-event (1.14.0)
     json (2.18.1)
+    language_server-protocol (3.17.0.5)
+    lint_roller (1.1.0)
     logger (1.7.0)
     marcel (1.1.0)
     metrics (0.15.0)
@@ -72,7 +76,29 @@ GEM
     multipart-post (2.4.1)
     net-http (0.9.1)
       uri (>= 0.11.1)
+    parallel (1.28.0)
+    parser (3.3.11.1)
+      ast (~> 2.4.1)
+      racc
+    prism (1.9.0)
+    racc (1.8.1)
     rainbow (3.1.1)
+    regexp_parser (2.11.3)
+    rubocop (1.86.0)
+      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.49.0, < 2.0)
+      ruby-progressbar (~> 1.7)
+      unicode-display_width (>= 2.4.0, < 4.0)
+    rubocop-ast (1.49.1)
+      parser (>= 3.3.7.2)
+      prism (~> 1.7)
+    ruby-progressbar (1.13.0)
     ruby_llm (1.8.2)
       base64
       event_stream_parser (~> 1)
@@ -84,8 +110,14 @@ GEM
       zeitwerk (~> 2)
     securerandom (0.4.1)
     traces (0.18.2)
+    type_toolkit (0.0.5)
+      lint_roller
+      rubocop (>= 1.72.0)
     tzinfo (2.0.6)
       concurrent-ruby (~> 1.0)
+    unicode-display_width (3.2.0)
+      unicode-emoji (~> 4.1)
+    unicode-emoji (4.2.0)
     uri (1.1.1)
     zeitwerk (2.7.3)
 

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

@@ -1,11 +1,12 @@
 PATH
   remote: ../..
   specs:
-    roast-ai (1.0.2)
+    roast-ai (1.1.0)
       activesupport (~> 8.0)
       async (>= 2.34)
       rainbow (>= 3.0.0)
       ruby_llm (>= 1.8)
+      type_toolkit (>= 0.0.5)
       zeitwerk (>= 2.6)
 
 PATH
@@ -133,6 +134,9 @@ GEM
     stringio (3.1.7)
     traces (0.18.2)
     tsort (0.2.0)
+    type_toolkit (0.0.5)
+      lint_roller
+      rubocop (>= 1.72.0)
     tzinfo (2.0.6)
       concurrent-ruby (~> 1.0)
     unicode-display_width (3.2.0)

data/examples/simple_pi_agent.rb

@@ -0,0 +1,18 @@
+# typed: true
+# frozen_string_literal: true
+
+#: self as Roast::Workflow
+
+config do
+  agent do
+    provider :pi
+    model "anthropic/claude-haiku-4-5-20251001"
+    append_system_prompt "Always respond in haiku form"
+    show_prompt!
+    dump_raw_agent_messages_to "tmp/pi-messages.log"
+  end
+end
+
+execute do
+  agent { "What is the world's largest lake?" }
+end

data/lib/roast/cog/output.rb

@@ -241,8 +241,9 @@ module Roast
           lines.reverse.each do |line|
             # Look for numbers with various separators, formats, and currency symbols (very permissive)
             # Matches: 123, 1,234, 1_234, 1 234, 1.23, -1.23, 1.23e10, 1.23e-10
+            # Reversed so that the last number on the line is tried first (matching the bottom-up line order)
             matches = line.scan(/-?[\d\s$¢£€¥.,_]+(?:[eE][+-]?\d+)?/) #: as Array[String]
-            candidates.concat(matches.map(&:strip))
+            candidates.concat(matches.map(&:strip).reverse)
           end
 
           candidates.compact.uniq

data/lib/roast/cog/registry.rb

@@ -51,13 +51,13 @@ module Roast
       #
       #: (singleton(Roast::Cog)) -> void
       def use(cog_class)
-        reg = create_registration(cog_class)
-        cogs[reg.first] = reg.second
+        name, klass = create_registration(cog_class)
+        cogs[name] = klass
       end
 
       private
 
-      #: (singleton(Roast::Cog)) -> Array(Symbol, singleton(Cog))
+      #: (singleton(Roast::Cog)) -> [Symbol, singleton(Cog)]
       def create_registration(cog_class)
         cog_class_name = cog_class.name
         raise CouldNotDeriveCogNameError if cog_class_name.nil?

data/lib/roast/cogs/agent/config.rb

@@ -5,7 +5,7 @@ module Roast
   module Cogs
     class Agent < Cog
       class Config < Cog::Config
-        VALID_PROVIDERS = [:claude].freeze #: Array[Symbol]
+        VALID_PROVIDERS = [:claude, :pi].freeze #: Array[Symbol]
 
         # Configure the cog to use a specified provider when invoking an agent
         #

data/lib/roast/cogs/agent/input.rb

@@ -9,10 +9,14 @@ module Roast
       # The agent cog requires a prompt that will be sent to the agent for processing.
       # Optionally, a session identifier can be provided to maintain context across multiple invocations.
       class Input < Cog::Input
-        # The prompt to send to the agent for processing
+        # The prompts to send to the agent for processing
         #
-        #: String?
-        attr_accessor :prompt
+        # When multiple prompts are specified, each subsequent prompt is passed to the agent as soon as it completes
+        # the previous one, in the same session throughout. This can be useful for helping to ensure the agent produces
+        # final outputs in the form you desire after performing a long and complex task.
+        #
+        #: Array[String]
+        attr_accessor :prompts
 
         # Optional session identifier for maintaining conversation context
         #
@@ -28,7 +32,7 @@ module Roast
         #: () -> void
         def initialize
           super
-          @prompt = nil #: String?
+          @prompts = [] #: Array[String]
         end
 
         # Validate that the input has all required parameters
@@ -37,41 +41,35 @@ module Roast
         #
         # #### See Also
         # - `coerce`
-        # - `valid_prompt!`
         #
         #: () -> void
         def validate!
-          valid_prompt!
+          raise Cog::Input::InvalidInputError, "At least one prompt is required" unless prompts.present?
+          raise Cog::Input::InvalidInputError, "Blank prompts are not allowed" if prompts.any?(&:blank?)
         end
 
         # Coerce the input from the return value of the input block
         #
         # If the input block returns a String, it will be used as the prompt value.
+        # If the input block returns an Array of Strings, the first will be used as the prompt and the
+        # rest will be used as finalizers.
         #
         # #### See Also
         # - `validate!`
         #
         #: (untyped) -> void
         def coerce(input_return_value)
-          if input_return_value.is_a?(String)
-            self.prompt ||= input_return_value
+          case input_return_value
+          when String
+            self.prompts = [input_return_value]
+          when Array
+            self.prompts = input_return_value.map(&:to_s)
           end
         end
 
-        # Get the validated prompt value
-        #
-        # Returns the prompt if it is present, otherwise raises an `InvalidInputError`.
-        #
-        # #### See Also
-        # - `prompt`
-        # - `validate!`
-        #
-        #: () -> String
-        def valid_prompt!
-          valid_prompt = @prompt
-          raise Cog::Input::InvalidInputError, "'prompt' is required" unless valid_prompt.present?
-
-          valid_prompt
+        #: (String) -> void
+        def prompt=(prompt)
+          @prompts = [prompt]
         end
       end
     end

data/lib/roast/cogs/agent/providers/claude/claude_invocation.rb

@@ -53,20 +53,23 @@ module Roast
               end
             end
 
-            #: (Agent::Config, Agent::Input) -> void
-            def initialize(config, input)
+            #: (Agent::Config, String, String?, ?fork_session: bool) -> void
+            def initialize(config, prompt, session, fork_session: true)
               @base_command = config.valid_command #: (String | Array[String])?
               @model = config.valid_model #: String?
               @append_system_prompt = config.valid_append_system_prompt #: String?
               @replace_system_prompt = config.valid_replace_system_prompt #: String?
               @apply_permissions = config.apply_permissions? #: bool
               @working_directory = config.valid_working_directory #: Pathname?
-              @prompt = input.valid_prompt! #: String
-              @session = input.session #: String?
               @context = Context.new #: Context
               @result = Result.new #: Result
               @raw_dump_file = config.valid_dump_raw_agent_messages_to_path #: Pathname?
+              @show_prompt = config.show_prompt? #: bool
               @show_progress = config.show_progress? #: bool
+              @show_response = config.show_response? #: bool
+              @prompt = prompt
+              @session = session
+              @fork_session = fork_session #: bool
             end
 
             #: () -> void
@@ -74,6 +77,7 @@ module Roast
               raise ClaudeAlreadyStartedError if started?
 
               @started = true
+              puts "[USER PROMPT] #{@prompt}" if @show_prompt
               _stdout, stderr, status = CommandRunner.execute(
                 command_line,
                 working_directory: @working_directory,
@@ -83,6 +87,7 @@ module Roast
 
               if status.success?
                 @completed = true
+                puts "[AGENT RESPONSE] #{@result.response}" if @show_response
               else
                 @failed = true
                 @result.success = false
@@ -180,7 +185,10 @@ module Roast
               command.push("--model", @model) if @model
               command.push("--system-prompt", @replace_system_prompt) if @replace_system_prompt
               command.push("--append-system-prompt", @append_system_prompt) if @append_system_prompt
-              command.push("--fork-session", "--resume", @session) if @session.present?
+              if @session.present?
+                command.push("--fork-session") if @fork_session
+                command.push("--resume", @session)
+              end
               command << "--dangerously-skip-permissions" unless @apply_permissions
               command
             end

data/lib/roast/cogs/agent/providers/claude.rb

@@ -18,9 +18,22 @@ module Roast
 
           #: (Agent::Input) -> Agent::Output
           def invoke(input)
-            invocation = ClaudeInvocation.new(@config, input)
-            invocation.run!
-            Output.new(invocation.result)
+            invocations = [] #: Array[ClaudeInvocation]
+            input.prompts.each do |prompt|
+              previous_session = invocations.last&.result&.session
+              invocation = ClaudeInvocation.new(
+                @config,
+                prompt,
+                previous_session || input.session,
+                fork_session: previous_session.nil?,
+              )
+              invocation.run!
+              invocations << invocation
+              break unless invocation.result.success
+            end
+            final_result = invocations.last.not_nil!.result
+            final_result.stats = invocations.filter_map { |i| i.result.stats }.reduce(:+) if invocations.size > 1
+            Output.new(final_result)
           end
         end
       end

data/lib/roast/cogs/agent/providers/pi/messages/tool_call_message.rb

@@ -0,0 +1,60 @@
+# typed: true
+# frozen_string_literal: true
+
+module Roast
+  module Cogs
+    class Agent < Cog
+      module Providers
+        class Pi < Provider
+          module Messages
+            # Represents a tool call made by the Pi agent
+            #
+            # In Pi's JSON protocol, tool calls appear as `toolcall_end` events within
+            # `message_update` messages, containing the tool name, id, and arguments.
+            class ToolCallMessage
+              #: String?
+              attr_reader :id
+
+              #: String?
+              attr_reader :name
+
+              #: Hash[Symbol, untyped]
+              attr_reader :arguments
+
+              #: (id: String?, name: String?, arguments: Hash[Symbol, untyped]) -> void
+              def initialize(id:, name:, arguments:)
+                @id = id
+                @name = name
+                @arguments = arguments
+              end
+
+              #: () -> String?
+              def format
+                return unless name
+
+                case name.to_s.downcase
+                when "bash"
+                  "BASH #{arguments[:command]}"
+                when "read"
+                  "READ #{arguments[:path]}"
+                when "edit"
+                  "EDIT #{arguments[:path]}"
+                when "write"
+                  "WRITE #{arguments[:path]}"
+                when "grep"
+                  "GREP #{arguments[:pattern]} #{arguments[:path]}"
+                when "find"
+                  "FIND #{arguments[:pattern]} #{arguments[:path]}"
+                when "ls"
+                  "LS #{arguments[:path]}"
+                else
+                  "TOOL [#{name}] #{arguments.inspect}"
+                end
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/roast/cogs/agent/providers/pi/messages/tool_result_message.rb

@@ -0,0 +1,57 @@
+# typed: true
+# frozen_string_literal: true
+
+module Roast
+  module Cogs
+    class Agent < Cog
+      module Providers
+        class Pi < Provider
+          module Messages
+            # Represents a tool execution result from the Pi agent
+            #
+            # In Pi's JSON protocol, tool results appear as `tool_execution_end` events
+            # containing the result content and error status.
+            class ToolResultMessage
+              #: String?
+              attr_reader :tool_call_id
+
+              #: String?
+              attr_reader :tool_name
+
+              #: String?
+              attr_reader :content
+
+              #: bool
+              attr_reader :is_error
+
+              #: (tool_call_id: String?, tool_name: String?, content: String?, is_error: bool) -> void
+              def initialize(tool_call_id:, tool_name:, content:, is_error:)
+                @tool_call_id = tool_call_id
+                @tool_name = tool_name
+                @content = content
+                @is_error = is_error
+              end
+
+              #: (PiInvocation::Context) -> String?
+              def format(context)
+                tool_call = context.tool_call(tool_call_id)
+                name = tool_name || tool_call&.name || "unknown"
+                status = is_error ? "ERROR" : "OK"
+
+                # Truncate long tool results for progress display
+                c = content
+                display_content = if c && c.length > 200
+                  "#{c[0..197]}..."
+                else
+                  c
+                end
+
+                "#{name.upcase} #{status}#{display_content ? " #{display_content}" : ""}"
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end