Significant-Gravitas
/
AutoGPT
mirror of https://github.com/Significant-Gravitas/AutoGPT.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

Fix commands with same name overwriting (#4226)

pull/3606/head^2
k-boikov 2023-05-15 21:41:11 +03:00 committed by GitHub
parent 16b7e7a91e
commit d06d8a609f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
autogpt/commands/command.py
View File

@ -3,6 +3,8 @@ import importlib
import inspect import inspect
from typing import Any, Callable, Optional from typing import Any, Callable, Optional
from autogpt.logs import logger
# Unique identifier for auto-gpt commands # Unique identifier for auto-gpt commands
AUTO_GPT_COMMAND_IDENTIFIER = "auto_gpt_command" AUTO_GPT_COMMAND_IDENTIFIER = "auto_gpt_command"
@ -59,6 +61,10 @@ class CommandRegistry:
return importlib.reload(module) return importlib.reload(module)
def register(self, cmd: Command) -> None: def register(self, cmd: Command) -> None:
if cmd.name in self.commands:
logger.warn(
f"Command '{cmd.name}' already registered and will be overwritten!"
)
self.commands[cmd.name] = cmd self.commands[cmd.name] = cmd
def unregister(self, command_name: str): def unregister(self, command_name: str):
@ -133,6 +139,11 @@ def command(
) -> Callable[..., Any]: ) -> Callable[..., Any]:
"""The command decorator is used to create Command objects from ordinary functions.""" """The command decorator is used to create Command objects from ordinary functions."""
if not enabled:
if disabled_reason is not None:
logger.debug(f"Command '{name}' is disabled: {disabled_reason}")
return lambda func: func
def decorator(func: Callable[..., Any]) -> Command: def decorator(func: Callable[..., Any]) -> Command:
cmd = Command( cmd = Command(
name=name, name=name,