Spaces:
Sleeping
Sleeping
File size: 1,538 Bytes
2028cec |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
"""
Project's config
"""
import os
from dotenv import load_dotenv
from pydantic_settings import BaseSettings, SettingsConfigDict
BASE_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), '../../'))
load_dotenv(os.path.join(BASE_DIR, '.env'))
class Settings(BaseSettings):
# model config
MODEL_ID: str = os.getenv('MODEL_ID')
MAX_NEW_TOKENS: int = int(os.getenv("MAX_NEW_TOKENS"))
MODEL_API_KEY: str = os.getenv('MODEL_API_KEY')
#openai config
OPENAI_API_KEY: str = os.getenv("OPENAI_API_KEY")
OPENAI_MODEL: str = os.getenv("OPENAI_MODEL")
OPENAI_EMBEDDING_MODEL: str = os.getenv("OPENAI_EMBEDDING_MODEL")
OPENAI_EMBEDDING_MODEL_DIMS: int = int(os.getenv("OPENAI_EMBEDDING_MODEL_DIMS"))
#helpers config
COHERE_API_KEY: str = os.getenv("COHERE_API_KEY")
MAX_NEW_TOKENS: int = int(os.getenv('MAX_NEW_TOKENS'))
MAX_OVERLAPS: int = int(os.getenv("MAX_OVERLAPS"))
# server config
SV_HOST: str = os.getenv('SV_HOST')
SV_PORT: int = int(os.getenv('SV_PORT'))
# embedding model config
EMBEDDING_MODEL: str = os.getenv('EMBEDDING_MODEL')
EMBEDDING_MODEL_API_KEY: str = os.getenv('EMBEDDING_MODEL_API_KEY')
EMBEDDING_MODEL_DIMENSIONS: int = int(os.getenv('EMBEDDING_MODEL_DIMENSIONS'))
#data and database path
CHROMA_DIR: str = os.getenv("CHROMA_DIR")
CHROMA_COLLECTION: str = os.getenv("CHROMA_COLLECTION")
CSS_PATH: str = os.getenv("CSS_PATH")
RAW_DATA_DIR: str = os.getenv("RAW_DATA_DIR")
settings = Settings()
|