j<ddlmZddlmZmZmZmZmZddlm Z m Z ddl m Z ddl mZGdded ZGd d e Zy ) ) annotations)AnyDictListMappingOptional)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun)LLM) BaseModelc<eZdZUdZdZded<dZded<dZded <y) Paramsz*Parameters for the Javelin AI Gateway LLM.gfloat temperatureNOptional[List[str]]stopz Optional[int] max_tokens)__name__ __module__ __qualname____doc__r__annotations__rru/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langchain_community/llms/javelin_ai_gateway.pyrrs$4K $D $ $J $rrallow)extraceZdZUdZded< dZded< dZded< dZd ed < dZded < dfd Z e dd Z e ddZ d ddZ d ddZe ddZxZS)JavelinAIGatewaya Javelin AI Gateway LLMs. To use, you should have the ``javelin_sdk`` python package installed. For more information, see https://docs.getjavelin.io Example: .. code-block:: python from langchain_community.llms import JavelinAIGateway completions = JavelinAIGateway( gateway_uri="", route="", params={ "temperature": 0.1 } ) strrouteNz Optional[Any]clientz Optional[str] gateway_urizOptional[Params]paramsjavelin_api_keyc  ddlm}m}t |di||j r% ||j |j|_yy#t$r tdwxYw#|$r}td|d}~wwxYw)Nr) JavelinClientUnauthorizedErrorz^Could not import javelin_sdk python package. Please install it with `pip install javelin_sdk`.)base_urlapi_keyzJavelin: Incorrect API Key.r) javelin_sdkr'r( ImportErrorsuper__init__r#r%r" ValueError)selfkwargsr'r(e __class__s rr.zJavelinAIGateway.__init__:s   "6"    G+!--t7K7K    D  % G !>?QF Gs"A #A$ A!$A:) A55A:c|j|j|jd|jr|jj }|Si}|S)z>Get the default parameters for calling Javelin AI Gateway API.)r#r!r%)r#r!r%r$dict)r0r$s r_default_paramsz JavelinAIGateway._default_paramsNs[ ++ZZ#33" &*[[t{{! "   79 "   rc|jS)zGet the identifying parameters.)r6r0s r_identifying_paramsz$JavelinAIGateway._identifying_paramsYs###rc d|i|jr|jjni}|xs$|jr|jjndx}r||d<|j(|jj |j |}n t d|j} |dddd S#t$rYy wxYw) z Call the Javelin AI Gateway API.promptNr query_body"Javelin client is not initialized. llm_responsechoicesrtext)r$r5rr" query_router!r/KeyError r0r;r run_managerr1datasresp resp_dicts r_callzJavelinAIGateway._call^s f %)[[t{{!b Ddkk$++**t E1 EDL ;; ";;**4::$*GDAB BIIK  ^,Y7:6B B  s, B:: CCcKd|i|jr|jjni}|xs$|jr|jjndx}r||d<|j0|jj |j |d{}n t d|j} |dddd S7/#t$rYy wxYww) z&Call async the Javelin AI Gateway API.r;Nrr<r>r?r@rrArB)r$r5rr" aquery_router!r/rDrEs r_acallzJavelinAIGateway._acallys f %)[[t{{!b Ddkk$++**t E1 EDL ;; "11$**1NNDAB BIIK  ^,Y7:6B BO  s6BCC C6 CC CCCCcy)zReturn type of llm.zjavelin-ai-gatewayrr8s r _llm_typezJavelinAIGateway._llm_types$r)r1r)returnzDict[str, Any])rQzMapping[str, Any])NN) r;r rrrFz"Optional[CallbackManagerForLLMRun]r1rrQr ) r;r rrrFz'Optional[AsyncCallbackManagerForLLMRun]r1rrQr )rQr )rrrrrr"r#r$r%r.propertyr6r9rKrNrP __classcell__)r3s@rrrs& J: FM (!%K%0#F #4%)O])5G($$%):> "8    <%)?C "=    6$$rrN) __future__rtypingrrrrrlangchain_core.callbacksr r #langchain_core.language_models.llmsr pydanticr rrrrrrYs7"554 %Yg%@$s@$r