o Zh @s&dZddlmZddlmZmZmZmZmZddl m Z m Z ddl m Z mZer:ddlmZddlmZddlmZd d d d e d edeee e fdeegeffddZd$dedd fddZ d%deedeededdd fddZd&deeddfddZd&d edeedd!fd"d#Zd S)'zUtilities to init Vertex AI.)metadata) TYPE_CHECKINGAnyCallableOptionalUnion)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun)BaseLLMcreate_base_retry_decorator ClientInfo) CredentialsImageN) max_retries run_managerllmrrreturncCsFddl}|jjj|jjj|jjj|jjj|jjjg}t|||d}|S)z0Create a retry decorator for Vertex / Palm LLMs.rN)Z error_typesrr) Zgoogle.api_coreZapi_core exceptionsZResourceExhaustedZServiceUnavailableZAbortedZDeadlineExceededZGoogleAPIErrorr )rrrgoogleerrors decoratorr]/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/utilities/vertexai.pycreate_retry_decorators r1.38.0minimum_expected_versioncCstd|)zRaise ImportError related to Vertex SDK being not available. Args: minimum_expected_version: The lowest expected version of the SDK. Raises: ImportError: an ImportError that mentions a required version of the SDK. zePlease, install or upgrade the google-cloud-aiplatform library: pip install google-cloud-aiplatform>=) ImportError)rrrrraise_vertex_import_error*s r projectlocation credentialsrcCs:zddl}Wn tytYnw|j|||ddS)aInit Vertex AI. Args: project: The default GCP project to use when making Vertex API calls. location: The default location to use when making API calls. credentials: The default custom credentials to use when making API calls. If not provided credentials will be ascertained from the environment. Raises: ImportError: If importing vertexai SDK did not succeed. rN)r!r"r#)vertexairr init)r!r"r#r$rrr init_vertexai8s    r&moduler c Csfzddlm}Wnty}ztd|d}~wwtd}|r(|d|n|}||d|dS) zReturn a custom user agent header. Args: module (Optional[str]): Optional. The module for a custom user agent header. Returns: google.api_core.gapic_v1.client_info.ClientInfo rr zPCould not import ClientInfo. Please, install it with pip install google-api-coreNZ langchain-z langchain/)client_library_version user_agent)$google.api_core.gapic_v1.client_infor rrversion)r'r excZlangchain_versionr)rrrget_client_infoUs"  r.pathrcCszddlm}Wn tytdwddlm}|j|d}|d}t|j|dd |dd d }t |d krEt d |d | |d S)z(Load an image from Google Cloud Storage.r)storagez5Could not import google-cloud-storage python package.r)r!/N)prefixrz"Found more than one candidate for !)Z google.cloudr0r"vertexai.preview.generative_modelsrZClientsplitlistZ list_blobsjoinlen ValueError from_bytesZdownload_as_bytes)r/r!r0rZ gcs_clientpiecesZblobsrrrload_image_from_gcsps    $ r>)r)NNN)N)__doc__ importlibrtypingrrrrrZlangchain_core.callbacksrr Z#langchain_core.language_models.llmsr r r+r Zgoogle.auth.credentialsrr6rintrstrr r&r.r>rrrrsJ