ijfddlmZddlZddlZddlmZmZddlmZGddZ GddeZ y) ) annotationsN)parserequest)ILanguageModelc8eZdZdZ dZdd dZd d dZdZd dZy) LibreTranslateAPIz!Connect to the LibreTranslate APIz$https://translate.argosopentech.com/Nc| |jn||_||_t|jdkDsJ|jddk7r|xjdz c_yy)zCreate a LibreTranslate API connection. Args: url: The url of the LibreTranslate endpoint. api_key: The API key. Nr/) DEFAULT_URLurlapi_keylen)selfr rs ]/home/spazio/kingdom_calls/work5/transenv/lib/python3.12/site-packages/argostranslate/apis.py__init__zLibreTranslateAPI.__init__sX(+{4## 488}q   88B<3  HHOH cl|jdz}|||d}|j|j|d<tj|}t j ||j }t j|}|jj} tj| dS)zTranslate string Args: q: The text to translate source: The source language code (ISO 639) target: The target language code (ISO 639) Returns: The translated text translate)qsourcetargetrdatatranslatedText r rr urlencoderRequestencodeurlopenreaddecodejsonloads) rrrrr params url_paramsreqresponse response_strs rrzLibreTranslateAPI.translate'shh$Ff= << # $ F9 __V, ooc (9(9(;<??3'}}--/ zz,'(899rcn|jdz}t}|j|j|d<tj|}t j ||j}t j|}|jj}tj|S)zRetrieve list of supported languages. Returns: A list of available languages ex. [{"code":"en", "name":"English"}] languagesrr) r dictrrrrrrr r!r"r#r$)rr r%r&r'r(r)s rr+zLibreTranslateAPI.languagesCs hh$ << # $ F9 __V, ooc (9(9(;<??3'}}--/ zz,''rcb|jdz}d|i}|j|j|d<tj|}t j ||j }t j|}|jj}tj|S)zDetect the language of a single text. Args: q: Text to detect Returns: The detected languages ex. [{"confidence": 0.6, "language": "en"}] detectrrrr)rrr r%r&r'r(r)s rr.zLibreTranslateAPI.detectZshh!q << # $ F9 __V, ooc (9(9(;<??3'}}--/ zz,''r)NN)r strrr/)enes)rr/rr/rr/returnr/)rr/) __name__ __module__ __qualname____doc__r rrr+r.rrrr s%+9K :8(.(rrceZdZddZddZy) OpenAIAPIc||_y)zeCreate an API connection. Args: api_key: The API key for the OpenAI API N)r)rrs rrzOpenAIAPI.__init__}s  rcd}|dd}dd|jzd}tj|j}t j |||d} t j |} |jj} tj| d d d S#t$r$}t|tjYd }~y d }~wwxYw#t$r$}t|tjYd }~y d }~wwxYw) zConnect to OpenAI API Args: prompt: The prompt to run inference on. Returns: The generated text z5https://api.openai.com/v1/engines/davinci/completionsd)prompt max_tokenszapplication/jsonzBearer )z Content-Type AuthorizationPOST)rheadersmethodNchoicesrtext)rr#dumpsrrrr Exceptionprintsysstderrr!r"r$) rr=r r%rAencoded_paramsr'r(er)s rinferzOpenAIAPI.infersF"#6/&5  F+224oocPVW s+H  #==?113L zz,' 215f==  !SZZ     !SZZ  s0B)-C) C2CC D"DDN)rr/)r=r/r2z str | None)r3r4r5rrLr7rrr9r9|s !>rr9) __future__rr#rHurllibrrargostranslate.modelsrrr9r7rrrPs-" !0h(h(d*>*>r