Add RevChatGPTChatbogConfigModel

Author: code-yeongyu

aishell/models/__init__.py

@@ -1,2 +1,3 @@
 from .language_model import LanguageModel as LanguageModel
 from .open_ai_response_model import OpenAIResponseModel as OpenAIResponseModel
+from .revchatgpt_chatbot_config_model import RevChatGPTChatbotConfigModel as RevChatGPTChatbotConfigModel

aishell/models/revchatgpt_chatbot_config_model.py

@@ -0,0 +1,23 @@
+from typing import Optional
+
+from pydantic import BaseModel, root_validator
+
+
+class RevChatGPTChatbotConfigModel(BaseModel):
+    email: Optional[str] = None
+    password: Optional[str] = None
+    session_token: Optional[str] = None
+    access_token: Optional[str] = None
+    paid: bool = False
+
+    @root_validator
+    def check_at_least_one_account_info(cls, values: dict[str, Optional[str]]):
+        IS_ACCOUNT_LOGIN = values.get('email') and values.get('password')
+        IS_TOKEN_AUTH = values.get('session_token') or values.get('access_token')
+        if not IS_ACCOUNT_LOGIN and not IS_TOKEN_AUTH:
+            raise ValueError('No information for authentication provided.')
+
+        return values
+
+    class Config:
+        frozen = True