File tree Expand file tree Collapse file tree 1 file changed +26
-0
lines changed
src/test/java/com/google/genai Expand file tree Collapse file tree 1 file changed +26
-0
lines changed Original file line number Diff line number Diff 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}
You can’t perform that action at this time.
0 commit comments