From c4b9e903c8e4bd1e3fb859dc2938fa7557d8c4a3 Mon Sep 17 00:00:00 2001
From: Kevin Hu <kevinhu.sh@gmail.com>
Date: Thu, 23 Jan 2025 11:45:22 +0800
Subject: [PATCH] Fix index not found for new user. (#4597)

### What problem does this PR solve?

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
---
 api/apps/kb_app.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/api/apps/kb_app.py b/api/apps/kb_app.py
index e2d7f2e8b..0cb511611 100644
--- a/api/apps/kb_app.py
+++ b/api/apps/kb_app.py
@@ -14,6 +14,7 @@
 #  limitations under the License.
 #
 import json
+import logging
 
 from flask import request
 from flask_login import login_required, current_user
@@ -286,13 +287,18 @@ def knowledge_graph(kb_id):
             message='No authorization.',
             code=settings.RetCode.AUTHENTICATION_ERROR
         )
-    e, kb = KnowledgebaseService.get_by_id(kb_id)
+    _, kb = KnowledgebaseService.get_by_id(kb_id)
     req = {
         "kb_id": [kb_id],
         "knowledge_graph_kwd": ["graph"]
     }
-    sres = settings.retrievaler.search(req, search.index_name(kb.tenant_id), [kb_id])
     obj = {"graph": {}, "mind_map": {}}
+    try:
+        sres = settings.retrievaler.search(req, search.index_name(kb.tenant_id), [kb_id])
+    except Exception as e:
+        logging.exception(e)
+        return get_json_result(data=obj)
+
     for id in sres.ids[:1]:
         ty = sres.field[id]["knowledge_graph_kwd"]
         try: