diff --git a/lang_agent/config.py b/lang_agent/config.py index 799ee1f..3bb3d67 100644 --- a/lang_agent/config.py +++ b/lang_agent/config.py @@ -143,6 +143,18 @@ class KeyConfig(InstantiateConfig): logger.info("ALI_API_KEY loaded from environ") +@dataclass +class LLMKeyConfig(KeyConfig): + llm_name: str = "qwen-plus" + """name of llm""" + + llm_provider:str = "openai" + """provider of the llm""" + + base_url:str = "https://dashscope.aliyuncs.com/compatible-mode/v1" + """base url; could be used to overwrite the baseurl in llm provider""" + + @dataclass class ToolConfig(InstantiateConfig): use_tool:bool = True