Function simplify_resumable_app_events

tests/unittests/testing_utils.py:139–153

(
    events: list[Event],
)

#
# Could be used to compare events for testing resumability.
def simplify_resumable_app_events(
    events: list[Event],
) -> list[(str, Union[types.Part, str])]:
  results = []
  for event in events:
    if event.content:
      results.append((event.author, simplify_content(event.content)))
    elif event.output and isinstance(event.output, (str, dict)):
      # Single_turn agents strip event.content and set event.output instead.
      results.append((event.author, event.output))
    elif event.actions.end_of_agent:
      results.append((event.author, END_OF_AGENT))
    elif event.actions.agent_state is not None:
      results.append((event.author, event.actions.agent_state))
  return results


# Simplifies the contents into a list of (author, simplified_content) tuples.