Ng4ddlZddlmZddlmZmZmZmZmZm Z m Z ddl Z ddl m Z mZddlmZddlmZmZmZmZmZddlmZmZmZddlmZdd l mZdd lmZGd d eZ Gd deZ!dS)N) HTTPStatus)Any AsyncIteratorDictIteratorListOptionalUnion)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun) BaseChatModel) AIMessageAIMessageChunk BaseMessage HumanMessage SystemMessage)ChatGenerationChatGenerationChunk ChatResult)Field)Response) HTTPErrorc*eZdZdeddfdZdefdZdS)MaritalkHTTPError request_objreturnNc||_ |}d|vr |d}nd|vr |d}n|}n#t$r |j}YnwxYw||_|j|_dS)Ndetailmessage)rjson Exceptiontextr status_code)selfr response_json api_messages d/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/chat_models/maritalk.py__init__zMaritalkHTTPError.__init__s& +',,..M=((+H5 m+++I6 +  + + +%*KKK +# &2s0:A Aclt|jj}d|jd|}|d|jz }|S)Nz HTTP Error: z - z Detail: )rr#phraser)r$status_code_meaningformatted_messages r'__str__zMaritalkHTTPError.__str__)sL()9::AU4+;UU@SUU8$,888  )__name__ __module__ __qualname__rr(strr-r.r'rrsQ3H33333 !!!!!!!r.rceZdZUdZeed< eed< edddZeed< ed d Z e ed < ed Z e ed< edddZ eed< edefdZdeedeeeeeeeeeeefffffdZ d!deedeeedeededef dZ d!deedeeedeededef dZ d!deedeeedeededeef dZ d!deedeeedeededeef dZ d!deedeeedeedede f dZ! d!deedeeedeedede f dZ"edeeeffd Z#dS)" ChatMaritalka]`MariTalk` Chat models API. This class allows interacting with the MariTalk chatbot API. To use it, you must provide an API key either through the constructor. Example: .. code-block:: python from langchain_community.chat_models import ChatMaritalk chat = ChatMaritalk(api_key="your_api_key_here") api_keymodelgffffff?gg?)defaultgtlt temperatureir)r8r9 max_tokensT)r8 do_samplegffffff?top_prcdS)z&Identifies the LLM type as 'maritalk'.maritalkr3r$s r' _llm_typezChatMaritalk._llm_typeVs zr.messagescg}|D]f}t|trd}n/t|trd}nt|trd}|||jdg|S)a> 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 O OG'<00 GY// "G]33   " "DW_#M#M N N N Nr.Nstop run_managerkwargsc Hd}dd|ji}||ng}||}||j|j|j|j|j|d|} tj|| |} | j r(| ddSt| ) 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 NrCr7r=r<r;r>stopping_tokens)r headersanswerNo answer found) r6rMr7r=r<r;r>requestspostokr getr) r$rCrNrOrPurlrWrVrLdataresponses r'_callzChatMaritalk._callvs8<"$94<$9$9:"&"2$$77AA(Z/+Z.     =4AAA ; .==??&&x1BCC C#H-- -r.c$K ddl}d}dd|ji}||ng}||} | |j|j|j|j|j|d|} |4d{V} | || |dd{V} dddd{Vn#1d{VswxYwY| j dkr(| d d St| #t$rtd wxYw) 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._acalls$"  LLL?C&(=t|(=(=>G&*&6ddBO";;HEEO,!^"o#/#2   D((**       f!'dGT"-""                            #s**}}**85FGGG'111   >  s6A,C50 B"C5" B,,C5/B,05C5&C55Dc +Kdd|ji}||ng}||}||j|j|j|j|j|dd|}tjdtj ||d} | j r| D]} | dr| ddd } | r[tj| } d | vrC| d } t#t%|  }|r|| | |VdSt)| )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._streams#$94<$9$9:"&"2$$77AA(Z/+Z.     = 9D!!     ; . ++-- ( (??9-- ($(LLC$@$@$G$G$P$PM$ (&*j&?&? ![00$/$7E$7(6u(E(E(E%%%E +Q + < rgrnr rsr# aiter_linesrurvrxrrryrrh)r$rCrNrOrPrfrWrVrLr_rir`rzr{r|r}rrs r'_astreamzChatMaritalk._astreams4  LLL&(=t|(=(=>G&*&6ddBO";;HEEO,!^"o#/#2   D((** : : : : : : :f!==AD))# ):::::::: +s22*2*>*>*@*@4444444$#x88 404 Xr0J0J #0 !426*]2K2KK'-'<'<0;F0C0C4B54Q4Q4Q1*1*1*,7).2=2N2N05U3O3.3.-.-.-.-.-.-.-./4 +A*@"099993::::::::::::::::::::::::::: : : : : : : : : : : : : : : : : : : : : : : : : : : : : : :8   >  saA+F5/3F"# E>E B"E>,F"> F F" F F"F5" F,,F5/F,0F55Gc |j|f||d|}t|}t|}t|gSN)rNrOrorp) generations)rarrrr$rCrNrOrP output_strr generations r' _generatezChatMaritalk._generateNsX TZWtWWPVWW J///#G444 zl3333r.cK|j|f||d|d{V}t|}t|}t|gSr)rjrrrrs r' _ageneratezChatMaritalk._agenerateZs'4;  [  r<rrAs r'_identifying_paramsz ChatMaritalk._identifying_paramshs(Z+Z/    r.)NN)$r/r0r1__doc__r2__annotations__rr;floatr<intr=boolr>propertyrBrrrr rrMr r rar rjrrr~rrrrrrr3r.r'r5r50s  LLL JJJ,sss;;;K;;;2eCA...J...@eD)))It)))?5##666E566653X[) d3c4c4S>.A(B#CCDDE F<%):> 2.2.{#2.tCy!2.67 2.  2. 2.2.2.2.n%)?C 44{#4tCy!4;< 4  4 4444r%):> /./.{#/.tCy!/.67 /.  /. % & /./././.h%)?C ;;{#;tCy!;;< ;  ; * + ;;;;@%):> 4 4{# 4tCy! 467 4  4  4 4 4 4%)?C 4 4{# 4tCy! 4;< 4  4  4 4 4 4  T#s(^    X    r.r5)"r httprtypingrrrrrr r rZlangchain_core.callbacksr r *langchain_core.language_models.chat_modelsr langchain_core.messagesrrrrrlangchain_core.outputsrrrpydanticrrrequests.exceptionsrrr5r3r.r'rs LLLLLLLLLLLLLLLLLLEDDDDDSRRRRRRRRR))))))!!!!! !!!0F F F F F =F F F F F r.