j4ddlZddlmZddlmZmZmZmZmZm Z m Z ddl Z ddl m Z mZddlmZddlmZmZmZmZmZddlmZmZmZddlmZdd l mZdd lmZGd d eZ Gd deZ!y)N) HTTPStatus)Any AsyncIteratorDictIteratorListOptionalUnion)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun) BaseChatModel) AIMessageAIMessageChunk BaseMessage HumanMessage SystemMessage)ChatGenerationChatGenerationChunk ChatResult)Field)Response) HTTPErrorc(eZdZdeddfdZdefdZy)MaritalkHTTPError request_objreturnNc||_ |j}d|vr|d}n d|vr|d}n|}||_|j |_y#t$r|j}Y0wxYw)Ndetailmessage)rjson Exceptiontextr status_code)selfr response_json api_messages r/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langchain_community/chat_models/maritalk.py__init__zMaritalkHTTPError.__init__sy& +',,.M=(+H5 m++I6 + # &22  +%**K +s&AA A ct|jj}d|jd|}|d|jz }|S)Nz HTTP Error: z - z Detail: )rr#phraser)r$status_code_meaningformatted_messages r'__str__zMaritalkHTTPError.__str__)sQ()9)9:AA*4+;+;* Parses messages from LangChain's format to the format expected by the MariTalk API. Parameters: messages (List[BaseMessage]): A list of messages in LangChain format to be parsed. Returns: A list of messages formatted for the MariTalk API. user assistantsystem)rolecontent) isinstancerrrappendrI)r$rCparsed_messagesrrHs r'parse_messages_for_modelz%ChatMaritalk.parse_messages_for_model[s_G'<0GY/"G]3  " "DW__#M N r.Nstop run_managerkwargsc pd}dd|ji}||ng}|j|}||j|j|j|j |j |d|} tj|| |} | jr | jjddSt| )a Sends the parsed messages to the MariTalk API and returns the generated response or an error message. This method makes an HTTP POST request to the MariTalk API with the provided messages and other parameters. If the request is successful and the API returns a response, this method returns a string containing the answer. If the request is rate-limited or encounters another error, it returns a string with the error message. Parameters: messages (List[BaseMessage]): Messages to send to the model. stop (Optional[List[str]]): Tokens that will signal the model to stop generating further tokens. Returns: str: If the API call is successful, returns the answer. If an error occurs (e.g., rate limiting), returns a string describing the error. +https://chat.maritaca.ai/api/chat/inference authorizationKey rCr7r=r<r;r>stopping_tokens)r headersanswerNo answer found) r6rMr7r=r<r;r>requestspostokr getr) r$rCrNrOrPurlrWrVrLdataresponses r'_callzChatMaritalk._callvs8<"d4<<.$9:"&"2$77A(ZZ//++ZZ.    ==4A ;;==?&&x1BC C#H- -r.cJK ddl}d}dd|ji}||ng}|j|} | |j|j|j |j |j|d|} |j4d{} | j|| |dd{} dddd{ jdk(r | jjd d St| 7k7P7B#1d{7swYRxYw#t$r td wxYww) a Asynchronously sends the parsed messages to the MariTalk API and returns the generated response or an error message. This method makes an HTTP POST request to the MariTalk API with the provided messages and other parameters using async I/O. If the request is successful and the API returns a response, this method returns a string containing the answer. If the request is rate-limited or encounters another error, it returns a string with the error message. rNrRrSrTrU)r rWtimeoutrXrYRCould not import httpx python package. Please install it with `pip install httpx`.)httpxr6rMr7r=r<r;r> AsyncClientr[r#r r]r ImportError) r$rCrNrOrPrfr^rWrVrLr_clientr`s r'_acallzChatMaritalk._acalls0$"  ?C&$t||n(=>G&*&6dBO";;HEO,!^^"oo#//#2  D((**f!'dGT"-"+* ##s*}}**85FGG'11++*** >  swD#BD C0D C6!C2"C6& D 1C422D $D#% D 2C64D 6D<C? =DD D  D#c +Kdd|ji}||ng}|j|}||j|j|j|j |j |dd|}tjdtj||d} | jr| jD]} | jds| jddjd } | s9tj | } d | vsS| d } t#t%|  }|r|j'| | |yt)| w)N AuthorizationrTTrCr7r=r<r;r>rVstreamrR)r_rWrnsdata: r.zutf-8r"rIrchunk)r6rMr7r=r<r;r>rZr[r dumpsr\ iter_lines startswithreplacedecodeloadsrron_llm_new_tokenr)r$rCrNrOrPrWrVrLr_r`line response_data parsed_datadeltarrs r'_streamzChatMaritalk._streamsA#d4<<.$9:"&"2$77A(ZZ//++ZZ.    == 9D!   ;; ++-??9-$(LLC$@$G$G$PM$&*jj&? ![0$/$7E$7(6u(E%E + + < rgrnr rsr# aiter_linesrurvrxrrryrrh)r$rCrNrOrPrfrWrVrLr_rir`rzr{r|r}rrs r'_astreamzChatMaritalk._astreams4  &$t||n(=>G&*&6dBO";;HEO,!^^"oo#//#2  D((**f!==AD)# ) ++s2*2*>*>*@4$#x804 Xr0J #026**]2KK'-'<0;F0C0C4B54Q1*,72=2N2N05U3O3.-.-./4 0995+4-.+A+****8 >  sGA?GE,G.F/5E.6F/9 FE4E0 E4!F4F F#3FE2 F,G.F/0E42F4F6 F/FF/F FF F/ G(F+)G.G/G5F8 6G=GGGGGc x|j|f||d|}t|}t|}t|gSN)rNrOrorp) generations)rarrrr$rCrNrOrP output_strr generations r' _generatezChatMaritalk._generateNsB TZZWtWPVW J/#G4 zl33r.cK|j|f||d|d{}t|}t|}t|gS7)wr)rjrrrrs r' _ageneratezChatMaritalk._agenerateZsZ'4;;  [ r<rrAs r'_identifying_paramsz ChatMaritalk._identifying_paramshs,ZZ++ZZ//   r.)NN)$r/r0r1__doc__r2__annotations__rr;floatr<intr=boolr>propertyrBrrrr rrMr r rar rjrrr~rrrrrrr3r.r'r5r50s L J,sss;K;2CA.J.@D)It)?##6E5653[) d3c4c4S>.A(B#CCDDE F<%):> 2.{#2.tCy!2.67 2.  2. 2.n%)?C 4{#4tCy!4;< 4  4 4r%):> /.{#/.tCy!/.67 /.  /. % & /.h%)?C ;{#;tCy!;;< ;  ; * + ;@%):> 4{# 4tCy! 467 4  4  4%)?C 4{# 4tCy! 4;< 4  4  4  T#s(^    r.r5)"r httprtypingrrrrrr r rZlangchain_core.callbacksr r *langchain_core.language_models.chat_modelsr langchain_core.messagesrrrrrlangchain_core.outputsrrrpydanticrrrequests.exceptionsrrr5r3r.r'rsX LLLESR)! !0F =F r.