VcRlAgent commited on
Commit
8fb6d95
·
1 Parent(s): fcb38d8

WorkWise backend GPU Data path

Browse files
Files changed (2) hide show
  1. app/config.py +6 -2
  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("FAISS_INDEX_PATH", "/home/user/app/workwise/data/faiss.index")
12
- FAISS_PAYLOADS_PATH: str = os.getenv("FAISS_PAYLOADS_PATH", "/home/user/app/workwise/data/faiss_payloads.json")
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]] = []