marketing assistant added

This commit is contained in:
kowshik24
2025-02-06 03:20:41 +06:00
commit 480f6f06c2
7 changed files with 246 additions and 0 deletions
+20
View File
@@ -0,0 +1,20 @@
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")