llm.rb 0.12.0 → 0.13.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2cd935a4ccd3b911e92b5ff54335cfc143247cbb5fe55214fd563551f7349da4
-  data.tar.gz: c76b36f2877c0cec7fdde54471a81ae19a4ec044158077742eba9acd26cd1483
+  metadata.gz: f83248fa3bce285d75952b726cc7d1b097e61b61fb77edd55f39168922a8614c
+  data.tar.gz: 52fa691fad7eaa2e77e24245aba97ea469688f8307f869d533fac47707aa6caf
 SHA512:
-  metadata.gz: f654d042a6f44cba15b2dc0049d3933aa442f631293be446486e524773ff01bfc0f13f89ecbf09659a175c3ff9f7c6512ae8bde5716f6935c8d3d05528d3e4e9
-  data.tar.gz: 73945113e01d89188301a1a7db921c8c07ac19a847fa71528bc2e62eb5162ac656aca02a5f17e02bc4918dc973c298f0021afbb14a4cb38e7f81705950c4ed5b
+  metadata.gz: d17e104601295bbbd8dd5bd34fe5378e96367aacb832ddb81b7f0fdfd3fa54f39d98dff6568f3190a4d979de4382a6ebd2eba8ee2fb2cce0e67db2478be9a1df
+  data.tar.gz: d2e6bb0f507bdcf140bd7f44f9edd40c057f061a9b4bc7123a87bdf6fa40a6f16a99642965bdaf81aaaeab5e995a01587c79cd47e19c6a2b5f86e456fa4813a9

data/README.md

@@ -3,7 +3,7 @@
 llm.rb is a zero-dependency Ruby toolkit for Large Language Models that
 includes OpenAI, Gemini, Anthropic, xAI (grok), DeepSeek, Ollama, and
 LlamaCpp. The toolkit includes full support for chat, streaming, tool calling,
-audio, images, files, and JSON Schema generation.
+audio, images, files, and structured outputs (JSON Schema).
 
 ## Features
 
@@ -22,7 +22,7 @@ audio, images, files, and JSON Schema generation.
 - 🗣️ Text-to-speech, transcription, and translation
 - 🖼️ Image generation, editing, and variation support
 - 📎 File uploads and prompt-aware file interaction
-- 💡 Multimodal prompts (text, images, PDFs, URLs, files)
+- 💡 Multimodal prompts (text, documents, audio, images, videos, URLs, etc)
 
 #### Embeddings
 - 🧮 Text embeddings and vector support
@@ -47,9 +47,9 @@ breaks things down by provider, so you can see exactly what’s supported where.
 | **Audio (TTS / Transcribe / Translate)** | ✅  | ❌        | ✅     | ❌       | ❌         | ❌     | ❌       |
 | **Image Generation & Editing**       | ✅     | ❌        | ✅     | ❌       | ✅         | ❌     | ❌       |
 | **File Uploads**                     | ✅     | ❌        | ✅     | ❌       | ❌         | ❌     | ❌       |
-| **Multimodal Prompts** *(text+image)* | ✅     | ✅        | ✅     | ✅       | ✅         | ✅     | ✅       |
+| **Multimodal Prompts** *(text, documents, audio, images, videos, URLs, etc)* | ✅     | ✅        | ✅     | ✅       | ✅         | ✅     | ✅       |
 | **Embeddings**                       | ✅     | ✅        | ✅     | ✅       | ❌         | ✅     | ✅       |
-| **Models API**                       g| ✅     | ✅        | ✅     | ✅       | ✅         | ✅     | ✅       |
+| **Models API**                       | ✅     | ✅        | ✅     | ✅       | ✅         | ✅     | ✅       |
 | **Local Model Support**              | ❌     | ❌        | ❌     | ❌       | ❌         | ✅     | ✅       |
 | **Vector Stores (RAG)**               | ✅     | ❌        | ❌     | ❌       | ❌         | ❌     | ❌       |
 | **Responses**                        | ✅     | ❌        | ❌     | ❌       | ❌         | ❌     | ❌       |
@@ -110,12 +110,12 @@ require "llm"
 
 llm  = LLM.openai(key: ENV["KEY"])
 bot  = LLM::Bot.new(llm)
