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
38 changes: 38 additions & 0 deletions agentplatform/_genai/agent_engines.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,11 +39,13 @@
from . import memories as memories_module
from . import a2a_tasks as a2a_tasks_module
from . import runtimes as runtimes_module
from . import feedback_entries as feedback_entries_module

_ = sessions_module
__ = memories_module
___ = a2a_tasks_module
____ = runtimes_module
_____ = feedback_entries_module


logger = logging.getLogger("agentplatform_genai.agentengines")
Expand Down Expand Up @@ -1516,6 +1518,7 @@ def _update(
_sandboxes = None
_sessions = None
_runtimes = None
_feedback_entries = None

@property
def runtimes(self) -> "runtimes_module.Runtimes":
Expand Down Expand Up @@ -1592,6 +1595,23 @@ def sessions(self) -> "sessions_module.Sessions":
) from e
return self._sessions.Sessions(self._api_client) # type: ignore[no-any-return]

@property
def feedback_entries(self) -> "feedback_entries_module.FeedbackEntries":
if self._feedback_entries is None:
try:
# We need to lazy load the feedback_entries module to handle the
# possibility of ImportError when dependencies are not installed.
self._feedback_entries = importlib.import_module(
".feedback_entries", __package__
)
except ImportError as e:
raise ImportError(
"The agent_engines.feedback_entries module requires additional "
"packages. Please install them using pip install "
"google-cloud-aiplatform[agent_engines]"
) from e
return self._feedback_entries.FeedbackEntries(self._api_client) # type: ignore[no-any-return]

def _list_pager(
self, *, config: Optional[types.ListAgentEngineConfigOrDict] = None
) -> Pager[types.ReasoningEngine]:
Expand Down Expand Up @@ -3930,6 +3950,7 @@ async def _update(
_memories = None
_sessions = None
_runtimes = None
_feedback_entries = None

async def delete(
self,
Expand Down Expand Up @@ -4019,6 +4040,23 @@ def sessions(self) -> "sessions_module.AsyncSessions":
) from e
return self._sessions.AsyncSessions(self._api_client) # type: ignore[no-any-return]

@property
def feedback_entries(self) -> "feedback_entries_module.AsyncFeedbackEntries":
if self._feedback_entries is None:
try:
# We need to lazy load the feedback_entries module to handle the
# possibility of ImportError when dependencies are not installed.
self._feedback_entries = importlib.import_module(
".feedback_entries", __package__
)
except ImportError as e:
raise ImportError(
"The agent_engines.feedback_entries module requires additional packages. "
"Please install them using pip install "
"google-cloud-aiplatform[agent_engines]"
) from e
return self._feedback_entries.AsyncFeedbackEntries(self._api_client) # type: ignore[no-any-return]

async def append_session_event(
self,
*,
Expand Down
Loading
Loading