Fix: Added a new regex to make sure the restoring logs are caught correctly.

[camiloCienet]

Summary
Refactored log validation utilities to support robust, multi-pattern regex filtering across Orbax and Post-Training DAGs, eliminating duplicate code and hardcoded query strings.

Key Changes
Validation Utilities (orbax/util/validation_util.py):

Replaced the singular text_filter string parameter with a text_filters list across all log validation functions.
Added a FilterMode flag enum (textPayload, jsonPayload_message) to strongly type which payload fields are searched.
Updated list_log_entries to dynamically construct OR-joined Google Cloud Logging regex (=~) queries from the provided filters.
Post-Training Facade (post_training/util/validation_util.py):

Re-exported the FilterMode enum to maintain clean, isolated imports for downstream post-training DAGs.
DAG Migrations:

Orbax: Updated 6 test DAGs (e.g., maxtext_emc_restore_local.py, maxtext_mtc_save_gcs.py) to utilize the new text_filters list and FilterMode flags.
Post-Training (maxtext_sft.py, maxtext_rl.py): Migrated to the new interface. Replaced brittle logical AND GCL string queries with robust, order-agnostic regex patterns to guarantee accurate JSON payload evaluations.

Fixes

It fixes b/530444528

Before submitting this PR, please make sure (put X in square brackets):

• I have performed a self-review of my code.
• I have necessary comments in my code, particularly in hard-to-understand areas.
• I have run one-shot tests and provided workload links above if applicable.
• I have made or will make corresponding changes to the doc if needed.

[alfredyu-cienet]

how about this

from enum import Flag, auto

class FilterMode(Flag):
    textPayload = auto()
    jsonPayload_message = auto()

in which textPayload will be 0001 (binary) and jsonPayload_message will be 0010 (binary)

def list_log_entries(
    ...,
    text_filters: list[str],
    filter_mode: FilterMode,
    ...,
):
  ...

  filters = []
  for txt in text_filters:
    if FilterMode.textPayload in filter_mode:
      filters.append(f"textPayload: {txt}")
    if FilterMode.jsonPayload_message in filter_mode:
      filters.append(f"jsonPayload.message: {txt}")

  text_filter = f"({" OR ".join(filters)})"

and then in this function, we can do somehting like this

reg_save_event = r"'event_type': 'save'"
reg_restor_event = r"'event_type': '(emergency_)?restore'"
reg_restoring = r"'restoring from this run\'s directory step (\d+)'"

list_log_entries(
    ...,
    text_filters=[reg_save_event, reg_restor_event, reg_restoring],
    filter_mode=FilterMode.textPayload | FilterMode.jsonPayload_message,
    ...,
)

...

message = ...
if re.search(reg_save_event, message):
  ... # do stuff
elif re.search(reg_restor_event, message) or re.search(reg_restoring, message):
  ... # do stuff