feat: Support Gemini batch inline request's metadata and add test coverage to safety setting

PiperOrigin-RevId: 817811792

Author: yinghsienwu

google/genai/batches.py

@@ -1071,6 +1071,9 @@ def _InlinedRequest_to_mldev(
         ],
     )
 
+  if getv(from_object, ['metadata']) is not None:
+    setv(to_object, ['metadata'], getv(from_object, ['metadata']))
+
   if getv(from_object, ['config']) is not None:
     setv(
         to_object,

google/genai/tests/batches/test_create_with_inlined_requests.py

@@ -26,17 +26,35 @@
 from ... import types
 from .. import pytest_helper
 
-_GEMINI_MODEL = 'gemini-1.5-flash-002'
+_GEMINI_MODEL = 'gemini-2.5-flash-lite'
 _DISPLAY_NAME = 'test_batch'
 
-_MLDEV_GEMINI_MODEL = 'gemini-2.0-flash'
+_MLDEV_GEMINI_MODEL = 'gemini-2.5-flash'
+
+_SAFETY_SETTINGS = [
+    {
+        'category': 'HARM_CATEGORY_HATE_SPEECH',
+        'threshold': 'BLOCK_ONLY_HIGH',
+    },
+    {
+        'category': 'HARM_CATEGORY_DANGEROUS_CONTENT',
+        'threshold': 'BLOCK_LOW_AND_ABOVE',
+    },
+]
+
 _INLINED_REQUEST = {
     'contents': [{
         'parts': [{
             'text': 'Hello!',
         }],
         'role': 'user',
     }],
+    'metadata': {
+        'key': 'request-1',
+    },
+    'config': {
+        'safety_settings': _SAFETY_SETTINGS,
+    },
 }
 _INLINED_TEXT_REQUEST_UNION = {
     'contents': [{

google/genai/types.py

@@ -11766,6 +11766,10 @@ class InlinedRequest(_common.BaseModel):
       description="""Content of the request.
       """,
   )
+  metadata: Optional[dict[str, str]] = Field(
+      default=None,
+      description="""The metadata to be associated with the request.""",
+  )
   config: Optional[GenerateContentConfig] = Field(
       default=None,
       description="""Configuration that contains optional model parameters.
@@ -11784,6 +11788,9 @@ class InlinedRequestDict(TypedDict, total=False):
   """Content of the request.
       """
 
+  metadata: Optional[dict[str, str]]
+  """The metadata to be associated with the request."""
+
   config: Optional[GenerateContentConfigDict]
   """Configuration that contains optional model parameters.
       """