From 9ff825f39d383f9f8944e5f815dd7de287937d3a Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Tue, 18 Feb 2025 09:09:22 +0800 Subject: [PATCH] Ignore exceptions when no index ahead. (#5047) ### What problem does this PR solve? ### Type of change - [x] Refactoring --- api/apps/kb_app.py | 2 +- rag/nlp/search.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/api/apps/kb_app.py b/api/apps/kb_app.py index 86584df3f..35e80b5e6 100644 --- a/api/apps/kb_app.py +++ b/api/apps/kb_app.py @@ -301,7 +301,7 @@ def knowledge_graph(kb_id): } obj = {"graph": {}, "mind_map": {}} - if not settings.docStoreConn.indexExist(search.index_name(kb.tenant_id)): + if not settings.docStoreConn.indexExist(search.index_name(kb.tenant_id), kb_id): return get_json_result(data=obj) sres = settings.retrievaler.search(req, search.index_name(kb.tenant_id), [kb_id]) if not len(sres.ids): diff --git a/rag/nlp/search.py b/rag/nlp/search.py index b463aa9a5..3580f2236 100644 --- a/rag/nlp/search.py +++ b/rag/nlp/search.py @@ -449,6 +449,8 @@ class Dealer: return res def all_tags(self, tenant_id: str, kb_ids: list[str], S=1000): + if not self.docStoreConn.indexExist(index_name(tenant_id), kb_ids[0]): + return [] res = self.dataStore.search([], [], {}, [], OrderByExpr(), 0, 0, index_name(tenant_id), kb_ids, ["tag_kwd"]) return self.dataStore.getAggregation(res, "tag_kwd")