a bg@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 Zeeed d d Zeee efd ddZee d ddZde eee eeee dddZdS)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)extra)valuesreturncCst|dd}zddlm}m}|d}|||d}|dsb|sFtd||d j|d }|j|d<|dtvrtd |dd td ||d<Wnt yt dYn0|S)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#v/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/llms/huggingface_hub.pyvalidate_environment?s>       z#HuggingFaceHub.validate_environment)rcCs$|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_type)promptstop run_managerkwargsrc Ks|jpi}i||}|jj||d|jd}t|}d|vrVtd|dt|j}t |t rx|d|} n||} |durt | |} | 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.") )inputs parameters)jsonrerrorzError raised by inference API: rN) rrpostrr0loadsdecoderr 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)rrr9r#r#r#r$rs0  $ r)r0typingrrrrrZlangchain_core._api.deprecationrZlangchain_core.callbacksrZ#langchain_core.language_models.llmsr Zlangchain_core.utilsr r Zpydanticr Zlangchain_community.llms.utilsr rrr#r#r#r$s&