o ȳg @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) error_typesrr) google.api_coreapi_core exceptionsResourceExhaustedServiceUnavailableAbortedDeadlineExceededGoogleAPIErrorr )rrrgoogleerrors decoratorr"b/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/langchain_community/utilities/vertexai.pycreate_retry_decorators r$1.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)r&r"r"r#raise_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+)vertexair'r(init)r)r*r+r,r"r"r# 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-coreN langchain-z langchain/)client_library_version user_agent)$google.api_core.gapic_v1.client_infor r'rversion)r/r exclangchain_versionr2r"r"r#get_client_infoUs"  r8pathrcCszddlm}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 !) google.cloudr:r'"vertexai.preview.generative_modelsrClientsplitlist list_blobsjoinlen ValueError from_bytesdownload_as_bytes)r9r)r:r gcs_clientpiecesblobsr"r"r#load_image_from_gcsps    $ rN)r%)NNN)N)__doc__ importlibrtypingrrrrrlangchain_core.callbacksrr #langchain_core.language_models.llmsr r r4r google.auth.credentialsrrArintr$strr(r.r8rNr"r"r"r#sJ