Skip to content

Commit 3ce1ebb

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
feat: add FeedbackService GenAI Python SDK
PiperOrigin-RevId: 914281684
1 parent bf32f5e commit 3ce1ebb

9 files changed

Lines changed: 3001 additions & 0 deletions

File tree

agentplatform/_genai/agent_engines.py

Lines changed: 38 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,11 +39,13 @@
3939
from . import memories as memories_module
4040
from . import a2a_tasks as a2a_tasks_module
4141
from . import runtimes as runtimes_module
42+
from . import feedback_entries as feedback_entries_module
4243

4344
_ = sessions_module
4445
__ = memories_module
4546
___ = a2a_tasks_module
4647
____ = runtimes_module
48+
_____ = feedback_entries_module
4749

4850

4951
logger = logging.getLogger("agentplatform_genai.agentengines")
@@ -1516,6 +1518,7 @@ def _update(
15161518
_sandboxes = None
15171519
_sessions = None
15181520
_runtimes = None
1521+
_feedback_entries = None
15191522

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

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+
15951615
def _list_pager(
15961616
self, *, config: Optional[types.ListAgentEngineConfigOrDict] = None
15971617
) -> Pager[types.ReasoningEngine]:
@@ -3930,6 +3950,7 @@ async def _update(
39303950
_memories = None
39313951
_sessions = None
39323952
_runtimes = None
3953+
_feedback_entries = None
39333954

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

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+
40224060
async def append_session_event(
40234061
self,
40244062
*,

0 commit comments

Comments
 (0)