mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-10-31 06:39:43 +00:00
23 lines
696 B
Python
23 lines
696 B
Python
from dataclasses import dataclass
|
|
|
|
from dbgpt.component import AppConfig
|
|
from dbgpt.util import BaseParameters
|
|
|
|
|
|
@dataclass
|
|
class BaseServeConfig(BaseParameters):
|
|
"""Base configuration class for serve"""
|
|
|
|
@classmethod
|
|
def from_app_config(cls, config: AppConfig, config_prefix: str):
|
|
"""Create a configuration object from a dictionary
|
|
|
|
Args:
|
|
config (AppConfig): Application configuration
|
|
config_prefix (str): Configuration prefix
|
|
"""
|
|
config_dict = config.get_all_by_prefix(config_prefix)
|
|
# remove prefix
|
|
config_dict = {k[len(config_prefix) :]: v for k, v in config_dict.items()}
|
|
return cls(**config_dict)
|