remove KeyConfig with LLMConfig
This commit is contained in:
@@ -125,11 +125,17 @@ class InstantiateConfig(PrintableConfig):
|
||||
def get_name(self):
|
||||
return self.__class__.__name__
|
||||
|
||||
|
||||
|
||||
|
||||
@dataclass
|
||||
class KeyConfig(InstantiateConfig):
|
||||
class LLMKeyConfig(InstantiateConfig):
|
||||
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"""
|
||||
|
||||
api_key:str = None
|
||||
"""api key for llm"""
|
||||
@@ -143,18 +149,6 @@ 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
|
||||
|
||||
Reference in New Issue
Block a user