29 lines
891 B
Python
29 lines
891 B
Python
import os
|
|
from pathlib import Path
|
|
|
|
# Base directory configuration
|
|
#BASE_DIR = Path(__file__).resolve().parent
|
|
BASE_DIR = Path("/root/ds_tjc")
|
|
|
|
# Data directories
|
|
UPLOAD_DIR = BASE_DIR / "marketing_data"
|
|
CHROMA_PATH = BASE_DIR / "chroma_index"
|
|
|
|
# Ensure directories exist
|
|
UPLOAD_DIR.mkdir(exist_ok=True)
|
|
for category in ["email", "books", "article", "social"]:
|
|
(UPLOAD_DIR / category).mkdir(exist_ok=True)
|
|
|
|
# Model configurations
|
|
MODEL_NAME = "BAAI/bge-large-en-v1.5"
|
|
RERANKER_NAME = "BAAI/bge-reranker-large"
|
|
GROQ_MODEL = "llama-3.3-70b-versatile"
|
|
COLLECTION_NAME = "marketing_docs"
|
|
|
|
# API configurations
|
|
API_KEY = "4BkwTtVd5VwhTiFDdG3NfzgATrCq7aD8AjnvWNeivirTntHgRvL6Xe84ULHcVTLB"
|
|
SERVER_URL = "https://ma.rommelcorral.com"
|
|
GROQ_API_KEY = "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"
|
|
|
|
# Valid document categories
|
|
VALID_CATEGORIES = ["email", "books", "article", "social"] |