riffer-rig 0.4.0 → 0.6.0

data/sig/generated/riffer/rig/events/turn_end.rbs

@@ -0,0 +1,22 @@
+# Generated from lib/riffer/rig/events/turn_end.rb with RBS::Inline
+
+class Riffer::Rig::Events::TurnEnd < Riffer::Rig::Events::Event
+  # @dynamic stop_reason, usage
+  attr_reader stop_reason: Symbol
+
+  attr_reader usage: ::Riffer::Providers::TokenUsage?
+
+  # @rbs stop_reason: Symbol
+  # @rbs usage: ::Riffer::Providers::TokenUsage?
+  # @rbs return: void
+  def initialize: (Symbol stop_reason, ::Riffer::Providers::TokenUsage? usage) -> void
+
+  # @rbs return: Float?
+  def cost: () -> Float?
+
+  # @rbs return: Symbol
+  def type: () -> Symbol
+
+  # @rbs return: Hash[Symbol, untyped]
+  def to_h: () -> Hash[Symbol, untyped]
+end

data/sig/generated/riffer/rig/events.rbs

@@ -0,0 +1,4 @@
+# Generated from lib/riffer/rig/events.rb with RBS::Inline
+
+module Riffer::Rig::Events
+end

data/sig/generated/riffer/rig/extension.rbs

@@ -0,0 +1,24 @@
+# Generated from lib/riffer/rig/extension.rb with RBS::Inline
+
+class Riffer::Rig::Extension
+  @name: String
+
+  @requires: Gem::Requirement?
+
+  @block: ^(::Riffer::Rig::Registrar) -> void
+
+  # @dynamic name, requires
+  attr_reader name: String
+
+  attr_reader requires: Gem::Requirement?
+
+  # @rbs name: String
+  # @rbs requires: String?
+  # @rbs &block: (::Riffer::Rig::Registrar) -> void
+  # @rbs return: void
+  def initialize: (String name, ?requires: String?) { (::Riffer::Rig::Registrar) -> void } -> void
+
+  # @rbs registrar: Riffer::Rig::Registrar
+  # @rbs return: void
+  def run: (Riffer::Rig::Registrar registrar) -> void
+end

data/sig/generated/riffer/rig/hosts/base.rbs

@@ -0,0 +1,26 @@
+# Generated from lib/riffer/rig/hosts/base.rb with RBS::Inline
+
+class Riffer::Rig::Hosts::Base
+  # @rbs return: Set[Symbol]
+  def capabilities: () -> Set[Symbol]
+
+  # @rbs question: String?
+  # @rbs options: Array[String]?
+  # @rbs secret: bool
+  # @rbs return: String?
+  def ask: (?String? question, ?options: Array[String]?, ?secret: bool) -> String?
+
+  # @rbs question: String?
+  # @rbs return: bool
+  def confirm: (?String? question) -> bool
+
+  # @rbs message: String?
+  # @rbs level: Symbol
+  # @rbs return: void
+  def notify: (?String? message, ?level: Symbol) -> void
+
+  # @rbs label: String?
+  # @rbs &: ? () -> void
+  # @rbs return: void
+  def progress: (?String? label) ?{ () -> void } -> void
+end

data/sig/generated/riffer/rig/hosts/mirror.rbs

@@ -0,0 +1,41 @@
+# Generated from lib/riffer/rig/hosts/mirror.rb with RBS::Inline
+
+class Riffer::Rig::Hosts::Mirror < Riffer::Rig::Hosts::Base
+  @host: Riffer::Rig::Hosts::Base
+
+  @queue: Array[Riffer::Rig::Events::Event]
+
+  # @dynamic capabilities
+  attr_reader capabilities: Set[Symbol]
+
+  # @rbs host: Riffer::Rig::Hosts::Base
+  # @rbs return: void
+  def initialize: (Riffer::Rig::Hosts::Base host) -> void
+
+  # @rbs question: String?
+  # @rbs options: Array[String]?
+  # @rbs secret: bool
+  # @rbs return: String?
+  def ask: (?String? question, ?options: Array[String]?, ?secret: bool) -> String?
+
+  # @rbs question: String?
+  # @rbs return: bool
+  def confirm: (?String? question) -> bool
+
+  # @rbs message: String?
+  # @rbs level: Symbol
+  # @rbs return: void
+  def notify: (?String? message, ?level: Symbol) -> void
+
+  # @rbs label: String?
+  # @rbs &block: ? () -> void
+  # @rbs return: void
+  def progress: (?String? label) ?{ () -> void } -> void
+
+  # @rbs event: Riffer::Rig::Events::Event
+  # @rbs return: void
+  def queue: (Riffer::Rig::Events::Event event) -> void
+
+  # @rbs return: Array[Riffer::Rig::Events::Event]
+  def drain: () -> Array[Riffer::Rig::Events::Event]
+end

