ollama-ruby 1.4.0 → 1.6.0

data/lib/ollama/options.rb

@@ -1,7 +1,21 @@
 require 'ollama/json_loader'
 
-# Options are explained in the parameters for the modelfile:
-# https://github.com/ollama/ollama/blob/main/docs/modelfile.md#parameter
+# A class that encapsulates configuration options for Ollama models.
+#
+# This class provides a structured way to define and manage various parameters
+# that can be passed to Ollama models during generation or chat operations. It
+# includes type validation to ensure that option values conform to expected
+# data types, making it easier to work with model configurations
+# programmatically.
+#
+# @example Creating an Options object with specific settings
+#   options = Ollama::Options.new(
+#     temperature: 0.7,
+#     num_ctx: 8192,
+#     top_p: 0.9
+#   )
+#
+# [Options are explained in the parameters for the modelfile.](https://github.com/ollama/ollama/blob/main/docs/modelfile.md#parameter)
 class Ollama::Options
   include Ollama::DTO
   extend Ollama::JSONLoader
@@ -69,6 +83,17 @@ class Ollama::Options
     end
   }
 
+  # The [] method creates a new instance of the class using a hash of attributes.
+  #
+  # This class method provides a convenient way to instantiate an object by
+  # passing a hash containing the desired attribute values. It converts the
+  # hash keys to symbols and forwards them as keyword arguments to the
+  # constructor.
+  #
+  # @param value [ Hash ] a hash containing the attribute names and their values
+  #
+  # @return [ self ] a new instance of the class initialized with the provided
+  # attributes
   def self.[](value)
     new(**value.to_h)
   end

data/lib/ollama/response.rb

@@ -1,4 +1,21 @@
+# A subclass of JSON::GenericObject that represents responses from the Ollama API.
+#
+# This class serves as a specialized response object that extends
+# JSON::GenericObject to provide structured access to API response data. It
+# maintains the ability to convert to JSON format while preserving the response
+# data in a hash-like structure.
+#
+# @example Accessing response data
+#   response = Ollama::Response.new(key: 'value')
+#   response[:key] # => 'value'
 class Ollama::Response < JSON::GenericObject
+  # The as_json method converts the object's attributes into a JSON-compatible hash.
+  #
+  # This method gathers all defined attributes of the object and constructs a
+  # hash representation, excluding any nil values or empty collections.
+  #
+  # @note This removes "json_class" attribute from hash for responses.
+  # @return [ Hash ] a hash containing the object's non-nil and non-empty attributes
   def as_json(*)
     to_hash
   end

data/lib/ollama/tool/function/parameters/property.rb

@@ -1,8 +1,48 @@
+# A class that represents a single property within the parameters specification
+# for a tool function.
+#
+# This class encapsulates the definition of an individual parameter, including
+# its data type, descriptive text, and optional enumeration values that define
+# valid inputs.
+#
+# @example Creating a property with type and description
+#   property = Ollama::Tool::Function::Parameters::Property.new(
+#     type: 'string',
+#     description: 'The location to get weather for'
+#   )
+#
+# @example Creating a property with an enumeration of valid values
+#   property = Ollama::Tool::Function::Parameters::Property.new(
+#     type: 'string',
+#     description: 'Temperature unit',
+#     enum: %w[celsius fahrenheit]
+#   )
 class Ollama::Tool::Function::Parameters::Property
   include Ollama::DTO
 
-  attr_reader :type, :description, :enum
+  # The type attribute reader returns the type associated with the object.
+  #
+  # @return [ String ] the type value stored in the instance variable
+  attr_reader :type
 
+  # The description attribute reader returns the description associated with
+  # the object.
+  #
+  # @return [ String ] the description value stored in the instance variable
+  attr_reader :description
+
+  # The enum attribute reader returns the enumeration values associated with the object.
+  #
+  # @return [ Array<String>, nil ] an array of valid string values that the
+  # property can take, or nil if not set
+  attr_reader :enum
+
+  # The initialize method sets up a new Property instance with the specified
+  # attributes.
+  #
+  # @param type [ String ] the data type of the property
+  # @param description [ String ] a detailed explanation of what the property represents
+  # @param enum [ Array<String>, nil ] an optional array of valid values that the property can take
   def initialize(type:, description:, enum: nil)
     @type, @description, @enum = type, description, Array(enum)
   end

