riffer-rig 0.3.0 → 0.5.0

data/lib/riffer/rig/ui/smoother.rb

@@ -17,7 +17,7 @@ class Riffer::Rig::UI::Smoother
 
   MIN_CHARS_PER_TICK = Rational(1, 6) #: Rational
 
-  # @rbs @io: untyped
+  # @rbs @io: IO
   # @rbs @theme: Riffer::Rig::UI::Theme
   # @rbs @clock: singleton(Kernel)
   # @rbs @backlog: String
@@ -25,9 +25,10 @@ class Riffer::Rig::UI::Smoother
   # @rbs @mutex: Mutex
   # @rbs @thread: Thread?
   # @rbs @stop: bool
+  # @rbs @newline_pending: bool
 
-  # @rbs io: untyped
-  # @rbs ?theme: Riffer::Rig::UI::Theme
+  # @rbs io: IO
+  # @rbs theme: Riffer::Rig::UI::Theme
   # @rbs clock: singleton(Kernel)
   # @rbs return: void
   def initialize(io: $stdout, theme: Riffer::Rig::UI::Theme.for(io), clock: Kernel)
@@ -39,6 +40,7 @@ class Riffer::Rig::UI::Smoother
     @mutex = Mutex.new
     @thread = nil
     @stop = false
+    @newline_pending = false
   end
 
   # @rbs return: void
@@ -58,11 +60,14 @@ class Riffer::Rig::UI::Smoother
   # @rbs content: String
   # @rbs return: self
   def <<(content)
-    if enabled?
-      @mutex.synchronize { @backlog << content }
-    else
-      @io.print(content)
-      @io.flush
+    @mutex.synchronize do
+      if enabled?
+        @backlog << content
+      else
+        @io.print(content)
+        @io.flush
+      end
+      @newline_pending = !content.end_with?("\n")
     end
     self
   end
@@ -79,19 +84,30 @@ class Riffer::Rig::UI::Smoother
       return if count.zero?
 
       @carry -= count
-      @io.print(@backlog.slice!(0, count))
+      chunk = @backlog.slice!(0, count) || ''
+      @newline_pending = !chunk.end_with?("\n")
+      @io.print(chunk)
       @io.flush
     end
   end
 
-  # Called before non-delta output so printed order matches stream order.
+  # Called before non-prose output so printed order matches stream order. The
+  # backlog is flushed, then any partial prose block the model left without a
+  # trailing newline gets one, so block spacing never depends on the model's
+  # last character.
   #
   # @rbs return: void
   def drain
     @mutex.synchronize do
-      return if @backlog.empty?
+      unless @backlog.empty?
+        backlog = @backlog.slice!(0, @backlog.length) || ''
+        @newline_pending = !backlog.end_with?("\n")
+        @io.print(backlog)
+      end
+      return unless @newline_pending
 
-      @io.print(@backlog.slice!(0, @backlog.length))
+      @newline_pending = false
+      @io.print("\n")
       @io.flush
     end
   end
@@ -113,6 +129,6 @@ class Riffer::Rig::UI::Smoother
 
   # @rbs return: bool
   def enabled?
-    @theme.enabled && @io.respond_to?(:tty?) && @io.tty?
+    @theme.enabled && @io.tty?
   end
 end

data/lib/riffer/rig/ui/theme.rb

@@ -15,10 +15,10 @@ class Riffer::Rig::UI::Theme
     @enabled = enabled
   end
 
-  # @rbs io: untyped
+  # @rbs io: IO
   # @rbs return: Riffer::Rig::UI::Theme
   def self.for(io)
-    new(enabled: io.respond_to?(:tty?) && io.tty? && !ENV.key?('NO_COLOR'))
+    new(enabled: io.tty? && !ENV.key?('NO_COLOR'))
   end
 
   # @rbs text: String

data/lib/riffer/rig/version.rb

@@ -2,6 +2,6 @@
 
 module Riffer
   module Rig
-    VERSION = '0.3.0'
+    VERSION = '0.5.0'
   end
 end

