o Zh@sddlZddlmZmZmZmZmZddlmZddl m Z ddl m Z ddl mZmZddlmZddlmZd d d d d d Zed dddGddde ZdS)N)AnyDictListMappingOptional) deprecated)CallbackManagerForLLMRun)LLM)get_from_dict_or_envpre_init) ConfigDict)enforce_stop_tokensZtranslation_textZ summary_textZgenerated_text) translationZ summarizationZconversationalztext-generationztext2text-generationz0.0.21z1.0z)langchain_huggingface.HuggingFaceEndpoint)ZremovalZalternative_importc @seZdZUdZdZeed<dZee ed< dZ ee ed< dZ ee ed< dZ ee ed<edd Zed ed efd d Zed ee effddZed e fddZ  dde deee deeded e f ddZdS)HuggingFaceHubaHuggingFaceHub models. ! This class is deprecated, you should use HuggingFaceEndpoint instead. To use, you should have the ``huggingface_hub`` python package installed, and the environment variable ``HUGGINGFACEHUB_API_TOKEN`` set with your API token, or pass it as a named parameter to the constructor. Supports `text-generation`, `text2text-generation`, `conversational`, `translation`, and `summarization`. Example: .. code-block:: python from langchain_community.llms import HuggingFaceHub hf = HuggingFaceHub(repo_id="gpt2", huggingfacehub_api_token="my-api-key") Nclientrepo_idtask model_kwargshuggingfacehub_api_tokenZforbid)extravaluesreturncCst|dd}zFddlm}m}|d}|||d}|ds1|s#td||d j|d }|j|d<|dtvrFtd |dd td ||d<W|St yWt dw)z?Validate that api key and python package exists in environment.rZHUGGINGFACEHUB_API_TOKENr)HfApiInferenceClientr)modeltokenrz1Must specify either `repo_id` or `task`, or both.)r)rzGot invalid task z, currently only z are supportedrzfCould not import huggingface_hub python package. Please install it with `pip install huggingface_hub`.) r huggingface_hubrr ValueError model_infoZ pipeline_tagVALID_TASKS_DICTkeys ImportError)clsrrrrrrrr#_/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/llms/huggingface_hub.pyvalidate_environment?s@      z#HuggingFaceHub.validate_environmentcCs$|jpi}i|j|jdd|iS)zGet the identifying parameters.)rrr)rrr)self _model_kwargsr#r#r$_identifying_paramsds  z"HuggingFaceHub._identifying_paramscCsdS)zReturn type of llm.rr#)r&r#r#r$ _llm_typemszHuggingFaceHub._llm_typepromptstop run_managerkwargsc Ks|jpi}i||}|jj||d|jd}t|}d|vr+td|dt|j}t |t r<|d|} n||} |durIt | |} | S)anCall out to HuggingFace Hub's inference 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 = hf("Tell me a joke.") )Zinputs parameters)jsonrerrorzError raised by inference API: rN) rrpostrr/loadsdecoderr isinstancelistr ) r&r*r+r,r-r'r.responseZ response_keytextr#r#r$_callrs      zHuggingFaceHub._call)NN)__name__ __module__ __qualname____doc__rr__annotations__rrstrrrdictrr Z model_configr rr%propertyrr(r)rrr8r#r#r#r$rs@  $ r)r/typingrrrrrZlangchain_core._api.deprecationrZlangchain_core.callbacksrZ#langchain_core.language_models.llmsr Zlangchain_core.utilsr r Zpydanticr Zlangchain_community.llms.utilsr rrr#r#r#r$s(