From a69225bb073cec6e05b246c7a30e35506618b5f4 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 5 Jul 2023 09:46:27 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20chore(loading.py):=20remove=20un?= =?UTF-8?q?used=20import=20statement=20=F0=9F=90=9B=20fix(loading.py):=20r?= =?UTF-8?q?emove=20unnecessary=20None=20default=20value=20for=20file=5Ffil?= =?UTF-8?q?ter=20parameter=20The=20unused=20import=20statement=20from=20la?= =?UTF-8?q?ngchain.prompts.base=20has=20been=20removed=20to=20improve=20co?= =?UTF-8?q?de=20cleanliness.=20The=20None=20default=20value=20for=20the=20?= =?UTF-8?q?file=5Ffilter=20parameter=20in=20the=20instantiate=5Fdocumentlo?= =?UTF-8?q?ader=20function=20has=20been=20removed=20as=20it=20is=20unneces?= =?UTF-8?q?sary=20and=20can=20be=20safely=20assumed=20that=20the=20paramet?= =?UTF-8?q?er=20will=20always=20be=20provided.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/interface/initialize/loading.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/interface/initialize/loading.py b/src/backend/langflow/interface/initialize/loading.py index 2985ab60c..cbac8278c 100644 --- a/src/backend/langflow/interface/initialize/loading.py +++ b/src/backend/langflow/interface/initialize/loading.py @@ -24,7 +24,6 @@ from langflow.utils import validate from langchain.chains.base import Chain from langchain.vectorstores.base import VectorStore from langchain.document_loaders.base import BaseLoader -from langchain.prompts.base import BasePromptTemplate def instantiate_class(node_type: str, base_type: str, params: Dict) -> Any: @@ -158,7 +157,7 @@ def instantiate_agent(class_object: Type[agent_module.Agent], params: Dict): return load_agent_executor(class_object, params) -def instantiate_prompt(node_type, class_object: Type[BasePromptTemplate], params: Dict): +def instantiate_prompt(node_type, class_object, params: Dict): if node_type == "ZeroShotPrompt": if "tools" not in params: params["tools"] = [] @@ -284,7 +283,7 @@ def instantiate_documentloader(class_object: Type[BaseLoader], params: Dict): # like lambda x: x.endswith(".txt") but as we don't know # anything besides the string, we will simply check if the string is # in x and if it is, we will return True - file_filter = params.pop("file_filter", None) + file_filter = params.pop("file_filter") extensions = file_filter.split(",") params["file_filter"] = lambda x: any( extension.strip() in x for extension in extensions