a bg=@svdZddlmZddlZddlmZmZmZmZm Z m Z m Z m Z m Z mZmZddlmZmZddlmZmZmZddlmZmZmZmZmZmZmZmZm Z m!Z!m"Z"ddl#m$Z$m%Z%m&Z&dd l'm(Z(m)Z)m*Z*m+Z+dd l,m-Z-m.Z.m/Z/m0Z0dd l1m2Z2m3Z3m4Z4m5Z5m6Z6e7e8Z9d d dddZ:d dddddZ;ddddddZGd#d d eZ?dS)$zJinaChat wrapper.) annotationsN) Any AsyncIteratorCallableDictIteratorListMappingOptionalTupleTypeUnion)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun) BaseChatModelagenerate_from_streamgenerate_from_stream) AIMessageAIMessageChunk BaseMessageBaseMessageChunk ChatMessageChatMessageChunkFunctionMessage HumanMessageHumanMessageChunk SystemMessageSystemMessageChunk)ChatGenerationChatGenerationChunk ChatResult)convert_to_secret_strget_from_dict_or_envget_pydantic_field_namespre_init) ConfigDictField SecretStrmodel_validator)before_sleep_logretryretry_if_exception_typestop_after_attemptwait_exponentialJinaChatCallable[[Any], Any])llmreturncCsrddl}d}d}tdt|jtd||dt|jjt|jjBt|jj Bt|jj Bt|jj Bt t tjdSNr<T) multiplierminmax)reraisestopwaitr* before_sleepopenair*r, max_retriesr-r+errorTimeoutZAPIErrorZAPIConnectionErrorZRateLimitErrorZServiceUnavailableErrorr)loggerloggingWARNING)r0r= min_seconds max_secondsrFv/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/chat_models/jinachat.py_create_retry_decorator=s$       rHr)r0kwargsr1cs4t}|dddfdd }|fi|IdHS)z0Use tenacity to retry the async completion call.rrIr1csjjfi|IdHSN)clientZacreaterIr0rFrG_completion_with_retryWsz6acompletion_with_retry.._completion_with_retryNrH)r0rIretry_decoratorrOrFrNrGacompletion_with_retrySsrRMapping[str, Any]zType[BaseMessageChunk]r)_dict default_classr1cCs|d}|dpd}|dks(|tkr2t|dS|dksB|tkrLt|dS|dks\|tkrft|dS|sr|tkr~t||dS||dSdS) NrolecontentuserrW assistantsystemrWrV)getrrrr)rTrUrVrWrFrFrG_convert_delta_to_message_chunk_s      r_r)rTr1cCsf|d}|dkrt|ddS|dkr<|dp0d}t|dS|dkrRt|ddSt|d|dSdS) NrVrYrWrZr[rXr\r])rrrr)rTrVrWrFrFrG_convert_dict_to_messageqs  r`dict)messager1cCst|tr|j|jd}nrt|tr2d|jd}nZt|trJd|jd}nBt|trbd|jd}n*t|tr~d|j|jd}nt d|d|j vr|j d|d<|S) N)rVrWrYr[r\function)rVnamerWzGot unknown type rd) isinstancerrVrWrrrrrd ValueErrorZadditional_kwargs)rbZ message_dictrFrFrG_convert_message_to_dict~s"      rgc@seZdZUdZeddddZeddddZd Zd e d <d Z d e d<e e dZ de d<d Zde d<d Zde d<dZde d<dZde d<d Zde d<eddZedd edd d!d"d#Zed$d$d!d%d&Zeddd'd(Zd)dd*d+Zd d d,d-d.Zd/d0d1d2d3ZdQd4d5d6d d7d8d9d:ZdRd4d5d6d d;d8dd?d@dAZdBd;dCdDdEZ dSd4d5dFd dGd8dHdIZ!dTd4d5dFd d;d8dJdKZ"edBddLdMZ#edNddOdPZ$d S)Ur.a`Jina AI` Chat models API. To use, you should have the ``openai`` python package installed, and the environment variable ``JINACHAT_API_KEY`` set to your API key, which you can generate at https://chat.jina.ai/api. Any parameters that are valid to be passed to the openai.create call can be passed in, even if not explicitly saved on this class. Example: .. code-block:: python from langchain_community.chat_models import JinaChat chat = JinaChat() zDict[str, str])r1cCsddiS)Njinachat_api_keyJINACHAT_API_KEYrFselfrFrFrG lc_secretsszJinaChat.lc_secretsboolcCsdS)z9Return whether this model can be serialized by Langchain.FrF)clsrFrFrGis_lc_serializableszJinaChat.is_lc_serializableNrrLgffffff?float temperature)default_factoryzDict[str, Any] model_kwargszOptional[SecretStr]rhz+Optional[Union[float, Tuple[float, float]]]request_timeoutintr>F streamingz Optional[int] max_tokensT)Zpopulate_by_namebefore)mode)valuesr1c Cst|}|di}t|D]P}||vr8td|d||vrtd|d|d|d||||<q||}|rtd|d ||d<|S) z>Build extra kwargs from additional params that were passed in.rszFound z supplied twice.z WARNING! z/ is not default parameter. zJ was transferred to model_kwargs. Please confirm that z is what you intended.z Parameters za should be specified explicitly. Instead they were passed in as part of `model_kwargs` parameter.) r#r^listrfrAwarningpop intersectionkeys)rnr{Zall_required_field_namesextra field_nameZinvalid_model_kwargsrFrFrG build_extras*   zJinaChat.build_extrarcCsjtt|dd|d<z ddl}Wnty:tdYn0z|j|d<WntydtdYn0|S)z?Validate that api key and python package exists in environment.rhrirNzTCould not import openai python package. Please install it with `pip install openai`.rLz`openai` has no `ChatCompletion` attribute, this is likely due to an old version of the openai package. Try upgrading it with `pip install --upgrade openai`.)r!r"r= ImportErrorZChatCompletionAttributeErrorrf)rnr{r=rFrFrGvalidate_environments       zJinaChat.validate_environmentcCs|j|j|j|jd|jS)z4Get the default parameters for calling JinaChat API.)rtrxstreamrq)rtrxrwrqrsrjrFrFrG_default_paramsszJinaChat._default_paramsr/cCsrddl}d}d}tdt|jtd||dt|jjt|jjBt|jj Bt|jj Bt|jj Bt t tjdSr2r<)rkr=rDrErFrFrGrHs$       z JinaChat._create_retry_decoratorrJc s.}|dddfdd }|fi|S)z*Use tenacity to retry the completion call.rrJcsjjfi|SrK)rLcreaterMrjrFrGrOsz>JinaChat.completion_with_retry.._completion_with_retryrP)rkrIrQrOrFrjrGcompletion_with_retryszJinaChat.completion_with_retryzList[Optional[dict]]ra) llm_outputsr1cCs\i}|D]J}|durq|d}|D]*\}}||vrH|||7<q&|||<q&qd|iS)N token_usage)items)rkrZoverall_token_usageoutputrkvrFrFrG_combine_llm_outputs s zJinaChat._combine_llm_outputszList[BaseMessage]zOptional[List[str]]z"Optional[CallbackManagerForLLMRun]zIterator[ChatGenerationChunk])messagesr9 run_managerrIr1c ks|||\}}i||ddi}t}|jfd|i|D]H}|ddd} t| |}|j}t|d} |r~|j|j| d| Vq\z2JinaChat._create_message_dicts..)ra_invocation_paramsrf)rkrr9rrrFrFrGrTs zJinaChat._create_message_dictsrS)rr1cCsJg}|dD]$}t|d}t|d}||q d|di}t||dS)Nrrbrrusage) generations llm_output)r`rappendr )rkrrresrbgenrrFrFrGr_s     zJinaChat._create_chat_resultz'Optional[AsyncCallbackManagerForLLMRun]z"AsyncIterator[ChatGenerationChunk]c Ks|||\}}i||ddi}t}t|fd|i|IdH2zV3dHW}|ddd} t| |}|j}t|d} |r|j|j| dIdH| VqB6dSr)rrrRr_rrrrWrrFrFrG_astreamhs"  zJinaChat._astreamc sl|jr,|jf|||d|}t|IdHS|||\}}i||}t|fd|i|IdH}||Sr)rwrrrrRrrrFrFrG _agenerate~s zJinaChat._ageneratecCs&|jo|jddd}i||jS)z,Get the parameters used to invoke the model.zhttps://api.chat.jina.ai/v1jinachat)Zapi_keyZapi_basemodel)rhZget_secret_valuer)rkZjinachat_credsrFrFrGrs zJinaChat._invocation_paramsstrcCsdS)zReturn type of chat model.rrFrjrFrFrG _llm_typeszJinaChat._llm_type)NN)NN)NN)NN)%__name__ __module__ __qualname____doc__propertyrl classmethodrorL__annotations__rqr&rarsrhrtr>rwrxr%Z model_configr(rr$rrrHrrrrrrrrrrrFrFrFrGr.sX             )@r __future__rrBtypingrrrrrrr r r r r Zlangchain_core.callbacksrrZ*langchain_core.language_models.chat_modelsrrrZlangchain_core.messagesrrrrrrrrrrrZlangchain_core.outputsrrr Zlangchain_core.utilsr!r"r#r$Zpydanticr%r&r'r(tenacityr)r*r+r,r- getLoggerrrArHrRr_r`rgr.rFrFrFrGs" 44