a bg@s dZddlmZddlmZmZmZmZmZddl m Z m Z ddl m Z mZertddlmZddlmZddlmZd d d e eeee e feegefd d dZded dddZd eeeeedd dddZd!eeddddZd"eeeddddZd S)#zUtilities to init Vertex AI.)metadata) TYPE_CHECKINGAnyCallableOptionalUnion)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun)BaseLLMcreate_base_retry_decorator ClientInfo) CredentialsImageN) max_retries run_manager)llmrrreturncCsFddl}|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 )rrrZgoogleerrors decoratorrt/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/utilities/vertexai.pycreate_retry_decorators r1.38.0)minimum_expected_versionrcCstd|dS)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>=N) ImportError)rrrrraise_vertex_import_error*s rr)projectlocation credentialsrcCs:z ddl}Wnty$tYn0|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")vertexairrinit)r r!r"r#rrr init_vertexai8s   r%r )modulerc Csrzddlm}Wn.ty>}ztd|WYd}~n d}~00td}|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-r)pathr rcCszddlm}Wnty*tdYn0ddlm}|j|d}|d}t|j|dd |dd d }t |d krt 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.cloudr/r"vertexai.preview.generative_modelsrZClientsplitlistZ list_blobsjoinlen ValueError from_bytesZdownload_as_bytes)r.r r/rZ gcs_clientpiecesZblobsrrrload_image_from_gcsps    $ r=)r)NNN)N)N)__doc__ importlibrtypingrrrrrZlangchain_core.callbacksrr Z#langchain_core.language_models.llmsr r r*r Zgoogle.auth.credentialsrr5rintrstrrr%r-r=rrrrs<