Spaces:
Runtime error
Runtime error
File size: 2,185 Bytes
a6fd1a3 e6cba72 7f9bd61 d3e9dc7 a6fd1a3 253c2e1 a6fd1a3 25e6e74 a6fd1a3 4f7e10b a6fd1a3 d3e9dc7 0bf745a 5760f5e |
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 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
# config.py
import os
# config.py
from dotenv import load_dotenv
load_dotenv()
API_HOST = os.environ.get("API_HOST")
API_PORT = os.environ.get("API_PORT")
ALLOWED_ORIGINS = os.environ.get("ALLOWED_ORIGINS")
# --- Cấu hình Đường dẫn ---
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
CORE_DATA_FOLDER = os.path.join(BASE_DIR, "data", "core")
# PENDING_UPLOADS_FOLDER = os.path.join(BASE_DIR, "data", "pending_uploads")
# PROCESSED_FILES_FOLDER = os.path.join(BASE_DIR, "data", "processed_files")
# FAILED_FILES_FOLDER = os.path.join(BASE_DIR, "data", "failed_files")
# PROCESSED_HASH_LOG = os.path.join(BASE_DIR, "data", "processed_hashes.log")
# PENDING_UPLOADS_FOLDER = '/tmp/pending_uploads'
LEGAL_DIC_FOLDER = os.path.join(BASE_DIR, "data", "dictionary")
# Cấu hình cho DB
WEAVIATE_URL = os.environ.get("WEAVIATE_URL")
WEAVIATE_COLLECTION_NAME = os.environ.get("WEAVIATE_COLLECTION_NAME")
WEAVIATE_API_KEY = os.environ.get("WEAVIATE_API_KEY")
# --- Cấu hình Model ---
EMBEDDING_MODEL_NAME = "bkai-foundation-models/vietnamese-bi-encoder"
RERANKER_MODEL_NAME = "cross-encoder/ms-marco-MiniLM-L-6-v2"
GROQ_MODEL_NAME = "llama-3.3-70b-versatile"
ALLOWED_EXTENSIONS = {".pdf", ".docx", ".doc"}
LEGAL_DOC_TYPES = ["Luật", "Bộ luật", "Nghị định", "Thông tư", "Quyết định", "Pháp lệnh", "Nghị quyết", "Chỉ thị", "Hiến pháp"]
MAX_CHUNK_SIZE = 1000 # Kích thước tối đa cho một chunk trước khi bị chia nhỏ hơn
CHUNK_OVERLAP = 200
REDIS_URL = os.environ.get("REDIS_URL")
SECRET_KEY = os.environ.get("SECRET_KEY")
ALGORITHM = os.environ.get("ALGORITHM", "HS256")
ACCESS_TOKEN_EXPIRE_MINUTES = os.environ.get("ACCESS_TOKEN_EXPIRE_MINUTES", 60)
LLAMA_CLOUD_API_KEYS=os.environ.get("LLAMA_CLOUD_API_KEYS")
GOOGLE_CLIENT_ID = os.environ.get("GOOGLE_CLIENT_ID")
GOOGLE_CLIENT_SECRET = os.environ.get("GOOGLE_CLIENT_SECRET")
FRONTEND_URL = os.environ.get("FRONTEND_URL")
APP_ENVIRONMENT = os.environ.get("APP_ENVIRONMENT")
# CHECKPOINT_FILE = "processed_files.log"
MONGODB_CLOUD_URI= os.environ.get("MONGODB_CLOUD_URI")
DB_NAME= os.environ.get("DB_NAME")
GOOGLE_API_KEYS=os.environ.get("GOOGLE_API_KEYS", "")
|