Spaces:
Sleeping
Sleeping
Debug retriever and search
Browse files- app/services/retriever.py +15 -0
- 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 |
)
|