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)
- 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 |
-
|
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:
|