Add enum parameter to number and integer schema methods

README.md

@@ -241,15 +241,17 @@ string :code, min_length: 3, max_length: 10
 
 ### Numbers
 
-Number types support the following properties:
+Number and integer types support the following properties:
 
+- `enum`: an array of allowed numeric values (e.g. `enum: [0, 1, 2]`)
 - `multiple_of`: a multiple of the number (e.g. `multiple_of: 0.01`)
 - `minimum`: the minimum value of the number (e.g. `minimum: 0`)
 - `maximum`: the maximum value of the number (e.g. `maximum: 100`)
 
 ```ruby
 number :price, minimum: 0, maximum: 100
 number :amount, multiple_of: 0.01
+integer :level, enum: [0, 1, 2]
 ```
 
 ### Booleans

lib/ruby_llm/schema/dsl/schema_builders.rb

@@ -16,23 +16,25 @@ def string_schema(description: nil, enum: nil, min_length: nil, max_length: nil,
           }.compact
         end
 
-        def number_schema(description: nil, minimum: nil, maximum: nil, multiple_of: nil)
+        def number_schema(description: nil, minimum: nil, maximum: nil, multiple_of: nil, enum: nil)
           {
             type: "number",
             description: description,
             minimum: minimum,
             maximum: maximum,
-            multipleOf: multiple_of
+            multipleOf: multiple_of,
+            enum: enum
           }.compact
         end
 
-        def integer_schema(description: nil, minimum: nil, maximum: nil, multiple_of: nil)
+        def integer_schema(description: nil, minimum: nil, maximum: nil, multiple_of: nil, enum: nil)
           {
             type: "integer",
             description: description,
             minimum: minimum,
             maximum: maximum,
-            multipleOf: multiple_of
+            multipleOf: multiple_of,
+            enum: enum
           }.compact
         end
 

spec/ruby_llm/schema/properties/numbers_spec.rb

@@ -18,6 +18,13 @@
     })
   end
 
+  it "supports number enum values" do
+    schema_class.number :score, enum: [0.5, 1.5]
+
+    properties = schema_class.properties
+    expect(properties[:score]).to eq({type: "number", enum: [0.5, 1.5]})
+  end
+
   it "supports number type with description" do
     schema_class.number :price, description: "Price field"
 
@@ -31,4 +38,11 @@
     properties = schema_class.properties
     expect(properties[:count]).to eq({type: "integer", description: "Count value"})
   end
+
+  it "supports integer enum values" do
+    schema_class.integer :level, enum: [0, 1, 2]
+
+    properties = schema_class.properties
+    expect(properties[:level]).to eq({type: "integer", enum: [0, 1, 2]})
+  end
 end