feat: Add environments service to GenAI SDK.

google/genai/_gaos/google_genai.py

@@ -52,8 +52,6 @@
 from .triggers import Triggers as GeneratedTriggers
 from .webhooks import AsyncWebhooks as GeneratedAsyncWebhooks
 from .webhooks import Webhooks as GeneratedWebhooks
-from .environments import AsyncEnvironments as GeneratedAsyncEnvironments
-from .environments import Environments as GeneratedEnvironments
 
 
 GOOGLE_GENAI_API_REVISION = _GOOGLE_GENAI_API_REVISION
@@ -761,72 +759,6 @@ async def list_executions(self, *args: Any, **kwargs: Any) -> Any:
             return await async_wrap_sdk_call(super().list_executions, *args, **kwargs)
 
 
-class GeminiNextGenEnvironments(GeneratedEnvironments):
-    """Public environments resource backed by the NextGen client."""
-
-    def __init__(self, api_client: Any):
-        sdk = build_google_genai_client(api_client)
-        super().__init__(sdk.sdk_configuration, parent_ref=sdk)
-
-    if not TYPE_CHECKING:
-        @property
-        def with_raw_response(self):
-            return _RawResponseAccessorProxy(super().with_raw_response)
-
-        @property
-        def with_streaming_response(self):
-            return _RawResponseAccessorProxy(super().with_streaming_response)
-
-        def create_environment(self, *args: Any, **kwargs: Any) -> Any:
-            return wrap_sdk_call(super().create_environment, *args, **kwargs)
-
-        def list_environments(self, *args: Any, **kwargs: Any) -> Any:
-            return wrap_sdk_call(super().list_environments, *args, **kwargs)
-
-        def get_environment(self, *args: Any, **kwargs: Any) -> Any:
-            return wrap_sdk_call(super().get_environment, *args, **kwargs)
-
-        def delete_environment(self, *args: Any, **kwargs: Any) -> Any:
-            return wrap_sdk_call(super().delete_environment, *args, **kwargs)
-
-        def get_environment_files(self, *args: Any, **kwargs: Any) -> Any:
-            return wrap_sdk_call(super().get_environment_files, *args, **kwargs)
-
-        # NOTE: update_environment, patch_environment are handled by fallback if they exist, but we assume they aren't generated based on our openapi.json.
-
-
-class AsyncGeminiNextGenEnvironments(GeneratedAsyncEnvironments):
-    """Async public environments resource backed by the NextGen client."""
-
-    def __init__(self, api_client: Any):
-        sdk = build_google_genai_async_client(api_client)
-        super().__init__(sdk.sdk_configuration, parent_ref=sdk)
-
-    if not TYPE_CHECKING:
-        @property
-        def with_raw_response(self):
-            return _AsyncRawResponseAccessorProxy(super().with_raw_response)
-
-        @property
-        def with_streaming_response(self):
-            return _AsyncRawResponseAccessorProxy(super().with_streaming_response)
-
-        async def create_environment(self, *args: Any, **kwargs: Any) -> Any:
-            return await async_wrap_sdk_call(super().create_environment, *args, **kwargs)
-
-        async def list_environments(self, *args: Any, **kwargs: Any) -> Any:
-            return await async_wrap_sdk_call(super().list_environments, *args, **kwargs)
-
-        async def get_environment(self, *args: Any, **kwargs: Any) -> Any:
-            return await async_wrap_sdk_call(super().get_environment, *args, **kwargs)
-
-        async def delete_environment(self, *args: Any, **kwargs: Any) -> Any:
-            return await async_wrap_sdk_call(super().delete_environment, *args, **kwargs)
-
-        async def get_environment_files(self, *args: Any, **kwargs: Any) -> Any:
-            return await async_wrap_sdk_call(super().get_environment_files, *args, **kwargs)
-
-
 def _add_output_properties_if_interaction(value: Any) -> Any:
     normalized = _normalize_interaction_shape(value)
     if normalized is None:
@@ -947,7 +879,6 @@ def _get_value(value: Any, name: str) -> Any:
         return value.get(name)
     return getattr(value, name, None)
 
-
 # Allowed create() body keys, derived from the generated request models so the
 # set tracks the schema; output-only fields are excluded.
 _CREATE_BODY_KEYS = frozenset(

google/genai/_gaos/models/__init__.py

@@ -61,10 +61,6 @@
         DeleteAgentRequest,
         DeleteAgentRequestParam,
     )
