Spaces:
Sleeping
Sleeping
| import os | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| os.environ["HF_HUB_DISABLE_SYMLINKS_WARNING"] = "1" | |
| BASE_DIR = os.path.dirname(os.path.abspath(__file__)) | |
| DATA_PATH = os.path.join(BASE_DIR, "DeepSeek_v4.json") | |
| CHROMA_DB_PATH = os.path.join(BASE_DIR, "chroma_db") | |
| EMBEDDING_MODEL = "BAAI/bge-small-en-v1.5" | |
| EMBEDDING_DIM = 384 | |
| EMBEDDING_DEVICE = "cpu" | |
| RERANKER_MODEL = "cross-encoder/ms-marco-MiniLM-L-6-v2" | |
| RERANKER_DEVICE = "cpu" | |
| CHUNK_SIZE = 512 | |
| CHUNK_OVERLAP = 64 | |
| BM25_TOP_K = 15 | |
| DENSE_TOP_K = 15 | |
| HYBRID_TOP_K = 15 | |
| RERANK_CANDIDATES = 10 | |
| RERANK_TOP_K = 7 | |
| FINAL_TOP_K = 5 | |
| HYBRID_ALPHA = 0.5 | |
| HYBRID_FUSION_METHOD = "normalized" | |
| CACHE_MAX_SIZE = 200 | |
| CACHE_TTL_SECONDS = 3600 | |
| HNSW_SPACE = "cosine" | |
| HNSW_EF_CONSTRUCTION = 200 | |
| HNSW_M = 32 | |
| HNSW_EF_SEARCH = 384 | |
| DUP_SIM_THRESHOLD = 0.92 | |
| DUP_NGRAM_SIZE = 5 | |
| DUP_NUM_PERM = 128 | |
| GROQ_API_KEY = os.getenv("GROQ_API_KEY", "") | |
| GROQ_MODEL = "llama-3.1-8b-instant" | |
| GROQ_TEMPERATURE = 0.1 | |
| GROQ_MAX_TOKENS = 2048 | |
| GROQ_TIMEOUT = 10 | |
| MAX_CONTEXT_TOKENS = 8192 | |
| SYSTEM_PROMPT_RATIO = 0.15 | |
| CHUNK_BUDGET_RATIO = 0.70 | |
| GENERATION_RATIO = 0.15 | |
| EVAL_NUM_QUESTIONS = 20 | |
| EVAL_RETRIEVAL_K = [1, 5, 10] | |
| GRADIO_PORT = 7860 | |
| GRADIO_SHARE = False | |