data/lib/riffer/rig.rb

@@ -8,3 +8,34 @@ loader = Zeitwerk::Loader.for_gem_extension(Riffer)
 loader.ignore("#{__dir__}/rig/version.rb")
 loader.inflector.inflect('cli' => 'CLI', 'repl' => 'REPL', 'ui' => 'UI')
 loader.setup
+
+module Riffer::Rig
+  # @rbs self.@extensions: Hash[String, Riffer::Rig::Extension]
+  @extensions = {} #: Hash[String, Riffer::Rig::Extension]
+
+  # Records an extension block in the process registry and returns the
+  # extension object. Re-recording a name replaces the block.
+  #
+  #   Riffer::Rig.extension('git') do |rig|
+  #     rig.tool GitLog
+  #   end
+  #
+  # @rbs name: String
+  # @rbs requires: String?
+  # @rbs &block: (::Riffer::Rig::Registrar) -> void
+  # @rbs return: Riffer::Rig::Extension
+  def self.extension(name, requires: nil, &)
+    ext = Extension.new(name, requires: requires, &)
+    if ext.requires && !ext.requires.satisfied_by?(Gem::Version.new(Riffer::Rig::VERSION))
+      raise Riffer::ArgumentError,
+            "extension #{name} requires riffer-rig #{ext.requires}, found #{Riffer::Rig::VERSION}"
+    end
+
+    @extensions[name] = ext
+  end
+
+  # @rbs return: Hash[String, Riffer::Rig::Extension]
+  def self.extensions
+    @extensions.dup
+  end
+end

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

@@ -3,10 +3,10 @@
 module Riffer::Rig::CLI
   PROVIDER_URLS: Hash[String, String]
 
-  # @rbs output: untyped
-  # @rbs input: untyped
+  # @rbs output: IO
+  # @rbs input: IO
   # @rbs return: Integer
-  def start: (?output: untyped, ?input: untyped) -> Integer
+  def start: (?output: IO, ?input: IO) -> Integer
 
   private
 
@@ -17,18 +17,18 @@ module Riffer::Rig::CLI
 
   # @rbs provider: String?
   # @rbs theme: Riffer::Rig::UI::Theme
-  # @rbs output: untyped
-  # @rbs input: untyped
+  # @rbs output: IO
+  # @rbs input: IO
   # @rbs return: String?
-  def onboard: (String? provider, Riffer::Rig::UI::Theme theme, output: untyped, input: untyped) -> String?
+  def onboard: (String? provider, Riffer::Rig::UI::Theme theme, output: IO, input: IO) -> String?
 
   # @rbs provider: String?
   # @rbs return: String
   def env_var_for: (String? provider) -> String
 
-  # @rbs input: untyped
+  # @rbs input: IO
   # @rbs return: String?
-  def read_secret: (untyped input) -> String?
+  def read_secret: (IO input) -> String?
 
   # @rbs theme: Riffer::Rig::UI::Theme
   # @rbs animator: Riffer::Rig::UI::Animator

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

@@ -47,9 +47,10 @@ module Riffer::Rig::Credentials
   def key_from_file: (String? provider, String path) -> String?
 
   # The parsed auth file, or an empty hash when the file is absent or
-  # malformed.
+  # malformed. Non-string values can't be keys, so they're dropped here rather
+  # than re-checked by every reader.
   #
   # @rbs path: String
-  # @rbs return: Hash[String, untyped]
-  def read: (String path) -> Hash[String, untyped]
+  # @rbs return: Hash[String, String]
+  def read: (String path) -> Hash[String, String]
 end

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

@@ -0,0 +1,28 @@
+# Generated from lib/riffer/rig/extension.rb with RBS::Inline
+
+# An extension recorded by <tt>Riffer::Rig.extension</tt>: a name, an optional
+# Gem::Requirement on riffer-rig, and the block that runs against a Runtime's
+# registrar. The process registry holds these objects; Runtime.new runs their
+# blocks.
+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/host.rbs

