Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 9 additions & 3 deletions vertexai/_genai/agent_engines.py
Original file line number Diff line number Diff line change
Expand Up @@ -1944,7 +1944,7 @@ def append_session_event(
DeprecationWarning,
stacklevel=2,
)
return self.sessions.events.append( # type: ignore[no-any-return]
return self.sessions.events.append(
name=name,
author=author,
invocation_id=invocation_id,
Expand All @@ -1967,7 +1967,7 @@ def list_session_events(
DeprecationWarning,
stacklevel=2,
)
return self.sessions.events.list(name=name, config=config) # type: ignore[no-any-return]
return self.sessions.events.list(name=name, config=config)


class AsyncAgentEngines(_api_module.BaseModule):
Expand Down Expand Up @@ -2436,7 +2436,13 @@ async def append_session_event(
DeprecationWarning,
stacklevel=2,
)
return await self.sessions.events.append(name=name, config=config) # type: ignore[no-any-return]
return await self.sessions.events.append(
name=name,
author=author,
invocation_id=invocation_id,
timestamp=timestamp,
config=config,
)

async def delete_memory(
self,
Expand Down
8 changes: 0 additions & 8 deletions vertexai/_genai/sessions.py
Original file line number Diff line number Diff line change
Expand Up @@ -608,10 +608,6 @@ def _update(
_events = None

@property
@_common.experimental_warning(
"The Vertex SDK GenAI agent_engines.sessions.events module is "
"experimental, and may change in future versions."
)
def events(self) -> "session_events_module.SessionEvents":
if self._events is None:
try:
Expand Down Expand Up @@ -1094,10 +1090,6 @@ async def _update(
_events = None

@property
@_common.experimental_warning(
"The Vertex SDK GenAI agent_engines.sessions.events module is "
"experimental, and may change in future versions."
)
def events(self) -> "session_events_module.AsyncSessionEvents":
if self._events is None:
try:
Expand Down
Loading