data/sig/generated/riffer/rig/hosts/null.rbs

@@ -0,0 +1,26 @@
+# Generated from lib/riffer/rig/hosts/null.rb with RBS::Inline
+
+class Riffer::Rig::Hosts::Null < Riffer::Rig::Hosts::Base
+  # @rbs return: Set[Symbol]
+  def capabilities: () -> Set[Symbol]
+
+  # @rbs question: String?
+  # @rbs options: Array[String]?
+  # @rbs secret: bool
+  # @rbs return: String?
+  def ask: (?String? question, ?options: Array[String]?, ?secret: bool) -> String?
+
+  # @rbs question: String?
+  # @rbs return: bool
+  def confirm: (?String? question) -> bool
+
+  # @rbs message: String?
+  # @rbs level: Symbol
+  # @rbs return: void
+  def notify: (?String? message, ?level: Symbol) -> void
+
+  # @rbs label: String?
+  # @rbs &block: ? () -> void
+  # @rbs return: void
+  def progress: (?String? label) ?{ () -> void } -> void
+end

data/sig/generated/riffer/rig/provider_setup/field.rbs

@@ -0,0 +1,22 @@
+# Generated from lib/riffer/rig/provider_setup/field.rb with RBS::Inline
+
+class Riffer::Rig::ProviderSetup::Field
+  # @dynamic name, env, secret, required, fallback
+  attr_reader name: Symbol
+
+  attr_reader env: Array[String]
+
+  attr_reader secret: bool
+
+  attr_reader required: bool
+
+  attr_reader fallback: (^() -> String?)?
+
+  # @rbs name: Symbol
+  # @rbs env: Array[String]
+  # @rbs secret: bool
+  # @rbs required: bool
+  # @rbs fallback: (^() -> String?)?
+  # @rbs return: void
+  def initialize: (name: Symbol, env: Array[String], secret: bool, required: bool, ?fallback: (^() -> String?)?) -> void
+end

data/sig/generated/riffer/rig/provider_setup.rbs

@@ -0,0 +1,30 @@
+# Generated from lib/riffer/rig/provider_setup.rb with RBS::Inline
+
+class Riffer::Rig::ProviderSetup
+  # @dynamic url, chain, fields
+  attr_reader url: String?
+
+  attr_reader chain: bool
+
+  attr_reader fields: Array[Riffer::Rig::ProviderSetup::Field]
+
+  # @rbs fields: Array[Riffer::Rig::ProviderSetup::Field]
+  # @rbs url: String?
+  # @rbs chain: bool
+  # @rbs return: void
+  def initialize: (fields: Array[Riffer::Rig::ProviderSetup::Field], ?url: String?, ?chain: bool) -> void
+
+  AWS_SHARED_CONFIG_REGION: ^() -> String?
+
+  # Upstream candidate: each riffer provider could declare its own credential
+  # fields, and this table would go.
+  TABLE: Hash[Symbol, Riffer::Rig::ProviderSetup]
+
+  # @rbs identifier: String | Symbol
+  # @rbs return: Riffer::Rig::ProviderSetup?
+  def self.[]: (String | Symbol identifier) -> Riffer::Rig::ProviderSetup?
+
+  # @rbs identifier: String | Symbol
+  # @rbs return: Riffer::Rig::ProviderSetup
+  def self.for: (String | Symbol identifier) -> Riffer::Rig::ProviderSetup
+end