@@ -0,0 +1,42 @@
+# Generated from lib/riffer/rig/host.rb with RBS::Inline
+
+# A host receives questions and out-of-band messages from a Runtime. It is
+# duck-typed: any object with #ask, #confirm, #notify, #progress and
+# #capabilities is a host.
+#
+#   class HeadlessHost
+#     def capabilities = Set[:notify, :progress].freeze
+#     def ask(*) = nil
+#     def confirm(*) = false
+#     def notify(message, level:) = warn("[#{level}] #{message}")
+#     def progress(label) = yield
+#   end
+#
+# Callers check #capabilities before asking, so a host that declines a
+# capability is never asked and the caller takes the declined path.
+# The do-nothing host: the default +host:+ for a Runtime. Every capability is
+# declined, so callers take the declined path without any UI attempt.
+class Riffer::Rig::Host
+  # @rbs return: Set[Symbol]
+  def capabilities: () -> Set[Symbol]
+
+  # @rbs _question: String?
+  # @rbs options: Array[String]?
+  # @rbs secret: bool
+  # @rbs return: String?
+  def ask: (?untyped _question, ?options: Array[String]?, ?secret: bool) -> String?
+
+  # @rbs _question: String?
+  # @rbs return: bool
+  def confirm: (?untyped _question) -> bool
+
+  # @rbs _message: String?
+  # @rbs level: Symbol
+  # @rbs return: void
+  def notify: (?untyped _message, ?level: Symbol) -> void
+
+  # @rbs _label: String?
+  # @rbs &block: ^() -> void
+  # @rbs return: void
+  def progress: (?untyped _label) ?{ (?) -> untyped } -> void
+end

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

@@ -0,0 +1,17 @@
+# Generated from lib/riffer/rig/registrar.rb with RBS::Inline
+
+# The per-Runtime registrar extension blocks run against. Seams are added one
+# at a time; today only #tool exists.
+class Riffer::Rig::Registrar
+  @tools: Array[singleton(Riffer::Tool)]
+
+  # @rbs return: void
+  def initialize: () -> void
+
+  # @rbs klass: singleton(Riffer::Tool)
+  # @rbs return: void
+  def tool: (singleton(Riffer::Tool) klass) -> void
+
+  # @rbs return: Array[singleton(Riffer::Tool)]
+  def tools: () -> Array[singleton(Riffer::Tool)]
+end

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

@@ -5,11 +5,9 @@ class Riffer::Rig::REPL
 
   SKILL_COMMAND: Regexp
 
-  @pending_tool_calls: Array[Riffer::StreamEvents::ToolCallDone]
+  @output: IO
 
-  @output: untyped
-
-  @input: untyped
+  @input: IO
 
   @theme: Riffer::Rig::UI::Theme
 
@@ -23,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
@@ -57,20 +55,23 @@ class Riffer::Rig::REPL
   # @rbs return: String
   def skill_block: (String name, String body) -> String
 
-  # Tool call lines print below their round's stats line to reflect the actual
-  # timeline (usage is reported when the model stream closes, after the calls
-  # were decided). ToolCallDone events are therefore held until the stats line
-  # or the next event forces them out. Callers must have the spinner stopped —
-  # the flush renders between their existing stop/start pair.
+  # 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 flush_pending_tool_calls: () -> void
+  def print_block: () { () -> String } -> void
 
-  # Tool results can land mid-animation (tool execution emits no stream events,
-  # so the indicator is up); stop it around the line so its next frame doesn't
-  # erase what we printed.
-  #
   # @rbs message: Riffer::Messages::Base
   # @rbs return: void
   def render_tool_result: (Riffer::Messages::Base message) -> void
+
+  # Every animator start sits behind a smoother drain: the smoother's backlog
+  # may still be trickling out from streamed prose, and spinner frames drawn
+  # mid-drain would carve `\r…\e[K` through a half-printed sentence.
+  #
+  # @rbs mode: Symbol
+  # @rbs return: void
+  def start_animator: (?Symbol mode) -> void
 end

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

