feat: Add AgentEngine Sessions module

matthew29tang · copybara-github · commit 368a8f82c73a · 2026-03-20T18:30:02.000-07:00
PiperOrigin-RevId: 887068191
diff --git a/tests/unit/vertexai/genai/replays/test_ae_session_events_append.py b/tests/unit/vertexai/genai/replays/test_ae_session_events_append.py
@@ -25,7 +25,7 @@ def test_append_session_event(client):
         name="reasoningEngines/2886612747586371584/sessions/6922431337672474624",
         author="test-user-123",
         invocation_id="test-invocation-id",
-        timestamp=datetime.datetime.fromtimestamp(1234567890, tz=datetime.timezone.utc),
+        timestamp=datetime.datetime.fromtimestamp(1234567860, tz=datetime.timezone.utc),
     )
     assert isinstance(session_event, types.AppendAgentEngineSessionEventResponse)
 

Original file line number	Diff line number	Diff line change
`@@ -25,7 +25,7 @@ def test_append_session_event(client):`
`25`	`25`	`name="reasoningEngines/2886612747586371584/sessions/6922431337672474624",`
`26`	`26`	`author="test-user-123",`
`27`	`27`	`invocation_id="test-invocation-id",`
`28`		`- timestamp=datetime.datetime.fromtimestamp(1234567890, tz=datetime.timezone.utc),`
	`28`	`+ timestamp=datetime.datetime.fromtimestamp(1234567860, tz=datetime.timezone.utc),`
`29`	`29`	`)`
`30`	`30`	`assert isinstance(session_event, types.AppendAgentEngineSessionEventResponse)`
`31`	`31`