a
    dgm                     @   sZ   d dl mZ d dl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Zd	S )
    )List)BaseLanguageModel)BaseTool)load_response_generator)TaskExecutor)load_chat_plannerc                   @   s4   e Zd ZdZeee dddZeedddZ	dS )	
HuggingGPTz&Agent for interacting with HuggingGPT.)llmtoolsc                 C   s(   || _ || _t|| _t|| _|  d S )N)r	   r
   r   chat_plannerr   response_generator)selfr	   r
    r   /var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_experimental/autonomous_agents/hugginggpt/hugginggpt.py__init__   s
    

zHuggingGPT.__init__)inputreturnc                 C   s@   | j j|| jdd}t|| _| j  | jd| ji}|S )N)r   Zhf_tools)inputsZtask_execution)r   planr
   r   Ztask_executorrunr   generate)r   r   r   responser   r   r   r      s    

zHuggingGPT.runN)
__name__
__module____qualname____doc__r   r   r   r   strr   r   r   r   r   r      s   r   N)typingr   Zlangchain.base_languager   Zlangchain_core.toolsr   ZFlangchain_experimental.autonomous_agents.hugginggpt.repsonse_generatorr   ZAlangchain_experimental.autonomous_agents.hugginggpt.task_executorr   Z@langchain_experimental.autonomous_agents.hugginggpt.task_plannerr   r   r   r   r   r   <module>   s   