Skip to content

Commit 131cfc0

Browse files
committed
Fix tests
1 parent 4957e92 commit 131cfc0

File tree

6 files changed

+244
-74
lines changed

6 files changed

+244
-74
lines changed

lib/llm/schema/array.rb

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,10 +8,10 @@ class LLM::Schema
88
# can act as constraints.
99
class Array < Leaf
1010
##
11-
# Returns an array for the given types
11+
# Returns an array for the given type
1212
# @return [LLM::Schema::Array]
13-
def self.[](*types)
14-
new(*types.map(&:new))
13+
def self.[](type)
14+
new(type.new)
1515
end
1616

1717
def initialize(items)

spec/fixtures/cassettes/gemini/chat/llm_schema_array.yml

Lines changed: 11 additions & 11 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

spec/fixtures/cassettes/gemini/chat/llm_schema_enum.yml

Lines changed: 92 additions & 10 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

spec/fixtures/cassettes/gemini/chat/llm_schema_object.yml

Lines changed: 91 additions & 9 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

spec/schema_spec.rb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
property :height, LLM::Schema::Number, "height description", required: true
1616
property :active, LLM::Schema::Boolean, "active description"
1717
property :location, LLM::Schema::Null, "location description"
18-
property :addresses, LLM::Schema::Array[LLM::Schema::String, LLM::Schema::Integer], "addresses description"
18+
property :addresses, LLM::Schema::Array[LLM::Schema::String], "addresses description"
1919
end
2020
end
2121

@@ -33,7 +33,7 @@
3333
array = schema.object["addresses"]
3434
schema = self.schema.schema
3535
expect(array).to eq(
36-
schema.array([schema.string, schema.integer]).description("addresses description")
36+
schema.array(schema.string).description("addresses description")
3737
)
3838
end
3939
end

0 commit comments

Comments
 (0)