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.

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

Unregister commands incompatible with current config. #4815

Merged
merged 6 commits into from
Jun 27, 2023
Merged
Changes from all commits
Commits
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/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -139,6 +139,20 @@ def run_auto_gpt(
for command_category in enabled_command_categories:
command_registry.import_commands(command_category)

# Unregister commands that are incompatible with the current config
incompatible_commands = []
for command in command_registry.commands.values():
if callable(command.enabled) and not command.enabled(config):
command.enabled = False
incompatible_commands.append(command)

for command in incompatible_commands:
command_registry.unregister(command.name)
logger.debug(
f"Unregistering incompatible command: {command.name}, "
f"reason - {command.disabled_reason or 'Disabled by current config.'}"
)

ai_name = ""
ai_config = construct_main_ai_config(config)
ai_config.command_registry = command_registry
Expand Down