data/sig/generated/riffer/rig/registrar.rbs

@@ -0,0 +1,25 @@
+# Generated from lib/riffer/rig/registrar.rb with RBS::Inline
+
+class Riffer::Rig::Registrar
+  @tools: Array[singleton(Riffer::Tool)]
+
+  @prompts: Hash[Symbol, ^(Riffer::Rig::Runtime) -> String?]
+
+  # @rbs return: void
+  def initialize: () -> void
+
+  # @rbs klass: singleton(Riffer::Tool)
+  # @rbs return: void
+  def tool: (singleton(Riffer::Tool) klass) -> void
+
+  # @rbs name: Symbol
+  # @rbs &block: (Riffer::Rig::Runtime) -> String?
+  # @rbs return: void
+  def prompt: (Symbol name) { (Riffer::Rig::Runtime) -> String? } -> void
+
+  # @rbs return: Array[singleton(Riffer::Tool)]
+  def tools: () -> Array[singleton(Riffer::Tool)]
+
+  # @rbs return: Hash[Symbol, ^(Riffer::Rig::Runtime) -> String?]
+  def prompts: () -> Hash[Symbol, ^(Riffer::Rig::Runtime) -> String?]
+end

data/sig/generated/riffer/rig/repl.rbs

@@ -5,9 +5,9 @@ class Riffer::Rig::REPL
 
   SKILL_COMMAND: Regexp
 
-  @output: untyped
+  @output: IO
 
-  @input: untyped
+  @input: IO
 
   @theme: Riffer::Rig::UI::Theme
 
@@ -21,16 +21,16 @@ class Riffer::Rig::REPL
 
   @agent: Riffer::Agent
 
-  # @rbs ?agent: Riffer::Agent
-  # @rbs ?renderer: Riffer::Rig::UI::Renderer
-  # @rbs input: untyped
-  # @rbs output: untyped
-  # @rbs ?theme: Riffer::Rig::UI::Theme
-  # @rbs ?animator: Riffer::Rig::UI::Animator
-  # @rbs ?smoother: Riffer::Rig::UI::Smoother
-  # @rbs ?cursor: Riffer::Rig::UI::Cursor
+  # @rbs agent: Riffer::Agent
+  # @rbs renderer: Riffer::Rig::UI::Renderer
+  # @rbs input: IO
+  # @rbs output: IO
+  # @rbs theme: Riffer::Rig::UI::Theme
+  # @rbs animator: Riffer::Rig::UI::Animator
+  # @rbs smoother: Riffer::Rig::UI::Smoother
+  # @rbs cursor: Riffer::Rig::UI::Cursor
   # @rbs return: void
-  def initialize: (agent: untyped, renderer: untyped, ?input: untyped, ?output: untyped, ?theme: untyped, ?animator: untyped, ?smoother: untyped, ?cursor: untyped) -> void
+  def initialize: (agent: Riffer::Agent, renderer: Riffer::Rig::UI::Renderer, ?input: IO, ?output: IO, ?theme: Riffer::Rig::UI::Theme, ?animator: Riffer::Rig::UI::Animator, ?smoother: Riffer::Rig::UI::Smoother, ?cursor: Riffer::Rig::UI::Cursor) -> void
 
   # @rbs return: Symbol
   def run: () -> Symbol
@@ -55,14 +55,15 @@ class Riffer::Rig::REPL
   # @rbs return: String
   def skill_block: (String name, String body) -> String
 
-  # Chrome lines that never pass through the renderer (errors, exit line) end
-  # their line with a newline but leave the renderer's tool-group state alone —
-  # they can only follow prose, never mid-group.
-  #
+  # @rbs &block: () -> String
   # @rbs return: void
-  def print_block: () ?{ (?) -> untyped } -> void
+  def print_block: () { () -> String } -> void
 
   # @rbs message: Riffer::Messages::Base
   # @rbs return: void
   def render_tool_result: (Riffer::Messages::Base message) -> void
+
+  # @rbs mode: Symbol
+  # @rbs return: void
+  def start_animator: (?Symbol mode) -> void
 end

