From 8cc04971208f3665973b792745352ff002abfbc2 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 29 Jun 2023 11:18:11 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=80=20refactor(base.py):=20add=20from?= =?UTF-8?q?=5Fmethod=5Fnodes=20dictionary=20to=20RetrieverCreator=20class?= =?UTF-8?q?=20The=20`from=5Fmethod=5Fnodes`=20dictionary=20is=20added=20to?= =?UTF-8?q?=20the=20`RetrieverCreator`=20class=20in=20order=20to=20map=20t?= =?UTF-8?q?he=20"MultiQueryRetriever"=20node=20to=20the=20"from=5Fllm"=20m?= =?UTF-8?q?ethod.=20This=20refactor=20improves=20the=20readability=20and?= =?UTF-8?q?=20maintainability=20of=20the=20code=20by=20explicitly=20defini?= =?UTF-8?q?ng=20the=20mapping=20between=20nodes=20and=20methods.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/interface/retrievers/base.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/backend/langflow/interface/retrievers/base.py b/src/backend/langflow/interface/retrievers/base.py index 3eb38861e..16f49191e 100644 --- a/src/backend/langflow/interface/retrievers/base.py +++ b/src/backend/langflow/interface/retrievers/base.py @@ -13,6 +13,8 @@ from langflow.utils.util import build_template_from_method class RetrieverCreator(LangChainTypeCreator): type_name: str = "retrievers" + from_method_nodes = {"MultiQueryRetriever": "from_llm"} + @property def frontend_node_class(self) -> Type[RetrieverFrontendNode]: return RetrieverFrontendNode