Ng?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/efffZ0dee/efdee/effdZ1de0deee/effdZ2deedeee/effdZ3GddeZ4dS)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-flashitemreturncv|ddkr,|d}t|trd|vr |d|d<|S)zProcess a single content item.type image_urlurl) isinstancedict)r&r*s `/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/chat_models/reka.pyprocess_content_itemr/.sK F|{""% i & & 15I+=+= )% 0D  Kcontentct|trd|dgSt|trg}|D]y}t|tr|d|d0t|tr#|t |ht d||St d)z\Process content to handle both text and media inputs, returning a list of content items.textr)r3zInvalid content item format: zInvalid content format)r,strlistappendr-r/ ValueError)r1resultr&s r.process_contentr:8s'3 31122 GT " " 3 I ID$$$ I vt<<====D$'' I 24889999 !G!G!GHHH 1222r0messagescg}d}|D]^}t|trB|1t|jtr|j}> *!,0088FBB"3A"666*KK->q-A&-IKK&a(00&,,FNCC"&  &=N!O!O P P P P  + + K,2K+@L <$3GO$D$D!*; Y'w888$6|D ')$!+EEI'o )* 5f =&*j:1F{1S&T&T++' )//0CDDDD-A \*   . . . .  - - K13L   $+$8"j(GO)DEE     )+    I$w--IIJJ J r0c 2eZdZUdZdZeed<dZeed<ee Z e ed<edZ e ed<dZeeed <d Zeed <dZeeed <d Ze ed<dZee ed<eeZee efed<edZdZeeee eeefge fed<e de!dee efdee effdZ"e#de$e effdZ%e#de fdZ& d,deedeee dee'dede(e)f d Z* d,deedeee dee+dede,e)f d!Z- d,deedeee dee'dede.f d"Z/ d,deedeee dee+dede.f d#Z0d$ee eeefde fd%Z1d&dd'd(e2eee efe3e4ee5fd)e d*eedede6e7eff fd+Z8xZ9S)-ChatRekaz Reka chat large language models.Nclient async_client)defaultmodel max_tokens temperatureF streamingdefault_request_timeout max_retries reka_api_key)default_factory model_kwargsforbid)extra token_counterbefore)modevaluesr'c|d}td|idd}||d< ddlm}m}|||d<|||d<n#t $rt dwxYw|S) zBValidate that API key and Python package exist in the environment.ri REKA_API_KEYr) AsyncRekaReka)api_keyr^r_zTCould not import Reka Python package. Please install it with `pip install reka-api`.)rLr reka.clientrtru ImportError)clsrqrirtrus r.validate_environmentzChatReka.validate_environmentszz.11 + \ *NN  ".~  3 3 3 3 3 3 3 3#t$   F8 &/Y$&&&F> " "   A    s &AA0cX|j|jd}|j |j|d<i||jS)z0Get the default parameters for calling Reka API.)rarcNrd)rarcrdrk)selfparamss r._default_paramszChatReka._default_paramssEZ/     '$($4F= !.&.D-..r0cdS)zReturn type of chat model.z reka-chat)r|s r. _llm_typezChatReka._llm_types {r0r;stop run_managerkwargsc+,Kt|}i|j|}|r||d<|jjjdd|i|}|D]T}|jdjj} tt| } |r| | | | VUdSNrr;rr1rT)chunkr) r[r~r^chat create_stream responsesrr1rron_llm_new_token r|r;rrrrRr}streamrr1 chat_chunks r._streamzChatReka._streams1:: 3D(3F3  "!F6N/!/QQQ&QQ  Eoa(.6G,^G5T5T5TUUUJ H,,WJ,GGG       r0c FKt|}i|j|}|r||d<|jjjdd|i|}|23d{V}|jdjj} tt| } |r| | | d{V| WVa6dSr) r[r~r_rrrrr1rrrrs r._astreamzChatReka._astreams1:: 3D(3F3  "!F6N5"'5WW}WPVWW!       %oa(.6G,^G5T5T5TUUUJ N!227*2MMMMMMMMM      "66sB c |jrt|j|f||d|St|}i|j|}|r||d<|jjjd d|i|}|jdj j r5|jdj j }tddd|Di} n+|jdj j } t| | nd } tt| g S) Nrrrr;rrAcbg|],}|jd|jtj|jdd-SrC)rDrE)rBr)rCrBrDrOrQrF.0tcs r. z&ChatReka._generate..W # # ##%%$.(*-1Z -F-F)) # # #r0r1rNrr generationsr)rerrr[r~r^rcreaterrTrArr1rr r|r;rrrrRr}responserArTr1s r. _generatezChatReka._generatesH > ' XTDkTTVTT 1:: 3D(3F3  "!F6N*4;#*LLMLVLL  a ( 3 P!+A.6AJ # ##- # # # #GG"(+3;G73FBOOOG~g'F'F'F&GHHHHr0cK|jr$t|j|f||d|d{VSt|}i|j|}|r||d<|jjjd d|i|d{V}|jdj j r5|jdj j }tddd|Di} n+|jdj j } t| | nd } tt| g S) Nrrr;rrrAcbg|],}|jd|jtj|jdd-Srrrs r.rz'ChatReka._agenerate..9rr0rrrrr)rerrr[r~r_rrrrTrArr1rrrs r. _ageneratezChatReka._agenerate"s~ > . hUT{UUfUU 1:: 3D(3F3  "!F6N6*/6XX XQWXXXXXXXX  a ( 3 P!+A.6AJ # ##- # # # #GG"(+3;G73FBOOOG~g'F'F'F&GHHHHr0inputc6|j||S ddl}n#t$rtdwxYw|d}t |t r"t ||St |tr]|j }t |t stdt|t ||St |trid}|D]b}|j }t |t stdt||t ||z }c|Stdt|)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`. cl100k_basez&Message content must be a string, got zUnsupported input type: )rntiktokenrx get_encodingr,r5lenencoderr1r8r)r6rK)r|rrencodingr1totalmsgs r.get_num_tokenszChatReka.get_num_tokensMs   )%%e,, ,  OOOO   A   ((77 eS ! ! Fxu--.. . { + + FmGgs++  LT']]LLxw//00 0 t $ $ FE 7 7+!'3//$PgPPX__W55666LDtE{{DDEE Es#=auto) tool_choicestricttoolsrrc fd|D}|dvrtd|d||d<d|D}tjdd|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. c2g|]}t|S))rr )rtoolrs r.rz'ChatReka.bind_tools..s4   <@ "4 7 7 7   r0)rnonerzInvalid tool_choice 'zB' provided. Tool choice must be one of: 'auto', 'none', or 'tool'.rcg|] }|d S)rCr)rrs r.rz'ChatReka.bind_tools..sHHH4 +HHHr0rr)r8superbind)r|rrrrformatted_tools __class__s ` r. bind_toolszChatReka.bind_tools}sL    DI    6 6 6I III  !,}IHHHHuww|<#>#>L$sCx.>>>:H---L 8%[${*;;<=sBC   _(###$sCx.T#s(^[$#2/c!2///X/3X%):> {#tCy!67    % & 0%)?C {#tCy!;<    * + 0%):> )I)I{#)ItCy!)I67 )I  )I  )I)I)I)I\%)?C )I)I{#)ItCy!)I;< )I  )I  )I)I)I)IV.FE#{Drs                           >===== SRRRRRRRRR------))))))555555HHHHHHBBBBBBBBBBBB!CeCc3h$7899: tCH~$sCx.3[3T$sCx.-A3333&BtK'8BT$sCx.=QBBBBJc=c=c=c=c=}c=c=c=c=c=r0