config.py

from pydantic_settings import BaseSettings


class ModelConfig(BaseSettings):
    MODEL_EMBEDDINGS: str = "sentence-transformers/all-MiniLM-L6-v2"
    AUTO_TOKENIZER: str = "meta-llama/Llama-2-7b-chat-hf"
    MODEL_LLM: str = "meta-llama/Llama-2-7b-chat-hf"


MODEL_CONFIG = ModelConfig()