VcRlAgent commited on
Commit
819f7c5
·
1 Parent(s): b27db4d

Debug retriever and search

Browse files
Files changed (2) hide show
  1. app/services/retriever.py +15 -0
  2. app/utils/logger.py +4 -2
app/services/retriever.py CHANGED
@@ -35,9 +35,24 @@ class RetrieverService:
35
  score_threshold=settings.SCORE_THRESHOLD
36
  )
37
 
 
38
  logger.debug(f"FAISS total vectors: {self.vector_store.index.ntotal}")
39
  D, I = self.vector_store.index.search(np.array([query_embedding]).astype("float32"), k=3)
40
  logger.debug(f"Distances: {D}, Indices: {I}")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
41
 
42
  #Qdrant
43
  # Search vector database
 
35
  score_threshold=settings.SCORE_THRESHOLD
36
  )
37
 
38
+ '''
39
  logger.debug(f"FAISS total vectors: {self.vector_store.index.ntotal}")
40
  D, I = self.vector_store.index.search(np.array([query_embedding]).astype("float32"), k=3)
41
  logger.debug(f"Distances: {D}, Indices: {I}")
42
+ '''
43
+ try:
44
+ logger.warning(f"FAISS index object: {self.vector_store.index}")
45
+ if self.vector_store.index is None:
46
+ logger.warning("❌ FAISS index is None")
47
+ else:
48
+ logger.warning(f"FAISS total vectors: {self.vector_store.index.ntotal}")
49
+ D, I = self.vector_store.index.search(
50
+ np.array([query_embedding]).astype("float32"), k=3
51
+ )
52
+ logger.warning(f"Distances: {D}, Indices: {I}")
53
+ except Exception as e:
54
+ import traceback
55
+ logger.error(f"FAISS search error: {e}\n{traceback.format_exc()}")
56
 
57
  #Qdrant
58
  # Search vector database
app/utils/logger.py CHANGED
@@ -5,11 +5,13 @@ import sys
5
  def setup_logger(name: str) -> logging.Logger:
6
  """Configure and return a logger instance"""
7
  logger = logging.getLogger(name)
8
- logger.setLevel(logging.INFO)
 
9
 
10
  if not logger.handlers:
11
  handler = logging.StreamHandler(sys.stdout)
12
- handler.setLevel(logging.INFO)
 
13
  formatter = logging.Formatter(
14
  '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
15
  )
 
5
  def setup_logger(name: str) -> logging.Logger:
6
  """Configure and return a logger instance"""
7
  logger = logging.getLogger(name)
8
+ #logger.setLevel(logging.INFO)
9
+ logger.setLevel(logging.DEBUG)
10
 
11
  if not logger.handlers:
12
  handler = logging.StreamHandler(sys.stdout)
13
+ #handler.setLevel(logging.INFO)
14
+ handler.setLevel(logging.DEBUG)
15
  formatter = logging.Formatter(
16
  '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
17
  )