Fix Google components init to use lazy loading pattern

Author: andreil-spec

langbuilder/src/backend/base/langbuilder/components/google/__init__.py

@@ -1,16 +1,35 @@
-from .gmail import GmailLoaderComponent
-from .google_bq_sql_executor import BigQueryExecutorComponent
-from .google_drive import GoogleDriveComponent
-from .google_drive_docs_parser import GoogleDriveDocsParserSA
-from .google_drive_docs_watcher import GoogleDriveDocsWatcher
-from .google_drive_search import GoogleDriveSearchComponent
-from .google_generative_ai import GoogleGenerativeAIComponent
-from .google_generative_ai_embeddings import GoogleGenerativeAIEmbeddingsComponent
-from .google_oauth_token import GoogleOAuthToken
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
+from langbuilder.components._importing import import_mod
+
+if TYPE_CHECKING:
+    from langbuilder.components.google.gmail import GmailLoaderComponent
+    from langbuilder.components.google.google_bq_sql_executor import BigQueryExecutorComponent
+    from langbuilder.components.google.google_drive import GoogleDriveComponent
+    from langbuilder.components.google.google_drive_docs_parser import GoogleDriveDocsParserSA
+    from langbuilder.components.google.google_drive_docs_watcher import GoogleDriveDocsWatcher
+    from langbuilder.components.google.google_drive_search import GoogleDriveSearchComponent
+    from langbuilder.components.google.google_generative_ai import GoogleGenerativeAIComponent
+    from langbuilder.components.google.google_generative_ai_embeddings import GoogleGenerativeAIEmbeddingsComponent
+    from langbuilder.components.google.google_oauth_token import GoogleOAuthToken
+
+_dynamic_imports = {
+    "GmailLoaderComponent": "gmail",
+    "BigQueryExecutorComponent": "google_bq_sql_executor",
+    "GoogleDriveComponent": "google_drive",
+    "GoogleDriveDocsParserSA": "google_drive_docs_parser",
+    "GoogleDriveDocsWatcher": "google_drive_docs_watcher",
+    "GoogleDriveSearchComponent": "google_drive_search",
+    "GoogleGenerativeAIComponent": "google_generative_ai",
+    "GoogleGenerativeAIEmbeddingsComponent": "google_generative_ai_embeddings",
+    "GoogleOAuthToken": "google_oauth_token",
+}
 
 __all__ = [
-    "BigQueryExecutorComponent",
     "GmailLoaderComponent",
+    "BigQueryExecutorComponent",
     "GoogleDriveComponent",
     "GoogleDriveDocsParserSA",
     "GoogleDriveDocsWatcher",
@@ -19,3 +38,21 @@
     "GoogleGenerativeAIEmbeddingsComponent",
     "GoogleOAuthToken",
 ]
+
+
+def __getattr__(attr_name: str) -> Any:
+    """Lazily import Google components on attribute access."""
+    if attr_name not in _dynamic_imports:
+        msg = f"module '{__name__}' has no attribute '{attr_name}'"
+        raise AttributeError(msg)
+    try:
+        result = import_mod(attr_name, _dynamic_imports[attr_name], __spec__.parent)
+    except (ModuleNotFoundError, ImportError, AttributeError) as e:
+        msg = f"Could not import '{attr_name}' from '{__name__}': {e}"
+        raise AttributeError(msg) from e
+    globals()[attr_name] = result
+    return result
+
+
+def __dir__() -> list[str]:
+    return list(__all__)