13 lines
315 B
Python
13 lines
315 B
Python
from pydantic_settings import BaseSettings
|
|
import os
|
|
|
|
class Settings(BaseSettings):
|
|
INDEX_FILE: str = os.getenv("INDEX_FILE", "index_data/index.faiss")
|
|
EMBEDDING_DIM: int = 768
|
|
TOP_K: int = 5
|
|
DOC_PATH: str = "docs/"
|
|
DEVICE: str = "cpu"
|
|
MODEL_NAME: str = "BAAI/bge-m3"
|
|
|
|
settings = Settings()
|