-    from .deleteenvironment import (
-        DeleteEnvironmentRequest,
-        DeleteEnvironmentRequestParam,
-    )
     from .deleteinteraction import (
         DeleteInteractionGlobals,
         DeleteInteractionGlobalsTypedDict,
@@ -89,11 +85,6 @@
         GetAgentRequest,
         GetAgentRequestParam,
     )
-    from .getenvironment import GetEnvironmentRequest, GetEnvironmentRequestParam
-    from .getenvironmentfiles import (
-        GetEnvironmentFilesRequest,
-        GetEnvironmentFilesRequestParam,
-    )
     from .getinteractionbyid import (
         GetInteractionByIDGlobals,
         GetInteractionByIDGlobalsTypedDict,
@@ -120,7 +111,6 @@
         ListAgentsRequest,
         ListAgentsRequestParam,
     )
-    from .listenvironments import ListEnvironmentsRequest, ListEnvironmentsRequestParam
     from .listtriggerexecutions import (
         ListTriggerExecutionsGlobals,
         ListTriggerExecutionsGlobalsTypedDict,
@@ -200,8 +190,6 @@
     "DeleteAgentGlobalsTypedDict",
     "DeleteAgentRequest",
     "DeleteAgentRequestParam",
-    "DeleteEnvironmentRequest",
-    "DeleteEnvironmentRequestParam",
     "DeleteInteractionGlobals",
     "DeleteInteractionGlobalsTypedDict",
     "DeleteInteractionRequest",
@@ -218,10 +206,6 @@
     "GetAgentGlobalsTypedDict",
     "GetAgentRequest",
     "GetAgentRequestParam",
-    "GetEnvironmentFilesRequest",
-    "GetEnvironmentFilesRequestParam",
-    "GetEnvironmentRequest",
-    "GetEnvironmentRequestParam",
     "GetInteractionByIDGlobals",
     "GetInteractionByIDGlobalsTypedDict",
     "GetInteractionByIDRequest",
@@ -240,8 +224,6 @@
     "ListAgentsGlobalsTypedDict",
     "ListAgentsRequest",
     "ListAgentsRequestParam",
-    "ListEnvironmentsRequest",
-    "ListEnvironmentsRequestParam",
     "ListTriggerExecutionsGlobals",
     "ListTriggerExecutionsGlobalsTypedDict",
     "ListTriggerExecutionsRequest",
@@ -305,8 +287,6 @@
     "DeleteAgentGlobalsTypedDict": ".deleteagent",
     "DeleteAgentRequest": ".deleteagent",
     "DeleteAgentRequestParam": ".deleteagent",
-    "DeleteEnvironmentRequest": ".deleteenvironment",
-    "DeleteEnvironmentRequestParam": ".deleteenvironment",
     "DeleteInteractionGlobals": ".deleteinteraction",
     "DeleteInteractionGlobalsTypedDict": ".deleteinteraction",
     "DeleteInteractionRequest": ".deleteinteraction",
@@ -323,10 +303,6 @@
     "GetAgentGlobalsTypedDict": ".getagent",
     "GetAgentRequest": ".getagent",
     "GetAgentRequestParam": ".getagent",
-    "GetEnvironmentRequest": ".getenvironment",
-    "GetEnvironmentRequestParam": ".getenvironment",
-    "GetEnvironmentFilesRequest": ".getenvironmentfiles",
-    "GetEnvironmentFilesRequestParam": ".getenvironmentfiles",
     "GetInteractionByIDGlobals": ".getinteractionbyid",
     "GetInteractionByIDGlobalsTypedDict": ".getinteractionbyid",
     "GetInteractionByIDRequest": ".getinteractionbyid",
@@ -345,8 +321,6 @@
     "ListAgentsGlobalsTypedDict": ".listagents",
     "ListAgentsRequest": ".listagents",
     "ListAgentsRequestParam": ".listagents",
-    "ListEnvironmentsRequest": ".listenvironments",
-    "ListEnvironmentsRequestParam": ".listenvironments",
     "ListTriggerExecutionsGlobals": ".listtriggerexecutions",
     "ListTriggerExecutionsGlobalsTypedDict": ".listtriggerexecutions",
     "ListTriggerExecutionsRequest": ".listtriggerexecutions",