FdohdZddlmZddlmZddlmZddlmZm Z ddl m Z Gdde Z d S) %Agent manager for managing GPT agents) annotations)Config) ChatSequence)Messagecreate_chat_completion) Singletonc:eZdZdZddZdd ZddZddZddZdS) AgentManagerrconfigrc0d|_i|_||_dS)Nr)next_keyagentsr )selfr s OC:\Users\Administrator\Downloads\Auto-GPT-master\autogpt\agent\agent_manager.py__init__zAgentManager.__init__ s"    taskstrcreation_promptmodelreturntuple[int, str]cftj|td|g}|jjD]_}|s||x}r|d|D`t||jj }| d|d}t|jjD]I\} }| s|d|Dx} r| rdnd} || | }J|r|dkr| d||j} |xjdz c_|t!||f|j| <|jjD],}|s||}-| |fS) aCreate a new agent and return its key Args: task: The task to perform creation_prompt: Prompt passed to the LLM at creation model: The model to use to run this agent Returns: The key of the new agent userc&g|]}tdi|Sr.0raw_msgs r z-AgentManager.create_agent..*& S S S!3!37!3!3 S S Srpromptr assistantc6g|]}|Srrawr!ms rr#z-AgentManager.create_agent..6 6Q6Q6Q1quuww6Q6Q6Qr )r for_modelrr pluginscan_handle_pre_instructionpre_instructionr+extendrcontentadd enumeratecan_handle_on_instructionon_instructionrlistrcan_handle_post_instructionpost_instruction) rrrrmessagespluginplugin_messages agent_reply plugins_replyi plugin_resultsepkeys r create_agentzAgentManager.create_agents )%'&/2R2R1STTk) U UF4466 "("8"8"H"HH U S S? S S STTT,DK      [+... "4;#677 G GIAv3355  & 5 56Q6Q6Q6Q6Q R RR} G'ddR#0 F# F} F F  5]b00 LLm 4 4 4m   $x..%8 Ck) ? ?F5577  11+>>KKKrrF str | intmessagec:|jt|\}}}tj||}|d||jjD]W}|s|d|Dx}r| d|DXt||jj }|d||} t|jjD]I\} }| s|d|Dx} r| rdnd} | | | } J| r| dkr|d| |jjD],}|s||}-|S) zSend a message to an agent and return its response Args: key: The key of the agent to message message: The message to send to the agent Returns: The agent's response rc6g|]}|Srr*r,s rr#z.AgentManager.message_agent..]s 9T9T9Ta!%%''9T9T9Trc&g|]}tdi|Srrr s rr#z.AgentManager.message_agent..^r$rr%r'c6g|]}|Srr*r,s rr#z.AgentManager.message_agent..kr.rr/r()rintrr1r7r r2r3r4r5rr6r8r9r:r<r=) rrFrIrr>rr?r@rArBrCrDrEs r message_agentzAgentManager.message_agentJs!% CHH 5h )%:: VW%%%k) U UF4466 "("8"89T9T89T9T9T"U"UU U S S? S S STTT-DK      [+...# "4;#677 G GIAv3355  & 5 56Q6Q6Q6Q6Q R RR} G'ddR#0 F# F} F F  5]b00 LLm 4 4 4k) ? ?F5577  11+>>KKrlist[tuple[str | int, str]]cHd|jDS)zkReturn a list of all agents Returns: A list of tuples of the form (key, task) c$g|] \}\}}}||fSrr)r!rFr_s rr#z,AgentManager.list_agents..s'III 1\dAqd IIIr)ritems)rs r list_agentszAgentManager.list_agentsys'JIT[5F5F5H5HIIIIrboolcT |jt|=dS#t$rYdSwxYw)zDelete an agent from the agent manager Args: key: The key of the agent to delete Returns: True if successful, False otherwise TF)rrNKeyError)rrFs r delete_agentzAgentManager.delete_agents>  CHH%4   55 s  ''N)r r)rrrrrrrr)rFrHrIrrr)rrP)rFrHrrV) __name__ __module__ __qualname____doc__rrGrOrUrYrrrr r s//1 1 1 1 f----^JJJJrr ) metaclassN) r] __future__rautogpt.configrautogpt.llm.baserautogpt.llm.chatrrautogpt.singletonr r rrrrds++""""""!!!!!!))))))<<<<<<<<''''''GGGGGYGGGGGGr