Add enum LanguageModel

Author: code-yeongyu

aishell/models/__init__.py

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

aishell/models/language_model.py

@@ -0,0 +1,9 @@
+from enum import auto
+
+from aishell.utils import StrEnum
+
+
+class LanguageModel(StrEnum):
+    OFFICIAL_CHATGPT = auto()
+    REVERSE_ENGINEERED_CHATGPT = auto()
+    GPT3 = auto()

aishell/utils/__init__.py

@@ -1 +1,2 @@
 from .make_executable_command import make_executable_command as make_executable_command
+from .str_enum import StrEnum as StrEnum

aishell/utils/str_enum.py

@@ -0,0 +1,14 @@
+from enum import Enum
+from typing import Any
+
+
+class StrEnum(str, Enum):
+
+    def _generate_next_value_(name: str, start: int, count: int, last_values: list[Any]):  # type: ignore
+        return name.lower()
+
+    def __repr__(self):
+        return self.name
+
+    def __str__(self):
+        return self.name