diff --git a/agent/canvas.py b/agent/canvas.py index b0e9edbf4..6c99b4d8e 100644 --- a/agent/canvas.py +++ b/agent/canvas.py @@ -481,7 +481,7 @@ class Canvas(Graph): except Exception as 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: self.retrieval = [{"chunks": {}, "doc_aggs": {}}] diff --git a/agent/tools/base.py b/agent/tools/base.py index 8e6b78dd0..0d946a696 100644 --- a/agent/tools/base.py +++ b/agent/tools/base.py @@ -166,7 +166,7 @@ class ToolBase(ComponentBase): "count": 1, "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))) def thoughts(self) -> str: diff --git a/agent/tools/crawler.py b/agent/tools/crawler.py index b3601b10b..869fae4a3 100644 --- a/agent/tools/crawler.py +++ b/agent/tools/crawler.py @@ -64,5 +64,5 @@ class Crawler(ToolBase, ABC): elif self._param.extract_type == 'markdown': return result.markdown elif self._param.extract_type == 'content': - result.extracted_content + return result.extracted_content return result.markdown diff --git a/agent/tools/deepl.py b/agent/tools/deepl.py index 31e92729c..41d12341d 100644 --- a/agent/tools/deepl.py +++ b/agent/tools/deepl.py @@ -43,7 +43,7 @@ class DeepLParam(ComponentParamBase): class DeepL(ComponentBase, ABC): - component_name = "GitHub" + component_name = "DeepL" def _run(self, history, **kwargs): ans = self.get_input() diff --git a/agent/tools/retrieval.py b/agent/tools/retrieval.py index 27ddfe4b2..a6e1e4fcd 100644 --- a/agent/tools/retrieval.py +++ b/agent/tools/retrieval.py @@ -163,7 +163,7 @@ class Retrieval(ToolBase, ABC): self.set_output("formalized_content", self._param.empty_response) 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)) self.set_output("formalized_content", form_cnt) return form_cnt