ReMe/memoryscope/enumeration/model_enum.py
2024-07-24 19:28:40 +08:00

17 lines
657 B
Python

from enum import Enum
class ModelEnum(str, Enum):
"""
An enumeration representing different types of models used within the system.
Members:
GENERATION_MODEL: Represents a model responsible for generating content.
EMBEDDING_MODEL: Represents a model tasked with creating embeddings, typically used for transforming data into a numerical form suitable for machine learning tasks.
RANK_MODEL: Denotes a model that specializes in ranking, often used to order items based on relevance or importance.
"""
GENERATION_MODEL = "generation_model"
EMBEDDING_MODEL = "embedding_model"
RANK_MODEL = "rank_model"