o
    Zh                     @   s`   d dl mZmZmZmZmZ d dlmZ d dlm	Z	 d dl
mZ d dlmZ G dd de	ZdS )	    )AnyDictListMappingOptional)CallbackManagerForLLMRun)LLM)pre_init)
ConfigDictc                   @   s   e Zd ZU dZdZeed< dZee	 ed< e
ddZede	de	fd	d
Zedeeef fddZedefddZ		ddedeee  dee dedef
ddZdS )ManifestWrapperz HazyResearch's Manifest library.Nclient
llm_kwargsZforbid)extravaluesreturnc                 C   s<   zddl m} t|d |stW |S  ty   tdw )z3Validate that python package exists in environment.r   )Manifestr   z[Could not import manifest python package. Please install it with `pip install manifest-ml`.)manifestr   
isinstance
ValueErrorImportError)clsr   r    r   X/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/llms/manifest.pyvalidate_environment   s   z$ManifestWrapper.validate_environmentc                 C   s"   | j pi }i | jj  |S )N)r   r   Zclient_poolZget_current_clientZget_model_params)selfkwargsr   r   r   _identifying_params"   s   
z#ManifestWrapper._identifying_paramsc                 C   s   dS )zReturn type of llm.r   r   )r   r   r   r   	_llm_type*   s   zManifestWrapper._llm_typepromptstoprun_managerr   c                 K   s\   |durt |dkrtd| | jpi }i ||}|dur$||d< | jj|fi |S )z!Call out to LLM through Manifest.N   z:Manifest currently only supports a single stop token, got Z
stop_token)lenNotImplementedErrorr   r   run)r   r   r   r    r   paramsr   r   r   _call/   s   
zManifestWrapper._call)NN)__name__
__module____qualname____doc__r   r   __annotations__r   r   r   r
   Zmodel_configr	   r   propertyr   strr   r   r   r   r&   r   r   r   r   r   	   s4   
 
r   N)typingr   r   r   r   r   Zlangchain_core.callbacksr   Z#langchain_core.language_models.llmsr   Zlangchain_core.utilsr	   Zpydanticr
   r   r   r   r   r   <module>   s    