Skip to content

Commit ef20d6b

Browse files
wanlin31copybara-github
authored andcommitted
feat: support Agent and Environment APIs.
PiperOrigin-RevId: 916890699
1 parent 6ed3020 commit ef20d6b

15 files changed

Lines changed: 1188 additions & 3 deletions

google/genai/_interactions/_client.py

Lines changed: 38 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,8 @@
4949
from ._client_adapter import GeminiNextGenAPIClientAdapter, AsyncGeminiNextGenAPIClientAdapter
5050

5151
if TYPE_CHECKING:
52-
from .resources import webhooks, interactions
52+
from .resources import agents, webhooks, interactions
53+
from .resources.agents import AgentsResource, AsyncAgentsResource
5354
from .resources.webhooks import WebhooksResource, AsyncWebhooksResource
5455
from .resources.interactions import InteractionsResource, AsyncInteractionsResource
5556

@@ -140,6 +141,12 @@ def webhooks(self) -> WebhooksResource:
140141

141142
return WebhooksResource(self)
142143

144+
@cached_property
145+
def agents(self) -> AgentsResource:
146+
from .resources.agents import AgentsResource
147+
148+
return AgentsResource(self)
149+
143150
@cached_property
144151
def with_raw_response(self) -> GeminiNextGenAPIClientWithRawResponse:
145152
return GeminiNextGenAPIClientWithRawResponse(self)
@@ -368,6 +375,12 @@ def webhooks(self) -> AsyncWebhooksResource:
368375

369376
return AsyncWebhooksResource(self)
370377

378+
@cached_property
379+
def agents(self) -> AsyncAgentsResource:
380+
from .resources.agents import AsyncAgentsResource
381+
382+
return AsyncAgentsResource(self)
383+
371384
@cached_property
372385
def with_raw_response(self) -> AsyncGeminiNextGenAPIClientWithRawResponse:
373386
return AsyncGeminiNextGenAPIClientWithRawResponse(self)
@@ -539,6 +552,12 @@ def webhooks(self) -> webhooks.WebhooksResourceWithRawResponse:
539552

540553
return WebhooksResourceWithRawResponse(self._client.webhooks)
541554

555+
@cached_property
556+
def agents(self) -> agents.AgentsResourceWithRawResponse:
557+
from .resources.agents import AgentsResourceWithRawResponse
558+
559+
return AgentsResourceWithRawResponse(self._client.agents)
560+
542561

543562
class AsyncGeminiNextGenAPIClientWithRawResponse:
544563
_client: AsyncGeminiNextGenAPIClient
@@ -558,6 +577,12 @@ def webhooks(self) -> webhooks.AsyncWebhooksResourceWithRawResponse:
558577

559578
return AsyncWebhooksResourceWithRawResponse(self._client.webhooks)
560579

580+
@cached_property
581+
def agents(self) -> agents.AsyncAgentsResourceWithRawResponse:
582+
from .resources.agents import AsyncAgentsResourceWithRawResponse
583+
584+
return AsyncAgentsResourceWithRawResponse(self._client.agents)
585+
561586

562587
class GeminiNextGenAPIClientWithStreamedResponse:
563588
_client: GeminiNextGenAPIClient
@@ -577,6 +602,12 @@ def webhooks(self) -> webhooks.WebhooksResourceWithStreamingResponse:
577602

578603
return WebhooksResourceWithStreamingResponse(self._client.webhooks)
579604

605+
@cached_property
606+
def agents(self) -> agents.AgentsResourceWithStreamingResponse:
607+
from .resources.agents import AgentsResourceWithStreamingResponse
608+
609+
return AgentsResourceWithStreamingResponse(self._client.agents)
610+
580611

581612
class AsyncGeminiNextGenAPIClientWithStreamedResponse:
582613
_client: AsyncGeminiNextGenAPIClient
@@ -596,6 +627,12 @@ def webhooks(self) -> webhooks.AsyncWebhooksResourceWithStreamingResponse:
596627

597628
return AsyncWebhooksResourceWithStreamingResponse(self._client.webhooks)
598629

630+
@cached_property
631+
def agents(self) -> agents.AsyncAgentsResourceWithStreamingResponse:
632+
from .resources.agents import AsyncAgentsResourceWithStreamingResponse
633+
634+
return AsyncAgentsResourceWithStreamingResponse(self._client.agents)
635+
599636

600637
Client = GeminiNextGenAPIClient
601638

google/genai/_interactions/resources/__init__.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,14 @@
1515

1616
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
1717

18+
from .agents import (
19+
AgentsResource,
20+
AsyncAgentsResource,
21+
AgentsResourceWithRawResponse,
22+
AsyncAgentsResourceWithRawResponse,
23+
AgentsResourceWithStreamingResponse,
24+
AsyncAgentsResourceWithStreamingResponse,
25+
)
1826
from .webhooks import (
1927
WebhooksResource,
2028
AsyncWebhooksResource,
@@ -45,4 +53,10 @@
4553
"AsyncWebhooksResourceWithRawResponse",
4654
"WebhooksResourceWithStreamingResponse",
4755
"AsyncWebhooksResourceWithStreamingResponse",
56+
"AgentsResource",
57+
"AsyncAgentsResource",
58+
"AgentsResourceWithRawResponse",
59+
"AsyncAgentsResourceWithRawResponse",
60+
"AgentsResourceWithStreamingResponse",
61+
"AsyncAgentsResourceWithStreamingResponse",
4862
]

0 commit comments

Comments
 (0)