@@ -0,0 +1,101 @@
+# Generated from lib/riffer/rig/runtime.rb with RBS::Inline
+
+# One Runtime builds its own Riffer::Agent from a per-instance
+# Riffer::Agent::Config and runs a prompt: streamed as events, or run to
+# completion and returned as riffer's Agent::Response.
+#
+#   runtime = Riffer::Rig::Runtime.new('mock/x')
+#   runtime.prompt('hello') { |event| ... }   # yields riffer StreamEvents
+#   runtime.prompt('hello').each { |event| }  # an Enumerator without a block
+#   response = runtime.ask('hello')           # a Riffer::Agent::Response
+#
+# Two Runtimes in one process share nothing but the process-wide extension
+# registry and riffer's provider repository. One prompt runs at a time; a
+# second while one is running raises Riffer::Rig::Runtime::BusyError. The
+# Runtime never renders, never prints, never reads the filesystem.
+class Riffer::Rig::Runtime
+  # Raised when a second prompt, ask or registrar build runs while one is
+  # already running on this Runtime.
+  class BusyError < StandardError
+  end
+
+  BASE_PROMPT_TEMPLATE: ::String
+
+  DEFAULT_NAME: ::String
+
+  # The legacy CodingAgent's default: an unlimited agent loop. riffer's own
+  # default (16) is too small for a general-purpose harness.
+  DEFAULT_MAX_STEPS: Integer?
+
+  interface _Host
+    def ask: (?String, ?options: Array[String]?, ?secret: bool) -> String?
+
+    def confirm: (?String) -> bool
+
+    def notify: (?String, ?level: Symbol) -> void
+
+    def progress: (?String) { () -> void } -> void
+
+    def capabilities: () -> Set[Symbol]
+  end
+
+  @registrar: Riffer::Rig::Registrar
+
+  @busy: bool
+
+  @settings: Hash[Symbol, untyped]
+
+  @host: _Host
+
+  @cwd: String
+
+  @agent: Riffer::Agent
+
+  # @dynamic agent, cwd, host, settings
+  attr_reader agent: Riffer::Agent
+
+  attr_reader cwd: String
+
+  attr_reader host: _Host
+
+  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: _Host
+  # @rbs cwd: String?
+  # @rbs name: String
+  # @rbs instructions: String?
+  # @rbs credentials: Hash[String, 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: _Host, ?cwd: String?, ?name: String, ?instructions: String?, ?credentials: Hash[String, String], ?pricing: Hash[String, Riffer::Rig::Settings::Pricing], ?max_steps: Integer?, ?snapshot: Hash[Symbol, untyped]?) -> void
+
+  # @rbs text: String
+  # @rbs &block: ?(Riffer::StreamEvents::Base) -> void
+  # @rbs return: (nil | Enumerator[Riffer::StreamEvents::Base, void])
+  def prompt: (String text) ?{ (Riffer::StreamEvents::Base) -> void } -> (nil | Enumerator[Riffer::StreamEvents::Base, void])
+
+  # @rbs text: String
+  # @rbs return: Riffer::Agent::Response
+  def ask: (String text) -> Riffer::Agent::Response
+
+  private
+
+  # @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 base_prompt: String
+  # @rbs return: String
+  def system_prompt: (String base_prompt) -> String
+end

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

@@ -0,0 +1,20 @@
+# 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]
+
+  # @dynamic model, reasoning, models
+  attr_reader model: String?
+
+  attr_reader reasoning: String?
+
+  attr_reader models: Hash[String, Riffer::Rig::Settings::Pricing]
+
+  # @rbs source: untyped
+  # @rbs return: void
+  def initialize: (untyped source) -> void
+end

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

@@ -14,6 +14,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

@@ -71,9 +71,9 @@ module Riffer::Rig::Settings
   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+.
@@ -83,10 +83,10 @@ module Riffer::Rig::Settings
   # @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
