Compare commits
1 Commits
| 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.
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user