jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor: Update ChromaVectorStoreComponent to use DropdownInput for search type

Browse source
This commit is contained in:
ogabrielluiz 2024-06-18 11:39:04 -03:00
commit 64584e98f4
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/backend/base/langflow/components/vectorstores/Chroma.py
View file

@ -8,7 +8,6 @@ from loguru import logger
from langflow.base.vectorstores.model import LCVectorStoreComponent from langflow.base.vectorstores.model import LCVectorStoreComponent
from langflow.base.vectorstores.utils import chroma_collection_to_data from langflow.base.vectorstores.utils import chroma_collection_to_data
from langflow.inputs import BoolInput, DataInput, DropdownInput, HandleInput, IntInput, MessageInput, StrInput from langflow.inputs import BoolInput, DataInput, DropdownInput, HandleInput, IntInput, MessageInput, StrInput
IntInput, MessageInput, StrInput)
from langflow.schema import Data from langflow.schema import Data
if TYPE_CHECKING: if TYPE_CHECKING:
@ -86,7 +85,7 @@ class ChromaVectorStoreComponent(LCVectorStoreComponent):
display_name="Search Type", display_name="Search Type",
options=["Similarity", "MMR"], options=["Similarity", "MMR"],
value="Similarity", value="Similarity",
advanced=True advanced=True,
), ),
IntInput( IntInput(
name="number_of_results", name="number_of_results",