apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-24 14:15:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: Add options to Config.__init__ to support externally specified options.

Browse source
This commit is contained in:
莘权 马 2023-08-23 14:56:53 +08:00
parent 937bd12a63
commit 9395d9f7dc
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
metagpt/config.py
View file

@ -3,8 +3,9 @@
"""
@Desc: Provide configuration, singleton.
@Modified By: mashenquan, replace `CONFIG` with `os.environ` to support personal config
@Desc: `os.environ` doesn't support personalization, while `Config` does.
`os.environ` doesn't support personalization, while `Config` does.
Hence, the parameter reading priority is `Config` first, and if not found, then `os.environ`.
@Modified By: mashenquan, 2023/8/23. Add `options` to `Config.__init__` to support externally specified options.
"""
import os
@ -43,10 +44,14 @@ class Config:
key_yaml_file = PROJECT_ROOT / "config/key.yaml"
default_yaml_file = PROJECT_ROOT / "config/config.yaml"
def __init__(self, yaml_file=default_yaml_file):
def __init__(self, yaml_file=default_yaml_file, options=None):
self._configs = {}
self._init_with_config_files_and_env(self._configs, yaml_file)
if options:
self._configs.update(options)
self._parse()
def _parse(self):
logger.info("Config loading done.")
self.global_proxy = self._get("GLOBAL_PROXY")
self.openai_api_key = self._get("OPENAI_API_KEY")