added new module 'ai.py'

chatmastermind/ai.py

@@ -1,7 +1,7 @@
 from dataclasses import dataclass
 from abc import abstractmethod
 from typing import Protocol, Optional, Union
-from .configuration import AIConfig
+from .configuration import Config
 from .message import Message
 from .chat import Chat
 
@@ -34,7 +34,7 @@ class AI(Protocol):
     """
 
     name: str
-    config: AIConfig
+    config: Config
 
     @abstractmethod
     def request(self,

chatmastermind/configuration.py

@@ -7,15 +7,7 @@ OpenAIConfigInst = TypeVar('OpenAIConfigInst', bound='OpenAIConfig')
 
 
 @dataclass
-class AIConfig:
+class OpenAIConfig():
-    """
-    The base class of all AI configurations.
-    """
-    name: str
-
-
-@dataclass
-class OpenAIConfig(AIConfig):
     """
     The OpenAI section of the configuration file.
     """
@@ -33,7 +25,6 @@ class OpenAIConfig(AIConfig):
         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']),
@@ -45,7 +36,7 @@ class OpenAIConfig(AIConfig):
 
 
 @dataclass
-class Config:
+class Config():
     """
     The configuration file structure.
     """