|
39 | 39 | ) |
40 | 40 | from agentplatform._genai import prompts as prompts_module |
41 | 41 | from agentplatform._genai import skills as skills_module |
| 42 | + from agentplatform._genai import ( |
| 43 | + model_garden as model_garden_module, |
| 44 | + ) |
42 | 45 | from agentplatform._genai import live as live_module |
43 | 46 | from agentplatform._genai import rag as rag_module |
44 | 47 |
|
@@ -85,6 +88,7 @@ def __init__(self, api_client: genai_client.BaseApiClient): # type: ignore[name |
85 | 88 | self._datasets: Optional[ModuleType] = None |
86 | 89 | self._skills: Optional[ModuleType] = None |
87 | 90 | self._rag: Optional[ModuleType] = None |
| 91 | + self._model_garden: Optional[ModuleType] = None |
88 | 92 |
|
89 | 93 | @property |
90 | 94 | @_common.experimental_warning( |
@@ -179,6 +183,19 @@ def rag(self) -> "rag_module.AsyncRag": |
179 | 183 | ) |
180 | 184 | return self._rag.AsyncRag(self._api_client) # type: ignore[no-any-return] |
181 | 185 |
|
| 186 | + @property |
| 187 | + @_common.experimental_warning( |
| 188 | + "The Model Garden module is experimental, and may change in future " |
| 189 | + "versions." |
| 190 | + ) |
| 191 | + def model_garden(self) -> "model_garden_module.AsyncModelGarden": |
| 192 | + if self._model_garden is None: |
| 193 | + self._model_garden = importlib.import_module( |
| 194 | + ".model_garden", |
| 195 | + __package__, |
| 196 | + ) |
| 197 | + return self._model_garden.AsyncModelGarden(self._api_client) # type: ignore[no-any-return] |
| 198 | + |
182 | 199 | async def aclose(self) -> None: |
183 | 200 | """Closes the async client explicitly. |
184 | 201 |
|
@@ -284,6 +301,7 @@ def __init__( |
284 | 301 | self._datasets: Optional[ModuleType] = None |
285 | 302 | self._skills: Optional[ModuleType] = None |
286 | 303 | self._rag: Optional[ModuleType] = None |
| 304 | + self._model_garden: Optional[ModuleType] = None |
287 | 305 |
|
288 | 306 | @property |
289 | 307 | def evals(self) -> "evals_module.Evals": |
@@ -402,3 +420,16 @@ def rag(self) -> "rag_module.Rag": |
402 | 420 | __package__, |
403 | 421 | ) |
404 | 422 | return self._rag.Rag(self._api_client) # type: ignore[no-any-return] |
| 423 | + |
| 424 | + @property |
| 425 | + @_common.experimental_warning( |
| 426 | + "The Model Garden module is experimental, and may change in future " |
| 427 | + "versions." |
| 428 | + ) |
| 429 | + def model_garden(self) -> "model_garden_module.ModelGarden": |
| 430 | + if self._model_garden is None: |
| 431 | + self._model_garden = importlib.import_module( |
| 432 | + ".model_garden", |
| 433 | + __package__, |
| 434 | + ) |
| 435 | + return self._model_garden.ModelGarden(self._api_client) # type: ignore[no-any-return] |
0 commit comments