mirror of
https://github.com/Sosokker/plain-rag.git
synced 2025-12-18 14:34:05 +01:00
18 lines
366 B
Python
18 lines
366 B
Python
from enum import Enum
|
|
|
|
|
|
class EmbeddingModelName(str, Enum):
|
|
MiniLMEmbeddingModel = "sentence-transformers/all-MiniLM-L6-v2"
|
|
|
|
|
|
class RerankerModelName(str, Enum):
|
|
MiniLMReranker = "cross-encoder/ms-marco-MiniLM-L-6-v2"
|
|
|
|
|
|
class LLMModelName(str, Enum):
|
|
GeminiFlash = "gemini/gemini-2.0-flash"
|
|
|
|
|
|
class VectorStoreType(str, Enum):
|
|
PGVECTOR = "pgvector"
|