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