U
    hW                     @   sT   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 G dd de	ZdS )    )AnyDictListMappingOptional)CallbackManagerForLLMRun)LLM)pre_initc                   @   s   e Zd ZU dZeed< dZee ed< G dd dZ	e
eeddd	Zeeeef d
ddZeed
ddZdeeee  ee eedddZdS )ManifestWrapperz HazyResearch's Manifest library.clientN
llm_kwargsc                   @   s   e Zd ZdZdS )zManifestWrapper.ConfigZforbidN)__name__
__module____qualname__extra r   r   E/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/llms/manifest.pyConfig   s   r   )valuesreturnc                 C   sD   z"ddl m} t|d |s tW n tk
r>   tdY nX |S )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   r   r   validate_environment   s    
z$ManifestWrapper.validate_environment)r   c                 C   s   | j p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   r   c                 K   sT   |dk	r"t |dkr"td| | jp*i }||}|dk	rD||d< | jj|f|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)r   r   r   __doc__r   __annotations__r   r   r   r   r	   r   propertyr   strr   r    r   r   r)   r   r   r   r   r
      s&   
  
r
   N)typingr   r   r   r   r   Zlangchain_core.callbacksr   Z#langchain_core.language_models.llmsr   Zlangchain_core.utilsr	   r
   r   r   r   r   <module>   s   