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