data/lib/ollama/tool/function/parameters.rb

@@ -1,8 +1,47 @@
+# A class that represents the parameters specification for a tool function in
+# Ollama API interactions.
+#
+# This class encapsulates the structure required for defining the parameters
+# that a function tool accepts. It includes the type of parameter object, the
+# properties of each parameter, and which parameters are required.
+#
+# @example Creating a parameters specification for a tool function
+#   property = Ollama::Tool::Function::Parameters::Property.new(
+#     type: 'string',
+#     description: 'The location to get weather for'
+#   )
+#   parameters = Ollama::Tool::Function::Parameters.new(
+#     type: 'object',
+#     properties: { location: property },
+#     required: %w[location]
+#   )
 class Ollama::Tool::Function::Parameters
   include Ollama::DTO
 
-  attr_reader :type, :properties, :required
+  # The type attribute reader returns the type associated with the object.
+  #
+  # @return [ String ] the type value stored in the instance variable
+  attr_reader :type
 
+  # The properties attribute reader returns the properties associated with the
+  # object.
+  #
+  # @return [ Hash ] the properties hash, or nil if not set
+  attr_reader :properties
+
+  # The required attribute reader returns the required parameter values
+  # associated with the object.
+  #
+  # @return [ Array<String>, nil ] an array of required parameter names, or nil
+  # if not set
+  attr_reader :required
+
+  # The initialize method sets up a new Parameters instance with the specified
+  # attributes.
+  #
+  # @param type [ String ] the type of parameter object
+  # @param properties [ Hash ] the properties of each parameter
+  # @param required [ Array<String> ] the names of required parameters
   def initialize(type:, properties:, required:)
     @type, @properties, @required =
       type, Hash(properties).transform_values(&:to_hash), Array(required)

data/lib/ollama/tool/function.rb

@@ -1,8 +1,51 @@
+# A class that represents a function definition for tool usage in Ollama API
+# interactions.
+#
+# This class encapsulates the structure required for defining functions that
+# can be passed to models to enable function calling capabilities. It includes
+# the function's name, description, parameters specification, and list of
+# required parameters.
+#
+# @example Creating a function definition for a tool
+#   function = Ollama::Tool::Function.new(
+#     name: 'get_current_weather',
+#     description: 'Get the current weather for a location',
+#     parameters: parameters,
+#     required: %w[location]
+#   )
 class Ollama::Tool::Function
   include Ollama::DTO
 
-  attr_reader :name, :description, :parameters, :required
+  # The name attribute reader returns the name associated with the object.
+  #
+  # @return [ String ] the name value stored in the instance variable
+  attr_reader :name
 
+  # The description attribute reader returns the description associated with
+  # the object.
+  #
+  # @return [ String ] the description value stored in the instance variable
+  attr_reader :description
+
+  # The parameters attribute reader returns the parameters associated with the
+  # object.
+  #
+  # @return [ Hash, nil ] the parameters hash, or nil if not set
+  attr_reader :parameters
+
+  # The required attribute reader returns the required parameter values
+  # associated with the object.
+  #
+  # @return [ Array<String>, nil ] an array of required parameter names, or nil if not set
+  attr_reader :required
+
+  # The initialize method sets up a new Tool::Function instance with the
+  # specified attributes.
+  #
+  # @param name [ String ] the name of the function
+  # @param description [ String ] a brief description of what the function does
+  # @param parameters [ Hash, nil ] optional parameters specification for the function
+  # @param required [ Array<String>, nil ] optional array of required parameter names
   def initialize(name:, description:, parameters: nil, required: nil)
     @name, @description, @parameters, @required =
       name, description, (Hash(parameters) if parameters),

data/lib/ollama/tool.rb

