|
39 | 39 | from . import memories as memories_module |
40 | 40 | from . import a2a_tasks as a2a_tasks_module |
41 | 41 | from . import runtimes as runtimes_module |
| 42 | + from . import feedback_entries as feedback_entries_module |
42 | 43 |
|
43 | 44 | _ = sessions_module |
44 | 45 | __ = memories_module |
45 | 46 | ___ = a2a_tasks_module |
46 | 47 | ____ = runtimes_module |
| 48 | + _____ = feedback_entries_module |
47 | 49 |
|
48 | 50 |
|
49 | 51 | logger = logging.getLogger("agentplatform_genai.agentengines") |
@@ -1516,6 +1518,7 @@ def _update( |
1516 | 1518 | _sandboxes = None |
1517 | 1519 | _sessions = None |
1518 | 1520 | _runtimes = None |
| 1521 | + _feedback_entries = None |
1519 | 1522 |
|
1520 | 1523 | @property |
1521 | 1524 | def runtimes(self) -> "runtimes_module.Runtimes": |
@@ -1592,6 +1595,23 @@ def sessions(self) -> "sessions_module.Sessions": |
1592 | 1595 | ) from e |
1593 | 1596 | return self._sessions.Sessions(self._api_client) # type: ignore[no-any-return] |
1594 | 1597 |
|
| 1598 | + @property |
| 1599 | + def feedback_entries(self) -> "feedback_entries_module.FeedbackEntries": |
| 1600 | + if self._feedback_entries is None: |
| 1601 | + try: |
| 1602 | + # We need to lazy load the feedback_entries module to handle the |
| 1603 | + # possibility of ImportError when dependencies are not installed. |
| 1604 | + self._feedback_entries = importlib.import_module( |
| 1605 | + ".feedback_entries", __package__ |
| 1606 | + ) |
| 1607 | + except ImportError as e: |
| 1608 | + raise ImportError( |
| 1609 | + "The agent_engines.feedback_entries module requires additional " |
| 1610 | + "packages. Please install them using pip install " |
| 1611 | + "google-cloud-aiplatform[agent_engines]" |
| 1612 | + ) from e |
| 1613 | + return self._feedback_entries.FeedbackEntries(self._api_client) # type: ignore[no-any-return] |
| 1614 | + |
1595 | 1615 | def _list_pager( |
1596 | 1616 | self, *, config: Optional[types.ListAgentEngineConfigOrDict] = None |
1597 | 1617 | ) -> Pager[types.ReasoningEngine]: |
@@ -3930,6 +3950,7 @@ async def _update( |
3930 | 3950 | _memories = None |
3931 | 3951 | _sessions = None |
3932 | 3952 | _runtimes = None |
| 3953 | + _feedback_entries = None |
3933 | 3954 |
|
3934 | 3955 | async def delete( |
3935 | 3956 | self, |
@@ -4019,6 +4040,23 @@ def sessions(self) -> "sessions_module.AsyncSessions": |
4019 | 4040 | ) from e |
4020 | 4041 | return self._sessions.AsyncSessions(self._api_client) # type: ignore[no-any-return] |
4021 | 4042 |
|
| 4043 | + @property |
| 4044 | + def feedback_entries(self) -> "feedback_entries_module.AsyncFeedbackEntries": |
| 4045 | + if self._feedback_entries is None: |
| 4046 | + try: |
| 4047 | + # We need to lazy load the feedback_entries module to handle the |
| 4048 | + # possibility of ImportError when dependencies are not installed. |
| 4049 | + self._feedback_entries = importlib.import_module( |
| 4050 | + ".feedback_entries", __package__ |
| 4051 | + ) |
| 4052 | + except ImportError as e: |
| 4053 | + raise ImportError( |
| 4054 | + "The agent_engines.feedback_entries module requires additional packages. " |
| 4055 | + "Please install them using pip install " |
| 4056 | + "google-cloud-aiplatform[agent_engines]" |
| 4057 | + ) from e |
| 4058 | + return self._feedback_entries.AsyncFeedbackEntries(self._api_client) # type: ignore[no-any-return] |
| 4059 | + |
4022 | 4060 | async def append_session_event( |
4023 | 4061 | self, |
4024 | 4062 | *, |
|
0 commit comments