Add initial project structure with configuration, utilities, and API endpoints
This commit is contained in:
@@ -1,20 +1,15 @@
|
||||
import os
|
||||
from pathlib import Path
|
||||
|
||||
# Base directory
|
||||
BASE_DIR = Path(__file__).parent.parent
|
||||
class Settings:
|
||||
MODEL_NAME = "BAAI/bge-large-en-v1.5"
|
||||
RERANKER_NAME = "BAAI/bge-reranker-large"
|
||||
GROQ_MODEL = "llama-3.3-70b-versatile"
|
||||
#DOCS_PATH = "/home/kowshik/work/ds_tjc/datasets/Client-Assets"
|
||||
DOCS_PATH = "/home/kowshik/work/ds_tjc/datasets/marketing_data"
|
||||
CHROMA_PATH = "/home/kowshik/work/ds_tjc/chroma_index"
|
||||
COLLECTION_NAME = "marketing_docs"
|
||||
API_KEY = "4BkwTtVd5VwhTiFDdG3NfzgATrCq7aD8AjnvWNeivirTntHgRvL6Xe84ULHcVTLB"
|
||||
SERVER_URL = "https://ma.rommelcorral.com"
|
||||
GROQ_API_KEY = "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"
|
||||
|
||||
# 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")
|
||||
settings = Settings()
|
||||
Reference in New Issue
Block a user