Skip to content

Commit 0830795

Browse files
sararobcopybara-github
authored andcommitted
chore: GenAI SDK client - fix mypy errors in client.py
PiperOrigin-RevId: 832340202
1 parent e9d9c31 commit 0830795

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

vertexai/_genai/client.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -289,7 +289,7 @@ def _get_api_client(
289289
)
290290

291291
@property
292-
def agent_engines(self):
292+
def agent_engines(self) -> "'agent_engines.AgentEngines'":
293293
if self._agent_engines is None:
294294
try:
295295
# We need to lazy load the agent_engines module to handle the
@@ -307,7 +307,7 @@ def agent_engines(self):
307307
return self._agent_engines.AgentEngines(self._api_client)
308308

309309
@property
310-
def prompts(self):
310+
def prompts(self) -> "'prompts.Prompts'":
311311
if self._prompts is None:
312312
# Lazy loading the prompts module
313313
self._prompts = importlib.import_module(
@@ -321,7 +321,7 @@ def prompts(self):
321321
"The Vertex SDK GenAI datasets module is experimental, "
322322
"and may change in future versions."
323323
)
324-
def datasets(self):
324+
def datasets(self) -> "'datasets.Datasets'":
325325
if self._datasets is None:
326326
self._datasets = importlib.import_module(
327327
".datasets",

0 commit comments

Comments
 (0)