llm.rb 0.4.0 → 0.4.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3d8311b461c49095ba393be6d3456bf7c3032a07e4480f4fd3e4cd9133f2b6e7
-  data.tar.gz: 5f9ce812a4a27e0982ea5072bca3f0494317ee0274987e57df792cc0e0d2fcfc
+  metadata.gz: 2ae98573410bfc74a61f0bf15f811c30800b6567c9e5fc9e4cbe50f4994a200f
+  data.tar.gz: 9776839912017f5f3bac8452670c60b3bc9eabecebe11c402458a36b78d958e7
 SHA512:
-  metadata.gz: 1d860cabe75a0718e0c06d686127d6d24c65b4dc8967aaf490b892bfad10bf88268791e9138b26e2d1c646a22e2eda7e74ad48e1fa31e9cb68c42de7ec53ac12
-  data.tar.gz: 8c2e14a316e87560e5d5dc3cdb416d151172ae12e2244bb813ff1bba61c582474607bd6920f82fa5f080e628b2dac319dd20d4f74eff55940f9c88ab7544b327
+  metadata.gz: 4252d2d861d409428067415340c2d36f2e75f34f3be9905a576a63455902d1faf4100d8c8c7060e683fe922ff095cc00fd56fd135b9589a746d9de716c66e5e5
+  data.tar.gz: 819676961e401aa5e27678e3d36af406362f0a68b0243e1e1f71f5f135286830dd1c009c47c69c26b6789a771da6a1fe72eb023f0f1833c20ffadac37f58ed1b

data/LICENSE

@@ -0,0 +1,17 @@
+Copyright (C) 2025
+Antar Azri <azantar@proton.me>
+0x1eef <0x1eef@proton.me>
+
+Permission to use, copy, modify, and/or distribute this
+software for any purpose with or without fee is hereby
+granted.
+
+THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS
+ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO
+EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
+INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER
+RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION,
+ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
+OF THIS SOFTWARE.

data/README.md

@@ -133,7 +133,7 @@ The interface is designed so you could drop in any other library in its place:
 require "llm"
 
 llm = LLM.openai(ENV["KEY"])
-schema = llm.schema.object({os: llm.schema.string.enum("OpenBSD", "FreeBSD", "NetBSD").required})
+schema = llm.schema.object({os: llm.schema.string.enum("OpenBSD", "FreeBSD", "NetBSD")})
 bot = LLM::Chat.new(llm, schema:)
 bot.chat "You secretly love NetBSD", :system
 bot.chat "What operating system is the best?", :user
