fix:repair custom llm config

This commit is contained in:
liujun 2024-03-14 08:32:18 +08:00
parent e1123bae82
commit b2cfea2c74

View file

@ -99,20 +99,21 @@ class Config(CLIParams, YamlModel):
dicts += [Config.read_yaml(path) for path in default_config_paths]
final = merge_dict(dicts)
return Config(**final)
@classmethod
def config(cls,config:dict):
def from_llm_config(cls, llm_config: dict):
"""user config llm
example:
llm_config = {"llm": {"api_type": "xxx", "api_key": "xxx", "model": "xxx"}
gpt4 = Config.config(llm_config)
llm_config = {"api_type": "xxx", "api_key": "xxx", "model": "xxx"}
gpt4 = Config.from_llm_config(llm_config)
A = Role(name="A", profile="Democratic candidate", goal="Win the election", actions=[a1], watch=[a2], config=gpt4)
"""
llm_config = LLMConfig.model_validate(llm_config)
dicts = [dict(os.environ)]
dicts += [config]
dicts += [{"llm": llm_config}]
final = merge_dict(dicts)
return Config(**final)
def update_via_cli(self, project_path, project_name, inc, reqa_file, max_auto_summarize_code):
"""update config via cli"""