fix the importing error

libs/community/langchain_community/utils/google.py

@@ -1,9 +1,37 @@
 """Utilities to use Google provided components."""
 
 from importlib import metadata
-from typing import Optional
+from typing import Any, Callable, Optional, Union
 
+import google.api_core
 from google.api_core.gapic_v1.client_info import ClientInfo
+from langchain_core.callbacks import (
+    AsyncCallbackManagerForLLMRun,
+    CallbackManagerForLLMRun,
+)
+from langchain_core.language_models.llms import create_base_retry_decorator
+
+
+def create_retry_decorator(
+    *,
+    max_retries: int = 1,
+    run_manager: Optional[
+        Union[AsyncCallbackManagerForLLMRun, CallbackManagerForLLMRun]
+    ] = None,
+) -> Callable[[Any], Any]:
+    """Creates a retry decorator for Vertex / Palm LLMs."""
+
+    errors = [
+        google.api_core.exceptions.ResourceExhausted,
+        google.api_core.exceptions.ServiceUnavailable,
+        google.api_core.exceptions.Aborted,
+        google.api_core.exceptions.DeadlineExceeded,
+        google.api_core.exceptions.GoogleAPIError,
+    ]
+    decorator = create_base_retry_decorator(
+        error_types=errors, max_retries=max_retries, run_manager=run_manager
+    )
+    return decorator
 
 
 def get_client_info(module: Optional[str] = None) -> "ClientInfo":