From 81794cc5b51d11edab28598fd01953960f57997c Mon Sep 17 00:00:00 2001 From: Jon David Schober Date: Tue, 23 Dec 2025 13:07:17 -0600 Subject: [PATCH] Add enum parameter to number and integer schema methods --- lib/ruby_llm/schema/dsl/schema_builders.rb | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/lib/ruby_llm/schema/dsl/schema_builders.rb b/lib/ruby_llm/schema/dsl/schema_builders.rb index 91ca52b..716fefb 100644 --- a/lib/ruby_llm/schema/dsl/schema_builders.rb +++ b/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