7|h ,dZddlmZddlmZmZmZmZmZddl m Z m Z ddl m Z mZerddlmZddlmZddlmZd d d d e d edeee e fdeegeffdZddedd fdZ ddeedeededdd fdZd deeddfdZd dedeeddfdZy )!zUtilities to init Vertex AI.)metadata) TYPE_CHECKINGAnyCallableOptionalUnion)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun)BaseLLMcreate_base_retry_decorator ClientInfo) CredentialsImageN) max_retries run_managerllmrrreturncdddl}|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 e/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/utilities/vertexai.pycreate_retry_decoratorr%s ""44""55""**""33""11 F, I minimum_expected_versionctd|)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"  00H/I K r&projectlocation credentialsrch ddl}j|||y#t$r tY*wxYw)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/s r$ init_vertexair18s="$ MM $!#$s 11modulerc ddlm}tjd}|r|d|n|}||d|S#t$r}td|d}~wwxYw) 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)r2rexclangchain_versionr6s r$get_client_infor<Us~C!((5+1  Qvh'7H 5 678   *  s5 A A  Apathrc t ddlm}ddlm}|j |}|j d}t|j|ddj|dd  }t|d kDrtd |d |j|djS#t$r tdwxYw)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_gcsrSpsS(90J ZZ_F &&vay&*9M&N OE 5zA~=dV1EFF 5  E!H668 99 SQRRSs B""B7)z1.38.0)NNN)N)__doc__ importlibrtypingrrrrrlangchain_core.callbacksr r #langchain_core.language_models.llmsr r r8rgoogle.auth.credentialsrrFrintr%strr*r1r<rSr&r$r]s"@@U?38    +-EEF seSj0  4 ""+/ c]sm-( :HSM\6 :c :HSM :W :r&