/ src / config / embedding.py
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