data/sig/generated/riffer/rig/runtime/cancel_flag.rbs

@@ -0,0 +1,19 @@
+# Generated from lib/riffer/rig/runtime/cancel_flag.rb with RBS::Inline
+
+class Riffer::Rig::Runtime::CancelFlag
+  @mutex: Thread::Mutex
+
+  @set: bool
+
+  # @rbs return: void
+  def initialize: () -> void
+
+  # @rbs return: void
+  def set: () -> void
+
+  # @rbs return: void
+  def clear: () -> void
+
+  # @rbs return: bool
+  def set?: () -> bool
+end

data/sig/generated/riffer/rig/runtime.rbs

@@ -0,0 +1,120 @@
+# Generated from lib/riffer/rig/runtime.rb with RBS::Inline
+
+# Never renders, prints or reads the filesystem: anything a host needs is a
+# Runtime feature, so embedders get it too.
+class Riffer::Rig::Runtime
+  class BusyError < StandardError
+  end
+
+  class ClosedError < StandardError
+  end
+
+  BASE_PROMPT_TEMPLATE: ::String
+
+  DEFAULT_NAME: ::String
+
+  # Unlimited: riffer's own default (16) is too small for a general-purpose
+  # harness.
+  DEFAULT_MAX_STEPS: Integer?
+
+  INTERRUPT_CANCELLED: Symbol
+
+  @agent: Riffer::Agent
+
+  @base_prompt: String
+
+  @cancel_flag: Riffer::Rig::Runtime::CancelFlag
+
+  @credentials: Hash[Symbol, Hash[Symbol, String]]
+
+  @cwd: String
+
+  @host: Riffer::Rig::Hosts::Mirror
+
+  @id: String
+
+  @settings: Hash[Symbol, untyped]
+
+  @busy: bool
+
+  @closed: bool
+
+  @session_start_pending: bool
+
+  @registrar: Riffer::Rig::Registrar
+
+  # @dynamic agent, credentials, cwd, host, id, settings
+  attr_reader agent: Riffer::Agent
+
+  attr_reader credentials: Hash[Symbol, Hash[Symbol, String]]
+
+  attr_reader cwd: String
+
+  attr_reader host: Riffer::Rig::Hosts::Mirror
+
+  attr_reader id: String
+
+  attr_reader settings: Hash[Symbol, untyped]
+
+  # @rbs model: String
+  # @rbs extensions: Array[Riffer::Rig::Extension]
+  # @rbs tools: Array[String]?
+  # @rbs settings: Hash[Symbol, untyped]
+  # @rbs host: Riffer::Rig::Hosts::Base
+  # @rbs cwd: String?
+  # @rbs name: String
+  # @rbs instructions: String?
+  # @rbs credentials: Hash[Symbol, Hash[Symbol, String]]
+  # @rbs pricing: Hash[String, Riffer::Rig::Settings::Pricing]
+  # @rbs max_steps: Integer?
+  # @rbs snapshot: Hash[Symbol, untyped]?
+  # @rbs return: void
+  def initialize: (String model, ?extensions: Array[Riffer::Rig::Extension], ?tools: Array[String]?, ?settings: Hash[Symbol, untyped], ?host: Riffer::Rig::Hosts::Base, ?cwd: String?, ?name: String, ?instructions: String?, ?credentials: Hash[Symbol, Hash[Symbol, String]], ?pricing: Hash[String, Riffer::Rig::Settings::Pricing], ?max_steps: Integer?, ?snapshot: Hash[Symbol, untyped]?) -> void
+
+  # @rbs text: String
+  # @rbs &block: ?(::Riffer::StreamEvents::Base | Riffer::Rig::Events::Event) -> void
+  # @rbs return: (nil | Enumerator[::Riffer::StreamEvents::Base | Riffer::Rig::Events::Event, Riffer::Agent::Response])
+  def prompt: (String text) ?{ (::Riffer::StreamEvents::Base | Riffer::Rig::Events::Event) -> void } -> (nil | Enumerator[::Riffer::StreamEvents::Base | Riffer::Rig::Events::Event, Riffer::Agent::Response])
+
+  # @rbs text: String
+  # @rbs return: Riffer::Agent::Response
+  def ask: (String text) -> Riffer::Agent::Response
+
+  # @rbs return: nil
+  def cancel: () -> nil
+
+  # @rbs return: void
+  def close: () -> void
+
+  private
+
+  # @rbs stream: Enumerator[Riffer::StreamEvents::Base, Riffer::Agent::Response]
+  # @rbs return: Enumerator[::Riffer::StreamEvents::Base | Riffer::Rig::Events::Event, Riffer::Agent::Response]
+  def wrap_stream: (Enumerator[Riffer::StreamEvents::Base, Riffer::Agent::Response] stream) -> Enumerator[::Riffer::StreamEvents::Base | Riffer::Rig::Events::Event, Riffer::Agent::Response]
+
+  # @rbs return: void
+  def interrupt_if_cancelled: () -> void
+
+  # @rbs outcome: Riffer::Agent::Outcome
+  # @rbs return: Symbol
+  def stop_reason: (Riffer::Agent::Outcome outcome) -> Symbol
+
+  # @rbs extensions: Array[Riffer::Rig::Extension]
+  # @rbs return: Riffer::Rig::Registrar
+  def build_registrar: (Array[Riffer::Rig::Extension] extensions) -> Riffer::Rig::Registrar
+
+  # @rbs registered: Array[singleton(Riffer::Tool)]
+  # @rbs allowlist: Array[String]?
+  # @rbs return: Array[singleton(Riffer::Tool)]
+  def select_tools: (Array[singleton(Riffer::Tool)] registered, Array[String]? allowlist) -> Array[singleton(Riffer::Tool)]
+
+  # @rbs return: void
+  def refresh_system_message: () -> void
+
+  # @rbs return: Array[String]
+  def rendered_sections: () -> Array[String]
+
+  # @rbs sections: Array[String]
+  # @rbs return: String
+  def system_prompt: (Array[String] sections) -> String
+end

