mirror of
https://github.com/Sosokker/plain-rag.git
synced 2025-12-18 22:44:03 +01:00
14 lines
264 B
Python
14 lines
264 B
Python
from enum import Enum
|
|
|
|
|
|
class EmbeddingModelName(str, Enum):
|
|
MiniLMEmbeddingModel = "MiniLMEmbeddingModel"
|
|
|
|
|
|
class RerankerModelName(str, Enum):
|
|
MiniLMReranker = "MiniLMReranker"
|
|
|
|
|
|
class LLMModelName(str, Enum):
|
|
GeminiFlash = "gemini/gemini-2.0-flash"
|