-url  = "https://upload.wikimedia.org/wikipedia/commons/thumb/9/9a/Cognac_glass.jpg/500px-Cognac_glass.jpg"
+url  = "https://en.wikipedia.org/wiki/Special:FilePath/Cognac_glass.jpg"
 msgs = bot.chat do |prompt|
   prompt.system "Your task is to answer all user queries"
   prompt.user ["Tell me about this URL", URI(url)]
-  prompt.user ["Tell me about this pdf", File.open("spec/fixtures/documents/freebsd.sysctl.pdf", "rb")]
-  prompt.user "Is the URL and PDF similar to each other?"
+  prompt.user ["Tell me about this PDF", File.open("handbook.pdf", "rb")]
+  prompt.user "Are the URL and PDF similar to each other?"
 end
 
 # At this point, we execute a single request
@@ -142,12 +142,12 @@ require "llm"
 
 llm = LLM.openai(key: ENV["KEY"])
 bot = LLM::Bot.new(llm)
-url = "https://upload.wikimedia.org/wikipedia/commons/thumb/9/9a/Cognac_glass.jpg/500px-Cognac_glass.jpg"
+url = "https://en.wikipedia.org/wiki/Special:FilePath/Cognac_glass.jpg"
 bot.chat(stream: $stdout) do |prompt|
   prompt.system "Your task is to answer all user queries"
   prompt.user ["Tell me about this URL", URI(url)]
