a bg@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@seZdZUdZdZeed<dZeed<dZ e ed<dZ eed <d Z e ed <d Z e ed <eed<dZeeed<eddZeddeeedddZeeeefdddZeeeefdddZeedddZd"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_urlZforbid)extrabefore)mode)valuesreturncCstt|dd|d<|S)z,Validate that api key exists in environment.rZFOREFRONTAI_API_KEYr )clsr r#r/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/llms/forefrontai.pyvalidate_environment5s 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_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_type)promptstop run_managerkwargsr!c Ksdd|j}tj|j|ddd|i|j|d}|}|ddd}|d ur`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)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 s<         r)typingrrrrrr7Zlangchain_core.callbacksrZ#langchain_core.language_models.llmsrZlangchain_core.utilsr r Zpydanticr r rZlangchain_community.llms.utilsrrr#r#r#r$s