Improve knowledge base loading logic and logging in app.py
Browse files
app.py
CHANGED
@@ -248,7 +248,7 @@ try:
|
|
248 |
llm, embeddings = init_models()
|
249 |
vector_store = None
|
250 |
|
251 |
-
if os.path.exists(VECTOR_STORE_PATH):
|
252 |
try:
|
253 |
vector_store = FAISS.load_local(
|
254 |
VECTOR_STORE_PATH,
|
@@ -257,8 +257,10 @@ try:
|
|
257 |
)
|
258 |
logger.info("Successfully loaded existing knowledge base")
|
259 |
except Exception as e:
|
260 |
-
logger.
|
261 |
-
|
|
|
|
|
262 |
|
263 |
if vector_store is None:
|
264 |
logger.info("Building new knowledge base...")
|
|
|
248 |
llm, embeddings = init_models()
|
249 |
vector_store = None
|
250 |
|
251 |
+
if os.path.exists(os.path.join(VECTOR_STORE_PATH, "index.faiss")):
|
252 |
try:
|
253 |
vector_store = FAISS.load_local(
|
254 |
VECTOR_STORE_PATH,
|
|
|
257 |
)
|
258 |
logger.info("Successfully loaded existing knowledge base")
|
259 |
except Exception as e:
|
260 |
+
logger.warning(f"Could not load existing knowledge base, will create new one: {str(e)}")
|
261 |
+
vector_store = None
|
262 |
+
else:
|
263 |
+
logger.info("No existing knowledge base found, will create new one")
|
264 |
|
265 |
if vector_store is None:
|
266 |
logger.info("Building new knowledge base...")
|