perf: lazily import the interactions API to speed up import google.genai

The top-level package eagerly imported the interactions submodule, and
client.py eagerly imported the _gaos backend, so every `import
google.genai` paid for hundreds of interactions-API modules even when the
caller only uses generate_content.

Defer both: make `interactions` a lazy module attribute via __getattr__,
and move the _gaos imports in client.py under TYPE_CHECKING plus into the
Client/AsyncClient properties that construct them. `from __future__ import
annotations` keeps the property return annotations valid without the eager
import. The _gaos modules now load only when .interactions/.agents/
.webhooks (or the nextgen client) are first accessed.

PiperOrigin-RevId: 945816846

Author: google-genai-bot

google/genai/__init__.py

@@ -15,12 +15,22 @@
 
 """Google Gen AI SDK"""
 
-from . import interactions
+import importlib
+from typing import Any
+
 from . import types
 from . import version
 from .client import Client
 
 
 __version__ = version.__version__
 
-__all__ = ['Client']
+__all__ = ['Client', 'interactions', 'types']
+
+
+def __getattr__(name: str) -> Any:
+  if name == 'interactions':
+    module = importlib.import_module('.interactions', __name__)
+    globals()[name] = module
+    return module
+  raise AttributeError(f'module {__name__!r} has no attribute {name!r}')

google/genai/client.py

@@ -13,13 +13,17 @@
 # limitations under the License.
 #
 
+from __future__ import annotations
+
 import asyncio
 import os
-from typing import Any, Optional, Union
+from typing import Any, Optional, TYPE_CHECKING, Union
+import warnings
 
 import google.auth
 import pydantic
 
+from . import _common
 from ._api_client import BaseApiClient
 from ._base_url import get_base_url
 from ._replay_api_client import ReplayApiClient
@@ -35,22 +39,17 @@
 from .tunings import AsyncTunings, Tunings
 from .types import HttpOptions, HttpOptionsDict, HttpRetryOptions
 
-import warnings
-
-from . import _common
-
-from ._gaos.google_genai import (
-    AsyncGeminiNextGenAgents,
-    AsyncGeminiNextGenInteractions,
-    AsyncGeminiNextGenWebhooks,
-    GeminiNextGenAgents,
-    GeminiNextGenInteractions,
-    GeminiNextGenWebhooks,
-    build_google_genai_async_client,
-    build_google_genai_client,
-)
-from ._gaos.sdk import AsyncGenAI as AsyncGeminiNextGenAPI
-from ._gaos.sdk import GenAI as GeminiNextGenAPI
+if TYPE_CHECKING:
+  from ._gaos.google_genai import (
+      AsyncGeminiNextGenAgents,
+      AsyncGeminiNextGenInteractions,
+      AsyncGeminiNextGenWebhooks,
+      GeminiNextGenAgents,
+      GeminiNextGenInteractions,
+      GeminiNextGenWebhooks,
+  )
+  from ._gaos.sdk import AsyncGenAI as AsyncGeminiNextGenAPI
+  from ._gaos.sdk import GenAI as GeminiNextGenAPI
 
 _agent_experimental_warned = False
 
@@ -77,6 +76,8 @@ def __init__(self, api_client: BaseApiClient):
 
   @property
   def _nextgen_client(self) -> AsyncGeminiNextGenAPI:
+    from ._gaos.google_genai import build_google_genai_async_client
+
     if self._nextgen_client_instance is None:
       self._nextgen_client_instance = build_google_genai_async_client(
           self._api_client
@@ -85,18 +86,24 @@ def _nextgen_client(self) -> AsyncGeminiNextGenAPI:
 
   @property
   def interactions(self) -> AsyncGeminiNextGenInteractions:
+    from ._gaos.google_genai import AsyncGeminiNextGenInteractions
+
     if self._interactions is None:
       self._interactions = AsyncGeminiNextGenInteractions(self._api_client)
     return self._interactions
 
   @property
   def webhooks(self) -> AsyncGeminiNextGenWebhooks:
+    from ._gaos.google_genai import AsyncGeminiNextGenWebhooks
+
     if self._webhooks is None:
       self._webhooks = AsyncGeminiNextGenWebhooks(self._api_client)
     return self._webhooks
 
   @property
   def agents(self) -> AsyncGeminiNextGenAgents:
+    from ._gaos.google_genai import AsyncGeminiNextGenAgents
+
     global _agent_experimental_warned
     if not _agent_experimental_warned:
       _agent_experimental_warned = True
@@ -400,6 +407,8 @@ def _get_api_client(
 
   @property
   def _nextgen_client(self) -> GeminiNextGenAPI:
+    from ._gaos.google_genai import build_google_genai_client
+
     if self._nextgen_client_instance is None:
       self._nextgen_client_instance = build_google_genai_client(
           self._api_client
@@ -408,18 +417,24 @@ def _nextgen_client(self) -> GeminiNextGenAPI:
 
   @property
   def interactions(self) -> GeminiNextGenInteractions:
+    from ._gaos.google_genai import GeminiNextGenInteractions
+
     if self._interactions is None:
       self._interactions = GeminiNextGenInteractions(self._api_client)
     return self._interactions
 
   @property
   def webhooks(self) -> GeminiNextGenWebhooks:
+    from ._gaos.google_genai import GeminiNextGenWebhooks
+
     if self._webhooks is None:
       self._webhooks = GeminiNextGenWebhooks(self._api_client)
     return self._webhooks
 
   @property
   def agents(self) -> GeminiNextGenAgents:
+    from ._gaos.google_genai import GeminiNextGenAgents
+
     global _agent_experimental_warned
     if not _agent_experimental_warned:
       _agent_experimental_warned = True