o Zh@sddlmZmZmZmZmZddlZddlmZddl m Z ddl m Z m Z ddlmZmZmZddlmZGdd d e ZdS) )AnyDictListMappingOptionalN)CallbackManagerForLLMRun)LLMconvert_to_secret_strget_from_dict_or_env) ConfigDict SecretStrmodel_validator)enforce_stop_tokensc @s eZdZUdZdZeed< dZeed< dZ e ed< dZ eed < d Z e ed < d Z e ed < eed<dZeeed< eddZeddededefddZedeeeffddZedeeeffddZedefddZ  d%ded eeed!eed"edef d#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_urlZforbid)extrabefore)modevaluesreturncCstt|dd|d<|S)z,Validate that api key exists in environment.rZFOREFRONTAI_API_KEYr )clsr r#[/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/llms/forefrontai.pyvalidate_environment5s z ForefrontAI.validate_environmentcCs|j|j|j|j|jdS)z7Get the default parameters for calling ForefrontAI API.rrrrrr&selfr#r#r$_default_params>s zForefrontAI._default_paramscCsid|ji|jS)zGet the identifying parameters.r)rr)r'r#r#r$_identifying_paramsIszForefrontAI._identifying_paramscCsdS)zReturn type of llm.Z forefrontair#r'r#r#r$ _llm_typeNszForefrontAI._llm_typepromptstop run_managerkwargsc Ksdd|j}tj|j|ddd|i|j|d}|}|ddd}|d ur0t||}|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)r4r) r(r,r-r.r/Z auth_valueresponseZ response_jsonr1r#r#r$_callSs zForefrontAI._call)NN) __name__ __module__ __qualname____doc__rstr__annotations__rfloatrintrrrr rrr Z model_configr classmethodrrr%propertyrr)r*r+rrr:r#r#r#r$r sT         r)typingrrrrrr7Zlangchain_core.callbacksrZ#langchain_core.language_models.llmsrZlangchain_core.utilsr r Zpydanticr r rZlangchain_community.llms.utilsrrr#r#r#r$s