a bg?@s|ddlZddlmZmZmZmZmZmZmZm Z m Z m Z m Z ddl mZmZddlmZddlmZmZmZddlmZmZmZmZmZmZddlmZmZm Z ddl!m"Z"dd l#m$Z$dd l%m&Z&dd l'm(Z(dd l)m*Z*m+Z+m,Z,m-Z-d Z.e e/ee e/ee/efffZ0ee/efee/efdddZ1e0eee/efdddZ2eeeee/efdddZ3GdddeZ4dS)N) Any AsyncIteratorCallableDictIteratorListMappingOptionalSequenceTypeUnion)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun)LanguageModelInput) BaseChatModelagenerate_from_streamgenerate_from_stream) AIMessageAIMessageChunk BaseMessage HumanMessage SystemMessage ToolMessage)ChatGenerationChatGenerationChunk ChatResult)Runnable)BaseTool)get_from_dict_or_envconvert_to_openai_tool) BaseModel ConfigDictFieldmodel_validatorz reka-flash)itemreturncCs6|ddkr2|d}t|tr2d|vr2|d|d<|S)zProcess a single content item.type image_urlurl) isinstancedict)r%r(r,r/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/chat_models/reka.pyprocess_content_item.s   r.)contentr&cCst|trd|dgSt|trvg}|D]H}t|trH|d|dq(t|trb|t|q(td|q(|StddS)z\Process content to handle both text and media inputs, returning a list of content items.textr'r0zInvalid content item format: zInvalid content formatN)r*strlistappendr+r. ValueError)r/resultr%r,r,r-process_content8s     r7)messagesr&c Csg}d}|D]}t|trL|durBt|jtr8|j}qJtdntdq t|trt|j}|r|rt|dtr|d ddkrd|dvr|d|dd|dd<n| dd|dd}| d |d q t|t rtd d i}|jr t|j}||d <d|j vrh|j d}g}|D]6}|d|ddt|ddd} | | q(||d<| |q t|trg} | |jtd|jid| d| d q tdt|q |S)z2Convert LangChain messages to Reka message format.Nz'SystemMessage content must be a string.z+Multiple system messages are not supported.rr'r0 r1user)roler/r;Z assistantr/ tool_callsidfunctionname arguments)r=r? parametersstatus) tool_call_idoutputZ tool_outputzUnsupported message type: )r*rr/r2 TypeErrorr5rr7r+getinsertr4radditional_kwargsjsonloadsrrCdumpsr') r8 reka_messagesZsystem_messagemessageZprocessed_contentZ reka_messager<Zformatted_tool_callsZ tool_callZformatted_tool_callZ content_listr,r,r-convert_to_reka_messagesKsr                  rNcsPeZdZUdZdZeed<dZeed<ee dZ e ed<eddZ e ed<dZeeed <d Zeed <dZeeed <d Ze ed<dZee ed<eedZee efed<eddZdZeeee eeefge fed<e dde!ee efee efdddZ"e#e$e efdddZ%e#e dddZ&d0eeeee ee'ee(e)dd d!Z*d1eeeee ee+ee,e)dd"d#Z-d2eeeee ee'ee.dd$d%Z/d3eeeee ee+ee.dd&d'Z0ee eeefe d(d)d*Z1d+dd,e2eee efe3e4ee5fe eeee6e7efd-fd.d/Z8Z9S)4ChatRekaz Reka chat large language models.Nclient async_client)defaultmodel max_tokens temperatureF streamingdefault_request_timeout max_retries reka_api_key)default_factory model_kwargsZforbid)extra token_counterbefore)mode)valuesr&cCsr|d}td|idd}||d<z0ddlm}m}||d|d<||d|d<WntyltdYn0|S) zBValidate that API key and Python package exist in the environment.r[Z REKA_API_KEYr) AsyncRekaReka)Zapi_keyrPrQzTCould not import Reka Python package. Please install it with `pip install reka-api`.)rFrZ reka.clientrcrd ImportError)clsrbr[rcrdr,r,r-validate_environments$     zChatReka.validate_environment)r&cCs0|j|jd}|jdur"|j|d<i||jS)z0Get the default parameters for calling Reka API.)rSrUNrV)rSrUrVr])selfparamsr,r,r-_default_paramss   zChatReka._default_paramscCsdS)zReturn type of chat model.z reka-chatr,)rhr,r,r- _llm_typeszChatReka._llm_type)r8stop run_managerkwargsr&c ks~t|}i|j|}|r"||d<|jjjfd|i|}|D]:}|jdjj} tt | dd} |rr|j | | d| Vq>dSNrlr8rr/rM)chunk) rNrjrPchat create_stream responsesrrr/rron_llm_new_token rhr8rlrmrnrLristreamrrr/Z chat_chunkr,r,r-_streamszChatReka._streamc Kst|}i|j|}|r"||d<|jjjfd|i|}|2zH3dHW}|jdjj} tt | dd} |r|j | | dIdH| Vq>6dSro) rNrjrQrsrtrurrr/rrrvrwr,r,r-_astreamszChatReka._astreamc Ks|jr"t|j|f||d|St|}i|j|}|rD||d<|jjjfd|i|}|jdj j r|jdj j }t dddd|Did } n$|jdj j } t | dur| ndd } t t| d gd S) Nrlrmrlr8rr<cSs*g|]"}|jd|jt|jddqSr>)r?r@)r=r'r>r=r?rIrKrA.0Ztcr,r,r- s  z&ChatReka._generate..r/rHrprqZ generations)rWrryrNrjrPrscreaterurMr<rr/rr rhr8rlrmrnrLriresponser<rMr/r,r,r- _generates* zChatReka._generatec s|jr(t|j|f||d|IdHSt|}i|j|}|rJ||d<|jjjfd|i|IdH}|jdj j r|jdj j }t dddd|Did } n$|jdj j } t | dur| ndd } t t| d gd S) Nr{rlr8rr|r<cSs*g|]"}|jd|jt|jddqSr}r~rr,r,r-r9s  z'ChatReka._agenerate..rrprqr)rWrrzrNrjrQrsrrurMr<rr/rrrr,r,r- _agenerate"s*  zChatReka._agenerate)inputr&cCs|jdur||Sz ddl}Wnty:tdYn0|d}t|tr^t||St|tr|j }t|tst dt |t||St|t rd}|D]8}|j }t|tst dt ||t||7}q|St dt |dS)aNCalculate number of tokens. Args: input: Either a string, a single BaseMessage, or a list of BaseMessages. Returns: int: Number of tokens in the input. Raises: ImportError: If tiktoken is not installed. ValueError: If message content is not a string. NrzXCould not import tiktoken python package. Please install it with `pip install tiktoken`.Z cl100k_basez&Message content must be a string, got zUnsupported input type: )r_tiktokenreZ get_encodingr*r2lenencoderr/r5r'r3rE)rhrrencodingr/totalmsgr,r,r-get_num_tokensMs:             zChatReka.get_num_tokensauto) tool_choicestrict)toolsrrrnr&c sVfdd|D}|dvr*td|d||d<dd|D}tjfd|i|S) aBind tool-like objects to this chat model. The `tool_choice` parameter controls how the model uses the tools you pass. There are three available options: - `"auto"`: Lets the model decide whether or not to invoke a tool. This is the recommended way to do function calling with our models. - `"none"`: Disables tool calling. In this case, even if you pass tools to the model, the model will not invoke any tools. - `"tool"`: Forces the model to invoke one or more of the tools it has been passed. Args: tools: A list of tool definitions to bind to this chat model. Supports any tool definition handled by :meth:`langchain_core.utils.function_calling.convert_to_openai_tool`. tool_choice: Controls how the model uses the tools you pass. Options are "auto", "none", or "tool". Defaults to "auto". strict: If True, model output is guaranteed to exactly match the JSON Schema provided in the tool definition. If False, input schema will not be validated and model output will not be validated. If None, ``strict`` argument will not be passed to the model. kwargs: Any additional parameters are passed directly to the model. Returns: Runnable: An executable chain or component. csg|]}t|dqS)rrrtoolrr,r-rsz'ChatReka.bind_tools..)rnonerzInvalid tool_choice 'zB' provided. Tool choice must be one of: 'auto', 'none', or 'tool'.rcSsg|] }|dqS)r>r,rr,r,r-rr)r5superbind)rhrrrrnZformatted_tools __class__rr- bind_tools}s&  zChatReka.bind_tools)NN)NN)NN)NN):__name__ __module__ __qualname____doc__rPr__annotations__rQr#DEFAULT_REKA_MODELrSr2rUintrVr floatrWboolrXrZr[r+r]rr"Z model_configr_rr rrr$ classmethodrgpropertyrrjrkrrrryr rrzrrrrr r r!rrrr __classcell__r,r,rr-rOs      $       .  +4 rO)5rItypingrrrrrrrr r r r Zlangchain_core.callbacksr rZlangchain_core.language_modelsrZ*langchain_core.language_models.chat_modelsrrrZlangchain_core.messagesrrrrrrZlangchain_core.outputsrrrZlangchain_core.runnablesrZlangchain_core.toolsrZlangchain_core.utilsrZ%langchain_core.utils.function_callingr Zpydanticr!r"r#r$rr2Z ContentTyper.r7rNrOr,r,r,r-s"4         E