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