diff --git a/langflow/__main__.py b/langflow/__main__.py index 4c19280c3..50d19f539 100644 --- a/langflow/__main__.py +++ b/langflow/__main__.py @@ -7,6 +7,7 @@ from pathlib import Path from langflow.server import LangflowApplication + def serve( workers: int = None, timeout: int = None, @@ -28,9 +29,12 @@ def serve( if not timeout: timeout = 60 - - options = {"bind": "0.0.0.0:5003", "workers": workers, "worker_class": "uvicorn.workers.UvicornWorker", "timeout": timeout} - + options = { + "bind": "0.0.0.0:5003", + "workers": workers, + "worker_class": "uvicorn.workers.UvicornWorker", + "timeout": timeout, + } LangflowApplication(app, options).run() diff --git a/langflow/backend/langflow_backend/custom/customs.py b/langflow/backend/langflow_backend/custom/customs.py index bd5c9aa59..cea81ac1a 100644 --- a/langflow/backend/langflow_backend/custom/customs.py +++ b/langflow/backend/langflow_backend/custom/customs.py @@ -3,7 +3,7 @@ from langchain.agents.mrkl import prompt def get_custom_prompts(): """Get custom prompts.""" - + return { "ZeroShotPrompt": { "template": { diff --git a/langflow/backend/langflow_backend/interface/loading.py b/langflow/backend/langflow_backend/interface/loading.py index 07ee5bf4a..f23aa3599 100644 --- a/langflow/backend/langflow_backend/interface/loading.py +++ b/langflow/backend/langflow_backend/interface/loading.py @@ -35,9 +35,6 @@ def replace_zero_shot_prompt_with_prompt_template(nodes): return nodes - - - def load_langchain_type_from_config(config: Dict[str, Any]): """Load langchain type from config""" # Get type list @@ -52,9 +49,6 @@ def load_langchain_type_from_config(config: Dict[str, Any]): raise ValueError("Type should be either agent, chain or llm") - - - def build_prompt_template(prompt, tools): """Build PromptTemplate from ZeroShotPrompt""" prefix = prompt["node"]["template"]["prefix"]["value"] diff --git a/langflow/backend/langflow_backend/utils/payload.py b/langflow/backend/langflow_backend/utils/payload.py index 7674e1cf9..4d183ba82 100644 --- a/langflow/backend/langflow_backend/utils/payload.py +++ b/langflow/backend/langflow_backend/utils/payload.py @@ -4,7 +4,8 @@ import re def extract_input_variables(data): """ - Extracts input variables from the template and adds them to the input_variables field. + Extracts input variables from the template + and adds them to the input_variables field. """ for node in data["nodes"]: with contextlib.suppress(Exception): @@ -59,7 +60,6 @@ def build_json(root, nodes, edges): elif "dict" in module_type: value = {} else: - children = [] for c in local_nodes: module_types = [c["data"]["type"]] diff --git a/langflow/backend/langflow_backend/utils/util.py b/langflow/backend/langflow_backend/utils/util.py index 2bc3b4afa..759e74740 100644 --- a/langflow/backend/langflow_backend/utils/util.py +++ b/langflow/backend/langflow_backend/utils/util.py @@ -3,7 +3,6 @@ import inspect import re import importlib -from langchain.agents.load_tools import * from langchain.agents.load_tools import ( _BASE_TOOLS, _LLM_TOOLS, diff --git a/langflow/server.py b/langflow/server.py index d245ec4f5..0e06bb9fb 100644 --- a/langflow/server.py +++ b/langflow/server.py @@ -8,10 +8,13 @@ class LangflowApplication(BaseApplication): super().__init__() def load_config(self): - config = {key: value for key, value in self.options.items() - if key in self.cfg.settings and value is not None} + config = { + key: value + for key, value in self.options.items() + if key in self.cfg.settings and value is not None + } for key, value in config.items(): self.cfg.set(key.lower(), value) def load(self): - return self.application \ No newline at end of file + return self.application