Skip to content

Commit

Permalink
refactor: update cached_vectorstore type and input order in Cassandra…
Browse files Browse the repository at this point in the history
…VectorStoreComponent
  • Loading branch information
ogabrielluiz committed Jul 2, 2024
1 parent 341c48a commit 79eec65
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
@@ -1,10 +1,11 @@
from typing import List

from langchain_community.vectorstores import Cassandra
from loguru import logger

from langflow.base.vectorstores.model import LCVectorStoreComponent
from langflow.helpers.data import docs_to_data
from langflow.inputs import DictInput, FloatInput, BoolInput
from langflow.inputs import BoolInput, DictInput, FloatInput
from langflow.io import (
DataInput,
DropdownInput,
Expand All @@ -15,7 +16,6 @@
SecretStrInput,
)
from langflow.schema import Data
from loguru import logger


class CassandraVectorStoreComponent(LCVectorStoreComponent):
Expand All @@ -24,7 +24,7 @@ class CassandraVectorStoreComponent(LCVectorStoreComponent):
documentation = "https://python.langchain.com/docs/modules/data_connection/vectorstores/integrations/cassandra"
icon = "Cassandra"

_cached_vectorstore: Cassandra = None
_cached_vectorstore: Cassandra | None = None

inputs = [
MessageTextInput(
Expand Down

0 comments on commit 79eec65

Please sign in to comment.