tj<ddlmZddlmZmZmZmZmZddlm Z m Z ddl m Z ddl mZGdded ZGd d e Zd S) ) annotations)AnyDictListMappingOptional)AsyncCallbackManagerForLLMRunCallbackManagerForLLMRun)LLM) BaseModelc>eZdZUdZdZded<dZded<dZded <dS) Paramsz*Parameters for the Javelin AI Gateway LLM.gfloat temperatureNOptional[List[str]]stopz Optional[int] max_tokens)__name__ __module__ __qualname____doc__r__annotations__rri/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langchain_community/llms/javelin_ai_gateway.pyrrsK44K $D$$$$ $J$$$$$$rrallow)extraceZdZUdZded< dZded< dZded< dZd ed < dZded < dfd Z e ddZ e ddZ d d!dZ d d"dZe d#dZxZS)$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_keykwargsrc  ddlm}m}n#t$rtdwxYwt jdi||jr< ||j|j|_dS#|$r}td|d}~wwxYwdS)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)selfr&r(r)e __class__s rr/zJavelinAIGateway.__init__:s             D   ""6"""   G G+m!-t7K % G G G !>??QF G  G Gs  & A((B-A==BreturnDict[str, Any]ct|j|j|jd|jr|jni}|S)z>Get the default parameters for calling Javelin AI Gateway API.)r#r!r%)r#r!r%r$dict)r1r$s r_default_paramsz JavelinAIGateway._default_paramsNsL +Z#3" " &*[8t{!!!b "   rMapping[str, Any]c|jS)zGet the identifying parameters.)r8r1s r_identifying_paramsz$JavelinAIGateway._identifying_paramsYs ##rpromptrr run_manager"Optional[CallbackManagerForLLMRun]c zd|i|jr|jni}|p|jr |jjndx}r||d<|j"|j|j|}nt d|} |dddd S#t$rYd SwxYw) z Call the Javelin AI Gateway API.r=Nr query_body"Javelin client is not initialized. llm_responsechoicesrtext)r$r7rr" query_router!r0KeyError r1r=rr>r&datasresp resp_dicts r_callzJavelinAIGateway._call^s f %)[8t{!!!b DdkC$+**t E1 DL ; ";**4:$*GGDDABB BIIKK  ^,Y7:6B B   22 sB,, B:9B:'Optional[AsyncCallbackManagerForLLMRun]cKd|i|jr|jni}|p|jr |jjndx}r||d<|j(|j|j|d{V}nt d|} |dddd S#t$rYd SwxYw) z&Call async the Javelin AI Gateway API.r=NrrArCrDrErrFrG)r$r7rr" aquery_router!r0rIrJs r_acallzJavelinAIGateway._acallys f %)[8t{!!!b DdkC$+**t E1 DL ; "11$*1NNNNNNNNDDABB BIIKK  ^,Y7:6B B   22 sB44 CCcdS)zReturn type of llm.zjavelin-ai-gatewayrr;s r _llm_typezJavelinAIGateway._llm_types $#r)r&r)r4r5)r4r9)NN) r=r rrr>r?r&rr4r ) r=r rrr>rPr&rr4r )r4r )rrrrrr"r#r$r%r/propertyr8r<rOrSrU __classcell__)r3s@rrrsO&JJJ: F    (!%K%%%%0#F####4%)O))))5GGGGGG(X$$$X$%):> <%)?C 6$$$X$$$$$rrN) __future__rtypingrrrrrlangchain_core.callbacksr r #langchain_core.language_models.llmsr pydanticr rrrrrr]s""""""55555555555555433333 %%%%%Yg%%%%@$@$@$@$@$s@$@$@$@$@$r