embedding.py
1 """Embedding model configuration.""" 2 3 4 from pydantic import Field 5 from pydantic_settings import BaseSettings 6 7 from src.embeddings.types import EmbeddingModelType 8 9 10 class EmbeddingConfig(BaseSettings): 11 """Embedding model configuration.""" 12 13 embed_name: EmbeddingModelType = Field( 14 default=EmbeddingModelType.HUGGINGFACE, 15 description="Embedding model type", 16 ) 17 embed_config: dict | None = Field( 18 default=None, 19 description="Additional model-specific keyword arguments", 20 ) 21