@@ -364,7 +364,7 @@ bot.messages.select(&:assistant?).each { print "[#{_1.role}] ", _1.content, "\n"
 Generally all providers accept text prompts but some providers can
 also understand URLs, and various file types (eg images, audio, video,
 etc). The llm.rb approach to multimodal prompts is to let you pass `URI`
-objects to describe links, `LLM::File` / `LLM::Response::File` objects
+objects to describe links, `LLM::File` | `LLM::Response::File` objects
 to describe files, `String` objects to describe text blobs, or an array
 of the aforementioned objects to describe multiple objects in a single
 prompt. Each object is a first class citizen that can be passed directly
@@ -372,9 +372,7 @@ to a prompt.
 
 For more depth and examples on how to use the multimodal API, please see
 the [provider-specific documentation](https://0x1eef.github.io/x/llm.rb/)
-for more provider-specific examples – there can be subtle differences
-between providers and even between APIs from the same provider that are
-not covered in the README:
+for more provider-specific examples:
 
 ```ruby
 #!/usr/bin/env ruby

data/lib/json/schema/array.rb

@@ -2,9 +2,8 @@
 
 class JSON::Schema
   class Array < Leaf
-    def initialize(items, **rest)
+    def initialize(*items)
       @items = items
-      super(**rest)
     end
 
     def to_h

data/lib/json/schema/leaf.rb

@@ -1,6 +1,12 @@
 # frozen_string_literal: true
 
 class JSON::Schema
+  ##
+  # The {JSON::Schema::Leaf JSON::Schema::Leaf} class is the
+  # superclass of all values that can appear in a JSON schema.
+  # See the instance methods of {JSON::Schema JSON::Schema} for
+  # an example of how to create instances of {JSON::Schema::Leaf JSON::Schema::Leaf}
+  # through its subclasses.
   class Leaf
     def initialize
       @description = nil
@@ -9,30 +15,51 @@ class JSON::Schema
       @required = nil
     end
 
+    ##
+    # Set the description of a leaf
+    # @param [String] str The description
+    # @return [JSON::Schema::Leaf]
     def description(str)
       tap { @description = str }
     end
 
+    ##
+    # Set the default value of a leaf
+    # @param [Object] value The default value
+    # @return [JSON::Schema::Leaf]
     def default(value)
       tap { @default = value }
     end
 
+    ##
+    # Set the allowed values of a leaf
+    # @param [Array] values The allowed values
+    # @return [JSON::Schema::Leaf]
     def enum(*values)
       tap { @enum = values }
     end
 
+    ##
+    # Denote a leaf as required
+    # @return [JSON::Schema::Leaf]
     def required
       tap { @required = true }
     end
 
+    ##
+    # @return [Hash]
     def to_h
       {description: @description, default: @default, enum: @enum}.compact
     end
 
+    ##
+    # @return [String]
     def to_json(options = {})
       to_h.to_json(options)
     end
 
+    ##
+    # @return [Boolean]
     def required?
       @required
     end

data/lib/json/schema/version.rb

@@ -0,0 +1,6 @@
+module JSON
+end unless defined?(JSON)
+
+class JSON::Schema
+  VERSION = "0.1.0"
+end

data/lib/json/schema.rb

@@ -3,7 +3,25 @@
 module JSON
 end unless defined?(JSON)
 
+##
+# The {JSON::Schema JSON::Schema} class represents a JSON schema,
+# and provides methods that let you describe and produce a schema
+# that can be used in various contexts that include the validation
+# and generation of JSON data.
+#
+# @see https://json-schema.org/ JSON Schema Specification
+# @see https://tour.json-schema.org/ JSON Schema Tour
+#
+# @example
+#  schema = JSON::Schema.new
+#  schema.object({
+#    name: schema.string.enum("John", "Jane").required,
+#    age: schema.integer.required,
+#    hobbies: schema.array(schema.string, schema.null).required,
+#    address: schema.object({street: schema.string}).required,
+#  })
 class JSON::Schema
+  require_relative "schema/version"
   require_relative "schema/leaf"
   require_relative "schema/object"
   require_relative "schema/array"
@@ -15,25 +33,22 @@ class JSON::Schema
 
   ##
   # Returns an object
-  # @param properties [Hash] A hash of properties
-  # @param rest [Hash] Any other options
+  # @param [Hash] properties A hash of properties
   # @return [JSON::Schema::Object]
-  def object(properties, **rest)
-    Object.new(properties, **rest)
+  def object(properties)
+    Object.new(properties)
   end
 
   ##
   # Returns an array
-  # @param items [Array] An array of items
-  # @param rest [Hash] Any other options
+  # @param [Array] items An array of items
   # @return [JSON::Schema::Array]
-  def array(items, **rest)
-    Array.new(items, **rest)
+  def array(*items)
+    Array.new(*items)
   end
 
   ##
   # Returns a string
-  # @param rest [Hash] Any other options
   # @return [JSON::Schema::String]
   def string(...)
     String.new(...)
@@ -41,7 +56,6 @@ class JSON::Schema
 
   ##
   # Returns a number
-  # @param rest [Hash] Any other options
   # @return [JSON::Schema::Number] a number
   def number(...)
     Number.new(...)
@@ -49,7 +63,6 @@ class JSON::Schema
 
   ##
   # Returns an integer
-  # @param rest [Hash] Any other options
   # @return [JSON::Schema::Integer]
   def integer(...)
     Integer.new(...)
@@ -57,7 +70,6 @@ class JSON::Schema
 
   ##
   # Returns a boolean
-  # @param rest [Hash] Any other options
   # @return [JSON::Schema::Boolean]
   def boolean(...)
     Boolean.new(...)
@@ -65,7 +77,6 @@ class JSON::Schema
 
   ##
   # Returns null
-  # @param rest [Hash] Any other options
   # @return [JSON::Schema::Null]
   def null(...)
     Null.new(...)

data/lib/llm/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module LLM
-  VERSION = "0.4.0"
+  VERSION = "0.4.2"
 end

data/llm.gemspec

@@ -21,10 +21,9 @@ Gem::Specification.new do |spec|
   spec.metadata["source_code_uri"] = "https://github.com/llmrb/llm"
 
   spec.files = Dir[
-    "README.md", "LICENSE.txt",
+    "README.md", "LICENSE",
     "lib/*.rb", "lib/**/*.rb",
-    "spec/*.rb", "spec/**/*.rb",
-    "share/llm/models/*.yml", "llm.gemspec"
+    "llm.gemspec"
   ]
   spec.require_paths = ["lib"]
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llm.rb
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.4.2
 platform: ruby
 authors:
 - Antar Azri
@@ -9,7 +9,7 @@ authors:
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-04-29 00:00:00.000000000 Z
+date: 2025-04-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: webmock
@@ -147,6 +147,7 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- LICENSE
 - README.md
 - lib/json/schema.rb
 - lib/json/schema/array.rb
@@ -157,6 +158,7 @@ files:
 - lib/json/schema/number.rb
 - lib/json/schema/object.rb
 - lib/json/schema/string.rb
+- lib/json/schema/version.rb
 - lib/llm.rb
 - lib/llm/buffer.rb
 - lib/llm/chat.rb
@@ -213,29 +215,6 @@ files:
 - lib/llm/utils.rb
 - lib/llm/version.rb
 - llm.gemspec
-- spec/anthropic/completion_spec.rb
-- spec/anthropic/embedding_spec.rb
-- spec/anthropic/models_spec.rb
-- spec/gemini/completion_spec.rb
-- spec/gemini/conversation_spec.rb
-- spec/gemini/embedding_spec.rb
-- spec/gemini/files_spec.rb
-- spec/gemini/images_spec.rb
-- spec/gemini/models_spec.rb
-- spec/llm/conversation_spec.rb
-- spec/ollama/completion_spec.rb
-- spec/ollama/conversation_spec.rb
-- spec/ollama/embedding_spec.rb
-- spec/ollama/models_spec.rb
-- spec/openai/audio_spec.rb
-- spec/openai/completion_spec.rb
-- spec/openai/embedding_spec.rb
-- spec/openai/files_spec.rb
-- spec/openai/images_spec.rb
-- spec/openai/models_spec.rb
-- spec/openai/responses_spec.rb
-- spec/readme_spec.rb
-- spec/setup.rb
 homepage: https://github.com/llmrb/llm
 licenses:
 - 0BSDL

data/spec/anthropic/completion_spec.rb

@@ -1,96 +0,0 @@
-# frozen_string_literal: true
-
-require "setup"
-
-RSpec.describe "LLM::Anthropic: completions" do
-  subject(:anthropic) { LLM.anthropic(token) }
-  let(:token) { ENV["LLM_SECRET"] || "TOKEN" }
-
-  context "when given a successful response",
-          vcr: {cassette_name: "anthropic/completions/successful_response"} do
-    subject(:response) { anthropic.complete("Hello, world", :user) }
-
-    it "returns a completion" do
-      expect(response).to be_a(LLM::Response::Completion)
-    end
-
-    it "returns a model" do
-      expect(response.model).to eq("claude-3-5-sonnet-20240620")
-    end
-
-    it "includes token usage" do
-      expect(response).to have_attributes(
-        prompt_tokens: 10,
-        completion_tokens: 30,
-        total_tokens: 40
-      )
-    end
-
-    context "with a choice" do
-      subject(:choice) { response.choices[0] }
-
-      it "has choices" do
-        expect(choice).to have_attributes(
-          role: "assistant",
-          content: "Hello! How can I assist you today? Feel free to ask me any questions or let me know if you need help with anything."
-        )
-      end
-
-      it "includes the response" do
-        expect(choice.extra[:response]).to eq(response)
-      end
-    end
-  end
-
-  context "when given a URI to an image",
-          vcr: {cassette_name: "anthropic/completions/successful_response_uri_image"} do
-    subject { response.choices[0].content.downcase[0..2] }
-    let(:response) do
-      anthropic.complete([
-        "Is this image the flag of brazil ? ",
-        "Answer with yes or no. ",
-        "Nothing else.",
-        uri
-      ], :user)
-    end
-    let(:uri) { URI("https://upload.wikimedia.org/wikipedia/en/thumb/0/05/Flag_of_Brazil.svg/250px-Flag_of_Brazil.svg.png") }
-
-    it "describes the image" do
-      is_expected.to eq("yes")
-    end
-  end
-
-  context "when given a local reference to an image",
-          vcr: {cassette_name: "anthropic/completions/successful_response_file_image"} do
-    subject { response.choices[0].content.downcase[0..2] }
-    let(:response) do
-      anthropic.complete([
-        "Is this image a representation of a blue book ?",
-        "Answer with yes or no.",
-        "Nothing else.",
-        file
-      ], :user)
-    end
-    let(:file) { LLM::File("spec/fixtures/images/bluebook.png") }
-
-    it "describes the image" do
-      is_expected.to eq("yes")
-    end
-  end
-
-  context "when given an unauthorized response",
-          vcr: {cassette_name: "anthropic/completions/unauthorized_response"} do
-    subject(:response) { anthropic.complete("Hello", :user) }
-    let(:token) { "BADTOKEN" }
-
-    it "raises an error" do
-      expect { response }.to raise_error(LLM::Error::Unauthorized)
-    end
-
-    it "includes the response" do
-      response
-    rescue LLM::Error::Unauthorized => ex
-      expect(ex.response).to be_kind_of(Net::HTTPResponse)
-    end
-  end
-end

data/spec/anthropic/embedding_spec.rb

@@ -1,25 +0,0 @@
-# frozen_string_literal: true
-
-require "setup"
-
-RSpec.describe "LLM::Anthropic: embeddings" do
-  let(:anthropic) { LLM.anthropic(token) }
-  let(:token) { ENV["LLM_SECRET"] || "TOKEN" }
-
-  context "when given a successful response",
-          vcr: {cassette_name: "anthropic/embeddings/successful_response"} do
-    subject(:response) { anthropic.embed("Hello, world", token:) }
-
-    it "returns an embedding" do
-      expect(response).to be_instance_of(LLM::Response::Embedding)
-    end
-
-    it "returns a model" do
-      expect(response.model).to eq("voyage-2")
-    end
-
-    it "has embeddings" do
-      expect(response.embeddings).to be_instance_of(Array)
-    end
-  end
-end

data/spec/anthropic/models_spec.rb

@@ -1,21 +0,0 @@
-# frozen_string_literal: true
-
-require "setup"
-
-RSpec.describe "LLM::Anthropic::Models" do
-  let(:token) { ENV["LLM_SECRET"] || "TOKEN" }
-  let(:provider) { LLM.anthropic(token) }
-
-  context "when given a successful list operation",
-          vcr: {cassette_name: "anthropic/models/successful_list"} do
-    subject { provider.models.all }
-
-    it "is successful" do
-      is_expected.to be_instance_of(LLM::Response::ModelList)
-    end
-
-    it "returns a list of models" do
-      expect(subject.models).to all(be_a(LLM::Model))
-    end
-  end
-end

data/spec/gemini/completion_spec.rb

@@ -1,85 +0,0 @@
-# frozen_string_literal: true
-
-require "setup"
-
-RSpec.describe "LLM::Gemini: completions" do
-  subject(:gemini) { LLM.gemini(token) }
-  let(:token) { ENV["LLM_SECRET"] || "TOKEN" }
-
-  context "when given a successful response",
-          vcr: {cassette_name: "gemini/completions/successful_response"} do
-    subject(:response) { gemini.complete("Hello!", :user) }
-
-    it "returns a completion" do
-      expect(response).to be_a(LLM::Response::Completion)
-    end
-
-    it "returns a model" do
-      expect(response.model).to eq("gemini-1.5-flash")
-    end
-
-    it "includes token usage" do
-      expect(response).to have_attributes(
-        prompt_tokens: 2,
-        completion_tokens: 11,
-        total_tokens: 13
-      )
-    end
-
-    context "with a choice" do
-      subject(:choice) { response.choices[0] }
-
-      it "has choices" do
-        expect(response).to be_a(LLM::Response::Completion).and have_attributes(
-          choices: [
-            have_attributes(
-              role: "model",
-              content: "Hello there! How can I help you today?\n"
-            )
-          ]
-        )
-      end
-
-      it "includes the response" do
-        expect(choice.extra[:response]).to eq(response)
-      end
-    end
-  end
-
-  context "when given a thread of messages",
-          vcr: {cassette_name: "gemini/completions/successful_response_thread"} do
-    subject(:response) do
-      gemini.complete "What is your name? What age are you?", :user, messages: [
-        {role: "user", content: "Answer all of my questions"},
-        {role: "user", content: "Your name is Pablo, you are 25 years old and you are my amigo"}
-      ]
-    end
-
-    it "has choices" do
-      expect(response).to have_attributes(
-        choices: [
-          have_attributes(
-            role: "model",
-            content: "My name is Pablo, and I am 25 years old.  ¡Amigo!\n"
-          )
-        ]
-      )
-    end
-  end
-
-  context "when given an unauthorized response",
-          vcr: {cassette_name: "gemini/completions/unauthorized_response"} do
-    subject(:response) { gemini.complete("Hello!", :user) }
-    let(:token) { "BADTOKEN" }
-
-    it "raises an error" do
-      expect { response }.to raise_error(LLM::Error::Unauthorized)
-    end
-
-    it "includes a response" do
-      response
-    rescue LLM::Error::Unauthorized => ex
-      expect(ex.response).to be_kind_of(Net::HTTPResponse)
-    end
-  end
-end

data/spec/gemini/conversation_spec.rb

@@ -1,31 +0,0 @@
-# frozen_string_literal: true
-
-require "setup"
-
-RSpec.describe "LLM::Chat: gemini" do
-  let(:described_class) { LLM::Chat }
-  let(:provider) { LLM.gemini(token) }
-  let(:token) { ENV["LLM_SECRET"] || "TOKEN" }
-  let(:conversation) { described_class.new(provider, **params).lazy }
-
-  context "when asked to describe an image",
-          vcr: {cassette_name: "gemini/conversations/multimodal_response"} do
-    subject { conversation.last_message }
-
-    let(:params) { {} }
-    let(:image) { LLM::File("spec/fixtures/images/bluebook.png") }
-
-    before do
-      conversation.chat(image, :user)
-      conversation.chat("Describe the image with a short sentance", :user)
-    end
-
-    it "describes the image" do
-      is_expected.to have_attributes(
-        role: "model",
-        content: "That's a simple illustration of a book " \
-                 "resting on a blue, X-shaped book stand.\n"
-      )
-    end
-  end
-end

data/spec/gemini/embedding_spec.rb

@@ -1,25 +0,0 @@
-# frozen_string_literal: true
-
-require "setup"
-
-RSpec.describe "LLM::OpenAI: embeddings" do
-  let(:gemini) { LLM.gemini(token) }
-  let(:token) { ENV["LLM_SECRET"] || "TOKEN" }
-
-  context "when given a successful response",
-          vcr: {cassette_name: "gemini/embeddings/successful_response"} do
-    subject(:response) { gemini.embed("Hello, world") }
-
-    it "returns an embedding" do
-      expect(response).to be_instance_of(LLM::Response::Embedding)
-    end
-
-    it "returns a model" do
-      expect(response.model).to eq("text-embedding-004")
-    end
-
-    it "has embeddings" do
-      expect(response.embeddings).to be_instance_of(Array)
-    end
-  end
-end