data/sig/generated/riffer/rig/settings/document.rbs

@@ -0,0 +1,24 @@
+# Generated from lib/riffer/rig/settings/document.rb with RBS::Inline
+
+class Riffer::Rig::Settings::Document
+  @model: String?
+
+  @reasoning: String?
+
+  @models: Hash[String, Riffer::Rig::Settings::Pricing]
+
+  @providers: Hash[String, Hash[String, String]]
+
+  # @dynamic model, reasoning, models, providers
+  attr_reader model: String?
+
+  attr_reader reasoning: String?
+
+  attr_reader models: Hash[String, Riffer::Rig::Settings::Pricing]
+
+  attr_reader providers: Hash[String, Hash[String, String]]
+
+  # @rbs source: untyped
+  # @rbs return: void
+  def initialize: (untyped source) -> void
+end

data/sig/generated/riffer/rig/settings/pricing.rbs

@@ -1,6 +1,5 @@
 # Generated from lib/riffer/rig/settings/pricing.rb with RBS::Inline
 
-# USD per million tokens, parsed from a model's entry in the settings file.
 class Riffer::Rig::Settings::Pricing
   # @dynamic input, output, cache_write, cache_read
   attr_reader input: Float
@@ -14,6 +13,10 @@ class Riffer::Rig::Settings::Pricing
   # @dynamic input, output, cache_write, cache_read
   attr_reader cache_read: Float
 
+  # @rbs entry: Hash[String, untyped]
+  # @rbs return: ::Riffer::Rig::Settings::Pricing
+  def self.from: (Hash[String, untyped] entry) -> ::Riffer::Rig::Settings::Pricing
+
   # @rbs input: Float
   # @rbs output: Float
   # @rbs cache_write: Float

data/sig/generated/riffer/rig/settings.rbs

@@ -1,34 +1,5 @@
 # Generated from lib/riffer/rig/settings.rb with RBS::Inline
 
