Spaces:
Sleeping
Sleeping
WorkWise backend GPU Data path
Browse files- app/config.py +6 -2
- app/services/vector_store.py +5 -0
app/config.py
CHANGED
|
@@ -4,12 +4,16 @@ from dotenv import load_dotenv
|
|
| 4 |
|
| 5 |
load_dotenv()
|
| 6 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
class Settings:
|
| 8 |
"""Application settings loaded from environment variables"""
|
| 9 |
|
| 10 |
# Faiss (local) configuration
|
| 11 |
-
FAISS_INDEX_PATH: str = os.getenv("
|
| 12 |
-
FAISS_PAYLOADS_PATH: str = os.getenv("
|
| 13 |
|
| 14 |
# Qdrant Configuration
|
| 15 |
QDRANT_URL: str = os.getenv("QDRANT_URL", "http://localhost:6333")
|
|
|
|
| 4 |
|
| 5 |
load_dotenv()
|
| 6 |
|
| 7 |
+
BASE_DIR = "/home/user/app"
|
| 8 |
+
DATA_DIR = os.path.join(BASE_DIR, "data")
|
| 9 |
+
os.makedirs(DATA_DIR, exist_ok=True)
|
| 10 |
+
|
| 11 |
class Settings:
|
| 12 |
"""Application settings loaded from environment variables"""
|
| 13 |
|
| 14 |
# Faiss (local) configuration
|
| 15 |
+
FAISS_INDEX_PATH: str = os.getenv("DATA_DIR", "faiss.index")
|
| 16 |
+
FAISS_PAYLOADS_PATH: str = os.getenv("DATA_DIR", "faiss_payloads.json")
|
| 17 |
|
| 18 |
# Qdrant Configuration
|
| 19 |
QDRANT_URL: str = os.getenv("QDRANT_URL", "http://localhost:6333")
|
app/services/vector_store.py
CHANGED
|
@@ -10,6 +10,7 @@ import numpy as np
|
|
| 10 |
|
| 11 |
logger = setup_logger(__name__)
|
| 12 |
|
|
|
|
| 13 |
def _normalize(vectors: np.ndarray) -> np.ndarray:
|
| 14 |
"""L2-normalize vectors so inner product equals cosine similarity."""
|
| 15 |
norms = np.linalg.norm(vectors, axis=1, keepdims=True) + 1e-12
|
|
@@ -24,6 +25,10 @@ class VectorStoreService:
|
|
| 24 |
"""
|
| 25 |
|
| 26 |
def __init__(self):
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
self.index: Optional[faiss.Index] = None
|
| 28 |
self.dimension: Optional[int] = None
|
| 29 |
self.payloads: List[Dict[str, Any]] = []
|
|
|
|
| 10 |
|
| 11 |
logger = setup_logger(__name__)
|
| 12 |
|
| 13 |
+
|
| 14 |
def _normalize(vectors: np.ndarray) -> np.ndarray:
|
| 15 |
"""L2-normalize vectors so inner product equals cosine similarity."""
|
| 16 |
norms = np.linalg.norm(vectors, axis=1, keepdims=True) + 1e-12
|
|
|
|
| 25 |
"""
|
| 26 |
|
| 27 |
def __init__(self):
|
| 28 |
+
|
| 29 |
+
print("FAISS_INDEX_PATH:", settings.FAISS_INDEX_PATH)
|
| 30 |
+
print("Working directory:", os.getcwd())
|
| 31 |
+
|
| 32 |
self.index: Optional[faiss.Index] = None
|
| 33 |
self.dimension: Optional[int] = None
|
| 34 |
self.payloads: List[Dict[str, Any]] = []
|