ollama-ruby 1.2.1 → 1.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0b6a67a947af787d3a67180531d7afb1390a009adf38616b55f32bc6c9301942
-  data.tar.gz: e7fb09439a06ede0a240f42f64048e88ac2d295fa50a8d3bc59dfed3bdb98b3a
+  metadata.gz: e26fc8b8ae9847d35f0422d1b4d8915c496b3a0d96b8ac202b7928c0f9ae7bd0
+  data.tar.gz: 802709860c8611397faf70b4f4fa7af41a0bebc2c5a642d97163b6337e6326e1
 SHA512:
-  metadata.gz: acc5e14fca8df34795a2cca585ff56157023b14dbd53c10995bf1751a39d2f137dbb9c5a26bbb2c58dc7a2ac1402d56ceea37eaf744189a5fc191f3532a0e57b
-  data.tar.gz: b8f0357fa9734eed66dd7b9013df7f6e464d3a0af7bd00acd11552dfffb15951b18d12bd2ab7c2c45736a735f4ee4f9ae2900d96415cf30bf9ff6a401e4dd3f8
+  metadata.gz: 26c44972749476a2fe6af10c97f7c18482d0ecca058e4ef3b416ec7e019d0109d6f5d24f7266cb37ddb420f59cd2c939a21bed44ab5db0180b1988bc4a8350de
+  data.tar.gz: 6be99709d688a678a4d6bbb12501cb0d758a5e794f76cb8310ce315c7956c53af456e8148bf05962fd4a8801c9f17f06187d84631182d03609920870aa8a84d4

data/CHANGES.md

@@ -1,5 +1,12 @@
 # Changes
 
+## 2025-07-06 v1.3.0
+
+* Added toggleable streaming in Markdown handler:
+  * Added conditional handler initialization
+  * Implemented toggleable streaming in Markdown handler
+  * Tested non-streaming (`stream: false`) behavior
+
 ## 2025-06-02 v1.2.1
 
 * Added thinking mode option to CLI:

data/bin/ollama_cli

@@ -8,24 +8,26 @@ require 'tins/xt/secure_write'
 require 'json'
 require 'tmpdir'
 
-class ChatStart
-  include Ollama::Handlers::Concern
-
-  def initialize(output: $stdout)
-    super
-    @output.sync = true
-    @content     = ''
-  end
+module Ollama::Handlers
+  class ChatStart
+    include Ollama::Handlers::Concern
+
+    def initialize(output: $stdout)
+      super
+      @output.sync = true
+      @content     = ''
+    end
 
-  attr_reader :content
+    attr_reader :content
 
-  def call(response)
-    if content = response.response
-      @content << content
-      @output << content
+    def call(response)
+      if content = response.response
+        @content << content
+        @output << content
+      end
+      response.done and @output.puts
+      self
     end
-    response.done and @output.puts
-    self
   end
 end
 
@@ -115,8 +117,15 @@ if ENV['DEBUG'].to_i == 1
   EOT
 end
 