@@ -1,8 +1,44 @@
+# Represents a tool definition used in interactions with the Ollama API.
+# This class encapsulates the structure required for defining tools that can be
+# passed to models to enable function calling capabilities. It includes the type
+# of tool and the associated function definition, which specifies the tool's name,
+# description, parameters, and required fields.
+#
+# @example Creating a tool with a function definition
+#   tool = Ollama::Tool.new(
+#     type: 'function',
+#     function: Ollama::Tool::Function.new(
+#       name: 'get_current_weather',
+#       description: 'Get the current weather for a location',
+#       parameters: Ollama::Tool::Function::Parameters.new(
+#         type: 'object',
+#         properties: { location: property },
+#         required: %w[location]
+#       )
+#     )
+#   )
 class Ollama::Tool
   include Ollama::DTO
 
-  attr_reader :type, :function
+  # The type attribute reader returns the type associated with the tool.
+  #
+  # @return [ String ] the type of tool, typically 'function' for function
+  # calling capabilities
+  attr_reader :type
 
+  # The function attribute reader returns the function definition associated
+  # with the tool.
+  #
+  # @return [ Hash ] the function definition as a hash, containing details such
+  # as the function's name, description, parameters, and required fields
+  attr_reader :function
+
+  # The initialize method sets up a new Tool instance with the specified type
+  # and function.
+  #
+  # @param type [ String ] the type of tool being created
+  # @param function [ Ollama::Tool::Function ] the function definition
+  # associated with the tool
   def initialize(type:, function:)
     @type, @function = type, function.to_hash
   end

data/lib/ollama/version.rb

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

data/lib/ollama.rb

@@ -5,6 +5,19 @@ require 'tins/xt/full'
 require 'tins/xt/hash_union'
 require 'tins/xt/string_camelize'
 
+# The main module for the Ollama Ruby client library.
+#
+# This module serves as the root namespace for all components of the Ollama
+# client library, providing access to core classes and functionality for
+# interacting with the Ollama API. It includes handlers, commands, data
+# transfer objects, and utility classes that enable communication with
+# Ollama servers.
+#
+# @example Accessing core components
+#   Ollama::Client
+#   Ollama::Handlers
+#   Ollama::Commands
+#   Ollama::DTO
 module Ollama
 end
 
@@ -25,6 +38,19 @@ require 'ollama/tool/function/parameters/property'
 require 'ollama/response'
 require 'ollama/options'
 
+# A module that groups together various command classes used to interact with
+# the Ollama API.
+#
+# This module serves as a namespace for all command implementations that
+# correspond to specific Ollama API endpoints. It provides the structure and
+# functionality needed to define and execute different types of operations such
+# as chat conversations, model generation, model management, and other
+# interactions with the Ollama server.
+#
+# @example Accessing command classes
+#   Ollama::Commands::Chat
+#   Ollama::Commands::Generate
+#   Ollama::Commands::Create
 class Ollama::Commands
 end
 require 'ollama/commands/generate'

data/ollama-ruby.gemspec

@@ -1,9 +1,9 @@
 # -*- encoding: utf-8 -*-
-# stub: ollama-ruby 1.4.0 ruby lib
+# stub: ollama-ruby 1.6.0 ruby lib
 
 Gem::Specification.new do |s|
   s.name = "ollama-ruby".freeze
-  s.version = "1.4.0".freeze
+  s.version = "1.6.0".freeze
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0".freeze) if s.respond_to? :required_rubygems_version=
   s.require_paths = ["lib".freeze]
@@ -11,9 +11,9 @@ Gem::Specification.new do |s|
   s.date = "1980-01-02"
   s.description = "Library that allows interacting with the Ollama API".freeze
   s.email = "flori@ping.de".freeze