-  prompt.user ["Tell me about this pdf", File.open("spec/fixtures/documents/freebsd.sysctl.pdf", "rb")]
-  prompt.user "Is the URL and PDF similar to each other?"
+  prompt.user ["Tell me about this PDF", File.open("handbook.pdf", "rb")]
+  prompt.user "Are the URL and PDF similar to each other?"
 end.to_a
 ```
 

data/lib/llm/bot/conversable.rb

@@ -12,8 +12,12 @@ class LLM::Bot
     # @param [Hash] params
     # @return [void]
     def async_response(prompt, params = {})
-      role = params.delete(:role)
-      @messages << [LLM::Message.new(role, prompt), @params.merge(params), :respond]
+      if Array === prompt and prompt.empty?
+        @messages
+      else
+        role = params.delete(:role)
+        @messages << [LLM::Message.new(role, prompt), @params.merge(params), :respond]
+      end
     end
 
     ##
@@ -22,8 +26,12 @@ class LLM::Bot
     # @param [Hash] params
     # @return [void]
     def async_completion(prompt, params = {})
-      role = params.delete(:role)
-      @messages.push [LLM::Message.new(role, prompt), @params.merge(params), :complete]
+      if Array === prompt and prompt.empty?
+        @messages
+      else
+        role = params.delete(:role)
+        @messages << [LLM::Message.new(role, prompt), @params.merge(params), :complete]
+      end
     end
   end
 end

data/lib/llm/bot/prompt/completion.rb

@@ -27,5 +27,23 @@ module LLM::Bot::Prompt
       params = defaults.merge(params)
       bot.chat prompt, params.merge(role: :user)
     end
+
+    ##
+    # @param [String] prompt
+    # @param [Hash] params (see LLM::Provider#complete)
+    # @return [LLM::Bot]
+    def assistant(prompt, params = {})
+      params = defaults.merge(params)
+      bot.chat prompt, params.merge(role: :assistant)
+    end
+
+    ##
+    # @param [String] prompt
+    # @param [Hash] params (see LLM::Provider#complete)
+    # @return [LLM::Bot]
+    def model(prompt, params = {})
+      params = defaults.merge(params)
+      bot.chat prompt, params.merge(role: :model)
+    end
   end
 end

data/lib/llm/bot/prompt/respond.rb

@@ -36,5 +36,14 @@ module LLM::Bot::Prompt
       params = defaults.merge(params)
       bot.respond prompt, params.merge(role: :user)
     end
+
+    ##
+    # @param [String] prompt
+    # @param [Hash] params (see LLM::Provider#complete)
+    # @return [LLM::Bot]
+    def assistant(prompt, params = {})
+      params = defaults.merge(params)
+      bot.chat prompt, params.merge(role: :assistant)
+    end
   end
 end

data/lib/llm/bot.rb

@@ -13,12 +13,12 @@ module LLM
   #
   #   llm  = LLM.openai(key: ENV["KEY"])
   #   bot  = LLM::Bot.new(llm)
-  #   url  = "https://upload.wikimedia.org/wikipedia/commons/thumb/9/9a/Cognac_glass.jpg/500px-Cognac_glass.jpg"
+  #   url  = "https://en.wikipedia.org/wiki/Special:FilePath/Cognac_glass.jpg"
   #   msgs = bot.chat do |prompt|
   #     prompt.system "Your task is to answer all user queries"
   #     prompt.user ["Tell me about this URL", URI(url)]
-  #     prompt.user ["Tell me about this pdf", File.open("freebsd_book.pdf", "rb")]
-  #     prompt.user "Is the URL and PDF similar to each other?"
+  #     prompt.user ["Tell me about this PDF", File.open("handbook.pdf", "rb")]
+  #     prompt.user "Are the URL and PDF similar to each other?"
   #   end
   #
   #   # At this point, we execute a single request

data/lib/llm/buffer.rb

@@ -55,7 +55,7 @@ module LLM
     end
 
     ##
-    # @param [[LLM::Message, Hash]] item
+    # @param [[LLM::Message, Hash, Symbol]] item
     #  A message and its parameters
     # @return [void]
     def <<(item)
@@ -80,6 +80,13 @@ module LLM
       "completed_count=#{@completed.size} pending_count=#{@pending.size}>"
     end
 
+    ##
+    # Returns true when the buffer is empty
+    # @return [Boolean]
+    def empty?
+      @pending.empty? and @completed.empty?
+    end
+
     private
 
     def empty!

data/lib/llm/error.rb

@@ -31,6 +31,10 @@ module LLM
   # HTTPTooManyRequests
   RateLimitError = Class.new(ResponseError)
 
+  ##
+  # HTTPServerError
+  ServerError = Class.new(ResponseError)
+
   ##
   # When an given an input object that is not understood
   FormatError = Class.new(Error)

data/lib/llm/function.rb

@@ -53,7 +53,7 @@ class LLM::Function
   # @yieldparam [LLM::Function] self The function object
   def initialize(name, &b)
     @name = name
-    @schema = JSON::Schema.new
+    @schema = LLM::Schema.new
     @called = false
     @cancelled = false
     yield(self)
@@ -72,7 +72,7 @@ class LLM::Function
   end
 
   ##
-  # @yieldparam [JSON::Schema] schema The schema object
+  # @yieldparam [LLM::Schema] schema The schema object
   # @return [void]
   def params
     @params = yield(@schema)

data/lib/llm/mime.rb

@@ -7,11 +7,8 @@ class LLM::Mime
   # Lookup a mime type
   # @return [String, nil]
   def self.[](key)
-    if key.respond_to?(:path)
-      types[File.extname(key.path)]
-    else
-      types[key]
-    end
+    key = key.respond_to?(:path) ? File.extname(key.path) : key
+    types[key] || "application/octet-stream"
   end
 
   ##
@@ -24,6 +21,15 @@ class LLM::Mime
       ".jpg" => "image/jpeg",
       ".jpeg" => "image/jpeg",
       ".webp" => "image/webp",
+      ".gif" => "image/gif",
+      ".bmp" => "image/bmp",
+      ".tif" => "image/tiff",
+      ".tiff" => "image/tiff",
+      ".svg" => "image/svg+xml",
+      ".ico" => "image/x-icon",
+      ".apng" => "image/apng",
+      ".jfif" => "image/jpeg",
+      ".heic" => "image/heic",
 
       # Videos
       ".flv" => "video/x-flv",
@@ -34,6 +40,12 @@ class LLM::Mime
       ".webm" => "video/webm",
       ".wmv" => "video/x-ms-wmv",
       ".3gp" => "video/3gpp",
+      ".avi" => "video/x-msvideo",
+      ".mkv" => "video/x-matroska",
+      ".ogv" => "video/ogg",
+      ".m4v" => "video/x-m4v",
+      ".m2ts" => "video/mp2t",
+      ".mts" => "video/mp2t",
 
       # Audio
       ".aac" => "audio/aac",
@@ -45,10 +57,80 @@ class LLM::Mime
       ".pcm" => "audio/L16",
       ".wav" => "audio/wav",
       ".weba" => "audio/webm",
+      ".oga" => "audio/ogg",
+      ".ogg" => "audio/ogg",
+      ".mid" => "audio/midi",
+      ".midi" => "audio/midi",
+      ".aiff" => "audio/aiff",
+      ".aif" => "audio/aiff",
+      ".amr" => "audio/amr",
+      ".mka" => "audio/x-matroska",
+      ".caf" => "audio/x-caf",
 
       # Documents
       ".pdf" => "application/pdf",
-      ".txt" => "text/plain"
+      ".txt" => "text/plain",
+      ".md" => "text/markdown",
+      ".markdown" => "text/markdown",
+      ".mkd" => "text/markdown",
+      ".doc" => "application/msword",
+      ".docx" => "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
+      ".xls" => "application/vnd.ms-excel",
+      ".xlsx" => "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
+      ".ppt" => "application/vnd.ms-powerpoint",
+      ".pptx" => "application/vnd.openxmlformats-officedocument.presentationml.presentation",
+      ".csv" => "text/csv",
+      ".json" => "application/json",
+      ".xml" => "application/xml",
+      ".html" => "text/html",
+      ".htm" => "text/html",
+      ".odt" => "application/vnd.oasis.opendocument.text",
+      ".odp" => "application/vnd.oasis.opendocument.presentation",
+      ".ods" => "application/vnd.oasis.opendocument.spreadsheet",
+      ".rtf" => "application/rtf",
+      ".epub" => "application/epub+zip",
+
+      # Code
+      ".js" => "application/javascript",
+      ".jsx" => "text/jsx",
+      ".ts" => "application/typescript",
+      ".tsx" => "text/tsx",
+      ".css" => "text/css",
+      ".c" => "text/x-c",
+      ".cpp" => "text/x-c++",
+      ".h" => "text/x-c",
+      ".rb" => "text/x-ruby",
+      ".py" => "text/x-python",
+      ".java" => "text/x-java-source",
+      ".sh" => "application/x-sh",
+      ".php" => "application/x-httpd-php",
+      ".yml" => "text/yaml",
+      ".yaml" => "text/yaml",
+      ".go" => "text/x-go",
+      ".rs" => "text/rust",
+
+      # Fonts
+      ".woff" => "font/woff",
+      ".woff2" => "font/woff2",
+      ".ttf" => "font/ttf",
+      ".otf" => "font/otf",
+
+      # Archives
+      ".zip" => "application/zip",
+      ".tar" => "application/x-tar",
+      ".gz" => "application/gzip",
+      ".bz2" => "application/x-bzip2",
+      ".xz" => "application/x-xz",
+      ".rar" => "application/vnd.rar",
+      ".7z" => "application/x-7z-compressed",
+      ".tar.gz" => "application/gzip",
+      ".tar.bz2" => "application/x-bzip2",
+      ".apk" => "application/vnd.android.package-archive",
+      ".exe" => "application/x-msdownload",
+
+      # Others
+      ".ics" => "text/calendar",
+      ".vcf" => "text/vcard"
     }
   end
 end

data/lib/llm/provider.rb

@@ -198,9 +198,9 @@ class LLM::Provider
 
   ##
   # Returns an object that can generate a JSON schema
-  # @return [JSON::Schema]
+  # @return [LLM::Schema]
   def schema
-    @schema ||= JSON::Schema.new
+    @schema ||= LLM::Schema.new
   end
 
   ##

data/lib/llm/providers/anthropic/error_handler.rb

@@ -22,6 +22,8 @@ class LLM::Anthropic
     #  Raises a subclass of {LLM::Error LLM::Error}
     def raise_error!
       case res
+      when Net::HTTPServerError
+        raise LLM::ServerError.new { _1.response = res }, "Server error"
       when Net::HTTPUnauthorized
         raise LLM::UnauthorizedError.new { _1.response = res }, "Authentication error"
       when Net::HTTPTooManyRequests

data/lib/llm/providers/gemini/error_handler.rb

@@ -22,6 +22,8 @@ class LLM::Gemini
     #  Raises a subclass of {LLM::Error LLM::Error}
     def raise_error!
       case res
+      when Net::HTTPServerError
+        raise LLM::ServerError.new { _1.response = res }, "Server error"
       when Net::HTTPBadRequest
         reason = body.dig("error", "details", 0, "reason")
         if reason == "API_KEY_INVALID"

data/lib/llm/providers/gemini/response/completion.rb

@@ -28,5 +28,7 @@ module LLM::Gemini::Response
         LLM::Object.new(function)
       end
     end
+
+    def candidates = body.candidates || []
   end
 end

data/lib/llm/providers/ollama/error_handler.rb

@@ -22,6 +22,8 @@ class LLM::Ollama
     #  Raises a subclass of {LLM::Error LLM::Error}
     def raise_error!
       case res
+      when Net::HTTPServerError
+        raise LLM::ServerError.new { _1.response = res }, "Server error"
       when Net::HTTPUnauthorized
         raise LLM::UnauthorizedError.new { _1.response = res }, "Authentication error"
       when Net::HTTPTooManyRequests

data/lib/llm/providers/openai/error_handler.rb

@@ -22,13 +22,25 @@ class LLM::OpenAI
     #  Raises a subclass of {LLM::Error LLM::Error}
     def raise_error!
       case res
+      when Net::HTTPServerError
+        raise LLM::ServerError.new { _1.response = res }, "Server error"
       when Net::HTTPUnauthorized
         raise LLM::UnauthorizedError.new { _1.response = res }, "Authentication error"
       when Net::HTTPTooManyRequests
         raise LLM::RateLimitError.new { _1.response = res }, "Too many requests"
       else
-        raise LLM::ResponseError.new { _1.response = res }, "Unexpected response"
+        error = body["error"] || {}
+        case error["type"]
+        when "server_error" then raise LLM::ServerError.new { _1.response = res }, error["message"]
+        else raise LLM::ResponseError.new { _1.response = res }, error["message"] || "Unexpected response"
+        end
       end
     end
+
+    private
+
+    def body
+      @body ||= JSON.parse(res.body)
+    end
   end
 end

data/lib/llm/providers/xai/images.rb

@@ -39,7 +39,7 @@ class LLM::XAI
     # @param [Hash] params Other parameters (see xAI docs)
     # @raise (see LLM::Provider#request)
     # @return [LLM::Response]
-    def create(model: "grok-2-image-1212", **)
+    def create(prompt:, model: "grok-2-image-1212", **params)
       super
     end
 

data/lib/llm/{json/schema → schema}/array.rb

@@ -1,10 +1,10 @@
 # frozen_string_literal: true
 
-class JSON::Schema
+class LLM::Schema
   ##
-  # The {JSON::Schema::Array JSON::Schema::Array} class represents an
+  # The {LLM::Schema::Array LLM::Schema::Array} class represents an
   # array value in a JSON schema. It is a subclass of
-  # {JSON::Schema::Leaf JSON::Schema::Leaf} and provides methods that
+  # {LLM::Schema::Leaf LLM::Schema::Leaf} and provides methods that
   # can act as constraints.
   class Array < Leaf
     def initialize(items)

data/lib/llm/{json/schema → schema}/boolean.rb

@@ -1,10 +1,10 @@
 # frozen_string_literal: true
 
-class JSON::Schema
+class LLM::Schema
   ##
-  # The {JSON::Schema::Boolean JSON::Schema::Boolean} class represents a
+  # The {LLM::Schema::Boolean LLM::Schema::Boolean} class represents a
   # boolean value in a JSON schema. It is a subclass of
-  # {JSON::Schema::Leaf JSON::Schema::Leaf}.
+  # {LLM::Schema::Leaf LLM::Schema::Leaf}.
   class Boolean < Leaf
     def to_h
       super.merge!({type: "boolean"})

data/lib/llm/{json/schema → schema}/integer.rb

@@ -1,16 +1,16 @@
 # frozen_string_literal: true
 
-class JSON::Schema
+class LLM::Schema
   ##
-  # The {JSON::Schema::Integer JSON::Schema::Integer} class represents a
+  # The {LLM::Schema::Integer LLM::Schema::Integer} class represents a
   # whole number value in a JSON schema. It is a subclass of
-  # {JSON::Schema::Leaf JSON::Schema::Leaf} and provides methods that
+  # {LLM::Schema::Leaf LLM::Schema::Leaf} and provides methods that
   # can act as constraints.
   class Integer < Leaf
     ##
     # Constrain the number to a minimum value
     # @param [Integer] i The minimum value
-    # @return [JSON::Schema::Number] Returns self
+    # @return [LLM::Schema::Number] Returns self
     def min(i)
       tap { @minimum = i }
     end
@@ -18,7 +18,7 @@ class JSON::Schema
     ##
     # Constrain the number to a maximum value
     # @param [Integer] i The maximum value
-    # @return [JSON::Schema::Number] Returns self
+    # @return [LLM::Schema::Number] Returns self
     def max(i)
       tap { @maximum = i }
     end
@@ -26,7 +26,7 @@ class JSON::Schema
     ##
     # Constrain the number to a multiple of a given value
     # @param [Integer] i The multiple
-    # @return [JSON::Schema::Number] Returns self
+    # @return [LLM::Schema::Number] Returns self
     def multiple_of(i)
       tap { @multiple_of = i }
     end

data/lib/llm/{json/schema → schema}/leaf.rb

@@ -1,11 +1,11 @@
 # frozen_string_literal: true
 
-class JSON::Schema
+class LLM::Schema
   ##
-  # The {JSON::Schema::Leaf JSON::Schema::Leaf} class is the
+  # The {LLM::Schema::Leaf LLM::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}
+  # See the instance methods of {LLM::Schema LLM::Schema} for
+  # an example of how to create instances of {LLM::Schema::Leaf LLM::Schema::Leaf}
   # through its subclasses.
   class Leaf
     def initialize
@@ -19,7 +19,7 @@ class JSON::Schema
     ##
     # Set the description of a leaf
     # @param [String] str The description
-    # @return [JSON::Schema::Leaf]
+    # @return [LLM::Schema::Leaf]
     def description(str)
       tap { @description = str }
     end
@@ -27,7 +27,7 @@ class JSON::Schema
     ##
     # Set the default value of a leaf
     # @param [Object] value The default value
-    # @return [JSON::Schema::Leaf]
+    # @return [LLM::Schema::Leaf]
     def default(value)
       tap { @default = value }
     end
@@ -36,7 +36,7 @@ class JSON::Schema
     # Set the allowed values of a leaf
     # @see https://tour.json-schema.org/content/02-Primitive-Types/07-Enumerated-Values-II Enumerated Values
     # @param [Array] values The allowed values
-    # @return [JSON::Schema::Leaf]
+    # @return [LLM::Schema::Leaf]
     def enum(*values)
       tap { @enum = values }
     end
@@ -45,14 +45,14 @@ class JSON::Schema
     # Set the value of a leaf to be a constant value
     # @see https://tour.json-schema.org/content/02-Primitive-Types/08-Defining-Constant-Values Constant Values
     # @param [Object] value The constant value
-    # @return [JSON::Schema::Leaf]
+    # @return [LLM::Schema::Leaf]
     def const(value)
       tap { @const = value }
     end
 
     ##
     # Denote a leaf as required
-    # @return [JSON::Schema::Leaf]
+    # @return [LLM::Schema::Leaf]
     def required
       tap { @required = true }
     end

data/lib/llm/{json/schema → schema}/null.rb

@@ -1,10 +1,10 @@
 # frozen_string_literal: true
 
-class JSON::Schema
+class LLM::Schema
   ##
-  # The {JSON::Schema::Null JSON::Schema::Null} class represents a
+  # The {LLM::Schema::Null LLM::Schema::Null} class represents a
   # null value in a JSON schema. It is a subclass of
-  # {JSON::Schema::Leaf JSON::Schema::Leaf}.
+  # {LLM::Schema::Leaf LLM::Schema::Leaf}.
   class Null < Leaf
     def to_h
       super.merge!({type: "null"})

data/lib/llm/{json/schema → schema}/number.rb

@@ -1,16 +1,16 @@
 # frozen_string_literal: true
 
-class JSON::Schema
+class LLM::Schema
   ##
-  # The {JSON::Schema::Number JSON::Schema::Number} class represents a
+  # The {LLM::Schema::Number LLM::Schema::Number} class represents a
   # a number (either whole or decimal) value in a JSON schema. It is a
-  # subclass of {JSON::Schema::Leaf JSON::Schema::Leaf} and provides
+  # subclass of {LLM::Schema::Leaf LLM::Schema::Leaf} and provides
   # methods that can act as constraints.
   class Number < Leaf
     ##
     # Constrain the number to a minimum value
     # @param [Integer, Float] i The minimum value
-    # @return [JSON::Schema::Number] Returns self
+    # @return [LLM::Schema::Number] Returns self
     def min(i)
       tap { @minimum = i }
     end
@@ -18,7 +18,7 @@ class JSON::Schema
     ##
     # Constrain the number to a maximum value
     # @param [Integer, Float] i The maximum value
-    # @return [JSON::Schema::Number] Returns self
+    # @return [LLM::Schema::Number] Returns self
     def max(i)
       tap { @maximum = i }
     end
@@ -26,7 +26,7 @@ class JSON::Schema
     ##
     # Constrain the number to a multiple of a given value
     # @param [Integer, Float] i The multiple
-    # @return [JSON::Schema::Number] Returns self
+    # @return [LLM::Schema::Number] Returns self
     def multiple_of(i)
       tap { @multiple_of = i }
     end

data/lib/llm/{json/schema → schema}/object.rb

@@ -1,10 +1,10 @@
 # frozen_string_literal: true
 
-class JSON::Schema
+class LLM::Schema
   ##
-  # The {JSON::Schema::Object JSON::Schema::Object} class represents an
+  # The {LLM::Schema::Object LLM::Schema::Object} class represents an
   # object value in a JSON schema. It is a subclass of
-  # {JSON::Schema::Leaf JSON::Schema::Leaf} and provides methods that
+  # {LLM::Schema::Leaf LLM::Schema::Leaf} and provides methods that
   # can act as constraints.
   class Object < Leaf
     def initialize(properties)

data/lib/llm/{json/schema → schema}/string.rb

@@ -1,16 +1,16 @@
 # frozen_string_literal: true
 
-class JSON::Schema
+class LLM::Schema
   ##
-  # The {JSON::Schema::String JSON::Schema::String} class represents a
+  # The {LLM::Schema::String LLM::Schema::String} class represents a
   # string value in a JSON schema. It is a subclass of
-  # {JSON::Schema::Leaf JSON::Schema::Leaf} and provides methods that
+  # {LLM::Schema::Leaf LLM::Schema::Leaf} and provides methods that
   # can act as constraints.
   class String < Leaf
     ##
     # Constrain the string to a minimum length
     # @param [Integer] i The minimum length
-    # @return [JSON::Schema::String] Returns self
+    # @return [LLM::Schema::String] Returns self
     def min(i)
       tap { @minimum = i }
     end
@@ -18,7 +18,7 @@ class JSON::Schema
     ##
     # Constrain the string to a maximum length
     # @param [Integer] i The maximum length
-    # @return [JSON::Schema::String] Returns self
+    # @return [LLM::Schema::String] Returns self
     def max(i)
       tap { @maximum = i }
     end

data/lib/llm/{json/schema → schema}/version.rb

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

data/lib/llm/{json/schema.rb → schema.rb}

@@ -1,10 +1,7 @@
 # frozen_string_literal: true
 
-module JSON
-end unless defined?(JSON)
-
 ##
-# The {JSON::Schema JSON::Schema} class represents a JSON schema,
+# The {LLM::Schema LLM::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.
@@ -13,14 +10,14 @@ end unless defined?(JSON)
 # @see https://tour.json-schema.org/ JSON Schema Tour
 #
 # @example
-#  schema = JSON::Schema.new
+#  schema = LLM::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
+class LLM::Schema
   require_relative "schema/version"
   require_relative "schema/leaf"
   require_relative "schema/object"
@@ -34,7 +31,7 @@ class JSON::Schema
   ##
   # Returns an object
   # @param [Hash] properties A hash of properties
-  # @return [JSON::Schema::Object]
+  # @return [LLM::Schema::Object]
   def object(properties)
     Object.new(properties)
   end
@@ -42,42 +39,42 @@ class JSON::Schema
   ##
   # Returns an array
   # @param [Array] items An array of items
-  # @return [JSON::Schema::Array]
+  # @return [LLM::Schema::Array]
   def array(*items)
     Array.new(*items)
   end
 
   ##
   # Returns a string
-  # @return [JSON::Schema::String]
+  # @return [LLM::Schema::String]
   def string
     String.new
   end
 
   ##
   # Returns a number
-  # @return [JSON::Schema::Number] a number
+  # @return [LLM::Schema::Number] a number
   def number
     Number.new
   end
 
   ##
   # Returns an integer
-  # @return [JSON::Schema::Integer]
+  # @return [LLM::Schema::Integer]
   def integer
     Integer.new
   end
 
   ##
   # Returns a boolean
-  # @return [JSON::Schema::Boolean]
+  # @return [LLM::Schema::Boolean]
   def boolean
     Boolean.new
   end
 
   ##
   # Returns null
-  # @return [JSON::Schema::Null]
+  # @return [LLM::Schema::Null]
   def null
     Null.new
   end

data/lib/llm/version.rb

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

data/lib/llm.rb

@@ -2,7 +2,7 @@
 
 module LLM
   require "stringio"
-  require_relative "llm/json/schema"
+  require_relative "llm/schema"
   require_relative "llm/object"
   require_relative "llm/version"
   require_relative "llm/utils"

data/llm.gemspec

@@ -12,7 +12,7 @@ Gem::Specification.new do |spec|
   llm.rb is a zero-dependency Ruby toolkit for Large Language Models that
   includes OpenAI, Gemini, Anthropic, xAI (grok), DeepSeek, Ollama, and
   LlamaCpp. The toolkit includes full support for chat, streaming, tool calling,
-  audio, images, files, and JSON Schema generation.
+  audio, images, files, and structured outputs (JSON Schema).
   SUMMARY
 
   spec.description = spec.summary

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llm.rb
 version: !ruby/object:Gem::Version
-  version: 0.12.0
+  version: 0.13.0
 platform: ruby
 authors:
 - Antar Azri
@@ -153,7 +153,7 @@ dependencies:
 description: llm.rb is a zero-dependency Ruby toolkit for Large Language Models that
   includes OpenAI, Gemini, Anthropic, xAI (grok), DeepSeek, Ollama, and LlamaCpp.
   The toolkit includes full support for chat, streaming, tool calling, audio, images,
-  files, and JSON Schema generation.
+  files, and structured outputs (JSON Schema).
 email:
 - azantar@proton.me
 - 0x1eef@proton.me
@@ -177,16 +177,6 @@ files:
 - lib/llm/eventstream/parser.rb
 - lib/llm/file.rb
 - lib/llm/function.rb
-- lib/llm/json/schema.rb
-- lib/llm/json/schema/array.rb
-- lib/llm/json/schema/boolean.rb
-- lib/llm/json/schema/integer.rb
-- lib/llm/json/schema/leaf.rb
-- lib/llm/json/schema/null.rb
-- lib/llm/json/schema/number.rb
-- lib/llm/json/schema/object.rb
-- lib/llm/json/schema/string.rb
-- lib/llm/json/schema/version.rb
 - lib/llm/message.rb
 - lib/llm/mime.rb
 - lib/llm/multipart.rb
@@ -250,6 +240,16 @@ files:
 - lib/llm/providers/xai.rb
 - lib/llm/providers/xai/images.rb
 - lib/llm/response.rb
+- lib/llm/schema.rb
+- lib/llm/schema/array.rb
+- lib/llm/schema/boolean.rb
+- lib/llm/schema/integer.rb
+- lib/llm/schema/leaf.rb
+- lib/llm/schema/null.rb
+- lib/llm/schema/number.rb
+- lib/llm/schema/object.rb
+- lib/llm/schema/string.rb
+- lib/llm/schema/version.rb
 - lib/llm/utils.rb
 - lib/llm/version.rb
 - llm.gemspec
@@ -278,5 +278,5 @@ specification_version: 4
 summary: llm.rb is a zero-dependency Ruby toolkit for Large Language Models that includes
   OpenAI, Gemini, Anthropic, xAI (grok), DeepSeek, Ollama, and LlamaCpp. The toolkit
   includes full support for chat, streaming, tool calling, audio, images, files, and
-  JSON Schema generation.
+  structured outputs (JSON Schema).
 test_files: []