gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

update names of doc engine helper functions

Browse source
This commit is contained in:
He Wang 2025-11-18 14:39:27 +08:00
parent 6e81cf2cbf
commit 08745e0304
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
rag/utils/ob_conn.py
View file

@ -1435,13 +1435,13 @@ class OBConnection(DocStoreConnection):
Helper functions for search result
"""
def getTotal(self, res) -> int:
def get_total(self, res) -> int:
return res.total
def getChunkIds(self, res) -> list[str]:
def get_chunk_ids(self, res) -> list[str]:
return [row["id"] for row in res.chunks]
def getFields(self, res, fields: list[str]) -> dict[str, dict]:
def get_fields(self, res, fields: list[str]) -> dict[str, dict]:
result = {}
for row in res.chunks:
data = {}
@ -1511,7 +1511,7 @@ class OBConnection(DocStoreConnection):
last_pos = token_pos
return re.sub(r'</em><em>', '', highlighted_txt)
def getHighlight(self, res, keywords: list[str], fieldnm: str):
def get_highlight(self, res, keywords: list[str], fieldnm: str):
ans = {}
if len(res.chunks) == 0 or len(keywords) == 0:
return ans
@ -1527,7 +1527,7 @@ class OBConnection(DocStoreConnection):
ans[d["id"]] = highlighted_txt
return ans
def getAggregation(self, res, fieldnm: str):
def get_aggregation(self, res, fieldnm: str):
if len(res.chunks) == 0:
return []