No public description

google/genai/_transformers.py

@@ -535,7 +535,6 @@ def process_schema(
 ):
   """Updates the schema and each sub-schema inplace to be API-compatible.
 
-  - Removes the `title` field from the schema if the client is not vertexai.
   - Inlines the $defs.
 
   Example of a schema before and after (with mldev):
@@ -579,21 +578,22 @@ def process_schema(
         'items': {
           'properties': {
             'continent': {
-                'type': 'string'
+              'title': 'Continent',
+              'type': 'string'
             },
             'gdp': {
-                'type': 'integer'}
+              'title': 'Gdp',
+              'type': 'integer'
             },
           }
           'required':['continent', 'gdp'],
+          'title': 'CountryInfo',
           'type': 'object'
         },
         'type': 'array'
     }
   """
   if not client.vertexai:
-    schema.pop('title', None)
-
     if schema.get('default') is not None:
       raise ValueError(
           'Default value is not supported in the response schema for the Gemini'

google/genai/caches.py

@@ -180,9 +180,6 @@ def _Schema_to_mldev(
   if getv(from_object, ['max_length']) is not None:
     raise ValueError('max_length parameter is not supported in Gemini API.')
 
-  if getv(from_object, ['title']) is not None:
-    raise ValueError('title parameter is not supported in Gemini API.')
-
   if getv(from_object, ['min_length']) is not None:
     raise ValueError('min_length parameter is not supported in Gemini API.')
 
@@ -235,6 +232,9 @@ def _Schema_to_mldev(
   if getv(from_object, ['required']) is not None:
     setv(to_object, ['required'], getv(from_object, ['required']))
 
+  if getv(from_object, ['title']) is not None:
+    setv(to_object, ['title'], getv(from_object, ['title']))
+
   if getv(from_object, ['type']) is not None:
     setv(to_object, ['type'], getv(from_object, ['type']))
 
@@ -259,9 +259,6 @@ def _Schema_to_vertex(
   if getv(from_object, ['max_length']) is not None:
     setv(to_object, ['maxLength'], getv(from_object, ['max_length']))
 
-  if getv(from_object, ['title']) is not None:
-    setv(to_object, ['title'], getv(from_object, ['title']))
-
   if getv(from_object, ['min_length']) is not None:
     setv(to_object, ['minLength'], getv(from_object, ['min_length']))
 
@@ -314,6 +311,9 @@ def _Schema_to_vertex(
   if getv(from_object, ['required']) is not None:
     setv(to_object, ['required'], getv(from_object, ['required']))
 
+  if getv(from_object, ['title']) is not None:
+    setv(to_object, ['title'], getv(from_object, ['title']))
+
   if getv(from_object, ['type']) is not None:
     setv(to_object, ['type'], getv(from_object, ['type']))
 

google/genai/models.py

@@ -181,9 +181,6 @@ def _Schema_to_mldev(
   if getv(from_object, ['max_length']) is not None:
     raise ValueError('max_length parameter is not supported in Gemini API.')
 
-  if getv(from_object, ['title']) is not None:
-    raise ValueError('title parameter is not supported in Gemini API.')
-
   if getv(from_object, ['min_length']) is not None:
     raise ValueError('min_length parameter is not supported in Gemini API.')
 
@@ -236,6 +233,9 @@ def _Schema_to_mldev(
   if getv(from_object, ['required']) is not None:
     setv(to_object, ['required'], getv(from_object, ['required']))
 
+  if getv(from_object, ['title']) is not None:
+    setv(to_object, ['title'], getv(from_object, ['title']))
+
   if getv(from_object, ['type']) is not None:
     setv(to_object, ['type'], getv(from_object, ['type']))
 
@@ -260,9 +260,6 @@ def _Schema_to_vertex(
   if getv(from_object, ['max_length']) is not None:
     setv(to_object, ['maxLength'], getv(from_object, ['max_length']))
 
-  if getv(from_object, ['title']) is not None:
-    setv(to_object, ['title'], getv(from_object, ['title']))
-
   if getv(from_object, ['min_length']) is not None:
     setv(to_object, ['minLength'], getv(from_object, ['min_length']))
 
@@ -315,6 +312,9 @@ def _Schema_to_vertex(
   if getv(from_object, ['required']) is not None:
     setv(to_object, ['required'], getv(from_object, ['required']))
 
+  if getv(from_object, ['title']) is not None:
+    setv(to_object, ['title'], getv(from_object, ['title']))
+
   if getv(from_object, ['type']) is not None:
     setv(to_object, ['type'], getv(from_object, ['type']))
 

google/genai/tests/transformers/test_schema.py

@@ -213,26 +213,8 @@ def test_schema_with_default_value_raises_for_mldev(client):
 
 
 def test_schema_with_any_of(client):
-  transformed_schema = _transformers.t_schema(
-      client, CountryInfoWithAnyOf
-  )
-  expected_schema_mldev = types.Schema(
-      properties={
-          'name': types.Schema(
-              type='STRING',
-          ),
-          'restaurants_per_capita': types.Schema(
-              any_of=[
-                  types.Schema(type='INTEGER'),
-                  types.Schema(type='NUMBER'),
-              ],
-          ),
-      },
-      type='OBJECT',
-      required=['name', 'restaurants_per_capita'],
-      property_ordering=['name', 'restaurants_per_capita'],
-  )
-  expected_schema_vertex = types.Schema(
+  transformed_schema = _transformers.t_schema(client, CountryInfoWithAnyOf)
+  expected_schema = types.Schema(
       properties={
           'name': types.Schema(
               type='STRING',
@@ -252,10 +234,7 @@ def test_schema_with_any_of(client):
       property_ordering=['name', 'restaurants_per_capita'],
   )
 
-  if client.vertexai:
-    assert transformed_schema == expected_schema_vertex
-  else:
-    assert transformed_schema == expected_schema_mldev
+  assert transformed_schema == expected_schema
 
 
 @pytest.mark.parametrize('use_vertex', [True, False])

google/genai/types.py

@@ -837,9 +837,6 @@ class Schema(_common.BaseModel):
       default=None,
       description="""Optional. Maximum length of the Type.STRING""",
   )
-  title: Optional[str] = Field(
-      default=None, description="""Optional. The title of the Schema."""
-  )
   min_length: Optional[int] = Field(
       default=None,
       description="""Optional. SCHEMA FIELDS FOR TYPE STRING Minimum length of the Type.STRING""",
@@ -903,6 +900,9 @@ class Schema(_common.BaseModel):
       default=None,
       description="""Optional. Required properties of Type.OBJECT.""",
   )
+  title: Optional[str] = Field(
+      default=None, description="""Optional. The title of the Schema."""
+  )
   type: Optional[Type] = Field(
       default=None, description="""Optional. The type of the data."""
   )
@@ -926,9 +926,6 @@ class SchemaDict(TypedDict, total=False):
   max_length: Optional[int]
   """Optional. Maximum length of the Type.STRING"""
 
-  title: Optional[str]
-  """Optional. The title of the Schema."""
-
   min_length: Optional[int]
   """Optional. SCHEMA FIELDS FOR TYPE STRING Minimum length of the Type.STRING"""
 
@@ -977,6 +974,9 @@ class SchemaDict(TypedDict, total=False):
   required: Optional[list[str]]
   """Optional. Required properties of Type.OBJECT."""
 
+  title: Optional[str]
+  """Optional. The title of the Schema."""
+
   type: Optional[Type]
   """Optional. The type of the data."""