fix: Set propertyOrdering when schema is specified as dict or types.Schema.

PiperOrigin-RevId: 752296364

Author: terryykoo

google/genai/_transformers.py

@@ -790,7 +790,7 @@ def t_schema(
   if not origin:
     return None
   if isinstance(origin, dict) and _is_type_dict_str_any(origin):
-    process_schema(origin, client, order_properties=False)
+    process_schema(origin, client)
     return types.Schema.model_validate(origin)
   if isinstance(origin, EnumMeta):
     return _process_enum(origin, client)
@@ -799,7 +799,7 @@ def t_schema(
       # response_schema value was coerced to an empty Schema instance because it did not adhere to the Schema field annotation
       raise ValueError(f'Unsupported schema type.')
     schema = origin.model_dump(exclude_unset=True)
-    process_schema(schema, client, order_properties=False)
+    process_schema(schema, client)
     return types.Schema.model_validate(schema)
 
   if (

google/genai/tests/transformers/test_schema.py

@@ -605,17 +605,25 @@ def test_t_schema_does_not_change_property_ordering_if_set(client):
   assert transformed_schema.property_ordering == custom_property_ordering
 
 
-def test_t_schema_does_not_set_property_ordering_for_json_schema(client):
-  """Tests t_schema doesn't set the property_ordering field for json schemas."""
+def test_t_schema_sets_property_ordering_for_json_schema(client):
+  """Tests t_schema sets the property_ordering field for json schemas."""
 
   schema = CountryInfo.model_json_schema()
 
   transformed_schema = _transformers.t_schema(client, schema)
-  assert transformed_schema.property_ordering is None
+  assert transformed_schema.property_ordering == [
+      'name',
+      'population',
+      'capital',
+      'continent',
+      'gdp',
+      'official_language',
+      'total_area_sq_mi',
+  ]
 
 
-def test_t_schema_does_not_set_property_ordering_for_schema_type(client):
-  """Tests t_schema doesn't set the property_ordering field for Schema types."""
+def test_t_schema_sets_property_ordering_for_schema_type(client):
+  """Tests t_schema sets the property_ordering field for Schema types."""
 
   schema = types.Schema(
       properties={
@@ -635,4 +643,4 @@ def test_t_schema_does_not_set_property_ordering_for_schema_type(client):
   )
 
   transformed_schema = _transformers.t_schema(client, schema)
-  assert transformed_schema.property_ordering is None
+  assert transformed_schema.property_ordering == ['name', 'population']