Kevin Hu commited on
Commit
cc4181b
·
1 Parent(s): 99634df

Fix index not found for new user. (#4597)

Browse files

### What problem does this PR solve?

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

Files changed (1) hide show
  1. api/apps/kb_app.py +8 -2
api/apps/kb_app.py CHANGED
@@ -14,6 +14,7 @@
14
  # limitations under the License.
15
  #
16
  import json
 
17
 
18
  from flask import request
19
  from flask_login import login_required, current_user
@@ -286,13 +287,18 @@ def knowledge_graph(kb_id):
286
  message='No authorization.',
287
  code=settings.RetCode.AUTHENTICATION_ERROR
288
  )
289
- e, kb = KnowledgebaseService.get_by_id(kb_id)
290
  req = {
291
  "kb_id": [kb_id],
292
  "knowledge_graph_kwd": ["graph"]
293
  }
294
- sres = settings.retrievaler.search(req, search.index_name(kb.tenant_id), [kb_id])
295
  obj = {"graph": {}, "mind_map": {}}
 
 
 
 
 
 
296
  for id in sres.ids[:1]:
297
  ty = sres.field[id]["knowledge_graph_kwd"]
298
  try:
 
14
  # limitations under the License.
15
  #
16
  import json
17
+ import logging
18
 
19
  from flask import request
20
  from flask_login import login_required, current_user
 
287
  message='No authorization.',
288
  code=settings.RetCode.AUTHENTICATION_ERROR
289
  )
290
+ _, kb = KnowledgebaseService.get_by_id(kb_id)
291
  req = {
292
  "kb_id": [kb_id],
293
  "knowledge_graph_kwd": ["graph"]
294
  }
 
295
  obj = {"graph": {}, "mind_map": {}}
296
+ try:
297
+ sres = settings.retrievaler.search(req, search.index_name(kb.tenant_id), [kb_id])
298
+ except Exception as e:
299
+ logging.exception(e)
300
+ return get_json_result(data=obj)
301
+
302
  for id in sres.ids[:1]:
303
  ty = sres.field[id]["knowledge_graph_kwd"]
304
  try: