Ng ^dZddlmZddlmZmZmZmZmZddl m Z m Z ddl m Z mZerddlmZddlmZddlmZd d d d e d edeee e fdeegeffdZddedd fdZ d deedeededdd fdZd!deeddfdZd!dedeeddfdZd S)"zUtilities to init Vertex AI.)metadata) TYPE_CHECKINGAnyCallableOptionalUnion)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun)BaseLLMcreate_base_retry_decorator ClientInfo) CredentialsImageN) max_retries run_managerllmrrreturncddl}|jjj|jjj|jjj|jjj|jjjg}t|||}|S)z0Create a retry decorator for Vertex / Palm LLMs.rN) error_typesrr) google.api_coreapi_core exceptionsResourceExhaustedServiceUnavailableAbortedDeadlineExceededGoogleAPIErrorr )rrrgoogleerrors decorators b/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/utilities/vertexai.pycreate_retry_decoratorr%su "4"5"*"3"1 F, I 1.38.0minimum_expected_versionc&td|)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(s r$raise_vertex_import_errorr+*s+  K0H K K  r&projectlocation credentialsrc~ ddl}n#t$rtYnwxYw||||dS)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.r0s r$ init_vertexair28so"$ $$$!#####$ MMs ""modulerc ddlm}n"#t$r}td|d}~wwxYwtjd}|r|d|n|}||d|S) 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_inforr*rversion)r3rexclangchain_versionr7s r$get_client_infor=UsCCCCCCC  *    !(55+1H ''v'''7H :58 688   s (#(pathrc  ddlm}n#t$rtdwxYwddlm}||}|d}t||dd |dd  }t|d krtd |d |j |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)r>r,r@r gcs_clientpiecesblobss r$load_image_from_gcsrTpsS((((((( SSSQRRRS88888800J ZZ__F &&vay&*9M9M&NN O OE 5zzA~~EdEEEFFF 5 E!H6688 9 99s #)r')NNN)N)__doc__ importlibrtypingrrrrrlangchain_core.callbacksr r #langchain_core.language_models.llmsr r r9rgoogle.auth.credentialsrrGrintr%strr+r2r=rTr&r$r^s""@@@@@@@@@@@@@@UTTTTTTT9??????333333888888    +-EEF seSj0   4    ""+/ c]sm-( :HSM\6 : :c :HSM :W : : : : : :r&