Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix commands with same name overwriting #4226

Merged
merged 4 commits into from
May 15, 2023
Merged

Fix commands with same name overwriting #4226

Changes from 2 commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
d4004b3
fix commands with same name overwriting
k-boikov May 15, 2023
d8e3afd
black dot
k-boikov May 15, 2023
c329043
removed debug for logging
k-boikov May 15, 2023
2294423
removed unneeded import
k-boikov May 15, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 14 additions & 0 deletions autogpt/commands/command.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,9 +3,14 @@
import inspect
from typing import Any, Callable, Optional

from autogpt.config import Config
from autogpt.logs import logger

# Unique identifier for auto-gpt commands
AUTO_GPT_COMMAND_IDENTIFIER = "auto_gpt_command"

cfg = Config()
k-boikov marked this conversation as resolved.
Show resolved Hide resolved


class Command:
"""A class representing a command.
Expand Down Expand Up @@ -59,6 +64,10 @@ def _reload_module(self, module: Any) -> Any:
return importlib.reload(module)

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

def unregister(self, command_name: str):
Expand Down Expand Up @@ -133,6 +142,11 @@ def command(
) -> Callable[..., Any]:
"""The command decorator is used to create Command objects from ordinary functions."""

if not enabled:
if disabled_reason is not None and cfg.debug_mode:
k-boikov marked this conversation as resolved.
Show resolved Hide resolved
logger.debug(f"Command '{name}' is disabled: {disabled_reason}")
return lambda func: func

def decorator(func: Callable[..., Any]) -> Command:
cmd = Command(
name=name,
Expand Down