+  # The parsed settings file, or an empty document when the file is absent or
   # malformed.
   #
   # @rbs path: String
-  # @rbs return: Hash[String, untyped]
-  def read: (String path) -> Hash[String, untyped]
+  # @rbs return: Document
+  def read: (String path) -> Document
 end

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

@@ -34,9 +34,9 @@ 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.
   #

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

@@ -7,9 +7,9 @@ class Riffer::Rig::Tools::Bash < Riffer::Tool
 
   # @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
 

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

@@ -5,7 +5,7 @@ class Riffer::Rig::Tools::Edit < Riffer::Tool
   # @rbs path: String
   # @rbs old_string: String
   # @rbs new_string: String
-  # @rbs ?replace_all: bool
+  # @rbs replace_all: bool
   # @rbs return: Riffer::Tools::Response
-  def call: (context: Riffer::Agent::Context?, path: String, old_string: String, new_string: String, ?replace_all: untyped) -> Riffer::Tools::Response
+  def call: (context: Riffer::Agent::Context?, path: String, old_string: String, new_string: String, ?replace_all: bool) -> Riffer::Tools::Response
 end

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

@@ -5,8 +5,8 @@ class Riffer::Rig::Tools::Read < Riffer::Tool
 
   # @rbs context: Riffer::Agent::Context?
   # @rbs path: String
-  # @rbs ?offset: Integer?
-  # @rbs ?limit: Integer
+  # @rbs offset: Integer?
+  # @rbs limit: Integer
   # @rbs return: Riffer::Tools::Response
-  def call: (context: Riffer::Agent::Context?, path: String, ?offset: untyped, ?limit: untyped) -> Riffer::Tools::Response
+  def call: (context: Riffer::Agent::Context?, path: String, ?offset: Integer?, ?limit: Integer) -> Riffer::Tools::Response
 end

data/sig/generated/riffer/rig/ui/animator.rbs

@@ -27,12 +27,12 @@ class Riffer::Rig::UI::Animator
 
   @theme: Riffer::Rig::UI::Theme
 
-  @io: untyped
+  @io: IO
 
-  # @rbs io: untyped
-  # @rbs ?theme: Riffer::Rig::UI::Theme
+  # @rbs io: IO
+  # @rbs theme: Riffer::Rig::UI::Theme
   # @rbs return: void
-  def initialize: (?io: untyped, ?theme: untyped) -> void
+  def initialize: (?io: IO, ?theme: Riffer::Rig::UI::Theme) -> void
 
   # When not on a TTY, prints the final frame once instead of animating.
   #
@@ -40,17 +40,17 @@ class Riffer::Rig::UI::Animator
   # @rbs return: void
   def reveal: (Array[Array[String]] frames) -> void
 
-  # @rbs ?mode: Symbol
+  # @rbs mode: Symbol
   # @rbs return: void
-  def start: (?untyped mode) -> void
+  def start: (?Symbol mode) -> void
 
   # @rbs return: void
   def stop: () -> void
 
   # @rbs tick: Integer
-  # @rbs ?label: String
+  # @rbs label: String
   # @rbs return: String
-  def equalizer: (Integer tick, ?untyped label) -> String
+  def equalizer: (Integer tick, ?String label) -> String
 
   private
 

data/sig/generated/riffer/rig/ui/cursor.rbs

@@ -7,14 +7,14 @@ class Riffer::Rig::UI::Cursor
 
   SHOW: String
 
-  @io: untyped
+  @io: IO
 
   @theme: Riffer::Rig::UI::Theme
 
-  # @rbs io: untyped
-  # @rbs ?theme: Riffer::Rig::UI::Theme
+  # @rbs io: IO
+  # @rbs theme: Riffer::Rig::UI::Theme
   # @rbs return: void
-  def initialize: (?io: untyped, ?theme: untyped) -> void
+  def initialize: (?io: IO, ?theme: Riffer::Rig::UI::Theme) -> void
 
   # @rbs return: void
   def hide: () -> void