Fd dddlZddlZddlmZmZddlmZddlmZddl m Z GddZ dS)N)AnyCallable)AUTO_GPT_COMMAND_IDENTIFIER)logger)CommandceZdZUdZeeefed<dZdede fdZ de de fdZ d edd fd Z d efd Z ddZdedede ffdZd ede fdZdefdZdedd fdZd S)CommandRegistrya The CommandRegistry class is a manager for a collection of Command objects. It allows the registration, modification, and retrieval of Command objects, as well as the scanning and loading of command plugins from a specified directory. commandsci|_dSNr )selfs SC:\Users\Administrator\Downloads\Auto-GPT-master\autogpt\models\command_registry.py__init__zCommandRegistry.__init__s  module_namereturnc*tj|Sr ) importlib import_module)rrs r_import_modulezCommandRegistry._import_modules&{333rmodulec*tj|Sr )rreload)rrs r_reload_modulezCommandRegistry._reload_modules'''rcmdNcz|j|jvrtjd|jd||j|j<dS)N Command 'z-' already registered and will be overwritten!)namer rwarn)rrs rregisterzCommandRegistry.registersK 8t} $ $ KSCHSSS   #& chr command_namecN||jvr |j|=dStd|d)Nr' not found in registry.r KeyError)rr"s r unregisterzCommandRegistry.unregister$s9 4= ( ( l+++M|MMMNN Nrc|jD]c}|j|}||j}||}t |dr||ddS)z#Reloads all loaded command plugins.r!N)r r __module__rhasattrr!)rcmd_namerrreloaded_modules rreload_commandszCommandRegistry.reload_commands*sx  / /H-)C((88F"11&99O 33 /((...  / /rr.c|j|Sr r )rrs r get_commandzCommandRegistry.get_command3s}T""rc d||jvrtd|d|j|}|di|S)Nrr$r%)rr"kwargscommands rcallzCommandRegistry.call6sK t} , ,M|MMMNN N- -w     rcdt|jD}d|S)zi Returns a string representation of all registered `Command` objects for use in a prompt cBg|]\}}|dzdt|S)z. )str).0idxrs r z2CommandRegistry.command_prompt..@sA   )1csQw $ $#c(( $ $   r ) enumerater valuesjoin)r commands_lists rcommand_promptzCommandRegistry.command_prompt<sJ  5>t}?S?S?U?U5V5V   yy'''rctj|}t|D]}t||}t |t r0t|t r||jWtj |r?t|tr*|tkr|}||dS)a Imports the specified Python module containing command plugins. This method imports the associated module and registers any functions or classes that are decorated with the `AUTO_GPT_COMMAND_IDENTIFIER` attribute as `Command` objects. The registered `Command` objects are then added to the `commands` dictionary of the `CommandRegistry` object. Args: module_name (str): The name of the module to import for command plugins. N) rrdirgetattrr*rr!r3inspectisclass issubclassr)rrr attr_nameattr cmd_instances rimport_commandszCommandRegistry.import_commandsEs(55V , ,I69--Dt899 ,g1?? , dl++++%% ,*4T7*C*C ,HLPW#tvv  l+++ , ,r)rN)__name__r) __qualname____doc__dictr8r__annotations__rrrrr!r'r-rr/r4rArKr1rrr r sj3<    4#4#4444(S(S((((&G&&&&&OsOOOO ////##c(:####!!3!!!! (((((,3,4,,,,,,rr ) rrEtypingrrautogpt.command_decoratorr autogpt.logsrautogpt.models.commandrr r1rrrUs AAAAAA******V,V,V,V,V,V,V,V,V,V,r