fix: Expose JOB_STATE_RUNNING and JOB_STATE_EXPIRED for Gemini Batches states

PiperOrigin-RevId: 811471392

Author: yinghsienwu

google/genai/_transformers.py

@@ -1114,12 +1114,16 @@ def t_job_state(state: str) -> str:
     return 'JOB_STATE_UNSPECIFIED'
   elif state == 'BATCH_STATE_PENDING':
     return 'JOB_STATE_PENDING'
+  elif state == 'BATCH_STATE_RUNNING':
+    return 'JOB_STATE_RUNNING'
   elif state == 'BATCH_STATE_SUCCEEDED':
     return 'JOB_STATE_SUCCEEDED'
   elif state == 'BATCH_STATE_FAILED':
     return 'JOB_STATE_FAILED'
   elif state == 'BATCH_STATE_CANCELLED':
     return 'JOB_STATE_CANCELLED'
+  elif state == 'BATCH_STATE_EXPIRED':
+    return 'JOB_STATE_EXPIRED'
   else:
     return state
 

google/genai/tests/transformers/test_t_batch.py

@@ -277,6 +277,8 @@ class TestJobState:
           ('BATCH_STATE_SUCCEEDED', 'JOB_STATE_SUCCEEDED'),
           ('BATCH_STATE_FAILED', 'JOB_STATE_FAILED'),
           ('BATCH_STATE_CANCELLED', 'JOB_STATE_CANCELLED'),
+          ('BATCH_STATE_EXPIRED', 'JOB_STATE_EXPIRED'),
+          ('BATCH_STATE_RUNNING', 'JOB_STATE_RUNNING'),
           ('BATCH_STATE_FOOBAR', 'BATCH_STATE_FOOBAR'),
       ],
   )