Fix

Author: danielfriis

lib/ruby_llm/schema/dsl.rb

@@ -14,4 +14,4 @@ module DSL
       include Utilities
     end
   end
-end
+end

lib/ruby_llm/schema/dsl/complex_types.rb

@@ -4,20 +4,20 @@ module RubyLLM
   class Schema
     module DSL
       module ComplexTypes
-        def object(name, required: true, **options, &block)
-          add_property(name, object_schema(**options, &block), required: required)
+        def object(name, description: nil, required: true, **options, &block)
+          add_property(name, object_schema(description: description, **options, &block), required: required)
         end
 
-        def array(name, required: true, **options, &block)
-          add_property(name, array_schema(**options, &block), required: required)
+        def array(name, description: nil, required: true, **options, &block)
+          add_property(name, array_schema(description: description, **options, &block), required: required)
         end
 
-        def any_of(name, required: true, **options, &block)
-          add_property(name, any_of_schema(**options, &block), required: required)
+        def any_of(name, description: nil, required: true, **options, &block)
+          add_property(name, any_of_schema(description: description, **options, &block), required: required)
         end
 
-        def optional(name, **options, &block)
-          any_of(name, **options) do
+        def optional(name, description: nil, &block)
+          any_of(name, description: description) do
             instance_eval(&block)
             null
           end

lib/ruby_llm/schema/dsl/primitive_types.rb

@@ -4,24 +4,24 @@ module RubyLLM
   class Schema
     module DSL
       module PrimitiveTypes
-        def string(name, required: true, **options)
-          add_property(name, string_schema(**options), required: required)
+        def string(name, description: nil, required: true, **options)
+          add_property(name, string_schema(description: description, **options), required: required)
         end
 
-        def number(name, required: true, **options)
-          add_property(name, number_schema(**options), required: required)
+        def number(name, description: nil, required: true, **options)
+          add_property(name, number_schema(description: description, **options), required: required)
         end
 
-        def integer(name, required: true, **options)
-          add_property(name, integer_schema(**options), required: required)
+        def integer(name, description: nil, required: true, **options)
+          add_property(name, integer_schema(description: description, **options), required: required)
         end
 
-        def boolean(name, required: true, **options)
-          add_property(name, boolean_schema(**options), required: required)
+        def boolean(name, description: nil, required: true, **options)
+          add_property(name, boolean_schema(description: description, **options), required: required)
         end
 
-        def null(name, required: true, **options)
-          add_property(name, null_schema(**options), required: required)
+        def null(name, description: nil, required: true, **options)
+          add_property(name, null_schema(description: description, **options), required: required)
         end
       end
     end

lib/ruby_llm/schema/dsl/schema_builders.rb

@@ -4,7 +4,7 @@ module RubyLLM
   class Schema
     module DSL
       module SchemaBuilders
-        def string_schema(enum: nil, description: nil, min_length: nil, max_length: nil, pattern: nil, format: nil)
+        def string_schema(description: nil, enum: nil, min_length: nil, max_length: nil, pattern: nil, format: nil)
           {
             type: "string",
             enum: enum,
@@ -44,7 +44,7 @@ def null_schema(description: nil)
           {type: "null", description: description}.compact
         end
 
-        def object_schema(reference: nil, description: nil, &block)
+        def object_schema(description: nil, reference: nil, &block)
           if reference
             reference(reference)
           else
@@ -66,7 +66,7 @@ def object_schema(reference: nil, description: nil, &block)
           end
         end
 
-        def array_schema(of: nil, description: nil, min_items: nil, max_items: nil, &block)
+        def array_schema(description: nil, of: nil, min_items: nil, max_items: nil, &block)
           items = determine_array_items(of, &block)
 
           {

lib/ruby_llm/schema/dsl/utilities.rb

@@ -45,11 +45,11 @@ def collect_schemas_from_block(&block)
           schema_builder = self
 
           context = Object.new
-          
+
           # Dynamically create methods for all schema builders
           schema_builder.methods.grep(/_schema$/).each do |schema_method|
-            type_name = schema_method.to_s.sub(/_schema$/, '')
-            
+            type_name = schema_method.to_s.sub(/_schema$/, "")
+
             context.define_singleton_method(type_name) do |name = nil, **options, &blk|
               schemas << schema_builder.send(schema_method, **options, &blk)
             end