pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 12:32:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix bugs for agent/tools. (#9930)

Browse Source 
### What problem does this PR solve?
1 Fix typos
2 Fix agent/tools/crawler.py return bug.
3 Fix agent/tools/deepl.py  component_name  bug.

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
- [x] Refactoring
- [x] Performance Improvement

Signed-off-by: zhanluxianshen <zhanluxianshen@163.com>
This commit is contained in:
湛露先生
2025-09-05 12:31:44 +08:00
committed by GitHub
parent 4e16936fa4
commit 6ff7cfe005
5 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
agent/canvas.py
View File

@ -481,7 +481,7 @@ class Canvas(Graph):
except Exception as e: except Exception as e:
logging.exception(e) logging.exception(e)
def add_refernce(self, chunks: list[object], doc_infos: list[object]): def add_reference(self, chunks: list[object], doc_infos: list[object]):
if not self.retrieval: if not self.retrieval:
self.retrieval = [{"chunks": {}, "doc_aggs": {}}] self.retrieval = [{"chunks": {}, "doc_aggs": {}}]

2
agent/tools/base.py
View File

@ -166,7 +166,7 @@ class ToolBase(ComponentBase):
"count": 1, "count": 1,
"url": url "url": url
}) })
self._canvas.add_refernce(chunks, aggs) self._canvas.add_reference(chunks, aggs)
self.set_output("formalized_content", "\n".join(kb_prompt({"chunks": chunks, "doc_aggs": aggs}, 200000, True))) self.set_output("formalized_content", "\n".join(kb_prompt({"chunks": chunks, "doc_aggs": aggs}, 200000, True)))
def thoughts(self) -> str: def thoughts(self) -> str:

2
agent/tools/crawler.py
View File

@ -64,5 +64,5 @@ class Crawler(ToolBase, ABC):
elif self._param.extract_type == 'markdown': elif self._param.extract_type == 'markdown':
return result.markdown return result.markdown
elif self._param.extract_type == 'content': elif self._param.extract_type == 'content':
result.extracted_content return result.extracted_content
return result.markdown return result.markdown

2
agent/tools/deepl.py
View File

@ -43,7 +43,7 @@ class DeepLParam(ComponentParamBase):
class DeepL(ComponentBase, ABC): class DeepL(ComponentBase, ABC):
component_name = "GitHub" component_name = "DeepL"
def _run(self, history, **kwargs): def _run(self, history, **kwargs):
ans = self.get_input() ans = self.get_input()

2
agent/tools/retrieval.py
View File

@ -163,7 +163,7 @@ class Retrieval(ToolBase, ABC):
self.set_output("formalized_content", self._param.empty_response) self.set_output("formalized_content", self._param.empty_response)
return return
self._canvas.add_refernce(kbinfos["chunks"], kbinfos["doc_aggs"]) self._canvas.add_reference(kbinfos["chunks"], kbinfos["doc_aggs"])
form_cnt = "\n".join(kb_prompt(kbinfos, 200000, True)) form_cnt = "\n".join(kb_prompt(kbinfos, 200000, True))
self.set_output("formalized_content", form_cnt) self.set_output("formalized_content", form_cnt)
return form_cnt return form_cnt