U hp@s|ddlmZmZmZmZmZddlZddlmZddl m Z ddl m Z m Z ddlmZmZddlmZGdd d e ZdS) )AnyDictListMappingOptionalN)CallbackManagerForLLMRun)LLM) SecretStrroot_validatorconvert_to_secret_strget_from_dict_or_env)enforce_stop_tokensc@seZdZUdZdZeed<dZeed<dZ e ed<dZ eed <d Z e ed <d Z e ed <eed<dZeeed<GdddZeddeedddZeeeefdddZeeeefdddZeedddZd"eeeeeeeedd d!ZdS)# ForefrontAIa2ForefrontAI large language models. To use, you should have the environment variable ``FOREFRONTAI_API_KEY`` set with your API key. Example: .. code-block:: python from langchain_community.llms import ForefrontAI forefrontai = ForefrontAI(endpoint_url="") endpoint_urlgffffff? temperaturelengthg?top_p(top_krepetition_penaltyforefrontai_api_keyNbase_urlc@seZdZdZdS)zForefrontAI.ConfigZforbidN)__name__ __module__ __qualname__extrar r H/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/llms/forefrontai.pyConfig1sr"T)pre)valuesreturncCstt|dd|d<|S)z,Validate that api key exists in environment.rZFOREFRONTAI_API_KEYr )clsr$r r r!validate_environment4s z ForefrontAI.validate_environment)r%cCs|j|j|j|j|jdS)z7Get the default parameters for calling ForefrontAI API.rrrrrr(selfr r r!_default_params<s zForefrontAI._default_paramscCsd|ji|jS)zGet the identifying parameters.r)rr+r)r r r!_identifying_paramsGszForefrontAI._identifying_paramscCsdS)zReturn type of llm.Z forefrontair r)r r r! _llm_typeLszForefrontAI._llm_type)promptstop run_managerkwargsr%c Ksbd|j}tj|j|ddd|i|j|d}|}|ddd}|d k r^t||}|S) arCall out to ForefrontAI's complete endpoint. Args: prompt: The prompt to pass into the model. stop: Optional list of stop words to use when generating. Returns: The string generated by the model. Example: .. code-block:: python response = ForefrontAI("Tell me a joke.") zBearer zapplication/json) Authorizationz Content-Typetext)urlheadersjsonresultr completionN)rZget_secret_valuerequestspostrr+r6r) r*r.r/r0r1Z auth_valueresponseZ response_jsonr3r r r!_callQs zForefrontAI._call)NN)rrr__doc__rstr__annotations__rfloatrintrrrr rrr"r rr'propertyrrr+r,r-rrr<r r r r!r s6         r)typingrrrrrr9Zlangchain_core.callbacksrZ#langchain_core.language_models.llmsrZlangchain_core.pydantic_v1r r Zlangchain_core.utilsr r Zlangchain_community.llms.utilsrrr r r r!s