tjW@ddlZddlmZmZmZmZmZmZmZm Z m Z m Z m Z m Z ddlmZmZddlmZddlmZmZmZddlmZmZmZmZmZmZddlmZm Z m!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 e0ee e0ee0efffZ1dee0efdee0effdZ2de1deee0effdZ3deedeee0effdZ4GddeZ5dS)N) Any AsyncIteratorCallableDictIteratorListLiteralMappingOptionalSequenceTypeUnion)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 b/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langchain_community/chat_models/reka.pyprocess_content_itemr0/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*r4zInvalid content item format: zInvalid content format)r-strlistappendr.r0 ValueError)r2resultr's r/process_contentr;9s'3 31122 GT " " 3 I ID$$$ I vt<<====D$'' I 24889999 !G!G!GHHH 1222r1messagescg}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 r1c ZeZdZUdZdZeed<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)eee e6d*fd+eedede7e8e9ff fd,Z:xZ;S).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.rj REKA_API_KEYr) AsyncRekaReka)api_keyr_r`zTCould not import Reka Python package. Please install it with `pip install reka-api`.)rMr reka.clientrurv ImportError)clsrrrjrurvs 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.)rbrdNre)rbrdrerl)selfparamss r/_default_paramszChatReka._default_paramssEZ/     '$($4F= !.&.D-..r1cdS)zReturn type of chat model.z reka-chat)r}s r/ _llm_typezChatReka._llm_types {r1r<stop run_managerkwargsc+,Kt|}i|j|}|r||d<|jjjdd|i|}|D]T}|jdjj} tt| } |r| | | | VUdSNrr<rr2rU)chunkr) r\rr_chat create_stream responsesrr2rron_llm_new_token r}r<rrrrSr~streamrr2 chat_chunks r/_streamzChatReka._streams1:: 3D(3F3  "!F6N/!/QQQ&QQ  Eoa(.6G,^G5T5T5TUUUJ H,,WJ,GGG       r1c FKt|}i|j|}|r||d<|jjjdd|i|}|23d{V}|jdjj} tt| } |r| | | d{V| WVa6dSr) r\rr`rrrrr2rrrrs 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<rrBcbg|],}|jd|jtj|jdd-SrD)rErF)rCr*rDrCrErPrRrG.0tcs r/ z&ChatReka._generate..W # # ##%%$.(*-1Z -F-F)) # # #r1r2rOrr generationsr)rfrrr\rr_rcreaterrUrBrr2rr r}r<rrrrSr~responserBrUr2s r/ _generatezChatReka._generatesH > ' XTDkTTVTT 1:: 3D(3F3  "!F6N*4;#*LLMLVLL  a ( 3 P!+A.6AJ # ##- # # # #GG"(+3;G73FBOOOG~g'F'F'F&GHHHHr1cK|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<rrrBcbg|],}|jd|jtj|jdd-Srrrs r/rz'ChatReka._agenerate..:rr1rrrrr)rfrrr\rr`rrrrUrBrr2rrrs r/ _ageneratezChatReka._agenerate#s~ > . hUT{UUfUU 1:: 3D(3F3  "!F6N6*/6XX XQWXXXXXXXX  a ( 3 P!+A.6AJ # ##- # # # #GG"(+3;G73FBOOOG~g'F'F'F&GHHHHr1inputc6|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: )rotiktokenry get_encodingr-r6lenencoderr2r9r*r7rL)r}rrencodingr2totalmsgs r/get_num_tokenszChatReka.get_num_tokensNs   )%%e,, ,  OOOO   A   ((77 eS ! ! Fxu--.. . { + + FmGgs++  LT']]LLxw//00 0 t $ $ FE 7 7+!'3//$PgPPX__W55666LDtE{{DDEE Es#=auto) tool_choicestricttoolsranyrc fd|D}|d}|dkrd}|dvrtd|d||d <d |D}tjd d |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   r1Nrrr)rnonerzInvalid tool_choice 'zB' provided. Tool choice must be one of: 'auto', 'none', or 'tool'.rcg|] }|d S)rDr)rrs r/rz'ChatReka.bind_tools..sHHH4 +HHHr1rr)r9superbind)r}rrrrformatted_tools __class__s ` r/ bind_toolszChatReka.bind_tools~sL    DI      K %   K 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.=QBBBBJg=g=g=g=g=}g=g=g=g=g=r1