class Ollama::Handlers::Print

def call(response)
  if content = response.response || response.message&.content
    @output.print content
  end
  response.done and @output.puts
  self
end

def initialize(output: $stdout)
  super
  @output.sync = true
end