阺hnYTUddlmZddlZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZddlmZddlmZddlZddlmZmZddlmZddl m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(dd l)m*Z*m+Z+m,Z,dd l-m.Z.m/Z/m0Z0dd l1m2Z2dd l3m4Z4m5Z5m6Z6m7Z7m8Z8dd l9m:Z:dZ;dem?Z?e;e?fzZ;n#e@$rdZ=dZ?YnwxYwejAeBZCGdde:ZDdGdZEdHdZFdHdZGdId ZHdJd#ZIdKd%ZJdKd&ZKdLdMd+ZLdNd-ZMdOd1ZN dPdQd7ZOdRd:ZPe!e&e$e*fdSdDZQGdEdFeZRdS)T) annotationsN)BytesIO) Any AsyncIteratorCallableDictIteratorListMappingOptionalSequenceTupleTypeUnioncast)urlparse)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun) BaseChatModel) AIMessageAIMessageChunk BaseMessage ChatMessageChatMessageChunk HumanMessageHumanMessageChunk SystemMessage)ChatGenerationChatGenerationChunk ChatResult)Field SecretStrroot_validator)get_from_dict_or_env)before_sleep_logretryretry_if_exception_typestop_after_attemptwait_exponential)GoogleGenerativeAIErrorr IMAGE_TYPES)ImageceZdZdZdS)ChatGoogleGenerativeAIErrora Custom exception class for errors associated with the `Google GenAI` API. This exception is raised when there are specific issues related to the Google genai API usage in the ChatGoogleGenerativeAI class, such as unsupported message types or roles. N)__name__ __module__ __qualname____doc__r+j/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langchain_google_genai/chat_models.pyr/r/Dsr4r/returnCallable[[Any], Any]c jddl}d}d}d}d}tdt|t|||t |jjjt |jjjzt |jjj ztttj  S) a Creates and returns a preconfigured tenacity retry decorator. The retry decorator is configured to handle specific Google API exceptions such as ResourceExhausted and ServiceUnavailable. It uses an exponential backoff strategy for retries. Returns: Callable[[Any], Any]: A retry decorator configured for handling specific Google API exceptions. rN< T) multiplierminmax)reraisestopwaitr& before_sleep)google.api_core.exceptionsr&r(r)r'api_core exceptionsResourceExhaustedServiceUnavailableGoogleAPIErrorr%loggerloggingWARNING)googler= min_seconds max_seconds max_retriess r5_create_retry_decoratorrQNs&%%%JKKK   , , + V V V #FO$>$P Q Q%fo&@&STT U%fo&@&OPP Q&fgo>>    r4generation_methodrkwargsrc \t}ddlm|dfd }|di|S)  Executes a chat generation method with retry logic using tenacity. This function is a wrapper that applies a retry mechanism to a provided chat generation function. It is useful for handling intermittent issues like network errors or temporary service unavailability. Args: generation_method (Callable): The chat generation method to be executed. **kwargs (Any): Additional keyword arguments to pass to the generation method. Returns: Any: The result from the chat generation method. rInvalidArgumentrSrr6ct di|S#$r}td||d}~wt$r}|d}~wwxYwNz%Invalid argument provided to Gemini: r+r/ ExceptionrSerWrRs r5_chat_with_retryz*_chat_with_retry.._chat_with_retrys{ $$..v.. .   -;;;    G s 7# 727rSrr6rr+rQrDrW)rRrSretry_decoratorr^rWs` @r5r^r^nsi.//O::::::      _    % %f % %%r4clKt}ddlm|dfd }|d i|d{VS) rUrrVrSrr6cK di|d{VS#$r}td||d}~wt$r}|d}~wwxYwrYrZr\s r5_achat_with_retryz,_achat_with_retry.._achat_with_retrys **44V44444444 4   -;;;    G s ?+ ?:?Nr_r+r`)rRrSrardrWs` @r5rdrds.//O::::::      _ #",,V,, , , , , , ,,r4partdictboolc d|vS)Ntyper+)res r5_is_openai_parts_formatrjs T>r4modelstrc d|vS)Nvisionr+rks r5_is_vision_modelrps u r4sc t|}t|j|jgS#t$r(}t d|Yd}~dSd}~wwxYw)NzUnable to parse URL: F)rallschemenetlocr[rJdebug)rqresultr]s r5_is_urlrxsp!FM6=1222  0Q00111uuuuus), AAAc,|dS)Nz data:image) startswith)rqs r5_is_b64r{s << % %%r4pathproject Optional[str]r-c 0 ddlm}n#t$rtdwxYwttd||}|d}t ||dd|dd }t|d krtd |d |d }tj t|S) Nr)storagezlgoogle-cloud-storage is required to load images from GCS. Install it with `pip install google-cloud-storage`KPIL is required to load images. Please install it with `pip install pillow`)r}/r9)prefixr:z"Found more than one candidate for !) google.cloudr ImportErrorPILClientsplitlist list_blobsjoinlen ValueErrordownload_as_bytesr-openr)r|r}r gcs_clientpiecesblobs img_bytess r5_load_image_from_gcsrs# (((((((     B     { (   00J ZZ__F &&vay&*9M9M&NN O OE 5zzA~~EdEEEFFFa**,,I 9>>'),, - --s # image_sourcec(ttd t|tr|St |r}|drt |Stj|}| tj t|j St|rY|dd\}}t!j|}tj t|St$j|rtj |St+d#t,$r}t+d|d}~wwxYw)Nrzgs://,r:z=The provided string is not a valid URL, base64, or file path.z-Unable to process the provided image source: )rr isinstancer,rxrzrrequestsgetraise_for_statusr-rrcontentr{rbase64 b64decodeosr|existsrr[)rresponse_encodeddatar]s r5 _url_to_pilrs~ { (   N lK 0 0   \ " " &&w// :+L999|L11H  % % ' ' '9>>'(*:";";<< < \ " " %++C33JAw#G,,D9>>'$--00 0 W^^L ) ) 9>>,// /O  NNNLLLMMMNs6E02E0"AE0;A'E0#=E0!E00 F:F  F raw_content&Union[str, Sequence[Union[str, dict]]]List[genai.types.PartType]cg}t|tr|gn|}|D]]}t|tr4|tj|Lt|t rt|r|ddkr|d|di|ddkra|d}t|trd|vrtd||d}|dt|itd|dt d ||Ptd |S) z:Converts a list of LangChain messages into a google parts.)textrir image_urlurlz#Unrecognized message image format: inline_dataz Unrecognized message part type: zz'_response_to_result..s911117V $$V,,111r4safety_ratings)rgeneration_infozKGemini produced an empty response. Continuing with empty message Feedback: r) generations llm_output)rrirr[rJrv candidatesrrrrrrnamerrr )rrrrrrrr]rrole_map candidater parts_contentmsgrs r5_response_to_resultrfsBJL L"8#;<<DD(EO-Z dLdMd@ZdS)NChatGoogleGenerativeAIa`Google Generative AI` Chat models API. To use, you must have either: 1. The ``GOOGLE_API_KEY``` environment variable set with your API key, or 2. Pass your API key using the google_api_key kwarg to the ChatGoogle constructor. Example: .. code-block:: python from langchain_google_genai import ChatGoogleGenerativeAI chat = ChatGoogleGenerativeAI(model="gemini-pro") chat.invoke("Write me a ballad about LangChain") .zBThe name of the model to use. Supported examples: - gemini-pro) descriptionrlrkNzMax output tokens)defaultrintmax_output_tokensrclientzOptional[SecretStr]google_api_keyzOptional[float] temperaturez Optional[int]top_ktop_pr:candidate_count)raliasnFrgrz0Client options to pass to the Google API client.zOptional[Dict]client_optionsz3A string, one of: [`rest`, `grpc`, `grpc_asyncio`].r~ transportceZdZdZdS)ChatGoogleGenerativeAI.ConfigTN)r0r1r2allow_population_by_field_namer+r4r5Configrs)-&&&r4rr6Dict[str, str]c ddiS)NrGOOGLE_API_KEYr+selfs r5 lc_secretsz!ChatGoogleGenerativeAI.lc_secretss "233r4cdS)Nzchat-google-generative-air+rs r5 _llm_typez ChatGoogleGenerativeAI._llm_types**r4c&|jduod|jvS)Ngeminirors r5 _is_geminiaiz#ChatGoogleGenerativeAI._is_geminiaisz%@(dj*@@r4cdS)NTr+rs r5is_lc_serializablez)ChatGoogleGenerativeAI.is_lc_serializablestr4valuesrcxt|dd}t|tr|}t j||d|d|d$d|dcxkrd ksntd |d $d|d cxkrd ksntd |d |d dkrtd|d}t j||d<|S)z@Validates params and passes them to google-generativeai package.rrrr)api_keyrrrNrr:z+temperature must be in the range [0.0, 1.0]rz%top_p must be in the range [0.0, 1.0]rztop_k must be positiverk) model_namer) r$rr"get_secret_valuer configurerrGenerativeModel)clsrrrks r5validate_environmentz+ChatGoogleGenerativeAI.validate_environments[. $&6   ni 0 0 ?+<<>>N "jj--!::&677    JJ} % % 1 .3333!3333JKK K ::g   *1w3L3L3L3L13L3L3L3LDEE E ::g   *vg!/C/C566 6w 0EBBBx r4Dict[str, Any]c8|j|j|j|jdS)zGet the identifying parameters.rkrrrrrs r5_identifying_paramsz*ChatGoogleGenerativeAI._identifying_paramss(Z+Z    r4rAOptional[List[str]]rSc d|j|j||j|j|jdD}d|vri||d}d|i|}|S)Nci|] \}}||| SNr+)rkvs r5 z:ChatGoogleGenerativeAI._prepare_params..s/    1} q}}r4)rrstop_sequencesrrrgeneration_config)rrrrritemspop)rrArS gen_configparamss r5_prepare_paramsz&ChatGoogleGenerativeAI._prepare_paramss   $(6#/"&%)%; egg     & ( (JJJ&**5H*I*IJJ%zO? ? ? ? ? #5? ? ? 9 9 9 9 9 9  #8,,,r4Iterator[ChatGenerationChunk]c+JK|||\}}}tdd|i||jdd}|D]j} t| tt t t} tt| j d} | V|r| | j kdSNr&rT)rRstream)rrrrrr+) r'r^r(rrrrrrron_llm_new_tokenr) rrrAr"rSrr*rrchunk _chat_resultgens r5_streamzChatGoogleGenerativeAI._streamJs!% 2 28$ 2 G Gg8H9 9 9 9 #/ 9 9 9 9   7 7E.'-+0 L*L,DQ,GHHCIII 7,,SX666 7 7r4"AsyncIterator[ChatGenerationChunk]c lK|||\}}}tdd|i||jddd{V23d{V}t|tt t t} tt| j d} | WV|r | | j d{Vw6dSr2) r'rdr.rrrrrrrr4r) rrrAr"rSrr*rr5r6r7s r5_astreamzChatGoogleGenerativeAI._astreames0!% 2 28$ 2 G Gg!2" " " " #5 " " " "        = = = = = = =% /'-+0 L*L,DQ,GHHCIIII =!2238<<<<<<<<rr*s r5r'z$ChatGoogleGenerativeAI._prepare_chatso &%d55f55% ,0,P   ++--{%%g%66tW$$r4)r6r)r6rl)r6rg)rrr6r)r6r)rArrSrr6r)NN) rr!rArr"r#rSrr6r ) rr!rArr"r,rSrr6r ) rr!rArr"r#rSrr6r0) rr!rArr"r,rSrr6r9r)rr!rArrSrr6r<) r0r1r2r3r!rk__annotations__rrrrrrrrrrpropertyrrr classmethodrr#rrr r+r/r8r;r'r+r4r5rrs#" E #U4=PQQQQQQQKKK*.N....#'K''''EE U1$5 6 6 6A6666I,1#1111&+U F&&&N %u I   I ........444X4+++X+AAAXA[^8   X .%):> - - - - -$%)?C - - - - -$%):> 77777<%)?C =====:%) % % % % % % %r4r)r6r7)rRrrSrr6r)rerfr6rg)rkrlr6rg)rqrlr6rgr)r|rlr}r~r6r-)rrlr6r-)rrr6r)F)rrrrgr6r)rrr6r) rrrrrrrrrrr6r )S __future__rrrKriortypingrrrrr r r r r rrrr urllib.parsergoogle.generativeai generativeairr langchain_core.callbacks.managerrr*langchain_core.language_models.chat_modelsrlangchain_core.messagesrrrrrrrrlangchain_core.outputsrrr langchain_core.pydantic_v1r!r"r#langchain_core.utilsr$tenacityr%r&r'r(r)langchain_google_genai._commonr*r,r@r PIL.Imager-r getLoggerr0rJr/rQr^rdrjrprxr{rrrrrrrr+r4r5rSsw"""""""  "!!!!!$#####EDDDDD                    SRRRRRRRRRGGGGGGGGGG555555CBBBBB JJJ(KK C EEE  8 $ $"9@&&&&B----B&&&&......NNNN:%%%%RTY+++++\2#,%1$/)7 3F3F3F3F3Flp%p%p%p%p%]p%p%p%p%p%s B11 B=<B=