fix test, linting, and vector_store folder

This commit is contained in:
Ibis Prevedello 2023-04-13 22:12:21 -03:00
commit 3d211fc487
14 changed files with 13 additions and 13 deletions

View file

@ -27,7 +27,7 @@ from langflow.interface.text_splitters.base import textsplitter_creator
from langflow.interface.toolkits.base import toolkits_creator
from langflow.interface.tools.base import tool_creator
from langflow.interface.tools.constants import FILE_TOOLS
from langflow.interface.vectorstore.base import vectorstore_creator
from langflow.interface.vector_store.base import vectorstore_creator
from langflow.interface.wrappers.base import wrapper_creator
from langflow.utils import payload

View file

@ -1,10 +1,10 @@
from abc import ABC, abstractmethod
from typing import Any, Dict, List, Optional, Type, Union
from langflow.utils.logger import logger
from pydantic import BaseModel
from langflow.template.base import FrontendNode, Template, TemplateField
from langflow.utils.logger import logger
# Assuming necessary imports for Field, Template, and FrontendNode classes

View file

@ -5,8 +5,8 @@ from langflow.interface.base import LangChainTypeCreator
from langflow.interface.custom_lists import chain_type_to_cls_dict
from langflow.settings import settings
from langflow.template.nodes import ChainFrontendNode
from langflow.utils.util import build_template_from_class
from langflow.utils.logger import logger
from langflow.utils.util import build_template_from_class
# Assuming necessary imports for Field, Template, and FrontendNode classes

View file

@ -3,8 +3,8 @@ from typing import Dict, List, Optional
from langflow.interface.base import LangChainTypeCreator
from langflow.interface.custom_lists import documentloaders_type_to_cls_dict
from langflow.settings import settings
from langflow.utils.util import build_template_from_class
from langflow.utils.logger import logger
from langflow.utils.util import build_template_from_class
def build_file_path_template(

View file

@ -3,8 +3,8 @@ from typing import Dict, List, Optional
from langflow.interface.base import LangChainTypeCreator
from langflow.interface.custom_lists import embedding_type_to_cls_dict
from langflow.settings import settings
from langflow.utils.util import build_template_from_class
from langflow.utils.logger import logger
from langflow.utils.util import build_template_from_class
class EmbeddingCreator(LangChainTypeCreator):

View file

@ -8,7 +8,7 @@ from langflow.interface.prompts.base import prompt_creator
from langflow.interface.text_splitters.base import textsplitter_creator
from langflow.interface.toolkits.base import toolkits_creator
from langflow.interface.tools.base import tool_creator
from langflow.interface.vectorstore.base import vectorstore_creator
from langflow.interface.vector_store.base import vectorstore_creator
from langflow.interface.wrappers.base import wrapper_creator

View file

@ -4,8 +4,8 @@ from langflow.interface.base import LangChainTypeCreator
from langflow.interface.custom_lists import llm_type_to_cls_dict
from langflow.settings import settings
from langflow.template.nodes import LLMFrontendNode
from langflow.utils.util import build_template_from_class
from langflow.utils.logger import logger
from langflow.utils.util import build_template_from_class
class LLMCreator(LangChainTypeCreator):

View file

@ -7,8 +7,8 @@ from langflow.interface.base import LangChainTypeCreator
from langflow.interface.importing.utils import import_class
from langflow.settings import settings
from langflow.template.nodes import PromptFrontendNode
from langflow.utils.util import build_template_from_class
from langflow.utils.logger import logger
from langflow.utils.util import build_template_from_class
class PromptCreator(LangChainTypeCreator):

View file

@ -5,8 +5,8 @@ from langchain.agents import agent_toolkits
from langflow.interface.base import LangChainTypeCreator
from langflow.interface.importing.utils import import_class, import_module
from langflow.settings import settings
from langflow.utils.util import build_template_from_class
from langflow.utils.logger import logger
from langflow.utils.util import build_template_from_class
class ToolkitCreator(LangChainTypeCreator):

View file

@ -8,7 +8,7 @@ from langflow.interface.prompts.base import prompt_creator
from langflow.interface.text_splitters.base import textsplitter_creator
from langflow.interface.toolkits.base import toolkits_creator
from langflow.interface.tools.base import tool_creator
from langflow.interface.vectorstore.base import vectorstore_creator
from langflow.interface.vector_store.base import vectorstore_creator
from langflow.interface.wrappers.base import wrapper_creator

View file

@ -3,8 +3,8 @@ from typing import Dict, List, Optional
from langflow.interface.base import LangChainTypeCreator
from langflow.interface.custom_lists import vectorstores_type_to_cls_dict
from langflow.settings import settings
from langflow.utils.util import build_template_from_class
from langflow.utils.logger import logger
from langflow.utils.util import build_template_from_class
class VectorstoreCreator(LangChainTypeCreator):

View file

@ -3,8 +3,8 @@ from typing import Dict, List, Optional
from langchain import requests
from langflow.interface.base import LangChainTypeCreator
from langflow.utils.util import build_template_from_class
from langflow.utils.logger import logger
from langflow.utils.util import build_template_from_class
class WrapperCreator(LangChainTypeCreator):

View file

@ -440,7 +440,7 @@ def test_chat_open_ai(client: TestClient):
assert template["_type"] == "ChatOpenAI"
assert (
model["description"]
== "Wrapper around OpenAI Chat large language models.To use, you should have the ``openai`` python package installed, and theenvironment variable ``OPENAI_API_KEY`` set with your API key.Any parameters that are valid to be passed to the openai.create call can be passedin, even if not explicitly saved on this class." # noqa E501
== "Wrapper around OpenAI Chat large language models." # noqa E501
)
assert set(model["base_classes"]) == {
"BaseChatModel",