Add initial project structure with configuration, utilities, and API endpoints

This commit is contained in:
2025-02-07 19:24:57 +06:00
parent 480f6f06c2
commit 87e7b99daa
21 changed files with 513 additions and 159 deletions
+12 -17
View File
@@ -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()