-  s.executables = ["ollama_console".freeze, "ollama_update".freeze, "ollama_cli".freeze]
+  s.executables = ["ollama_console".freeze, "ollama_update".freeze, "ollama_cli".freeze, "ollama_browse".freeze]
   s.extra_rdoc_files = ["README.md".freeze, "lib/ollama.rb".freeze, "lib/ollama/client.rb".freeze, "lib/ollama/client/command.rb".freeze, "lib/ollama/client/configuration/config.rb".freeze, "lib/ollama/client/doc.rb".freeze, "lib/ollama/commands/chat.rb".freeze, "lib/ollama/commands/copy.rb".freeze, "lib/ollama/commands/create.rb".freeze, "lib/ollama/commands/delete.rb".freeze, "lib/ollama/commands/embed.rb".freeze, "lib/ollama/commands/embeddings.rb".freeze, "lib/ollama/commands/generate.rb".freeze, "lib/ollama/commands/ps.rb".freeze, "lib/ollama/commands/pull.rb".freeze, "lib/ollama/commands/push.rb".freeze, "lib/ollama/commands/show.rb".freeze, "lib/ollama/commands/tags.rb".freeze, "lib/ollama/commands/version.rb".freeze, "lib/ollama/dto.rb".freeze, "lib/ollama/errors.rb".freeze, "lib/ollama/handlers.rb".freeze, "lib/ollama/handlers/collector.rb".freeze, "lib/ollama/handlers/concern.rb".freeze, "lib/ollama/handlers/dump_json.rb".freeze, "lib/ollama/handlers/dump_yaml.rb".freeze, "lib/ollama/handlers/markdown.rb".freeze, "lib/ollama/handlers/nop.rb".freeze, "lib/ollama/handlers/print.rb".freeze, "lib/ollama/handlers/progress.rb".freeze, "lib/ollama/handlers/say.rb".freeze, "lib/ollama/handlers/single.rb".freeze, "lib/ollama/image.rb".freeze, "lib/ollama/json_loader.rb".freeze, "lib/ollama/message.rb".freeze, "lib/ollama/options.rb".freeze, "lib/ollama/response.rb".freeze, "lib/ollama/tool.rb".freeze, "lib/ollama/tool/function.rb".freeze, "lib/ollama/tool/function/parameters.rb".freeze, "lib/ollama/tool/function/parameters/property.rb".freeze, "lib/ollama/version.rb".freeze]
