Compare commits
1 Commits
05ae13c147
...
f3dfbc627e
| Author | SHA1 | Date | |
|---|---|---|---|
| f3dfbc627e |
@ -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,
|
||||
|
||||
@ -7,15 +7,7 @@ OpenAIConfigInst = TypeVar('OpenAIConfigInst', bound='OpenAIConfig')
|
||||
|
||||
|
||||
@dataclass
|
||||
class AIConfig:
|
||||
"""
|
||||
The base class of all AI configurations.
|
||||
"""
|
||||
name: str
|
||||
|
||||
|
||||
@dataclass
|
||||
class OpenAIConfig(AIConfig):
|
||||
class OpenAIConfig():
|
||||
"""
|
||||
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.
|
||||
"""
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user