-handler = Object.const_get(opts[?H])
-handler == ChatStart and handler = handler.new
+handler = Ollama::Handlers.const_get(opts[?H])
+handler = case
+          when handler == Ollama::Handlers::ChatStart
+            handler.new
+          when handler == Ollama::Handlers::Markdown
+            handler.new(stream: !!opts[?S])
+          else
+            handler
+          end
 
 Client.new(base_url:, read_timeout: 120).generate(
   model:,
@@ -128,7 +137,7 @@ Client.new(base_url:, read_timeout: 120).generate(
   &handler
 )
 
-if handler.is_a?(ChatStart)
+if handler.is_a?(Ollama::Handlers::ChatStart)
   filename = File.join(Dir.tmpdir, 'chat_start_%u.json' % $$)
   File.secure_write(filename) do |out|
     JSON.dump(

data/lib/ollama/handlers/markdown.rb

@@ -5,17 +5,23 @@ class Ollama::Handlers::Markdown
   include Ollama::Handlers::Concern
   include Term::ANSIColor
 
-  def initialize(output: $stdout)
-    super
-    @output.sync = true
+  def initialize(output: $stdout, stream: true)
+    super(output:)
+    @stream      = stream
+    @output.sync = @stream
     @content     = ''
   end
 
   def call(response)
     if content = response.response || response.message&.content
-      @content << content
-      markdown_content = Kramdown::ANSI.parse(@content)
-      @output.print clear_screen, move_home, markdown_content
+      if @stream
+        @content << content
+        markdown_content = Kramdown::ANSI.parse(@content)
+        @output.print clear_screen, move_home, markdown_content
+      else
+        markdown_content = Kramdown::ANSI.parse(content)
+        @output.print markdown_content
+      end
     end
     self
   end

data/lib/ollama/version.rb

@@ -1,6 +1,6 @@
 module Ollama
   # Ollama version
-  VERSION         = '1.2.1'
+  VERSION         = '1.3.0'
   VERSION_ARRAY   = VERSION.split('.').map(&:to_i) # :nodoc:
   VERSION_MAJOR   = VERSION_ARRAY[0] # :nodoc:
   VERSION_MINOR   = VERSION_ARRAY[1] # :nodoc:

data/ollama-ruby.gemspec

@@ -1,9 +1,9 @@
 # -*- encoding: utf-8 -*-
-# stub: ollama-ruby 1.2.1 ruby lib
+# stub: ollama-ruby 1.3.0 ruby lib
 
 Gem::Specification.new do |s|
   s.name = "ollama-ruby".freeze
-  s.version = "1.2.1".freeze
+  s.version = "1.3.0".freeze
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0".freeze) if s.respond_to? :required_rubygems_version=
   s.require_paths = ["lib".freeze]
@@ -18,7 +18,7 @@ Gem::Specification.new do |s|
   s.licenses = ["MIT".freeze]
   s.rdoc_options = ["--title".freeze, "Ollama-ruby - Interacting with the Ollama API".freeze, "--main".freeze, "README.md".freeze]
   s.required_ruby_version = Gem::Requirement.new("~> 3.1".freeze)
-  s.rubygems_version = "3.6.7".freeze
+  s.rubygems_version = "3.6.9".freeze
   s.summary = "Interacting with the Ollama API".freeze
   s.test_files = ["spec/ollama/client/doc_spec.rb".freeze, "spec/ollama/client_spec.rb".freeze, "spec/ollama/commands/chat_spec.rb".freeze, "spec/ollama/commands/copy_spec.rb".freeze, "spec/ollama/commands/create_spec.rb".freeze, "spec/ollama/commands/delete_spec.rb".freeze, "spec/ollama/commands/embed_spec.rb".freeze, "spec/ollama/commands/embeddings_spec.rb".freeze, "spec/ollama/commands/generate_spec.rb".freeze, "spec/ollama/commands/ps_spec.rb".freeze, "spec/ollama/commands/pull_spec.rb".freeze, "spec/ollama/commands/push_spec.rb".freeze, "spec/ollama/commands/show_spec.rb".freeze, "spec/ollama/commands/tags_spec.rb".freeze, "spec/ollama/commands/version_spec.rb".freeze, "spec/ollama/handlers/collector_spec.rb".freeze, "spec/ollama/handlers/dump_json_spec.rb".freeze, "spec/ollama/handlers/dump_yaml_spec.rb".freeze, "spec/ollama/handlers/markdown_spec.rb".freeze, "spec/ollama/handlers/nop_spec.rb".freeze, "spec/ollama/handlers/print_spec.rb".freeze, "spec/ollama/handlers/progress_spec.rb".freeze, "spec/ollama/handlers/say_spec.rb".freeze, "spec/ollama/handlers/single_spec.rb".freeze, "spec/ollama/image_spec.rb".freeze, "spec/ollama/message_spec.rb".freeze, "spec/ollama/options_spec.rb".freeze, "spec/ollama/tool_spec.rb".freeze, "spec/spec_helper.rb".freeze]
 

data/spec/ollama/handlers/markdown_spec.rb

@@ -41,4 +41,14 @@ RSpec.describe Ollama::Handlers::Markdown do
     response = double('response', response: nil, message: nil, done: true)
     markdown.call(response)
   end
+
+  it 'displays markdown response in one go when stream is false' do
+    output = double('output', :sync= => false)
+    expect(output).to receive(:print).with(ansi).once
+    markdown = described_class.new(output:, stream: false)
+    response = double('response', response: md, done: true)
+    markdown.call(response)
+    response = double('response', response: nil, message: nil, done: true)
+    markdown.call(response)
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ollama-ruby
 version: !ruby/object:Gem::Version
-  version: 1.2.1
+  version: 1.3.0
 platform: ruby
 authors:
 - Florian Frank
@@ -362,7 +362,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.7
+rubygems_version: 3.6.9
 specification_version: 4
 summary: Interacting with the Ollama API
 test_files: