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: