| from os import getenv | |
| from dotenv import load_dotenv | |
| from pydantic import BaseSettings | |
| class Config(BaseSettings): | |
| """Configuration for the application.""" | |
| # Pinecone | |
| pinecone_api_key: str | |
| pinecone_env: str | |
| pinecone_index: str | |
| # LLM | |
| embedding_dim: int | |
| embedding_version: str | |
| embedding_dir: str | |
| model_name: str | |
| # HF | |
| password: str | |
| load_dotenv() | |
| config = Config( | |
| pinecone_api_key=getenv("PINECONE_API_KEY"), | |
| pinecone_env=getenv("PINECONE_ENV"), | |
| pinecone_index=getenv("PINECONE_INDEX"), | |
| embedding_dim=int(getenv("EMBEDDING_DIM")), | |
| embedding_version=getenv("EMBEDDING_VERSION"), | |
| embedding_dir=getenv("EMBEDDING_DIR"), | |
| model_name=getenv("MODEL_NAME"), | |
| password=getenv("PASSWORD"), | |
| ) | |