Skip to content

Commit 530fe76

Browse files
fix(viking-kb): fix viking db got none responses (#447)
1 parent c237559 commit 530fe76

File tree

1 file changed

+23
-11
lines changed

1 file changed

+23
-11
lines changed

veadk/knowledgebase/backends/vikingdb_knowledge_backend.py

Lines changed: 23 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -137,8 +137,6 @@ class VikingDBKnowledgeBackend(BaseKnowledgebaseBackend):
137137
_viking_sdk_client = None
138138

139139
def model_post_init(self, __context: Any) -> None:
140-
self._set_service_info()
141-
142140
self.precheck_index_naming()
143141

144142
# check whether collection exist, if not, create it
@@ -572,17 +570,31 @@ def _search_knowledge(
572570
post_processing=post_precessing,
573571
)
574572

573+
logger.debug(
574+
f"Search knowledge {self.index} using project {self.volcengine_project} original response: {response}"
575+
)
576+
575577
entries = []
576-
for result in response.get("result_list", []):
577-
doc_meta_raw_str = result.get("doc_info", {}).get("doc_meta")
578-
doc_meta_list = json.loads(doc_meta_raw_str) if doc_meta_raw_str else []
579-
metadata = {}
580-
for meta in doc_meta_list:
581-
metadata[meta["field_name"]] = meta["field_value"]
582-
583-
entries.append(
584-
KnowledgebaseEntry(content=result.get("content", ""), metadata=metadata)
578+
if not response.get("result_list", []):
579+
logger.warning(
580+
f"Search knowledge {self.index} using project {self.volcengine_project} got empty response."
581+
)
582+
else:
583+
logger.debug(
584+
f"Search knowledge {self.index} using project {self.volcengine_project} got {len(response.get('result_list', []))} results."
585585
)
586+
for result in response.get("result_list", []):
587+
doc_meta_raw_str = result.get("doc_info", {}).get("doc_meta")
588+
doc_meta_list = json.loads(doc_meta_raw_str) if doc_meta_raw_str else []
589+
metadata = {}
590+
for meta in doc_meta_list:
591+
metadata[meta["field_name"]] = meta["field_value"]
592+
593+
entries.append(
594+
KnowledgebaseEntry(
595+
content=result.get("content", ""), metadata=metadata
596+
)
597+
)
586598

587599
return entries
588600

0 commit comments

Comments
 (0)