diff --git a/chatmastermind/configuration.py b/chatmastermind/configuration.py index 5ae32d6..79daa72 100644 --- a/chatmastermind/configuration.py +++ b/chatmastermind/configuration.py @@ -7,7 +7,15 @@ OpenAIConfigInst = TypeVar('OpenAIConfigInst', bound='OpenAIConfig') @dataclass -class OpenAIConfig(): +class AIConfig: + """ + The base class of all AI configurations. + """ + name: str + + +@dataclass +class OpenAIConfig(AIConfig): """ The OpenAI section of the configuration file. """ @@ -25,6 +33,7 @@ class OpenAIConfig(): Create OpenAIConfig from a dict. """ return cls( + name='OpenAI', api_key=str(source['api_key']), model=str(source['model']), max_tokens=int(source['max_tokens']), @@ -36,7 +45,7 @@ class OpenAIConfig(): @dataclass -class Config(): +class Config: """ The configuration file structure. """