-  s.files = [".yardopts".freeze, "CHANGES.md".freeze, "Gemfile".freeze, "LICENSE".freeze, "README.md".freeze, "Rakefile".freeze, "bin/ollama_browse".freeze, "bin/ollama_cli".freeze, "bin/ollama_console".freeze, "bin/ollama_update".freeze, "lib/ollama.rb".freeze, "lib/ollama/client.rb".freeze, "lib/ollama/client/command.rb".freeze, "lib/ollama/client/configuration/config.rb".freeze, "lib/ollama/client/doc.rb".freeze, "lib/ollama/commands/chat.rb".freeze, "lib/ollama/commands/copy.rb".freeze, "lib/ollama/commands/create.rb".freeze, "lib/ollama/commands/delete.rb".freeze, "lib/ollama/commands/embed.rb".freeze, "lib/ollama/commands/embeddings.rb".freeze, "lib/ollama/commands/generate.rb".freeze, "lib/ollama/commands/ps.rb".freeze, "lib/ollama/commands/pull.rb".freeze, "lib/ollama/commands/push.rb".freeze, "lib/ollama/commands/show.rb".freeze, "lib/ollama/commands/tags.rb".freeze, "lib/ollama/commands/version.rb".freeze, "lib/ollama/dto.rb".freeze, "lib/ollama/errors.rb".freeze, "lib/ollama/handlers.rb".freeze, "lib/ollama/handlers/collector.rb".freeze, "lib/ollama/handlers/concern.rb".freeze, "lib/ollama/handlers/dump_json.rb".freeze, "lib/ollama/handlers/dump_yaml.rb".freeze, "lib/ollama/handlers/markdown.rb".freeze, "lib/ollama/handlers/nop.rb".freeze, "lib/ollama/handlers/print.rb".freeze, "lib/ollama/handlers/progress.rb".freeze, "lib/ollama/handlers/say.rb".freeze, "lib/ollama/handlers/single.rb".freeze, "lib/ollama/image.rb".freeze, "lib/ollama/json_loader.rb".freeze, "lib/ollama/message.rb".freeze, "lib/ollama/options.rb".freeze, "lib/ollama/response.rb".freeze, "lib/ollama/tool.rb".freeze, "lib/ollama/tool/function.rb".freeze, "lib/ollama/tool/function/parameters.rb".freeze, "lib/ollama/tool/function/parameters/property.rb".freeze, "lib/ollama/version.rb".freeze, "ollama-ruby.gemspec".freeze, "spec/assets/client.json".freeze, "spec/assets/kitten.jpg".freeze, "spec/assets/options.json".freeze, "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, "tmp/.keep".freeze]
+  s.files = [".contexts/code_comment.rb".freeze, ".contexts/full.rb".freeze, ".contexts/info.rb".freeze, ".contexts/lib.rb".freeze, ".contexts/yard.md".freeze, ".yardopts".freeze, "CHANGES.md".freeze, "Gemfile".freeze, "LICENSE".freeze, "README.md".freeze, "Rakefile".freeze, "bin/ollama_browse".freeze, "bin/ollama_cli".freeze, "bin/ollama_console".freeze, "bin/ollama_update".freeze, "lib/ollama.rb".freeze, "lib/ollama/client.rb".freeze, "lib/ollama/client/command.rb".freeze, "lib/ollama/client/configuration/config.rb".freeze, "lib/ollama/client/doc.rb".freeze, "lib/ollama/commands/chat.rb".freeze, "lib/ollama/commands/copy.rb".freeze, "lib/ollama/commands/create.rb".freeze, "lib/ollama/commands/delete.rb".freeze, "lib/ollama/commands/embed.rb".freeze, "lib/ollama/commands/embeddings.rb".freeze, "lib/ollama/commands/generate.rb".freeze, "lib/ollama/commands/ps.rb".freeze, "lib/ollama/commands/pull.rb".freeze, "lib/ollama/commands/push.rb".freeze, "lib/ollama/commands/show.rb".freeze, "lib/ollama/commands/tags.rb".freeze, "lib/ollama/commands/version.rb".freeze, "lib/ollama/dto.rb".freeze, "lib/ollama/errors.rb".freeze, "lib/ollama/handlers.rb".freeze, "lib/ollama/handlers/collector.rb".freeze, "lib/ollama/handlers/concern.rb".freeze, "lib/ollama/handlers/dump_json.rb".freeze, "lib/ollama/handlers/dump_yaml.rb".freeze, "lib/ollama/handlers/markdown.rb".freeze, "lib/ollama/handlers/nop.rb".freeze, "lib/ollama/handlers/print.rb".freeze, "lib/ollama/handlers/progress.rb".freeze, "lib/ollama/handlers/say.rb".freeze, "lib/ollama/handlers/single.rb".freeze, "lib/ollama/image.rb".freeze, "lib/ollama/json_loader.rb".freeze, "lib/ollama/message.rb".freeze, "lib/ollama/options.rb".freeze, "lib/ollama/response.rb".freeze, "lib/ollama/tool.rb".freeze, "lib/ollama/tool/function.rb".freeze, "lib/ollama/tool/function/parameters.rb".freeze, "lib/ollama/tool/function/parameters/property.rb".freeze, "lib/ollama/version.rb".freeze, "ollama-ruby.gemspec".freeze, "spec/assets/client.json".freeze, "spec/assets/kitten.jpg".freeze, "spec/assets/options.json".freeze, "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, "tmp/.keep".freeze]
   s.homepage = "https://github.com/flori/ollama-ruby".freeze
   s.licenses = ["MIT".freeze]
   s.rdoc_options = ["--title".freeze, "Ollama-ruby - Interacting with the Ollama API".freeze, "--main".freeze, "README.md".freeze]
@@ -24,17 +24,18 @@ Gem::Specification.new do |s|
 
   s.specification_version = 4
 
-  s.add_development_dependency(%q<gem_hadar>.freeze, ["~> 1.20".freeze])
+  s.add_development_dependency(%q<gem_hadar>.freeze, ["~> 2.0".freeze])
   s.add_development_dependency(%q<all_images>.freeze, ["~> 0.6".freeze])
   s.add_development_dependency(%q<rspec>.freeze, ["~> 3.2".freeze])
   s.add_development_dependency(%q<kramdown>.freeze, ["~> 2.0".freeze])
   s.add_development_dependency(%q<webmock>.freeze, [">= 0".freeze])
   s.add_development_dependency(%q<debug>.freeze, [">= 0".freeze])
   s.add_development_dependency(%q<simplecov>.freeze, [">= 0".freeze])
+  s.add_development_dependency(%q<context_spook>.freeze, [">= 0".freeze])
   s.add_runtime_dependency(%q<excon>.freeze, ["~> 1.0".freeze])
   s.add_runtime_dependency(%q<infobar>.freeze, ["~> 0.8".freeze])
   s.add_runtime_dependency(%q<json>.freeze, ["~> 2.0".freeze])
-  s.add_runtime_dependency(%q<tins>.freeze, ["~> 1.34".freeze])
+  s.add_runtime_dependency(%q<tins>.freeze, ["~> 1.41".freeze])
   s.add_runtime_dependency(%q<term-ansicolor>.freeze, ["~> 1.11".freeze])
   s.add_runtime_dependency(%q<kramdown-ansi>.freeze, ["~> 0.0".freeze, ">= 0.0.1".freeze])
   s.add_runtime_dependency(%q<ostruct>.freeze, ["~> 0.0".freeze])

data/spec/ollama/client/doc_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Client::Doc do
+describe Ollama::Client::Doc do
   it 'can document commands' do
     expect(Ollama::Client::Doc.new(:generate).to_s).to match(/generate-a-completion/)
   end

data/spec/ollama/client_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Client do
+describe Ollama::Client do
   let :base_url do
     'https://ai.foo.bar'
   end
@@ -35,9 +35,13 @@ RSpec.describe Ollama::Client do
   end
 
   it 'can be configured via environment variable' do
+    old_base_url = ENV['OLLAMA_URL']
+    ENV.delete('OLLAMA_URL')
     expect { described_class.new }.to raise_error(ArgumentError)
     ENV['OLLAMA_URL'] = base_url
     expect(described_class.new).to be_a described_class
+  ensure
+    ENV['OLLAMA_URL'] = old_base_url
   end
 
   it 'can disable ssl peer verification' do
@@ -145,6 +149,20 @@ RSpec.describe Ollama::Client do
       ollama.generate(model: 'llama3.1', prompt: 'Hello World')
     end
 
+    it 'can soldier on with parse errors and output warning' do
+      expect(excon).to receive(:send).with(
+        :post,
+        body:  '{"model":"llama3.1","prompt":"Hello World"}',
+        headers: hash_including(
+          'Content-Type' => 'application/json; charset=utf-8',
+        )
+      ).and_return(double(status: 200, body: '{i am so broken}'))
+      expect(ollama).to receive(:warn).with(
+        "Caught JSON::ParserError: expected object key, got 'i' at line 1 column 2"
+      )
+      expect(ollama.generate(model: 'llama3.1', prompt: 'Hello World')).to be nil
+    end
+
     it 'can generate with stream' do
       expect(excon).to receive(:send).with(
         :post,

data/spec/ollama/commands/chat_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Commands::Chat do
+describe Ollama::Commands::Chat do
   it 'can be instantiated' do
     messages = [
       Ollama::Message.new(role: 'user', content: "Let's play Global Thermonuclear War.")

data/spec/ollama/commands/copy_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Commands::Copy do
+describe Ollama::Commands::Copy do
   it 'can be instantiated' do
     copy = described_class.new(source: 'llama3.1', destination: 'camell3')
     expect(copy).to be_a described_class

data/spec/ollama/commands/create_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Commands::Create do
+describe Ollama::Commands::Create do
   it 'can be instantiated' do
     create = described_class.new(model: 'llama3.1', stream: true)
     expect(create).to be_a described_class

data/spec/ollama/commands/delete_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Commands::Delete do
+describe Ollama::Commands::Delete do
   it 'can be instantiated' do
     delete = described_class.new(model: 'llama3.1')
     expect(delete).to be_a described_class

data/spec/ollama/commands/embed_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Commands::Embed do
+describe Ollama::Commands::Embed do
   it 'can be instantiated' do
     embed = described_class.new(
       model: 'all-minilm',

data/spec/ollama/commands/embeddings_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Commands::Embeddings do
+describe Ollama::Commands::Embeddings do
   it 'can be instantiated' do
     embeddings = described_class.new(
       model: 'mxbai-embed-large',

data/spec/ollama/commands/generate_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Commands::Generate do
+describe Ollama::Commands::Generate do
   it 'can be instantiated' do
     generate = described_class.new(model: 'llama3.1', prompt: 'Hello World')
     expect(generate).to be_a described_class

data/spec/ollama/commands/ps_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Commands::Ps do
+describe Ollama::Commands::Ps do
   let :ps do
     described_class.new
   end

data/spec/ollama/commands/pull_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Commands::Pull do
+describe Ollama::Commands::Pull do
   it 'can be instantiated' do
     pull = described_class.new(model: 'llama3.1')
     expect(pull).to be_a described_class

data/spec/ollama/commands/push_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Commands::Push do
+describe Ollama::Commands::Push do
   it 'can be instantiated' do
     push = described_class.new(model: 'llama3.1')
     expect(push).to be_a described_class

data/spec/ollama/commands/show_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Commands::Show do
+describe Ollama::Commands::Show do
   it 'can be instantiated' do
     show = described_class.new(model: 'llama3.1')
     expect(show).to be_a described_class

data/spec/ollama/commands/tags_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Commands::Tags do
+describe Ollama::Commands::Tags do
   let :tags do
     described_class.new
   end

data/spec/ollama/commands/version_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Commands::Version do
+describe Ollama::Commands::Version do
   let :version do
     described_class.new
   end

data/spec/ollama/handlers/collector_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Handlers::Collector do
+describe Ollama::Handlers::Collector do
   it 'has .call' do
     expect_any_instance_of(described_class).to receive(:call).with(:foo)
     described_class.call(:foo)

data/spec/ollama/handlers/dump_json_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Handlers::DumpJSON do
+describe Ollama::Handlers::DumpJSON do
   it 'has .call' do
     expect_any_instance_of(described_class).to receive(:call).with(:foo)
     described_class.call(:foo)

data/spec/ollama/handlers/dump_yaml_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Handlers::DumpYAML do
+describe Ollama::Handlers::DumpYAML do
   it 'has .call' do
     expect_any_instance_of(described_class).to receive(:call).with(:foo)
     described_class.call(:foo)

data/spec/ollama/handlers/markdown_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Handlers::Markdown do
+describe Ollama::Handlers::Markdown do
   it 'has .call' do
     expect_any_instance_of(described_class).to receive(:call).with(:foo)
     described_class.call(:foo)

data/spec/ollama/handlers/nop_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Handlers::NOP do
+describe Ollama::Handlers::NOP do
   it 'has .to_proc' do
     expect_any_instance_of(described_class).to receive(:call).with(:foo)
     described_class.call(:foo)
@@ -10,6 +10,6 @@ RSpec.describe Ollama::Handlers::NOP do
     nop = described_class.new(output:)
     response = Ollama::Response[foo: 'testing']
     nop.call(response)
-    expect(nop.result).to be_nil
+    expect(nop.result).to be nil
   end
 end

data/spec/ollama/handlers/print_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Handlers::Print do
+describe Ollama::Handlers::Print do
   it 'has .to_proc' do
     expect_any_instance_of(described_class).to receive(:call).with(:foo)
     described_class.call(:foo)

data/spec/ollama/handlers/progress_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Handlers::Progress do
+describe Ollama::Handlers::Progress do
   it 'has .to_proc' do
     expect_any_instance_of(described_class).to receive(:call).with(:foo)
     described_class.call(:foo)

data/spec/ollama/handlers/say_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Handlers::Say do
+describe Ollama::Handlers::Say do
   let :say do
     described_class.new
   end

data/spec/ollama/handlers/single_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Handlers::Single do
+describe Ollama::Handlers::Single do
   it 'has .call' do
     expect_any_instance_of(described_class).to receive(:call).with(:foo)
     described_class.call(:foo)

data/spec/ollama/image_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe Ollama::Image do
+describe Ollama::Image do
   let :image do
     described_class.for_filename(asset('kitten.jpg'))
   end