20 lines
539 B
Python
20 lines
539 B
Python
|
import os
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
# Base directory
|
||
|
|
BASE_DIR = Path(__file__).parent.parent
|
||
|
|
|
||
|
|
# Configuration
|
||
|
|
MODEL_NAME = "BAAI/bge-large-en-v1.5"
|
||
|
|
RERANKER_NAME = "BAAI/bge-reranker-base"
|
||
|
|
GROQ_MODEL = "llama-3.3-70b-versatile"
|
||
|
|
DOCS_PATH = BASE_DIR / "client_assets"
|
||
|
|
CHROMA_PATH = BASE_DIR / "chroma_index"
|
||
|
|
COLLECTION_NAME = "marketing_docs"
|
||
|
|
|
||
|
|
# Create directories if they don't exist
|
||
|
|
DOCS_PATH.mkdir(exist_ok=True)
|
||
|
|
CHROMA_PATH.mkdir(exist_ok=True)
|
||
|
|
|
||
|
|
# Groq API Key (Set through environment variable)
|
||
|
|
GROQ_API_KEY = os.getenv("GROQ_API_KEY")
|