-# Reads and writes the riffer-rig user settings file at
-# <tt>~/.riffer/settings.json</tt>.
-#
-# Example file:
-#
-#   {
-#     "model": "anthropic/claude-sonnet-4-6",
-#     "reasoning": "low",
-#     "models": {
-#       "anthropic/claude-sonnet-4-6": {
-#         "input": 3.0,
-#         "output": 15.0,
-#         "cache_write": 3.75,
-#         "cache_read": 0.3
-#       }
-#     }
-#   }
-#
-# All keys are optional. Missing pricing means cost display is suppressed.
-# <tt>"reasoning"</tt> is translated to the appropriate provider-specific
-# parameter on each API call; accepted values depend on the provider:
-#
-# - Anthropic: <tt>"low"</tt>, <tt>"medium"</tt>, <tt>"high"</tt>,
-#   <tt>"xhigh"</tt>, <tt>"max"</tt>
-# - OpenAI / OpenRouter: <tt>"low"</tt>, <tt>"medium"</tt>, <tt>"high"</tt>,
-#   <tt>"xhigh"</tt>
-#
-# Omitting the key (or supplying an unrecognised value) leaves the model's
-# default reasoning behaviour unchanged.
 module Riffer::Rig::Settings
   PATH: String
 
@@ -36,57 +7,65 @@ module Riffer::Rig::Settings
 
   REASONING_LEVELS_BY_PROVIDER: Hash[String, Array[String]]
 
-  # Returns the configured model string, or +DEFAULT_MODEL+ if not set.
-  #
   # @rbs path: String
   # @rbs return: String
   def model: (?path: String) -> String
 
-  # Returns model options for the configured model and reasoning level, ready
-  # to pass directly to the Riffer agent's +model_options+.
-  #
   # @rbs path: String
   # @rbs return: Hash[Symbol, untyped]
   def model_options: (?path: String) -> Hash[Symbol, untyped]
 
-  # Returns the provider prefix for +model_string+, e.g. <tt>"anthropic"</tt>
-  # for <tt>"anthropic/claude-sonnet-4-6"</tt>. Returns +nil+ if the model
-  # string contains no slash.
-  #
   # @rbs model_string: String
   # @rbs return: String?
   def provider_for: (String model_string) -> String?
 
-  # Returns the pricing for +model+, or +nil+ if not configured.
-  #
   # @rbs model: String
   # @rbs path: String
   # @rbs return: Pricing?
   def pricing_for: (String model, ?path: String) -> Pricing?
 
+  # @rbs identifier: String
+  # @rbs path: String
+  # @rbs return: Hash[String, String]
+  def provider_fields: (String identifier, ?path: String) -> Hash[String, String]
+
+  # @rbs identifier: String
+  # @rbs fields: Hash[String, String]
+  # @rbs path: String
+  # @rbs return: void
+  def store_provider: (String identifier, Hash[String, String] fields, ?path: String) -> void
+
+  # @rbs identifier: String
+  # @rbs path: String
+  # @rbs return: void
+  def remove_provider: (String identifier, ?path: String) -> void
+
   private
 
+  # @rbs path: String
+  # @rbs &: (Hash[String, Hash[String, String]]) -> Hash[String, Hash[String, String]]
+  # @rbs return: void
+  def update_providers: (String path) { (Hash[String, Hash[String, String]]) -> Hash[String, Hash[String, String]] } -> void
+
   # @rbs provider: String?
   # @rbs return: Hash[Symbol, untyped]
   def base_options: (String? provider) -> Hash[Symbol, untyped]
 
   # @rbs path: String
-  # @rbs ?provider: String?
+  # @rbs provider: String?
   # @rbs return: String?
-  def reasoning_for: (?path: String, ?provider: untyped) -> String?
+  def reasoning_for: (?path: String, ?provider: String?) -> String?
 
-  # Maps a reasoning level to the provider-specific model option hash expected
-  # by Riffer. Returns an empty hash when +level+ is +nil+.
-  #
   # @rbs level: String?
   # @rbs provider: String?
   # @rbs return: Hash[Symbol, untyped]
   def reasoning_options: (String? level, String? provider) -> Hash[Symbol, untyped]
 
