From 064cf8120f2f2e6fc95092daa476e4344e47680e Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Thu, 30 Mar 2023 16:34:52 -0300 Subject: [PATCH] fix: change typing for linting and formatting --- src/backend/langflow/graph/graph.py | 2 -- src/backend/langflow/graph/nodes.py | 3 --- src/backend/langflow/interface/signature.py | 1 - src/backend/langflow/settings.py | 16 ++++++++-------- src/backend/langflow/utils/util.py | 2 +- 5 files changed, 9 insertions(+), 15 deletions(-) diff --git a/src/backend/langflow/graph/graph.py b/src/backend/langflow/graph/graph.py index 91b225986..de8469641 100644 --- a/src/backend/langflow/graph/graph.py +++ b/src/backend/langflow/graph/graph.py @@ -1,5 +1,3 @@ -from copy import deepcopy -import types from typing import Dict, List, Union from langflow.utils import payload from langflow.interface.listing import ALL_TOOLS_NAMES diff --git a/src/backend/langflow/graph/nodes.py b/src/backend/langflow/graph/nodes.py index 963b6cb45..43350b3d9 100644 --- a/src/backend/langflow/graph/nodes.py +++ b/src/backend/langflow/graph/nodes.py @@ -1,9 +1,6 @@ from copy import deepcopy -import types from typing import Any, Dict, List, Optional, Union -from langflow.interface.listing import ALL_TYPES_DICT, TOOLS_DICT -from langflow.interface import loading from langflow.graph.base import Node diff --git a/src/backend/langflow/interface/signature.py b/src/backend/langflow/interface/signature.py index 80a544d31..f9de437a7 100644 --- a/src/backend/langflow/interface/signature.py +++ b/src/backend/langflow/interface/signature.py @@ -13,7 +13,6 @@ from langflow.interface.custom_lists import ( llm_type_to_cls_dict, memory_type_to_cls_dict, toolkit_type_to_cls_dict, - toolkit_type_to_loader_dict, wrapper_type_to_cls_dict, ) diff --git a/src/backend/langflow/settings.py b/src/backend/langflow/settings.py index f4dd4ae30..2c8d3e8c6 100644 --- a/src/backend/langflow/settings.py +++ b/src/backend/langflow/settings.py @@ -1,18 +1,18 @@ import os -from typing import List, Optional +from typing import List import yaml from pydantic import BaseSettings, Field, root_validator class Settings(BaseSettings): - chains: Optional[List[str]] = Field(...) - agents: Optional[List[str]] = Field(...) - prompts: Optional[List[str]] = Field(...) - llms: Optional[List[str]] = Field(...) - tools: Optional[List[str]] = Field(...) - memories: Optional[List[str]] = Field(...) - dev: bool = Field(...) + chains: List[str] = Field(default=[]) + agents: List[str] = Field(default=[]) + prompts: List[str] = Field(default=[]) + llms: List[str] = Field(default=[]) + tools: List[str] = Field(default=[]) + memories: List[str] = Field(default=[]) + dev: bool = Field(default=False) class Config: validate_assignment = True diff --git a/src/backend/langflow/utils/util.py b/src/backend/langflow/utils/util.py index c7dfc230e..10048c831 100644 --- a/src/backend/langflow/utils/util.py +++ b/src/backend/langflow/utils/util.py @@ -55,7 +55,7 @@ def build_template_from_parameters( return { "template": format_dict(variables, name), - "description": docs["Description"], + "description": docs["Description"], # type: ignore "base_classes": base_classes, }