refactor: Update VectaraVectorStoreComponent to handle embedding input and add documents to vector store

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-24 08:58:38 -03:00
commit 8c7063e0ec

View file

@ -1,11 +1,11 @@
from typing import List, TYPE_CHECKING from typing import TYPE_CHECKING, List
from langchain_community.vectorstores import Vectara from langchain_community.vectorstores import Vectara
from loguru import logger 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, SecretStrInput, StrInput, TextInput from langflow.io import HandleInput, IntInput, MessageTextInput, SecretStrInput, StrInput
from langflow.schema import Data from langflow.schema import Data
if TYPE_CHECKING: if TYPE_CHECKING:
@ -37,7 +37,7 @@ class VectaraVectorStoreComponent(LCVectorStoreComponent):
input_types=["Document", "Data"], input_types=["Document", "Data"],
is_list=True, is_list=True,
), ),
TextInput( MessageTextInput(
name="search_query", name="search_query",
display_name="Search Query", display_name="Search Query",
), ),