-  # The parsed settings file, or an empty hash when the file is absent or
-  # malformed.
-  #
+  # @rbs path: String
+  # @rbs return: Document
+  def read: (String path) -> Document
+
   # @rbs path: String
   # @rbs return: Hash[String, untyped]
-  def read: (String path) -> Hash[String, untyped]
+  def read_source: (String path) -> Hash[String, untyped]
 end

data/sig/generated/riffer/rig/token_tally.rbs

@@ -1,16 +1,5 @@
 # Generated from lib/riffer/rig/token_tally.rb with RBS::Inline
 
-# Accumulates token usage across multiple turns in a session and computes an
-# estimated cost based on a pricing object sourced from
-# +Riffer::Rig::Settings+.
-#
-# Instantiate once per REPL session, call +add+ after each turn, then read
-# +total_tokens+ and +estimated_cost+ for display.
-#
-#   pricing = Riffer::Rig::Settings.pricing_for('anthropic/claude-sonnet-4-6')
-#   tally = Riffer::Rig::TokenTally.new(pricing: pricing)
-#   tally.add(usage)
-#   tally.estimated_cost  # => 0.000042  (USD), or nil if pricing is nil
 class Riffer::Rig::TokenTally
   @input_tokens: Integer
 
@@ -34,28 +23,20 @@ class Riffer::Rig::TokenTally
   # @dynamic input_tokens, output_tokens, cache_write_tokens, cache_read_tokens
   attr_reader cache_read_tokens: untyped
 
-  # @rbs ?pricing: Riffer::Rig::Settings::Pricing?
+  # @rbs pricing: Riffer::Rig::Settings::Pricing?
   # @rbs return: void
-  def initialize: (?pricing: untyped) -> void
+  def initialize: (?pricing: Riffer::Rig::Settings::Pricing?) -> void
 
-  # Accumulates token counts from a +Riffer::Providers::TokenUsage+ object.
-  #
   # @rbs usage: Riffer::Providers::TokenUsage
   # @rbs return: void
   def add: (Riffer::Providers::TokenUsage usage) -> void
 
-  # Returns the total token count across all categories.
-  #
   # @rbs return: Integer
   def total_tokens: () -> Integer
 
-  # Returns +true+ if any tokens have been counted.
-  #
   # @rbs return: bool
   def any?: () -> bool
 
-  # Returns the estimated cost in USD, or +nil+ if no pricing was provided.
-  #
   # @rbs return: Float?
   def estimated_cost: () -> Float?
 end

data/sig/generated/riffer/rig/tools/bash.rbs

@@ -5,18 +5,30 @@ class Riffer::Rig::Tools::Bash < Riffer::Tool
 
   MAX_OUTPUT_BYTES: Integer
 
+  POLL_SECONDS: Float
+
   # @rbs context: Riffer::Agent::Context?
   # @rbs command: String
-  # @rbs ?timeout_ms: Integer
+  # @rbs timeout_ms: Integer
   # @rbs return: Riffer::Tools::Response
-  def call: (context: Riffer::Agent::Context?, command: String, ?timeout_ms: untyped) -> Riffer::Tools::Response
+  def call: (context: Riffer::Agent::Context?, command: String, ?timeout_ms: Integer) -> Riffer::Tools::Response
 
   private
 
   # @rbs command: String
   # @rbs timeout_seconds: Float
+  # @rbs cancel_flag: Riffer::Rig::Runtime::CancelFlag?
   # @rbs return: [String, Integer]
-  def run: (String command, Float timeout_seconds) -> [ String, Integer ]
+  def run: (String command, Float timeout_seconds, Riffer::Rig::Runtime::CancelFlag? cancel_flag) -> [ String, Integer ]
+
+  # @rbs wait_thread: Process::Waiter
+  # @rbs deadline: Float
+  # @rbs cancel_flag: Riffer::Rig::Runtime::CancelFlag?
+  # @rbs return: Symbol
+  def await: (Process::Waiter wait_thread, Float deadline, Riffer::Rig::Runtime::CancelFlag? cancel_flag) -> Symbol
+
+  # @rbs return: Float
+  def monotonic_now: () -> Float
 
   # @rbs pid: Integer
   # @rbs return: Integer