fix: run ruff

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-24 08:42:51 -03:00
commit d11e7c6fc5

View file

@ -5,12 +5,13 @@ from loguru import logger
from langflow.base.vectorstores.model import LCVectorStoreComponent from langflow.base.vectorstores.model import LCVectorStoreComponent
from langflow.helpers.data import docs_to_data from langflow.helpers.data import docs_to_data
from langflow.io import HandleInput, IntInput, Output, SecretStrInput, StrInput, TextInput from langflow.io import HandleInput, IntInput, SecretStrInput, StrInput, TextInput
from langflow.schema import Data from langflow.schema import Data
if TYPE_CHECKING: if TYPE_CHECKING:
from langchain_community.vectorstores import Vectara from langchain_community.vectorstores import Vectara
class VectaraVectorStoreComponent(LCVectorStoreComponent): class VectaraVectorStoreComponent(LCVectorStoreComponent):
""" """
Vectara Vector Store with search capabilities Vectara Vector Store with search capabilities
@ -56,9 +57,7 @@ class VectaraVectorStoreComponent(LCVectorStoreComponent):
try: try:
from langchain_community.vectorstores import Vectara from langchain_community.vectorstores import Vectara
except ImportError: except ImportError:
raise ImportError( raise ImportError("Could not import Vectara. Please install it with `pip install langchain-community`.")
"Could not import Vectara. Please install it with `pip install langchain-community`."
)
vectara = Vectara( vectara = Vectara(
vectara_customer_id=self.vectara_customer_id, vectara_customer_id=self.vectara_customer_id,