ruby_llm-schema 0.1.7 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4706281dcd180f4d6e2b385cc42b79d73629d8eb8a7b5723ddde3b2786a5022e
-  data.tar.gz: eb083d6fa029a0b107f7a728c86e9f8342a1b9d0c3d0a46d8d4d33adaacd756a
+  metadata.gz: 2a5add3bd7f57004c32b88a3fb676370622b0d94931b3e8ad46b2aa021da96f7
+  data.tar.gz: 1d371d0595ba66e8e0a31b42f524be07569cf837eb2a044a560cad8f412ea17d
 SHA512:
-  metadata.gz: bc52ea4b47664561909278ece104676ae5053572aa791fe6985deb1ecdf40421e49b3b02529ba2a02cbb1df068484d6d162ec0d4c8d525105111ced98451fcbf
-  data.tar.gz: df555cdab1fb1c786ee0425fb5a1c0dd61bd5ed6c35dfc53bb355cdad27b462c5dbd84cb5d8fb32db6fdbe2c364916f0c8d2ed32565bff87c8ca17b79915ee03
+  metadata.gz: e558d54dbe37fde9df32d7c0aa3116d95687e8f1b76e9d6260f10025242671ce67bdb5f44f6020a757dc38ec693e764c11e20388420c6fd65f7ccc7efa391394
+  data.tar.gz: d6ba597b33675f573bb3b9d728f128a2bfc431a47e6c87ef2ed37d8c5866a8475eae5f20eb1b670edf31dfd3bdf7d7e79a6329ef68550538930eb4a8aa13030e

data/README.md

@@ -1,5 +1,9 @@
 # RubyLLM::Schema
 
+[![Gem Version](https://badge.fury.io/rb/ruby_llm-schema.svg)](https://rubygems.org/gems/ruby_llm-schema)
+[![GitHub license](https://img.shields.io/badge/license-MIT-blue.svg)](https://github.com/danielfriis/ruby_llm-schema/blob/main/LICENSE.txt)
+[![CI](https://github.com/danielfriis/ruby_llm-schema/actions/workflows/ci.yml/badge.svg?branch=main)](https://github.com/danielfriis/ruby_llm-schema/actions/workflows/ci.yml)
+
 A Ruby DSL for creating JSON schemas with a clean, Rails-inspired API. Perfect for defining structured data schemas for LLM function calling or structured outputs.
 
 ## Use Cases
@@ -267,9 +271,14 @@ class MySchema < RubyLLM::Schema
     string :longitude
   end
   
+  # Using a reference in an array
   array :coordinates, of: :location
-  
-  object :home_location do
+
+  # Using a reference in an object via the `reference` option
+  object :home_location, reference: :location
+
+  # Using a reference in an object via block
+  object :user do
     reference :location
   end
 end

data/lib/ruby_llm/schema/dsl.rb

@@ -41,8 +41,8 @@ module RubyLLM
       end
 
       # Complex type methods
-      def object(name = nil, description: nil, required: true, &block)
-        add_property(name, build_property_schema(:object, description: description, &block), required: required)
+      def object(name = nil, reference: nil, description: nil, required: true, &block)
+        add_property(name, build_property_schema(:object, description: description, reference: reference, &block), required: required)
       end
 
       def array(name, of: nil, description: nil, required: true, min_items: nil, max_items: nil, &block)
@@ -123,16 +123,26 @@ module RubyLLM
         when :null
           {type: "null", description: options[:description]}.compact
         when :object
+          # If the reference option is provided, return the reference
+          return reference(options[:reference]) if options[:reference]
+
           sub_schema = Class.new(Schema)
-          sub_schema.class_eval(&)
 
-          {
-            type: "object",
-            properties: sub_schema.properties,
-            required: sub_schema.required_properties,
-            additionalProperties: additional_properties,
-            description: options[:description]
-          }.compact
+          # Evaluate the block and capture the result
+          result = sub_schema.class_eval(&)
+
+          # If the block returned a reference and no properties were added, use the reference
+          if result.is_a?(Hash) && result["$ref"] && sub_schema.properties.empty?
+            result.merge(options[:description] ? {description: options[:description]} : {})
+          else
+            {
+              type: "object",
+              properties: sub_schema.properties,
+              required: sub_schema.required_properties,
+              additionalProperties: additional_properties,
+              description: options[:description]
+            }.compact
+          end
         when :any_of
           schemas = collect_property_schemas_from_block(&)
           {

data/lib/ruby_llm/schema/version.rb

@@ -2,6 +2,6 @@
 
 module RubyLlm
   class Schema
-    VERSION = "0.1.7"
+    VERSION = "0.1.8"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm-schema
 version: !ruby/object:Gem::Version
-  version: 0.1.7
+  version: 0.1.8
 platform: ruby
 authors:
 - Daniel Friis