Skip to content

Commit b5012ba

Browse files
yyyu-googlecopybara-github
authored andcommitted
test: verify schema handling in transformer
PiperOrigin-RevId: 745676854
1 parent ce9e7b0 commit b5012ba

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed

src/test/java/com/google/genai/TransformerTest.java

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,4 +88,30 @@ public void testTSchema_Items_success() {
8888
assertEquals("STRING", transformedSchema.items().get().type().get());
8989
assertEquals("ARRAY", transformedSchema.type().get());
9090
}
91+
92+
@Test
93+
public void testTSchema_Required_sucess() {
94+
Schema schema = Schema.builder()
95+
.type("ARRAY")
96+
.items(
97+
Schema.builder()
98+
.type("OBJECT")
99+
.properties(
100+
ImmutableMap.of(
101+
"recipe_name",
102+
Schema.builder().type("STRING").build(),
103+
"ingredients",
104+
Schema.builder()
105+
.type("ARRAY")
106+
.items(Schema.builder().type("STRING").build())
107+
.build()
108+
))
109+
.required(ImmutableList.of("recipe_name", "ingredients"))
110+
.build())
111+
.build();
112+
Schema transformedSchema = Transformers.tSchema(GEMINI_API_CLIENT, schema);
113+
assertEquals(2, transformedSchema.items().get().required().get().size());
114+
assertEquals("recipe_name", transformedSchema.items().get().required().get().get(0));
115+
assertEquals("ingredients", transformedSchema.items().get().required().get().get(1));
116+
}
91117
}

0 commit comments

Comments
 (0)