aboutsummaryrefslogtreecommitdiff
path: root/autogpt/plugins/plugins_config.py
diff options
context:
space:
mode:
Diffstat (limited to 'autogpt/plugins/plugins_config.py')
-rw-r--r--autogpt/plugins/plugins_config.py112
1 files changed, 0 insertions, 112 deletions
diff --git a/autogpt/plugins/plugins_config.py b/autogpt/plugins/plugins_config.py
deleted file mode 100644
index 13b871303..000000000
--- a/autogpt/plugins/plugins_config.py
+++ /dev/null
@@ -1,112 +0,0 @@
-from __future__ import annotations
-
-import os
-from typing import Union
-
-import yaml
-from pydantic import BaseModel
-
-from autogpt.logs import logger
-from autogpt.plugins.plugin_config import PluginConfig
-
-
-class PluginsConfig(BaseModel):
- """Class for holding configuration of all plugins"""
-
- plugins: dict[str, PluginConfig]
-
- def __repr__(self):
- return f"PluginsConfig({self.plugins})"
-
- def get(self, name: str) -> Union[PluginConfig, None]:
- return self.plugins.get(name)
-
- def is_enabled(self, name) -> bool:
- plugin_config = self.plugins.get(name)
- return plugin_config is not None and plugin_config.enabled
-
- @classmethod
- def load_config(
- cls,
- plugins_config_file: str,
- plugins_denylist: list[str],
- plugins_allowlist: list[str],
- ) -> "PluginsConfig":
- empty_config = cls(plugins={})
-
- try:
- config_data = cls.deserialize_config_file(
- plugins_config_file,
- plugins_denylist,
- plugins_allowlist,
- )
- if type(config_data) != dict:
- logger.error(
- f"Expected plugins config to be a dict, got {type(config_data)}, continuing without plugins"
- )
- return empty_config
- return cls(plugins=config_data)
-
- except BaseException as e:
- logger.error(
- f"Plugin config is invalid, continuing without plugins. Error: {e}"
- )
- return empty_config
-
- @classmethod
- def deserialize_config_file(
- cls,
- plugins_config_file: str,
- plugins_denylist: list[str],
- plugins_allowlist: list[str],
- ) -> dict[str, PluginConfig]:
- if not os.path.exists(plugins_config_file):
- logger.warn("plugins_config.yaml does not exist, creating base config.")
- cls.create_empty_plugins_config(
- plugins_config_file,
- plugins_denylist,
- plugins_allowlist,
- )
-
- with open(plugins_config_file, "r") as f:
- plugins_config = yaml.load(f, Loader=yaml.FullLoader)
-
- plugins = {}
- for name, plugin in plugins_config.items():
- if type(plugin) == dict:
- plugins[name] = PluginConfig(
- name=name,
- enabled=plugin.get("enabled", False),
- config=plugin.get("config", {}),
- )
- elif type(plugin) == PluginConfig:
- plugins[name] = plugin
- else:
- raise ValueError(f"Invalid plugin config data type: {type(plugin)}")
- return plugins
-
- @staticmethod
- def create_empty_plugins_config(
- plugins_config_file: str,
- plugins_denylist: list[str],
- plugins_allowlist: list[str],
- ):
- """Create an empty plugins_config.yaml file. Fill it with values from old env variables."""
- base_config = {}
-
- logger.debug(f"Legacy plugin denylist: {plugins_denylist}")
- logger.debug(f"Legacy plugin allowlist: {plugins_allowlist}")
-
- # Backwards-compatibility shim
- for plugin_name in plugins_denylist:
- base_config[plugin_name] = {"enabled": False, "config": {}}
-
- for plugin_name in plugins_allowlist:
- base_config[plugin_name] = {"enabled": True, "config": {}}
-
- logger.debug(f"Constructed base plugins config: {base_config}")
-
- logger.debug(f"Creating plugin config file {plugins_config_file}")
- with open(plugins_config_file, "w+") as f:
- f.write(yaml.dump(base_config))
- return base_config